mirror of
https://github.com/TriliumNext/Notes.git
synced 2025-07-29 02:52:27 +08:00
146 lines
4.5 KiB
TypeScript
146 lines
4.5 KiB
TypeScript
/**
|
|
* @module
|
|
*
|
|
* Options are key-value pairs that are used to store information such as user preferences (for example
|
|
* the current theme, sync server information), but also information about the state of the application.
|
|
*
|
|
* Although options internally are represented as strings, their value can be interpreted as a number or
|
|
* boolean by calling the appropriate methods from this service (e.g. {@link #getOptionInt}).\
|
|
*
|
|
* Generally options are shared across multiple instances of the application via the sync mechanism,
|
|
* however it is possible to have options that are local to an instance. For example, the user can select
|
|
* a theme on a device and it will not affect other devices.
|
|
*/
|
|
|
|
import becca from "../becca/becca.js";
|
|
import BOption from "../becca/entities/boption.js";
|
|
import type { OptionRow } from "@triliumnext/commons";
|
|
import type { FilterOptionsByType, OptionDefinitions, OptionMap, OptionNames } from "@triliumnext/commons";
|
|
import sql from "./sql.js";
|
|
|
|
function getOptionOrNull(name: OptionNames): string | null {
|
|
let option;
|
|
|
|
if (becca.loaded) {
|
|
option = becca.getOption(name);
|
|
} else {
|
|
// e.g. in initial sync becca is not loaded because DB is not initialized
|
|
try {
|
|
option = sql.getRow<OptionRow>("SELECT * FROM options WHERE name = ?", [name]);
|
|
} catch (e: unknown) {
|
|
// DB is not initialized.
|
|
return null;
|
|
}
|
|
}
|
|
|
|
return option ? option.value : null;
|
|
}
|
|
|
|
function getOption(name: OptionNames) {
|
|
const val = getOptionOrNull(name);
|
|
|
|
if (val === null) {
|
|
throw new Error(`Option '${name}' doesn't exist`);
|
|
}
|
|
|
|
return val;
|
|
}
|
|
|
|
function getOptionInt(name: FilterOptionsByType<number>, defaultValue?: number): number {
|
|
const val = getOption(name);
|
|
|
|
const intVal = parseInt(val);
|
|
|
|
if (isNaN(intVal)) {
|
|
if (defaultValue === undefined) {
|
|
throw new Error(`Could not parse '${val}' into integer for option '${name}'`);
|
|
} else {
|
|
return defaultValue;
|
|
}
|
|
}
|
|
|
|
return intVal;
|
|
}
|
|
|
|
function getOptionBool(name: FilterOptionsByType<boolean>): boolean {
|
|
const val = getOption(name);
|
|
|
|
if (typeof val !== "string" || !["true", "false"].includes(val)) {
|
|
throw new Error(`Could not parse '${val}' into boolean for option '${name}'`);
|
|
}
|
|
|
|
return val === "true";
|
|
}
|
|
|
|
function setOption<T extends OptionNames>(name: T, value: string | OptionDefinitions[T]) {
|
|
const option = becca.getOption(name);
|
|
|
|
if (option) {
|
|
option.value = value as string;
|
|
|
|
option.save();
|
|
} else {
|
|
createOption(name, value, false);
|
|
}
|
|
|
|
// Clear current AI provider when AI-related options change
|
|
const aiOptions = [
|
|
'aiSelectedProvider', 'openaiApiKey', 'openaiBaseUrl', 'openaiDefaultModel',
|
|
'anthropicApiKey', 'anthropicBaseUrl', 'anthropicDefaultModel',
|
|
'ollamaBaseUrl', 'ollamaDefaultModel'
|
|
];
|
|
|
|
if (aiOptions.includes(name)) {
|
|
// Import dynamically to avoid circular dependencies
|
|
setImmediate(async () => {
|
|
try {
|
|
const aiServiceManager = (await import('./llm/ai_service_manager.js')).default;
|
|
aiServiceManager.getInstance().clearCurrentProvider();
|
|
console.log(`Cleared AI provider after ${name} option changed`);
|
|
} catch (error) {
|
|
console.log(`Could not clear AI provider: ${error}`);
|
|
}
|
|
});
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Creates a new option in the database, with the given name, value and whether it should be synced.
|
|
*
|
|
* @param name the name of the option to be created.
|
|
* @param value the value of the option, as a string. It can then be interpreted as other types such as a number of boolean.
|
|
* @param isSynced `true` if the value should be synced across multiple instances (e.g. locale) or `false` if it should be local-only (e.g. theme).
|
|
*/
|
|
function createOption<T extends OptionNames>(name: T, value: string | OptionDefinitions[T], isSynced: boolean) {
|
|
new BOption({
|
|
name: name,
|
|
value: value as string,
|
|
isSynced: isSynced
|
|
}).save();
|
|
}
|
|
|
|
function getOptions() {
|
|
return Object.values(becca.options);
|
|
}
|
|
|
|
function getOptionMap() {
|
|
const map: Record<string, string> = {};
|
|
|
|
for (const option of Object.values(becca.options)) {
|
|
map[option.name] = option.value;
|
|
}
|
|
|
|
return map as OptionMap;
|
|
}
|
|
|
|
export default {
|
|
getOption,
|
|
getOptionInt,
|
|
getOptionBool,
|
|
setOption,
|
|
createOption,
|
|
getOptions,
|
|
getOptionMap,
|
|
getOptionOrNull
|
|
};
|