views.py

from flask import Flask, render_template, request, redirect, url_for, flash
from flask_login import current_user, login_user, logout_user, login_required
from flask_socketio import emit, join_room, leave_room
from models import User, Note, Message, Room
from app import app, db, socketio
import uuid
from datetime import datetime

@app.route('/') # 路由
@app.route('/home')
def home():
    if current_user.is_authenticated:
        friends = db.session.query(User.user_ID).distinct().join(
            Room, 
            db.or_(
                db.and_(Room.user_ID1 == User.user_ID, Room.user_ID2 == current_user.user_ID),
                db.and_(Room.user_ID2 == User.user_ID, Room.user_ID1 == current_user.user_ID)
            )
        ).all()
        friendsID = [friend[0] for friend in friends]
        friendsID.append(current_user.user_ID)
        notes = Note.query.filter(Note.user_ID.in_(friendsID)).all()
        rooms = Room.query.filter(db.or_(Room.user_ID1 == current_user.user_ID, Room.user_ID2 == current_user.user_ID)).all()
        print(notes)
        return render_template('home.html', notes=notes, rooms=rooms, user_id=current_user.user_ID)
    
    return render_template('login.html')

@app.route('/register', methods=['GET', 'POST'])
def register(): # 註冊
    if request.method == 'POST':
        user_id = request.form.get('user_id')
        name = request.form.get('name')
        password = request.form.get('password')
        passwordConfirm = request.form.get('password_confirm')
        
        # 檢查使用者是否存在
        if User.query.filter_by(user_ID=user_id).first():
            flash('使用者帳號已存在', 'error')
            return redirect(url_for('register'))
        
        if passwordConfirm != password:
            flash('密碼不一致', 'error')
            return redirect(url_for('register'))
        
        try:
            # 建立新使用者
            newUser = User(user_ID=user_id, name=name, password=password)
            db.session.add(newUser)
            db.session.commit()
            flash('註冊成功！', 'success')
            return redirect(url_for('login'))
        except Exception as e:
            User.session.rollback()
            flash('註冊失敗，請稍後再試', 'warning')
            return redirect(url_for('register'))
            
    return render_template('register.html')

@app.route('/login', methods=['GET', 'POST'])
def login(): # 登入
    if request.method == 'POST':
        user_id = request.form.get('user_id')
        password = request.form.get('password')
        
        # 檢查使用者是否存在
        user = User.query.filter_by(user_ID=user_id).first()
        if user and user.password == password:
            login_user(user)
            return redirect(url_for('home', user_id=user_id))
        elif not user:
            flash('帳號或密碼錯誤', 'error')
        else:
            flash('帳號或密碼錯誤', 'error')
    
    return render_template('login.html')

@login_required
@app.route('/logout', methods=['GET'])
def logout(): # 登出
    logout_user()
    return redirect(url_for('home'))

@login_required
@app.route('/addNote', methods=['GET', 'POST'])
def addNote(): # 新增便利貼
    if request.method == 'POST':
        content = request.form.get('content')
        user_id = current_user.user_ID
        
        if not Note.query.filter_by(user_ID=user_id).first():
            # 建立新便利貼
            newNote = Note(note_ID=str(uuid.uuid4()), content=content, time=datetime.now(), user_ID=user_id)
            db.session.add(newNote)
            db.session.commit()
            flash('新增便利貼成功！', 'success')
        else:
            # 更新便利貼
            updateNote = Note.query.filter_by(user_ID=user_id).first()
            updateNote.content = content
            updateNote.time = datetime.now()
            db.session.commit()
            flash('更新便利貼成功！', 'success')
        
        return redirect(url_for('home', user_id=user_id)) # 修改完重新導向至 home.html

@login_required
@app.route('/createRoom', methods=['GET', 'POST'])
def createRoom(): # 建立聊天室
    if request.method == 'POST':
        user_id1 = current_user.user_ID
        user_id2 = request.form.get('user_ID2')

        if Room.query.filter_by(user_ID1=user_id1, user_ID2=user_id2).first():
            flash('聊天室已存在！', 'error')
            return redirect(url_for('room', room_no=Room.query.filter_by(user_ID1=user_id1, user_ID2=user_id2).first().room_no))
        elif Room.query.filter_by(user_ID1=user_id2, user_ID2=user_id1).first():
            flash('聊天室已存在！', 'error')
            return redirect(url_for('room', room_no=Room.query.filter_by(user_ID1=user_id2, user_ID2=user_id1).first().room_no))
        else:
            # 建立新聊天室
            newRoom = Room(room_no=str(uuid.uuid4()), user_ID1=user_id1, user_ID2=user_id2)
            db.session.add(newRoom)
            db.session.commit()
            flash('建立聊天室成功！', 'success')
            return redirect(url_for('room', room_no=newRoom.room_no))

@login_required
@app.route('/room/<string:room_no>', methods=['GET', 'POST'])
def room(room_no): # 聊天室畫面
    room = Room.query.filter_by(room_no=room_no).first()
    if not room:
        flash('聊天室不存在！', 'error')
        return redirect(url_for('home'))
    
    if current_user.is_authenticated:
        friends = db.session.query(User.user_ID).distinct().join(
            Room, 
            db.or_(
                db.and_(Room.user_ID1 == User.user_ID, Room.user_ID2 == current_user.user_ID),
                db.and_(Room.user_ID2 == User.user_ID, Room.user_ID1 == current_user.user_ID)
            )
        ).all()
        friendsID = [friend[0] for friend in friends]
        friendsID.append(current_user.user_ID)
        messages = Message.query.filter_by(room_no=room_no).all()
        notes = Note.query.filter(Note.user_ID.in_(friendsID)).all()
        rooms = Room.query.filter(db.or_(Room.user_ID1 == current_user.user_ID, Room.user_ID2 == current_user.user_ID)).all()

    return render_template('room.html', room=room, room_no=room_no, messages=messages, user_id=current_user.user_ID, notes=notes, rooms=rooms)

@socketio.on('join')
@login_required
def onJoin(data):
    try:
        room = data.get('room')
        if room:
            join_room(room)
    except Exception as e:
        print(e)

@socketio.on('sendMessage')
@login_required
# @app.route('/sendMessage', methods=['GET', 'POST'])
def sendMessage(data):
    try:
        room_no = data.get('room')
        message = data.get('message')
        sender = current_user.user_ID
        time = datetime.now()

        newMessage = Message(room_no=room_no, sender=sender, time=time, message=message)
        db.session.add(newMessage)
        db.session.commit()

        emit('receiveMessage', {'message': message, 'sender': sender}, room=room_no)
    
    except Exception as e:
        db.session.rollback()
        print(e)