from flask import Flask, render_template, request, flash, redirect, url_for
import pymysql
from pymysql.cursors import Cursor
import os
from typing import Dict

app = Flask(__name__)
app.secret_key = os.environ.get('SECRET_KEY', 'OPTIONALSECRETKEY')

def get_db():
    return pymysql.connect(
        host='localhost', user='kejingfan', 
        password='xxxxxxxx', database='DBLab_7_1',
        charset='utf8mb4',
        cursorclass=pymysql.cursors.DictCursor
    )

@app.route("/")
def index():
    return render_template("index.html")

@app.route("/signup.html", methods=('GET', 'POST'))
def signup():
    if request.method == 'GET':
        return render_template('signup.html')
    
    if request.method == 'POST':
        id = request.form['cardCode']
        name = request.form['name']
        phone_number = request.form['mobileNo']
        password = request.form['encryptedPassword']

        db = get_db()
        cursor = db.cursor()

        try:
            cursor.callproc('RegisterPassenger', (id, name, phone_number, password, "@result_message"))
            cursor.fetchall()
            cursor.execute("SELECT @_RegisterPassenger_4;")
            result_message = cursor.fetchone()['@_RegisterPassenger_4']
            print(result_message)
            flash(result_message)
            db.commit()
        except pymysql.MySQLError as e:
            db.rollback()
            if e.args[0] == 1644:  # SQLSTATE 45000 corresponds to error code 1644
                flash("乘客已存在，无法重复注册")
            else:
                print(e)
                flash("数据库异常，注册失败")
        db.close()
        return redirect(url_for('index'))

def verify_user(cursor: Cursor, id: str, password: str) -> str:
    try:
        cursor.callproc('VerifyUser', (id, password, "@verify_status"))
        cursor.fetchall()
        cursor.execute("SELECT @_VerifyUser_2;")
        verify_status = cursor.fetchone()['@_VerifyUser_2']
    except pymysql.MySQLError as e:
        print(e)
        if e.args[0] == 1644:  # SQLSTATE 45000 corresponds to error code 1644
            return "NO_USER"
        return "DB_ERROR"
    return verify_status

class ModifyInfo:
    def __init__(self, form: Dict[str, str]):
        self.id = form['cardCode']
        modifyType = form['modifyType']
        self.new_password = form['encryptedNewPassword']
        self.phone_number = form['mobileNo'] if form['mobileNo'] != "" else "11111111111"
        modifyType2command = {
            '1': 'delete account',
            '2': 'modify Password',
            '3': 'modify Phone_Number'
        }
        self.command = modifyType2command[modifyType]

    def get_args(self):
        return (self.id, self.command, self.new_password, self.phone_number, "@result_message")

    def get_ok_message(self, cursor):
        cursor.execute("SELECT @_ModifyPassengerInfo_4;")
        return cursor.fetchone()['@_ModifyPassengerInfo_4']

@app.route("/modify.html", methods=('GET', 'POST'))
def modify():
    if request.method == 'GET':
        return render_template('modify.html')
    
    if request.method == 'POST':
        id = request.form['cardCode']
        password = request.form['encryptedPassword']
        db = get_db()
        cursor = db.cursor()

        verify_info = verify_user(cursor, id, password)
        if verify_info == "NO_USER":
            flash("您未注册过，无法修改账号")
            db.close()
            return redirect(url_for('signup'))
        elif verify_info == "WRONG_PASSWORD":
            flash("密码错误")
            db.close()
            return redirect(url_for('modify'))
        elif verify_info == "DB_ERROR":
            db.close()
            return redirect(url_for('modify'))

        modifyInfo = ModifyInfo(request.form)
        try:
            cursor.callproc('ModifyPassengerInfo', modifyInfo.get_args())
            cursor.fetchall()
            db.commit()
            flash(modifyInfo.get_ok_message(cursor))
        except pymysql.MySQLError as e:
            db.rollback()
            if e.args[0] == 1644:  # SQLSTATE 45000 corresponds to error code 1644
                flash("用户不存在，无法修改")
            else:
                print(e)
                flash("数据库异常，修改失败")
        db.close()
        return redirect(url_for('index'))

if __name__ == "__main__":
    app.run(debug=True)