73 lines
1.8 KiB
TypeScript
73 lines
1.8 KiB
TypeScript
import { SQLiteDatabase } from "expo-sqlite";
|
|
import { getDb } from "./db";
|
|
|
|
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: { [key: string]: string } | null = this.db.getFirstSync(`SELECT ${key} from settings LIMIT 1`)
|
|
|
|
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
|
|
this.db.runSync(`INSERT INTO OR UPDATE
|
|
settings
|
|
(${key})
|
|
VALUES
|
|
(?)
|
|
WHERE
|
|
${key} IS NOT NULL`, 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");
|
|
}
|
|
|
|
static async getDefault() {
|
|
return new Settings(await getDb())
|
|
}
|
|
|
|
} |