import { getDb } from "./db";
import { Knex } from "knex";

export class Settings {

    static KEYS = [
        "host_language",
        "libretranslate_base_url",
        'ui_direction',
        "whisper_model",
    ]

    constructor(public db: Knex) {

    }

    private async getValue(key: string) {
        // For security...ensure key is a valid value
        if (!Settings.KEYS.includes(key)) {
            throw new Error(`Invalid setting: '${key}'`)
        }

        const row = await this.db("settings").select(key).limit(1).first();
        if (!(row && row[key])) return undefined;
        return row[key];
    }


    private async setValue(key: string, value: any) {
        if (!Settings.KEYS.includes(key)) {
            throw new Error(`Invalid setting: '${key}'`);
        }

        // Check if the key already exists
        const [exists] = await this.db("settings").select(1).whereNotNull(key).limit(1);

        if (exists) {
            // Update the existing column
            await this.db("settings").update({ [key]: value });
        } else {
            // Insert a new value into the specified column
            const insertData: { [key: string]: any } = {};
            insertData[key] = value;
            await this.db("settings").insert(insertData);
        }
    }

    async setHostLanguage(value: string) {
        await this.setValue("host_language", value);
    }

    async getHostLanguage() {
        return await this.getValue("host_language")
    }

    async setLibretranslateBaseUrl(value : string) {
        await this.setValue("libretranslate_base_url", value)
    }

    async getLibretranslateBaseUrl() {
        return await this.getValue("libretranslate_base_url")
    }

    async setWhisperModel(value : string) {
        await this.setValue("whisper_model", value);
    }

    async getWhisperModel() {
        return await this.getValue("whisper_model");
    }

    static async getDefault() {
        return new Settings(await getDb())
    }

}