* feat: Add shimmer text animation for processing state indicators * feat: Redesign CollapsibleContentBlock component with improved UX * feat: Add conditional setting display support with dependsOn field * feat: Add showAgenticTurnStats setting for per-turn statistics * feat: Update ChatMessageAgenticContent with improved UI and new features * feat: Enhance file read tool UI/UX * feat: Refine styling of collapsible content and code preview blocks * feat: add terminal variant to CollapsibleContentBlock * feat: add built-in tools UI registry * feat: extract ChatMessageReasoningBlock and ChatMessageToolCallBlock * refactor: simplify ChatMessageAgenticContent to use extracted blocks * fix: correct markdown content block margin spacing * fix: reorganize SettingsChatFields layout and reset button positioning * fix: use direct map access in agentic store session methods * refactor: remove reasoning preview/throttle system from CollapsibleContentBlock * feat: add auto-scroll to reasoning block and remove showThoughtInProgress * feat: add ChatMessageToolCallDateTime component and support for new tool types * feat: improve auto-scroll reliability in reasoning block with RAF coalescing and MutationObserver * feat: show MCP server favicon for tools without a built-in icon * feat: add search-results parsing utilities and tests * feat: add ChatMessageToolCallSearchResults component * feat: integrate search results rendering into ChatMessageAgenticContent * feat: display tool call input alongside output in ChatMessageToolCallBlock * style: use muted foreground color in reasoning block content * chore: Format * feat: Refine reasoning block layout and make pending thoughts display configurable * feat: Stream tool call code blocks with auto-scroll and handle partial JSON * feat: add streaming permission gate infrastructure * feat: wire permission gate into the agentic loop * fix: bail out on abort and skip already-approved tool calls * fix: clear partial tool calls on abort and savePartialResponse * test: cover partial tool call cleanup end-to-end * refactor: Remove streaming permission gate logic * fix: Correct autoscroll and streaming gates for tool calls and reasoning blocks * refactor: Chat Message Assistant componentization * fix: Show health metadata for disabled MCP servers and promote connections on enable * fix: Inherit global enabled state for missing MCP per-chat overrides * refactor: Cleanup * refactor: Split ChatMessageToolCallBlock into dedicated components * feat: Add live streaming and auto-scroll for tool execution output * feat: Add line numbers and change markers to file edit diffs * chore: Formatting * feat: Add type definitions and utilities for recommended MCP servers * feat: Add recommended MCP servers configuration and storage key * feat: Add McpServerCardCompact component for recommended servers * feat: Add recommended servers section to Add New Server dialog * feat: Update McpServerForm to support authorization requirements * feat: Add select-none classes for text selection prevention * feat: Add recommended MCP server icon assets * refactor: Store dismissed MCP recommendations as a boolean flag * feat: Render tool results as JSON or Markdown based on detected content type * feat: UI improvement * feat: Render search block early and update heading to show execution state * fix: Prevent non-web-search tools from triggering the search UI block * refactor: Cleanup * refactor: Extract hardcoded icon size classes into shared constants * refactor: Extract hardcoded tool result separator into a shared constant * refactor: Tool Calls UI/logic * refactor: Cleanup * refactor: Cleanup * refactor: Cleanup
741 lines
23 KiB
TypeScript
741 lines
23 KiB
TypeScript
/**
|
|
* Migration Service - Unified data migration hook
|
|
*
|
|
* Centralizes all data migrations (localStorage, IndexedDB, legacy formats) into a single
|
|
* initialization point. Each migration copies data to new format WITHOUT deleting the old.
|
|
*
|
|
* **Architecture:**
|
|
* - Migrations are defined as objects with `id` and `run()` methods
|
|
* - Migration state is tracked in localStorage to avoid re-running
|
|
* - `runAllMigrations()` should be called once at app startup
|
|
* - All migrations are NON-DESTRUCTIVE - legacy data is preserved for downgrade compatibility
|
|
*
|
|
* **Current Migrations:**
|
|
* 1. localStorage prefix: Copy LlamaCppWebui.* → LlamaUi.* (both preserved)
|
|
* 2. IndexedDB database: Copy LlamacppWebui → LlamaUi (both preserved)
|
|
* 3. Legacy message format: Transform in-place (preserves structure, migrates markers)
|
|
* 4. Theme key: Copy standalone `theme` → config object (both preserved)
|
|
*/
|
|
|
|
import Dexie from 'dexie';
|
|
import {
|
|
STORAGE_APP_NAME,
|
|
STORAGE_APP_NAME_DEPRECATED,
|
|
DB_APP_NAME_DEPRECATED,
|
|
CONFIG_LOCALSTORAGE_KEY,
|
|
IDXDB_TABLES,
|
|
IDXDB_STORES,
|
|
NEW_TO_DEPRECATED_MAP
|
|
} from '$lib/constants';
|
|
import { LEGACY_AGENTIC_REGEX, LEGACY_REASONING_TAGS } from '$lib/constants/agentic';
|
|
import { SETTINGS_KEYS } from '$lib/constants/settings-keys';
|
|
import { MessageRole } from '$lib/enums';
|
|
|
|
// Types
|
|
|
|
interface Migration {
|
|
/** Unique identifier for this migration */
|
|
id: string;
|
|
/** Human-readable description */
|
|
description: string;
|
|
/** Run the migration forward (non-destructive - copies, doesn't delete) */
|
|
run(): Promise<void>;
|
|
}
|
|
|
|
interface MigrationState {
|
|
completed: string[];
|
|
failed: string[];
|
|
lastRun: string;
|
|
}
|
|
|
|
// Constants
|
|
|
|
const MIGRATION_STATE_KEY = `${STORAGE_APP_NAME}.migration-state`;
|
|
const MIGRATION_STATE_VERSION = 1;
|
|
|
|
// State Management
|
|
|
|
function getMigrationState(): MigrationState {
|
|
try {
|
|
const raw = localStorage.getItem(MIGRATION_STATE_KEY);
|
|
if (!raw) return { completed: [], failed: [], lastRun: '' };
|
|
const parsed = JSON.parse(raw);
|
|
if (parsed.version !== MIGRATION_STATE_VERSION) {
|
|
return { completed: [], failed: [], lastRun: '' };
|
|
}
|
|
return {
|
|
completed: parsed.completed ?? [],
|
|
failed: parsed.failed ?? [],
|
|
lastRun: parsed.lastRun ?? ''
|
|
};
|
|
} catch {
|
|
return { completed: [], failed: [], lastRun: '' };
|
|
}
|
|
}
|
|
|
|
function saveMigrationState(state: MigrationState): void {
|
|
localStorage.setItem(
|
|
MIGRATION_STATE_KEY,
|
|
JSON.stringify({
|
|
version: MIGRATION_STATE_VERSION,
|
|
...state,
|
|
lastRun: new Date().toISOString()
|
|
})
|
|
);
|
|
}
|
|
|
|
function isMigrationCompleted(id: string): boolean {
|
|
const state = getMigrationState();
|
|
return state.completed.includes(id);
|
|
}
|
|
|
|
function markMigrationCompleted(id: string): void {
|
|
const state = getMigrationState();
|
|
if (!state.completed.includes(id)) {
|
|
state.completed.push(id);
|
|
}
|
|
state.failed = state.failed.filter((f) => f !== id);
|
|
saveMigrationState(state);
|
|
}
|
|
|
|
function markMigrationFailed(id: string): void {
|
|
const state = getMigrationState();
|
|
if (!state.failed.includes(id)) {
|
|
state.failed.push(id);
|
|
}
|
|
saveMigrationState(state);
|
|
}
|
|
|
|
// Migration 1: LocalStorage Key Prefix (Non-Destructive)
|
|
|
|
const LOCALSTORAGE_MIGRATION_ID = 'localstorage-prefix-v1';
|
|
|
|
const localStorageMigration: Migration = {
|
|
id: LOCALSTORAGE_MIGRATION_ID,
|
|
description: 'Copy localStorage keys from LlamaCppWebui to LlamaUi prefix (non-destructive)',
|
|
|
|
async run(): Promise<void> {
|
|
// Non-destructive: copy to new key, but KEEP the old key
|
|
for (const [newKey, deprecatedKey] of Object.entries(NEW_TO_DEPRECATED_MAP)) {
|
|
// Only migrate if new key doesn't already exist
|
|
const newValue = localStorage.getItem(newKey);
|
|
if (newValue !== null) {
|
|
if (import.meta.env.DEV && import.meta.env.VITE_DEBUG)
|
|
console.log(`[Migration] localStorage: ${newKey} already exists, skipping`);
|
|
continue;
|
|
}
|
|
|
|
const oldValue = localStorage.getItem(deprecatedKey);
|
|
if (oldValue !== null) {
|
|
localStorage.setItem(newKey, oldValue);
|
|
// Keep old key for downgrade compatibility - DO NOT DELETE
|
|
if (import.meta.env.DEV && import.meta.env.VITE_DEBUG) {
|
|
console.log(
|
|
`[Migration] localStorage: copied ${deprecatedKey} → ${newKey} (preserved old)`
|
|
);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
};
|
|
|
|
// Migration 2: IndexedDB Database Name (Non-Destructive)
|
|
|
|
const IDXDB_MIGRATION_ID = 'idxdb-database-v1';
|
|
|
|
const idxdbMigration: Migration = {
|
|
id: IDXDB_MIGRATION_ID,
|
|
description: 'Copy IndexedDB from LlamacppWebui to LlamaUi database (non-destructive)',
|
|
|
|
async run(): Promise<void> {
|
|
const oldDbNames = await Dexie.getDatabaseNames();
|
|
if (!oldDbNames.includes(DB_APP_NAME_DEPRECATED)) {
|
|
if (import.meta.env.DEV && import.meta.env.VITE_DEBUG)
|
|
console.log('[Migration] IndexedDB: no old database found, skipping');
|
|
return;
|
|
}
|
|
|
|
// Check if new database already has data
|
|
const newDb = new Dexie(STORAGE_APP_NAME);
|
|
newDb.version(1).stores(IDXDB_STORES);
|
|
const existingConvs = await newDb.table(IDXDB_TABLES.conversations).count();
|
|
if (existingConvs > 0) {
|
|
if (import.meta.env.DEV && import.meta.env.VITE_DEBUG)
|
|
console.log('[Migration] IndexedDB: new database already has data, skipping');
|
|
return;
|
|
}
|
|
|
|
if (import.meta.env.DEV && import.meta.env.VITE_DEBUG)
|
|
console.log('[Migration] IndexedDB: copying from', DB_APP_NAME_DEPRECATED);
|
|
|
|
const oldDb = new Dexie(DB_APP_NAME_DEPRECATED);
|
|
oldDb.version(1).stores(IDXDB_STORES);
|
|
|
|
const conversations = await oldDb.table(IDXDB_TABLES.conversations).toArray();
|
|
const messages = await oldDb.table(IDXDB_TABLES.messages).toArray();
|
|
|
|
if (conversations.length > 0) {
|
|
await newDb.table(IDXDB_TABLES.conversations).bulkAdd(conversations);
|
|
if (import.meta.env.DEV && import.meta.env.VITE_DEBUG)
|
|
console.log(`[Migration] IndexedDB: copied ${conversations.length} conversations`);
|
|
}
|
|
if (messages.length > 0) {
|
|
await newDb.table(IDXDB_TABLES.messages).bulkAdd(messages);
|
|
if (import.meta.env.DEV && import.meta.env.VITE_DEBUG)
|
|
console.log(`[Migration] IndexedDB: copied ${messages.length} messages`);
|
|
}
|
|
|
|
// Non-destructive: DO NOT delete old database - keep for downgrade compatibility
|
|
if (import.meta.env.DEV && import.meta.env.VITE_DEBUG)
|
|
console.log('[Migration] IndexedDB: preserved old database for downgrade compatibility');
|
|
}
|
|
};
|
|
|
|
// Migration 3: Legacy Message Format
|
|
|
|
const LEGACY_MESSAGE_MIGRATION_ID = 'legacy-message-format-v2';
|
|
|
|
interface ParsedTurn {
|
|
textBefore: string;
|
|
toolCalls: Array<{ name: string; args: string; result: string }>;
|
|
}
|
|
|
|
function parseLegacyToolCalls(content: string): ParsedTurn[] {
|
|
const turns: ParsedTurn[] = [];
|
|
const regex = new RegExp(LEGACY_AGENTIC_REGEX.COMPLETED_TOOL_CALL.source, 'g');
|
|
|
|
let lastIndex = 0;
|
|
let currentTurn: ParsedTurn = { textBefore: '', toolCalls: [] };
|
|
let match;
|
|
|
|
while ((match = regex.exec(content)) !== null) {
|
|
const textBefore = content.slice(lastIndex, match.index).trim();
|
|
|
|
if (textBefore && currentTurn.toolCalls.length > 0) {
|
|
turns.push(currentTurn);
|
|
currentTurn = { textBefore, toolCalls: [] };
|
|
} else if (textBefore && currentTurn.toolCalls.length === 0) {
|
|
currentTurn.textBefore = textBefore;
|
|
}
|
|
|
|
currentTurn.toolCalls.push({
|
|
name: match[1],
|
|
args: match[2],
|
|
result: match[3].replace(/^\n+|\n+$/g, '')
|
|
});
|
|
|
|
lastIndex = match.index + match[0].length;
|
|
}
|
|
|
|
const remainingText = content.slice(lastIndex).trim();
|
|
|
|
if (currentTurn.toolCalls.length > 0) {
|
|
turns.push(currentTurn);
|
|
}
|
|
|
|
if (remainingText) {
|
|
const cleanRemaining = remainingText
|
|
.replace(LEGACY_AGENTIC_REGEX.AGENTIC_TOOL_CALL_OPEN, '')
|
|
.trim();
|
|
if (cleanRemaining) {
|
|
turns.push({ textBefore: cleanRemaining, toolCalls: [] });
|
|
}
|
|
}
|
|
|
|
if (turns.length === 0) {
|
|
turns.push({ textBefore: content.trim(), toolCalls: [] });
|
|
}
|
|
|
|
return turns;
|
|
}
|
|
|
|
function extractLegacyReasoning(content: string): { reasoning: string; cleanContent: string } {
|
|
let reasoning = '';
|
|
let cleanContent = content;
|
|
|
|
const re = new RegExp(LEGACY_AGENTIC_REGEX.REASONING_EXTRACT.source, 'g');
|
|
let match;
|
|
while ((match = re.exec(content)) !== null) {
|
|
reasoning += match[1];
|
|
}
|
|
|
|
cleanContent = cleanContent
|
|
.replace(new RegExp(LEGACY_AGENTIC_REGEX.REASONING_BLOCK.source, 'g'), '')
|
|
.replace(LEGACY_AGENTIC_REGEX.REASONING_OPEN, '');
|
|
|
|
return { reasoning, cleanContent };
|
|
}
|
|
|
|
function hasLegacyMarkers(content: string): boolean {
|
|
return LEGACY_AGENTIC_REGEX.HAS_LEGACY_MARKERS.test(content);
|
|
}
|
|
|
|
let DatabaseService: typeof import('./database.service').DatabaseService | null = null;
|
|
|
|
async function getDatabaseService() {
|
|
if (!DatabaseService) {
|
|
const module = await import('./database.service');
|
|
DatabaseService = module.DatabaseService;
|
|
}
|
|
return DatabaseService;
|
|
}
|
|
|
|
const legacyMessageMigration: Migration = {
|
|
id: LEGACY_MESSAGE_MIGRATION_ID,
|
|
description: 'Migrate legacy marker-based messages to structured format',
|
|
|
|
async run(): Promise<void> {
|
|
const db = await getDatabaseService();
|
|
const conversations = await db.getAllConversations();
|
|
let migratedCount = 0;
|
|
|
|
for (const conv of conversations) {
|
|
const allMessages = await db.getConversationMessages(conv.id);
|
|
|
|
for (const message of allMessages) {
|
|
if (message.role !== MessageRole.ASSISTANT) {
|
|
if (message.content?.includes(LEGACY_REASONING_TAGS.START)) {
|
|
const { reasoning, cleanContent } = extractLegacyReasoning(message.content);
|
|
await db.updateMessage(message.id, {
|
|
content: cleanContent.trim(),
|
|
reasoningContent: reasoning || undefined
|
|
});
|
|
migratedCount++;
|
|
}
|
|
continue;
|
|
}
|
|
|
|
if (!hasLegacyMarkers(message.content ?? '')) continue;
|
|
|
|
const { reasoning, cleanContent } = extractLegacyReasoning(message.content);
|
|
const turns = parseLegacyToolCalls(cleanContent);
|
|
|
|
let existingToolCalls: Array<{
|
|
id: string;
|
|
function?: { name: string; arguments: string };
|
|
}> = [];
|
|
if (message.toolCalls) {
|
|
try {
|
|
existingToolCalls = JSON.parse(message.toolCalls);
|
|
} catch {
|
|
// Ignore
|
|
}
|
|
}
|
|
|
|
const firstTurn = turns[0];
|
|
if (!firstTurn) continue;
|
|
|
|
const firstTurnToolCalls = firstTurn.toolCalls.map((tc, i) => {
|
|
const existing =
|
|
existingToolCalls.find((e) => e.function?.name === tc.name) || existingToolCalls[i];
|
|
return {
|
|
id: existing?.id || `legacy_tool_${i}`,
|
|
type: 'function' as const,
|
|
function: { name: tc.name, arguments: tc.args }
|
|
};
|
|
});
|
|
|
|
await db.updateMessage(message.id, {
|
|
content: firstTurn.textBefore,
|
|
reasoningContent: reasoning || undefined,
|
|
toolCalls: firstTurnToolCalls.length > 0 ? JSON.stringify(firstTurnToolCalls) : ''
|
|
});
|
|
|
|
let currentParentId = message.id;
|
|
let toolCallIdCounter = existingToolCalls.length;
|
|
|
|
for (let i = 0; i < firstTurn.toolCalls.length; i++) {
|
|
const tc = firstTurn.toolCalls[i];
|
|
const toolCallId = firstTurnToolCalls[i]?.id || `legacy_tool_${i}`;
|
|
|
|
const toolMsg = await db.createMessageBranch(
|
|
{
|
|
convId: conv.id,
|
|
type: 'text',
|
|
role: MessageRole.TOOL,
|
|
content: tc.result,
|
|
toolCallId,
|
|
timestamp: message.timestamp + i + 1,
|
|
toolCalls: '',
|
|
children: []
|
|
},
|
|
currentParentId
|
|
);
|
|
currentParentId = toolMsg.id;
|
|
}
|
|
|
|
for (let turnIdx = 1; turnIdx < turns.length; turnIdx++) {
|
|
const turn = turns[turnIdx];
|
|
|
|
const turnToolCalls = turn.toolCalls.map((tc, i) => {
|
|
const idx = toolCallIdCounter + i;
|
|
const existing = existingToolCalls[idx];
|
|
return {
|
|
id: existing?.id || `legacy_tool_${idx}`,
|
|
type: 'function' as const,
|
|
function: { name: tc.name, arguments: tc.args }
|
|
};
|
|
});
|
|
toolCallIdCounter += turn.toolCalls.length;
|
|
|
|
const assistantMsg = await db.createMessageBranch(
|
|
{
|
|
convId: conv.id,
|
|
type: 'text',
|
|
role: MessageRole.ASSISTANT,
|
|
content: turn.textBefore,
|
|
timestamp: message.timestamp + turnIdx * 100,
|
|
toolCalls: turnToolCalls.length > 0 ? JSON.stringify(turnToolCalls) : '',
|
|
children: [],
|
|
model: message.model
|
|
},
|
|
currentParentId
|
|
);
|
|
currentParentId = assistantMsg.id;
|
|
|
|
for (let i = 0; i < turn.toolCalls.length; i++) {
|
|
const tc = turn.toolCalls[i];
|
|
const toolCallId = turnToolCalls[i]?.id || `legacy_tool_${toolCallIdCounter + i}`;
|
|
|
|
const toolMsg = await db.createMessageBranch(
|
|
{
|
|
convId: conv.id,
|
|
type: 'text',
|
|
role: MessageRole.TOOL,
|
|
content: tc.result,
|
|
toolCallId,
|
|
timestamp: message.timestamp + turnIdx * 100 + i + 1,
|
|
toolCalls: '',
|
|
children: []
|
|
},
|
|
currentParentId
|
|
);
|
|
currentParentId = toolMsg.id;
|
|
}
|
|
}
|
|
|
|
if (message.children.length > 0 && currentParentId !== message.id) {
|
|
for (const childId of message.children) {
|
|
const child = allMessages.find((m) => m.id === childId);
|
|
if (!child) continue;
|
|
if (child.role !== MessageRole.TOOL) {
|
|
await db.updateMessage(childId, { parent: currentParentId });
|
|
}
|
|
}
|
|
await db.updateMessage(message.id, { children: [] });
|
|
}
|
|
|
|
migratedCount++;
|
|
}
|
|
}
|
|
|
|
if (import.meta.env.DEV && import.meta.env.VITE_DEBUG)
|
|
console.log(`[Migration] Legacy messages: migrated ${migratedCount} messages`);
|
|
}
|
|
};
|
|
|
|
// Migration 4: Theme Key (Non-Destructive)
|
|
|
|
const THEME_MIGRATION_ID = 'theme-key-v1';
|
|
|
|
const themeMigration: Migration = {
|
|
id: THEME_MIGRATION_ID,
|
|
description: 'Copy standalone theme key to config object (non-destructive)',
|
|
|
|
async run(): Promise<void> {
|
|
const legacyTheme = localStorage.getItem('theme');
|
|
if (legacyTheme === null) {
|
|
if (import.meta.env.DEV && import.meta.env.VITE_DEBUG)
|
|
console.log('[Migration] Theme: no legacy theme key found, skipping');
|
|
return;
|
|
}
|
|
|
|
// Check if config already has theme
|
|
const configRaw = localStorage.getItem(CONFIG_LOCALSTORAGE_KEY);
|
|
const config = configRaw ? JSON.parse(configRaw) : {};
|
|
|
|
if (SETTINGS_KEYS.THEME in config) {
|
|
if (import.meta.env.DEV && import.meta.env.VITE_DEBUG)
|
|
console.log('[Migration] Theme: config already has theme, skipping');
|
|
return;
|
|
}
|
|
|
|
config[SETTINGS_KEYS.THEME] = legacyTheme;
|
|
localStorage.setItem(CONFIG_LOCALSTORAGE_KEY, JSON.stringify(config));
|
|
|
|
// Non-destructive: DO NOT delete legacy theme key - keep for downgrade compatibility
|
|
if (import.meta.env.DEV && import.meta.env.VITE_DEBUG)
|
|
console.log(`[Migration] Theme: copied standalone theme to config (preserved old key)`);
|
|
}
|
|
};
|
|
|
|
// Migration Registry & Runner
|
|
|
|
const CUSTOM_JSON_MIGRATION_ID = 'custom-json-key-v1';
|
|
|
|
const customJsonKeyMigration: Migration = {
|
|
id: CUSTOM_JSON_MIGRATION_ID,
|
|
description: 'Copy legacy custom config key to customJson (non-destructive)',
|
|
|
|
async run(): Promise<void> {
|
|
const configRaw = localStorage.getItem(CONFIG_LOCALSTORAGE_KEY);
|
|
if (configRaw === null) return;
|
|
|
|
const config = JSON.parse(configRaw);
|
|
|
|
if (!('custom' in config)) return;
|
|
if (SETTINGS_KEYS.CUSTOM_JSON in config) return;
|
|
|
|
config[SETTINGS_KEYS.CUSTOM_JSON] = config.custom;
|
|
localStorage.setItem(CONFIG_LOCALSTORAGE_KEY, JSON.stringify(config));
|
|
|
|
// Non-destructive: keep the legacy custom key for downgrade compatibility
|
|
if (import.meta.env.DEV && import.meta.env.VITE_DEBUG)
|
|
console.log(`[Migration] Custom JSON: copied custom to customJson (preserved old key)`);
|
|
}
|
|
};
|
|
|
|
const MCP_DEFAULT_ENABLED_MIGRATION_ID = 'mcp-default-enabled-to-config-v1';
|
|
|
|
const LEGACY_MCP_DEFAULT_ENABLED_KEY = `${STORAGE_APP_NAME}.mcpDefaultEnabled`;
|
|
const DEPRECATED_LEGACY_MCP_DEFAULT_ENABLED_KEY = `${STORAGE_APP_NAME_DEPRECATED}.mcpDefaultEnabled`;
|
|
|
|
const mcpDefaultEnabledMigration: Migration = {
|
|
id: MCP_DEFAULT_ENABLED_MIGRATION_ID,
|
|
description:
|
|
'Copy mcpDefaultEnabled localStorage key into settings config (preserves legacy keys)',
|
|
|
|
async run(): Promise<void> {
|
|
const raw =
|
|
localStorage.getItem(LEGACY_MCP_DEFAULT_ENABLED_KEY) ??
|
|
localStorage.getItem(DEPRECATED_LEGACY_MCP_DEFAULT_ENABLED_KEY);
|
|
|
|
// Legacy keys intentionally left in place so a downgrade keeps reading them.
|
|
|
|
if (raw === null) {
|
|
if (import.meta.env.DEV && import.meta.env.VITE_DEBUG)
|
|
console.log('[Migration] MCP default enabled: no legacy key found, skipping');
|
|
return;
|
|
}
|
|
|
|
const configRaw = localStorage.getItem(CONFIG_LOCALSTORAGE_KEY);
|
|
const config = configRaw ? JSON.parse(configRaw) : {};
|
|
|
|
// Don't overwrite an existing config entry — current data wins.
|
|
if (MCP_DEFAULT_OVERRIDES_LEGACY_KEY in config) {
|
|
if (import.meta.env.DEV && import.meta.env.VITE_DEBUG)
|
|
console.log('[Migration] MCP default enabled: config already has overrides, skipping');
|
|
return;
|
|
}
|
|
|
|
try {
|
|
const parsed = JSON.parse(raw);
|
|
if (!Array.isArray(parsed)) return;
|
|
const valid = parsed.every(
|
|
(o) =>
|
|
typeof o === 'object' &&
|
|
o !== null &&
|
|
typeof (o as Record<string, unknown>).serverId === 'string' &&
|
|
typeof (o as Record<string, unknown>).enabled === 'boolean'
|
|
);
|
|
if (!valid) return;
|
|
} catch {
|
|
return;
|
|
}
|
|
|
|
config[MCP_DEFAULT_OVERRIDES_LEGACY_KEY] = raw;
|
|
localStorage.setItem(CONFIG_LOCALSTORAGE_KEY, JSON.stringify(config));
|
|
|
|
if (import.meta.env.DEV && import.meta.env.VITE_DEBUG)
|
|
console.log('[Migration] MCP default enabled: moved legacy key into config');
|
|
}
|
|
};
|
|
|
|
const CONFIG_TYPES_MIGRATION_ID = 'config-type-normalization-v1';
|
|
|
|
const configTypesMigration: Migration = {
|
|
id: CONFIG_TYPES_MIGRATION_ID,
|
|
description: 'Coerce legacy string-encoded booleans in persisted config to real booleans',
|
|
|
|
async run(): Promise<void> {
|
|
const configRaw = localStorage.getItem(CONFIG_LOCALSTORAGE_KEY);
|
|
if (configRaw === null) return;
|
|
|
|
const config = JSON.parse(configRaw);
|
|
let changed = false;
|
|
|
|
// Pre-schema configs persisted booleans as "true"/"false" strings; the strict server
|
|
// schema rejects them. No config string field holds exactly "true"/"false", so the
|
|
// match is unambiguous.
|
|
for (const key of Object.keys(config)) {
|
|
if (config[key] === 'true') {
|
|
config[key] = true;
|
|
changed = true;
|
|
} else if (config[key] === 'false') {
|
|
config[key] = false;
|
|
changed = true;
|
|
}
|
|
}
|
|
|
|
if (changed) {
|
|
localStorage.setItem(CONFIG_LOCALSTORAGE_KEY, JSON.stringify(config));
|
|
}
|
|
|
|
if (import.meta.env.DEV && import.meta.env.VITE_DEBUG)
|
|
console.log(`[Migration] Config types: coerced string booleans (changed=${changed})`);
|
|
}
|
|
};
|
|
|
|
const MCP_DEFAULT_OVERRIDES_LEGACY_KEY = `${STORAGE_APP_NAME}.mcpDefaultServerOverrides`;
|
|
const MCP_DEFAULT_OVERRIDES_MERGE_MIGRATION_ID = 'mcp-default-overrides-merge-v1';
|
|
|
|
/**
|
|
* Folds `mcpDefaultServerOverrides` (the legacy "default for new chats" list,
|
|
* JSON-encoded as `[{ serverId, enabled }, ...]`) into `mcpServers[i].enabled`.
|
|
* The legacy override key is intentionally left in the config so a downgrade
|
|
* keeps reading it. Runs after `mcpDefaultEnabledMigration` so any legacy
|
|
* standalone overrides are already inside the config.
|
|
*/
|
|
const mcpDefaultOverridesMergeMigration: Migration = {
|
|
id: MCP_DEFAULT_OVERRIDES_MERGE_MIGRATION_ID,
|
|
description:
|
|
'Merge mcpDefaultServerOverrides entries onto mcpServers[i].enabled (preserves legacy key)',
|
|
|
|
async run(): Promise<void> {
|
|
const configRaw = localStorage.getItem(CONFIG_LOCALSTORAGE_KEY);
|
|
if (configRaw === null) return;
|
|
|
|
const config = JSON.parse(configRaw);
|
|
const raw = config[MCP_DEFAULT_OVERRIDES_LEGACY_KEY];
|
|
|
|
if (typeof raw !== 'string' || raw.length === 0) {
|
|
if (import.meta.env.DEV && import.meta.env.VITE_DEBUG)
|
|
console.log('[Migration] MCP default overrides merge: nothing to merge');
|
|
return;
|
|
}
|
|
|
|
let overrides: { serverId: string; enabled: boolean }[];
|
|
try {
|
|
const parsed = JSON.parse(raw);
|
|
if (!Array.isArray(parsed)) return;
|
|
overrides = parsed.filter(
|
|
(o) =>
|
|
typeof o === 'object' &&
|
|
o !== null &&
|
|
typeof (o as Record<string, unknown>).serverId === 'string' &&
|
|
typeof (o as Record<string, unknown>).enabled === 'boolean'
|
|
) as { serverId: string; enabled: boolean }[];
|
|
} catch {
|
|
return;
|
|
}
|
|
|
|
const serversRaw = config[SETTINGS_KEYS.MCP_SERVERS];
|
|
let servers: { id: string; enabled?: boolean }[];
|
|
try {
|
|
servers = typeof serversRaw === 'string' ? JSON.parse(serversRaw) : [];
|
|
} catch {
|
|
return;
|
|
}
|
|
|
|
if (!Array.isArray(servers)) servers = [];
|
|
|
|
let serversChanged = false;
|
|
const knownIds = new Set(servers.map((s) => s.id));
|
|
for (const override of overrides) {
|
|
if (!knownIds.has(override.serverId)) continue;
|
|
const index = servers.findIndex((s) => s.id === override.serverId);
|
|
|
|
if (index >= 0 && servers[index].enabled !== override.enabled) {
|
|
servers[index] = { ...servers[index], enabled: override.enabled };
|
|
serversChanged = true;
|
|
}
|
|
}
|
|
|
|
if (serversChanged) {
|
|
config[SETTINGS_KEYS.MCP_SERVERS] = JSON.stringify(servers);
|
|
localStorage.setItem(CONFIG_LOCALSTORAGE_KEY, JSON.stringify(config));
|
|
}
|
|
|
|
if (import.meta.env.DEV && import.meta.env.VITE_DEBUG)
|
|
console.log(
|
|
`[Migration] MCP default overrides merge: applied=${overrides.length} serversChanged=${serversChanged} (legacy key preserved)`
|
|
);
|
|
}
|
|
};
|
|
|
|
const migrations: Migration[] = [
|
|
localStorageMigration,
|
|
idxdbMigration,
|
|
legacyMessageMigration,
|
|
themeMigration,
|
|
customJsonKeyMigration,
|
|
mcpDefaultEnabledMigration,
|
|
mcpDefaultOverridesMergeMigration,
|
|
configTypesMigration
|
|
];
|
|
|
|
export const MigrationService = {
|
|
/**
|
|
* Get all registered migrations
|
|
*/
|
|
getMigrations(): Migration[] {
|
|
return [...migrations];
|
|
},
|
|
|
|
/**
|
|
* Check if a specific migration has been completed
|
|
*/
|
|
isCompleted(id: string): boolean {
|
|
return isMigrationCompleted(id);
|
|
},
|
|
|
|
/**
|
|
* Get current migration state
|
|
*/
|
|
getState(): MigrationState {
|
|
return getMigrationState();
|
|
},
|
|
|
|
/**
|
|
* Reset migration state (use with caution - migrations will run again)
|
|
*/
|
|
resetState(): void {
|
|
localStorage.removeItem(MIGRATION_STATE_KEY);
|
|
if (import.meta.env.DEV && import.meta.env.VITE_DEBUG)
|
|
console.log('[Migration] State reset - all migrations will run again');
|
|
},
|
|
|
|
/**
|
|
* Run all pending migrations (non-destructive - preserves legacy data)
|
|
* Should be called once at app initialization
|
|
*/
|
|
async runAllMigrations(): Promise<void> {
|
|
const state = getMigrationState();
|
|
if (import.meta.env.DEV && import.meta.env.VITE_DEBUG)
|
|
console.log('[Migration] Starting migration run, state:', state);
|
|
|
|
for (const migration of migrations) {
|
|
if (isMigrationCompleted(migration.id)) {
|
|
if (import.meta.env.DEV && import.meta.env.VITE_DEBUG)
|
|
console.log(`[Migration] ${migration.id}: already completed, skipping`);
|
|
continue;
|
|
}
|
|
|
|
try {
|
|
if (import.meta.env.DEV && import.meta.env.VITE_DEBUG)
|
|
console.log(`[Migration] ${migration.id}: running...`);
|
|
await migration.run();
|
|
markMigrationCompleted(migration.id);
|
|
if (import.meta.env.DEV && import.meta.env.VITE_DEBUG)
|
|
console.log(`[Migration] ${migration.id}: completed successfully`);
|
|
} catch (error) {
|
|
console.error(`[Migration] ${migration.id}: failed`, error);
|
|
markMigrationFailed(migration.id);
|
|
}
|
|
}
|
|
|
|
if (import.meta.env.DEV && import.meta.env.VITE_DEBUG)
|
|
console.log('[Migration] All migrations complete');
|
|
}
|
|
};
|