import { DatabaseSync } from "node:sqlite"
// import { Buffer } from "node:buffer"
import path from 'node:path'

import config from '../config.ts'
import ChatBean from './ChatBean.ts'
import { SQLInputValue } from "node:sqlite"
import chalk from "chalk"
import User from "./User.ts"
import ChatType from "./ChatType.ts"

/**
 * Chat.ts - Wrapper and manager
 * Wrap with ChatBean to directly update database
 * Manage the database by itself (static)
 */
export default class Chat {
    static table_name: string = "Chat"
    private static database: DatabaseSync = Chat.init()
    private static init(): DatabaseSync {
        const db: DatabaseSync = new DatabaseSync(path.join(config.data_path, 'Chats.db'))
        db.exec(`
            CREATE TABLE IF NOT EXISTS ${Chat.table_name} (
                /* 序号 */ count INTEGER PRIMARY KEY AUTOINCREMENT,
                /* 類型 */ type TEXT NOT NULL,
                /* Chat ID */ id TEXT NOT NULL,
                /* 標題 (群組) */ title TEXT,
                /* 頭像 (群組) */ avatar BLOB,
                /* 成員 */ members_list TEXT,
                /* 设置 */ settings TEXT NOT NULL
            );
       `)
        return db
    }

    protected static findAllBeansByCondition(condition: string, ...args: SQLInputValue[]): ChatBean[] {
        return this.database.prepare(`SELECT * FROM ${Chat.table_name} WHERE ${condition}`).all(...args) as unknown as ChatBean[]
    }

    static findById(id: string) {
        const beans = this.findAllBeansByCondition('id = ?', id)
        if (beans.length == 0)
            return null
        else if (beans.length > 1)
            console.error(chalk.red(`警告: 查询 id = ${id} 时, 查询到多个相同 ID 的 Chat`))
        return new Chat(beans[0])
    }

    static create(chatId: string, type: ChatType) {
        const chat = new Chat(
            Chat.findAllBeansByCondition(
                'count = ?',
                Chat.database.prepare(`INSERT INTO ${Chat.table_name} (
                    type,
                    id,
                    title,
                    avatar,
                    members_list,
                    settings
                ) VALUES (?, ?, ?, ?, ?, ?);`).run(
                    type,
                    chatId,
                    null,
                    null,
                    "[]",
                    "{}"
                ).lastInsertRowid
            )[0]
        )
        return chat
    }

    declare bean: ChatBean
    constructor(bean: ChatBean) {
        this.bean = bean
    }
    setAttr(key: string, value: SQLInputValue): void {
        Chat.database.prepare(`UPDATE ${Chat.table_name} SET ${key} = ? WHERE id = ?`).run(value, this.bean.id)
        this.bean[key] = value
    }

    getMembersList() {
        return JSON.parse(this.bean.members_list) as string[]
    }
    addMember(userId: string) {
        const ls = this.getMembersList()
        ls.push(userId)
        this.setMembers(ls)
    }
    setMembers(members: string[]) {
        this.setAttr("members_list", JSON.stringify(members))
    }
    removeMembers(members: string[]) {
        const ls = this.getMembersList().filter((v) => !members.includes(v))
        this.setAttr("members_list", JSON.stringify(ls))
    }
    getAnotherUserForPrivate(userMySelf: User) {
        const user_a_id = this.getMembersList()[0]
        const user_b_id = this.getMembersList()[0]
        // 注意: 這裏已經確定了 Chat, 不需要再指定對方用戶
        if (user_a_id == userMySelf.bean.id)
            return User.findById(user_b_id as string)
        if (user_b_id == userMySelf.bean.id)
            return userMySelf

        return null
    }
    getTitle(userMySelf?: User) {
        if (this.bean.type == 'group') return this.bean.title
        if (this.bean.type == 'private') return this.getAnotherUserForPrivate(userMySelf as User)?.getNickName()
    }
    getAvatarFileHash(userMySelf?: User) {
        if (this.bean.type == 'group') return this.bean.avatar_file_hash
        if (this.bean.type == 'private') return this.getAnotherUserForPrivate(userMySelf as User)?.getAvatarFileHash()
    }
}