76 lines
2.0 KiB
TypeScript
76 lines
2.0 KiB
TypeScript
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())
|
|
}
|
|
|
|
} |