from db.pgsql import PostgreSQL
from db.system_config_dao import SystemConfigDAO

def check_user_login(user_id: str, password: str) -> bool:
    db = PostgreSQL()
    try:
        db.connect()
        sql = "SELECT 1 FROM sys_user WHERE user_id=%s AND password=md5(%s)"
        result = db.execute_query(sql, (user_id, password))
        return bool(len(result) > 0)
    finally:
        db.disconnect()

def get_user_info(user_id: str):
    db = PostgreSQL()
    try:
        db.connect()
        sql = """
            SELECT user_id, nick_name AS user_name, position_id, 
                   f_getname('GETCORPEXP', '', '', org_id) AS corp_name, 
                   org_id AS corp_id, default_org_id
            FROM sys_user
            WHERE user_id = %s
        """
        result = db.execute_query(sql, (user_id,))
        if result:
            user_name = result[0]['user_name']
            corp_name = result[0]['corp_name']
            corp_id = result[0]['corp_id']
            position_id = result[0]['position_id']
            default_org_id = result[0]['default_org_id']
            
            # 如果有默认账套ID，使用默认账套ID替代登录账套ID
            if default_org_id:
                corp_id = default_org_id
                # 查询默认账套的名称
                corp_name_sql = """
                    SELECT corp_exp
                    FROM sys_org
                    WHERE org_id = %s
                """
                corp_result = db.execute_query(corp_name_sql, (corp_id,))
                if corp_result:
                    corp_name = corp_result[0]['corp_exp']
            
            # 将用户的当前账套ID保存到内存中
            SystemConfigDAO.save_user_selected_corp(user_id, corp_id)
            
            return user_name, corp_name, corp_id, position_id
        return None, None, None, None
    finally:
        db.disconnect()