47 lines
1.6 KiB
TypeScript
47 lines
1.6 KiB
TypeScript
|
|
import { create } from "zustand";
|
||
|
|
import { persist } from "zustand/middleware";
|
||
|
|
import type { AppSettings, ProviderConfig } from "@/lib/tauriCommands";
|
||
|
|
|
||
|
|
interface SettingsState extends AppSettings {
|
||
|
|
addProvider: (provider: ProviderConfig) => void;
|
||
|
|
updateProvider: (index: number, provider: ProviderConfig) => void;
|
||
|
|
removeProvider: (index: number) => void;
|
||
|
|
setActiveProvider: (name: string) => void;
|
||
|
|
setTheme: (theme: "light" | "dark") => void;
|
||
|
|
getActiveProvider: () => ProviderConfig | undefined;
|
||
|
|
}
|
||
|
|
|
||
|
|
export const useSettingsStore = create<SettingsState>()(
|
||
|
|
persist(
|
||
|
|
(set, get) => ({
|
||
|
|
theme: "dark" as const,
|
||
|
|
ai_providers: [] as ProviderConfig[],
|
||
|
|
active_provider: undefined,
|
||
|
|
default_provider: "ollama",
|
||
|
|
default_model: "llama3.2:3b",
|
||
|
|
ollama_url: "http://localhost:11434",
|
||
|
|
|
||
|
|
addProvider: (provider) =>
|
||
|
|
set((state) => ({ ai_providers: [...state.ai_providers, provider] })),
|
||
|
|
updateProvider: (index, provider) =>
|
||
|
|
set((state) => {
|
||
|
|
const providers = [...state.ai_providers];
|
||
|
|
providers[index] = provider;
|
||
|
|
return { ai_providers: providers };
|
||
|
|
}),
|
||
|
|
removeProvider: (index) =>
|
||
|
|
set((state) => ({
|
||
|
|
ai_providers: state.ai_providers.filter((_, i) => i !== index),
|
||
|
|
})),
|
||
|
|
setActiveProvider: (name) => set({ active_provider: name }),
|
||
|
|
setTheme: (theme) => set({ theme }),
|
||
|
|
getActiveProvider: () => {
|
||
|
|
const state = get();
|
||
|
|
return state.ai_providers.find((p) => p.name === state.active_provider)
|
||
|
|
?? state.ai_providers[0];
|
||
|
|
},
|
||
|
|
}),
|
||
|
|
{ name: "tftsr-settings" }
|
||
|
|
)
|
||
|
|
);
|