import json
from abc import ABC, abstractmethod

import streamlit as st
import streamlit_authenticator as stauth
import yaml


# 策略介面
class DataHandlerStrategy(ABC):
    """資料處理策略介面"""

    @abstractmethod
    def read_data(self, path):
        """讀取資料"""
        pass

    @abstractmethod
    def write_data(self, path, data):
        """寫入資料"""
        pass


# YAML 策略
class YamlHandler(DataHandlerStrategy):
    """YAML 資料處理策略"""

    def read_data(self, path):
        """讀取 YAML 資料"""
        with open(path, "r") as f:
            return yaml.safe_load(f)

    def write_data(self, path, data):
        """寫入 YAML 資料"""
        with open(path, "w") as f:
            yaml.safe_dump(data, f)


# JSON 策略
class JsonHandler(DataHandlerStrategy):
    """JSON 資料處理策略"""

    def read_data(self, path):
        """讀取 JSON 資料"""
        with open(path, "r") as f:
            return json.load(f)

    def write_data(self, path, data):
        """寫入 JSON 資料"""
        with open(path, "w") as f:
            json.dump(data, f, ensure_ascii=False, indent=4)


class AuthMixin:
    """驗證混合類別"""

    def __init__(self, config_path="./config.yaml", handler: DataHandlerStrategy = YamlHandler()):
        self.config_path = config_path
        self.handler = handler
        self.config = self.load_config(config_path)
        self.authenticator = self.create_authenticator()

    def load_config(self, config_path):
        """載入配置檔案"""
        try:
            return self.handler.read_data(config_path)
        except FileNotFoundError:
            st.error(f"The configuration file {config_path} was not found.")
            st.stop()

    def create_authenticator(self):
        """建立驗證器"""
        config = self.config
        return stauth.Authenticate(
            config["credentials"],
            config["cookie"]["name"],
            config["cookie"]["key"],
            config["cookie"]["expiry_days"],
            config["preauthorized"],
        )

    def login(self):
        """登入"""
        fields = {
            "Form name": "Login",
            "Username": "Username",
            "Password": "Password",
            "Login": "Login",
        }
        name, authentication_status, username = self.authenticator.login(location="main", fields=fields)
        print(f"Name: {name}, Auth Status: {authentication_status}, Username: {username}")
        return name, authentication_status, username

    def logout(self):
        """登出"""
        self.authenticator.logout("Logout", "main", key="unique_key")

    def reset_password(self, username):
        """重設密碼"""
        fields = {
            "Form name": "Reset password",
            "Username": "Username",
            "Password": "New password",
            "Repeat password": "Repeat password",
            "Submit": "Submit",
        }
        try:
            if self.authenticator.reset_password(username, fields=fields):
                st.success("Password modified successfully")
                self.update_config()  # 在成功修改密碼後更新配置文件
        except Exception as e:
            st.error(e)

    def update_config(self):
        """更新配置檔案"""
        try:
            self.handler.write_data(self.config_path, self.config)
        except FileNotFoundError:
            st.error(f"The configuration file {self.config_path} was not found for writing.")

    def is_authenticated(self):
        """檢查是否已驗證"""
        _, authentication_status, _ = self.login()
        return authentication_status is True


class MyApp(AuthMixin):
    """應用程式主類別"""

    def __init__(self, config_path=None, handler=None):
        if config_path and handler:
            AuthMixin.__init__(self, config_path, handler)
        self.run()

    def run(self):
        """執行應用程式"""

        if hasattr(self, "authenticator"):
            self.handle_authenticated()
        else:
            self.display_content()

    def handle_authenticated(self):
        """處理已驗證使用者"""

        name, authentication_status, username = self.login()

        if authentication_status:  # 如果驗證狀態為真
            self.logout()  # 顯示 "Logout" 按鈕，連接到 "main"，並使用特定的 key
            st.write(f"Welcome *{name}*")  # 顯示歡迎訊息，使用 st.session_state 中的 name 屬性
            self.display_content()
            self.reset_password(username)  # 嘗試重設密碼
        elif authentication_status is False:  # 如果驗證狀態為假
            st.error("Username/password is incorrect")  # 顯示錯誤訊息，提示使用者帳號或密碼不正確
        elif authentication_status is None:  # 如果驗證狀態為空
            st.warning("Please enter your username and password")  # 顯示警告訊息，提示使用者輸入帳號和密碼

    def display_content(self):
        """顯示內容"""
        st.title("Some content")  # 顯示標題為 "Some content"


if __name__ == "__main__":
    # 使用 AuthMixin 進行驗證
    # MyApp(config_path="./config.yaml", handler=YamlHandler())

    # 使用 JSON 處理器進行驗證
    # MyApp(config_path="./config.json", handler=JsonHandler())

    # 不使用 AuthMixin 進行驗證
    MyApp()