Spaces:
Build error
Build error
File size: 4,084 Bytes
87a665c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 | /**
* Shared helpers for managing system-level connections.
* Used by both the admin settings UI and the desktop event handler
* to ensure consistent add/remove logic.
*/
import { getOpenAIConfig, updateOpenAIConfig } from '$lib/apis/openai';
import { getTerminalServerConnections, setTerminalServerConnections } from '$lib/apis/configs';
// ─── OpenAI Connections ─────────────────────────────────
/**
* Add an OpenAI-compatible API connection at the system level.
* Mirrors the logic in admin/Settings/Connections.svelte.
*/
export const addOpenAIConnection = async (
token: string,
connection: { url: string; key?: string; config?: object }
) => {
const current = await getOpenAIConfig(token);
const urls = current?.OPENAI_API_BASE_URLS ?? [];
const keys = current?.OPENAI_API_KEYS ?? [];
const configs = current?.OPENAI_API_CONFIGS ?? {};
const normalizedUrl = connection.url.replace(/\/$/, '');
// Don't add duplicates
if (urls.map((u: string) => u.replace(/\/$/, '')).includes(normalizedUrl)) {
return current;
}
urls.push(normalizedUrl);
keys.push(connection.key ?? '');
if (connection.config) {
configs[(urls.length - 1).toString()] = connection.config;
}
return await updateOpenAIConfig(token, {
ENABLE_OPENAI_API: current?.ENABLE_OPENAI_API ?? true,
OPENAI_API_BASE_URLS: urls,
OPENAI_API_KEYS: keys,
OPENAI_API_CONFIGS: configs
});
};
/**
* Remove an OpenAI-compatible API connection by URL at the system level.
* Re-indexes OPENAI_API_CONFIGS to match the admin delete pattern.
*/
export const removeOpenAIConnection = async (token: string, url: string) => {
const current = await getOpenAIConfig(token);
const urls: string[] = current?.OPENAI_API_BASE_URLS ?? [];
const keys: string[] = current?.OPENAI_API_KEYS ?? [];
const configs: Record<string, any> = current?.OPENAI_API_CONFIGS ?? {};
const normalizedUrl = url.replace(/\/$/, '');
const idx = urls.findIndex((u: string) => u.replace(/\/$/, '') === normalizedUrl);
if (idx === -1) return current;
const newUrls = urls.filter((_: string, i: number) => i !== idx);
const newKeys = keys.filter((_: string, i: number) => i !== idx);
// Re-index configs (mirrors admin/Settings/Connections.svelte onDelete)
const newConfigs: Record<string, any> = {};
newUrls.forEach((_: string, newIdx: number) => {
newConfigs[newIdx] = configs[newIdx < idx ? newIdx : newIdx + 1];
});
return await updateOpenAIConfig(token, {
ENABLE_OPENAI_API: current?.ENABLE_OPENAI_API ?? true,
OPENAI_API_BASE_URLS: newUrls,
OPENAI_API_KEYS: newKeys,
OPENAI_API_CONFIGS: newConfigs
});
};
// ─── Terminal Server Connections ────────────────────────
/**
* Add a terminal server connection at the system level.
* Mirrors the logic in admin/Settings/Integrations.svelte.
*/
export const addTerminalConnection = async (
token: string,
connection: { url: string; key?: string; name?: string; auth_type?: string }
) => {
const current = await getTerminalServerConnections(token);
const servers = current?.TERMINAL_SERVER_CONNECTIONS ?? [];
// Don't add duplicates
if (servers.find((s: any) => s.url === connection.url)) {
return current;
}
servers.push({
url: connection.url,
key: connection.key ?? '',
auth_type: connection.auth_type ?? 'bearer',
name: connection.name ?? 'Open Terminal',
enabled: true
});
return await setTerminalServerConnections(token, {
TERMINAL_SERVER_CONNECTIONS: servers
});
};
/**
* Remove a terminal server connection by URL at the system level.
*/
export const removeTerminalConnection = async (token: string, url: string) => {
const current = await getTerminalServerConnections(token);
const servers = current?.TERMINAL_SERVER_CONNECTIONS ?? [];
const filtered = servers.filter((s: any) => s.url !== url);
if (filtered.length === servers.length) return current; // nothing to remove
return await setTerminalServerConnections(token, {
TERMINAL_SERVER_CONNECTIONS: filtered
});
};
|