from flask import Blueprint, Response, jsonify, request

from app.extensions import db
from app.models.session import Session
from app.models.task import Task
from paper.errors import NeedPasswordException
from paper.parser import PaperParser

sessions = Blueprint("sessions", __name__, url_prefix="/sessions")

@sessions.route("/", methods=["POST"])
async def create_session(**kwargs):
    data = kwargs or request.json or request.form
    data["name"] = data["name"].replace("/", "").replace("\\", "") 

    session = Session.query.filter_by(name=data.get("name")).first()

    if not session:
        session = Session(
            name=data.get("name"),
            authorized=False
        )
        
        db.session.add(session)
    
    response = {"status": "ok"}

    async with PaperParser(data.get("name")) as parser:
        try:
            result = await parser.sign_in(**data)

            if hasattr(result, "phone_code_hash"):
                response = {
                    "status": "error",
                    "message": "Need secret code from telegram",
                    "action": "code",
                    "phone_hash": result.phone_code_hash,
                }
            else:
                session.authorized = True

        except NeedPasswordException:
            response = {
                "status": "error",
                "message": "Need 2FA Password",
                "action": "password",
            }
        
        except Exception as e:
            response = {
                "status": "exception",
                "message": str(e),
            }
        
        finally:
            db.session.commit()

    return jsonify(response)

@sessions.route("/", methods=["GET"])
def get_sessions(**kwargs):
    return Session.query.filter_by(authorized=True).all()

@sessions.route("/<int:id>", methods=["DELETE"])
async def remove_session(id: int):
    session: Session = Session.query.get_or_404(id)
    
    if session.has_depending_tasks():
        return 'Есть зависимые задачи', 409
    
    async with PaperParser(session.name) as parser:
        try:
            await parser.client.log_out()
        
        except Exception as e:
            return Response(e, 500)
        
    db.session.delete(session)
    db.session.commit()
    
    return Response('ok', status=200)