70 lines
1.8 KiB
TypeScript
70 lines
1.8 KiB
TypeScript
import { SQLiteDatabase } from "expo-sqlite";
|
|
import { getDb } from "./db";
|
|
import { WhisperFile, whisper_model_tag_t } from "./whisper";
|
|
|
|
export class Settings {
|
|
|
|
static KEYS = [
|
|
"host_language",
|
|
"libretranslate_base_url",
|
|
'ui_direction',
|
|
"whisper_model",
|
|
]
|
|
|
|
constructor(public db: SQLiteDatabase) {
|
|
|
|
}
|
|
|
|
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: { value: string } | null = this.db.getFirstSync(`SELECT value FROM settings WHERE key = ?`, key)
|
|
|
|
return row?.value;
|
|
}
|
|
|
|
|
|
private async setValue(key: string, value: any) {
|
|
if (!Settings.KEYS.includes(key)) {
|
|
throw new Error(`Invalid setting: '${key}'`);
|
|
}
|
|
|
|
// Check if the key already exists
|
|
this.db.runSync(`INSERT OR REPLACE INTO
|
|
settings
|
|
(key, value)
|
|
VALUES
|
|
(?, ?)`, key, value);
|
|
}
|
|
|
|
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") as whisper_model_tag_t;
|
|
}
|
|
|
|
static async getDefault() {
|
|
return new Settings(await getDb())
|
|
}
|
|
} |