mirror of
https://github.com/siteboon/claudecodeui.git
synced 2026-05-28 23:15:33 +08:00
Model selection was acting like a provider-level preference. That made resumed sessions drift back to a default or request-time model. Users expect /models changes made inside a conversation to affect that session. Store explicit session choices in app-owned ~/.cloudcli state. This avoids editing provider transcripts or native provider config. Resolve the effective model before launching each provider runtime. Claude, Cursor, Codex, Gemini, and OpenCode now honor stored resume choices. Expose a backend active-model change endpoint for existing sessions. The models modal can now distinguish default changes from session overrides. It also shows when a selected model will apply on the next response. For Claude, stop probing active model state by resuming with a dummy prompt. Read the indexed JSONL transcript from the end instead. This preserves provider history while honoring /model stdout or model fields. Add service tests for adapter delegation and resume-model precedence. The tests keep cache state, override state, and requested fallback separate.
1148 lines
35 KiB
TypeScript
1148 lines
35 KiB
TypeScript
import { randomUUID } from 'node:crypto';
|
|
import fs from 'node:fs';
|
|
import {
|
|
access,
|
|
lstat,
|
|
mkdir,
|
|
readFile,
|
|
readdir,
|
|
readlink,
|
|
realpath,
|
|
stat,
|
|
writeFile,
|
|
} from 'node:fs/promises';
|
|
import os from 'node:os';
|
|
import path from 'node:path';
|
|
import readline from 'node:readline';
|
|
|
|
import type { NextFunction, Request, RequestHandler, Response } from 'express';
|
|
|
|
import { parseFrontMatter } from '@/shared/frontmatter.js';
|
|
import type {
|
|
AnyRecord,
|
|
ApiSuccessShape,
|
|
AppErrorOptions,
|
|
LLMProvider,
|
|
NormalizedMessage,
|
|
ProviderChangeActiveModelInput,
|
|
ProviderCurrentActiveModel,
|
|
ProviderModelsDefinition,
|
|
ProviderSessionActiveModelChange,
|
|
ProviderSkillSource,
|
|
WorkspacePathValidationResult,
|
|
} from '@/shared/types.js';
|
|
|
|
//----------------- NORMALIZED MESSAGE HELPER INPUT TYPES ------------
|
|
/**
|
|
* Input payload accepted by `createNormalizedMessage`.
|
|
*
|
|
* Callers provide provider-specific fields plus the required `kind/provider`
|
|
* pair; this helper fills missing envelope fields (`id`, `sessionId`,
|
|
* `timestamp`) in a consistent way.
|
|
*/
|
|
type NormalizedMessageInput =
|
|
{
|
|
kind: NormalizedMessage['kind'];
|
|
provider: NormalizedMessage['provider'];
|
|
id?: string | null;
|
|
sessionId?: string | null;
|
|
timestamp?: string | null;
|
|
} & Record<string, unknown>;
|
|
|
|
// ---------------------------
|
|
//----------------- HTTP HANDLER UTILITIES ------------
|
|
/**
|
|
* Wraps arbitrary data in the standard API success envelope.
|
|
*
|
|
* Use this helper in route handlers to keep successful JSON responses consistent
|
|
* across endpoints.
|
|
*/
|
|
export function createApiSuccessResponse<TData>(
|
|
data: TData,
|
|
): ApiSuccessShape<TData> {
|
|
return {
|
|
success: true,
|
|
data,
|
|
};
|
|
}
|
|
|
|
/**
|
|
* Converts an async Express handler into a standard `RequestHandler` and routes
|
|
* rejected promises to Express error middleware.
|
|
*
|
|
* Use this to avoid repeating `try/catch(next)` in every async route.
|
|
*/
|
|
export function asyncHandler(
|
|
handler: (req: Request, res: Response, next: NextFunction) => Promise<unknown>
|
|
): RequestHandler {
|
|
return (req, res, next) => {
|
|
void Promise.resolve(handler(req, res, next)).catch(next);
|
|
};
|
|
}
|
|
|
|
// ---------------------------
|
|
//----------------- SHARED ERROR UTILITIES ------------
|
|
/**
|
|
* Shared application error with HTTP status and machine-readable code metadata.
|
|
*
|
|
* Throw this from service/route layers when the caller should receive a
|
|
* controlled error response rather than a generic 500.
|
|
*/
|
|
export class AppError extends Error {
|
|
readonly code: string;
|
|
readonly statusCode: number;
|
|
readonly details?: unknown;
|
|
|
|
constructor(message: string, options: AppErrorOptions = {}) {
|
|
super(message);
|
|
this.name = 'AppError';
|
|
this.code = options.code ?? 'INTERNAL_ERROR';
|
|
this.statusCode = options.statusCode ?? 500;
|
|
this.details = options.details;
|
|
}
|
|
}
|
|
|
|
// ---------------------------
|
|
//----------------- WORKSPACE PATH VALIDATION UTILITIES ------------
|
|
/**
|
|
* Root directory that all workspace/project paths must stay under.
|
|
*
|
|
* This is resolved from `WORKSPACES_ROOT` when configured; otherwise it falls
|
|
* back to the current user's home directory.
|
|
*/
|
|
export const WORKSPACES_ROOT = process.env.WORKSPACES_ROOT || os.homedir();
|
|
|
|
/**
|
|
* System-critical paths that must never be used as workspace roots.
|
|
*
|
|
* The validation helper blocks these values directly and also blocks paths
|
|
* nested under them (with explicit allow-list exceptions where necessary).
|
|
*/
|
|
export const FORBIDDEN_WORKSPACE_PATHS = [
|
|
// Unix
|
|
'/',
|
|
'/etc',
|
|
'/bin',
|
|
'/sbin',
|
|
'/usr',
|
|
'/dev',
|
|
'/proc',
|
|
'/sys',
|
|
'/var',
|
|
'/boot',
|
|
'/root',
|
|
'/lib',
|
|
'/lib64',
|
|
'/opt',
|
|
'/tmp',
|
|
'/run',
|
|
// Windows
|
|
'C:\\Windows',
|
|
'C:\\Program Files',
|
|
'C:\\Program Files (x86)',
|
|
'C:\\ProgramData',
|
|
'C:\\System Volume Information',
|
|
'C:\\$Recycle.Bin',
|
|
];
|
|
|
|
function stripWindowsLongPathPrefix(inputPath: string): string {
|
|
if (inputPath.startsWith('\\\\?\\UNC\\')) {
|
|
return `\\\\${inputPath.slice('\\\\?\\UNC\\'.length)}`;
|
|
}
|
|
|
|
if (inputPath.startsWith('\\\\?\\')) {
|
|
return inputPath.slice('\\\\?\\'.length);
|
|
}
|
|
|
|
return inputPath;
|
|
}
|
|
|
|
function shouldUseWindowsPathNormalization(inputPath: string): boolean {
|
|
if (process.platform === 'win32') {
|
|
return true;
|
|
}
|
|
|
|
return inputPath.startsWith('\\\\') || /^[a-zA-Z]:([\\/]|$)/.test(inputPath);
|
|
}
|
|
|
|
/**
|
|
* Canonicalizes project/workspace paths for stable DB keys and comparisons.
|
|
*
|
|
* Normalization rules:
|
|
* - trim whitespace
|
|
* - strip Windows long-path prefixes (`\\?\` and `\\?\UNC\`)
|
|
* - normalize path separators and dot segments
|
|
* - trim trailing separators except for filesystem roots
|
|
*/
|
|
export function normalizeProjectPath(inputPath: string): string {
|
|
if (typeof inputPath !== 'string') {
|
|
return '';
|
|
}
|
|
|
|
const trimmed = inputPath.trim();
|
|
if (!trimmed) {
|
|
return '';
|
|
}
|
|
|
|
const withoutLongPrefix = stripWindowsLongPathPrefix(trimmed);
|
|
const useWindowsPathRules = shouldUseWindowsPathNormalization(withoutLongPrefix);
|
|
const normalized = useWindowsPathRules
|
|
? path.win32.normalize(withoutLongPrefix)
|
|
: path.posix.normalize(withoutLongPrefix);
|
|
|
|
if (!normalized) {
|
|
return '';
|
|
}
|
|
|
|
const parser = useWindowsPathRules ? path.win32 : path.posix;
|
|
const root = parser.parse(normalized).root;
|
|
if (normalized === root) {
|
|
return normalized;
|
|
}
|
|
|
|
return normalized.replace(/[\\/]+$/, '');
|
|
}
|
|
|
|
/**
|
|
* Validates that a user-supplied workspace path is safe to use.
|
|
*
|
|
* Call this before any filesystem mutation that creates or registers projects.
|
|
* The function resolves symlinks, enforces `WORKSPACES_ROOT` containment, and
|
|
* blocks known system directories.
|
|
*/
|
|
export async function validateWorkspacePath(requestedPath: string): Promise<WorkspacePathValidationResult> {
|
|
try {
|
|
const normalizedRequestedPath = normalizeProjectPath(requestedPath);
|
|
if (!normalizedRequestedPath) {
|
|
return {
|
|
valid: false,
|
|
error: 'Workspace path is required',
|
|
};
|
|
}
|
|
|
|
const absolutePath = path.resolve(normalizedRequestedPath);
|
|
const normalizedPath = normalizeProjectPath(absolutePath);
|
|
|
|
if (FORBIDDEN_WORKSPACE_PATHS.includes(normalizedPath) || normalizedPath === '/') {
|
|
return {
|
|
valid: false,
|
|
error: 'Cannot use system-critical directories as workspace locations',
|
|
};
|
|
}
|
|
|
|
for (const forbiddenPath of FORBIDDEN_WORKSPACE_PATHS) {
|
|
const normalizedForbiddenPath = normalizeProjectPath(forbiddenPath);
|
|
if (
|
|
normalizedPath === normalizedForbiddenPath
|
|
|| normalizedPath.startsWith(`${normalizedForbiddenPath}${path.sep}`)
|
|
) {
|
|
// Allow specific user-writable folders under /var.
|
|
if (
|
|
normalizedForbiddenPath === '/var'
|
|
&& (normalizedPath.startsWith('/var/tmp') || normalizedPath.startsWith('/var/folders'))
|
|
) {
|
|
continue;
|
|
}
|
|
|
|
return {
|
|
valid: false,
|
|
error: `Cannot create workspace in system directory: ${forbiddenPath}`,
|
|
};
|
|
}
|
|
}
|
|
|
|
let resolvedPath = normalizeProjectPath(absolutePath);
|
|
try {
|
|
await access(absolutePath);
|
|
resolvedPath = normalizeProjectPath(await realpath(absolutePath));
|
|
} catch (error) {
|
|
const fileError = error as NodeJS.ErrnoException;
|
|
if (fileError.code !== 'ENOENT') {
|
|
throw fileError;
|
|
}
|
|
|
|
const parentPath = path.dirname(absolutePath);
|
|
try {
|
|
const parentRealPath = await realpath(parentPath);
|
|
resolvedPath = normalizeProjectPath(path.join(parentRealPath, path.basename(absolutePath)));
|
|
} catch (parentError) {
|
|
const parentFileError = parentError as NodeJS.ErrnoException;
|
|
if (parentFileError.code !== 'ENOENT') {
|
|
throw parentFileError;
|
|
}
|
|
}
|
|
}
|
|
|
|
const resolvedWorkspaceRoot = normalizeProjectPath(await realpath(WORKSPACES_ROOT));
|
|
if (
|
|
!resolvedPath.startsWith(`${resolvedWorkspaceRoot}${path.sep}`)
|
|
&& resolvedPath !== resolvedWorkspaceRoot
|
|
) {
|
|
return {
|
|
valid: false,
|
|
error: `Workspace path must be within the allowed workspace root: ${WORKSPACES_ROOT}`,
|
|
};
|
|
}
|
|
|
|
try {
|
|
await access(absolutePath);
|
|
const pathStats = await lstat(absolutePath);
|
|
if (pathStats.isSymbolicLink()) {
|
|
const symlinkTarget = await readlink(absolutePath);
|
|
const resolvedSymlinkPath = path.resolve(path.dirname(absolutePath), symlinkTarget);
|
|
const realSymlinkPath = await realpath(resolvedSymlinkPath);
|
|
if (
|
|
!realSymlinkPath.startsWith(`${resolvedWorkspaceRoot}${path.sep}`)
|
|
&& realSymlinkPath !== resolvedWorkspaceRoot
|
|
) {
|
|
return {
|
|
valid: false,
|
|
error: 'Symlink target is outside the allowed workspace root',
|
|
};
|
|
}
|
|
}
|
|
} catch (error) {
|
|
const fileError = error as NodeJS.ErrnoException;
|
|
if (fileError.code !== 'ENOENT') {
|
|
throw fileError;
|
|
}
|
|
}
|
|
|
|
return {
|
|
valid: true,
|
|
resolvedPath,
|
|
};
|
|
} catch (error) {
|
|
return {
|
|
valid: false,
|
|
error: `Path validation failed: ${(error as Error).message}`,
|
|
};
|
|
}
|
|
}
|
|
|
|
// ---------------------------
|
|
//----------------- NORMALIZED PROVIDER MESSAGE UTILITIES ------------
|
|
/**
|
|
* Generates a stable unique id for normalized provider messages.
|
|
*/
|
|
export function generateMessageId(prefix = 'msg'): string {
|
|
return `${prefix}_${randomUUID()}`;
|
|
}
|
|
|
|
/**
|
|
* Creates a normalized provider message and fills the shared envelope fields.
|
|
*
|
|
* Provider adapters and live SDK handlers pass through provider-specific fields,
|
|
* while this helper guarantees every emitted event has an id, session id,
|
|
* timestamp, and provider marker.
|
|
*/
|
|
export function createNormalizedMessage(fields: NormalizedMessageInput): NormalizedMessage {
|
|
return {
|
|
...fields,
|
|
id: fields.id || generateMessageId(fields.kind),
|
|
sessionId: fields.sessionId || '',
|
|
timestamp: fields.timestamp || new Date().toISOString(),
|
|
provider: fields.provider,
|
|
};
|
|
}
|
|
|
|
// ---------------------------
|
|
//----------------- MCP CONFIG PARSING UTILITIES ------------
|
|
/**
|
|
* Safely narrows an unknown value to a plain object record.
|
|
*
|
|
* This deliberately rejects arrays, `null`, and primitive values so callers can
|
|
* treat the returned value as a JSON-style object map without repeating the same
|
|
* defensive shape checks at every config read site.
|
|
*/
|
|
export const readObjectRecord = (value: any): AnyRecord | null => {
|
|
if (!value || typeof value !== 'object' || Array.isArray(value)) {
|
|
return null;
|
|
}
|
|
|
|
return value as AnyRecord;
|
|
};
|
|
|
|
/**
|
|
* Reads an optional string from unknown input and normalizes empty or whitespace-only
|
|
* values to `undefined`.
|
|
*
|
|
* This is useful when parsing config files where a field may be missing, present
|
|
* with the wrong type, or present as an empty string that should be treated as
|
|
* "not configured".
|
|
*/
|
|
export const readOptionalString = (value: unknown): string | undefined => {
|
|
if (typeof value !== 'string') {
|
|
return undefined;
|
|
}
|
|
|
|
const normalized = value.trim();
|
|
return normalized.length > 0 ? normalized : undefined;
|
|
};
|
|
|
|
/**
|
|
* Reads an optional string array from unknown input.
|
|
*
|
|
* Non-array values are ignored, and any array entries that are not strings are
|
|
* filtered out. This lets provider config readers consume loosely shaped JSON/TOML
|
|
* data without failing on incidental invalid members.
|
|
*/
|
|
export const readStringArray = (value: unknown): string[] | undefined => {
|
|
if (!Array.isArray(value)) {
|
|
return undefined;
|
|
}
|
|
|
|
return value.filter((entry): entry is string => typeof entry === 'string');
|
|
};
|
|
|
|
/**
|
|
* Reads an optional string-to-string map from unknown input.
|
|
*
|
|
* The function first ensures the source value is a plain object, then keeps only
|
|
* keys whose values are strings. If no valid entries remain, it returns `undefined`
|
|
* so callers can distinguish "no usable map" from an empty object that was
|
|
* intentionally authored downstream.
|
|
*/
|
|
export const readStringRecord = (value: unknown): Record<string, string> | undefined => {
|
|
const record = readObjectRecord(value);
|
|
if (!record) {
|
|
return undefined;
|
|
}
|
|
|
|
const normalized: Record<string, string> = {};
|
|
for (const [key, entry] of Object.entries(record)) {
|
|
if (typeof entry === 'string') {
|
|
normalized[key] = entry;
|
|
}
|
|
}
|
|
|
|
return Object.keys(normalized).length > 0 ? normalized : undefined;
|
|
};
|
|
|
|
// ---------------------------
|
|
//----------------- PROVIDER MODEL LOOKUP UTILITIES ------------
|
|
/**
|
|
* Builds the standard "default current model" result used when a provider
|
|
* cannot resolve a session-backed active model.
|
|
*
|
|
* Provider model adapters should call this after loading their supported model
|
|
* catalog so the fallback stays aligned with the provider's current `DEFAULT`
|
|
* selection instead of drifting to a hard-coded duplicate.
|
|
*/
|
|
export function buildDefaultProviderCurrentActiveModel(
|
|
models: ProviderModelsDefinition,
|
|
): ProviderCurrentActiveModel {
|
|
return {
|
|
model: models.DEFAULT,
|
|
};
|
|
}
|
|
|
|
// ---------------------------
|
|
//----------------- PROVIDER SESSION MODEL CHANGE UTILITIES ------------
|
|
type ProviderSessionActiveModelChangeCacheEntry = ProviderSessionActiveModelChange & {
|
|
updatedAt: string;
|
|
};
|
|
|
|
type ProviderSessionActiveModelChangeCacheFile = {
|
|
version: number;
|
|
entries: Record<string, ProviderSessionActiveModelChangeCacheEntry>;
|
|
};
|
|
|
|
const PROVIDER_SESSION_ACTIVE_MODEL_CHANGE_CACHE_VERSION = 1;
|
|
|
|
/**
|
|
* Resolves the backend-owned cache file used for session-scoped resume model
|
|
* overrides.
|
|
*
|
|
* The file lives under `~/.cloudcli` because these overrides are an application
|
|
* concern rather than a provider-native config file. Providers, routes, and
|
|
* runtime command launchers should all use this helper instead of re-creating
|
|
* the path so the storage location stays consistent.
|
|
*/
|
|
export function getProviderSessionActiveModelChangesPath(): string {
|
|
return path.join(os.homedir(), '.cloudcli', 'provider-session-active-model-changes.json');
|
|
}
|
|
|
|
const buildProviderSessionActiveModelChangeKey = (
|
|
provider: LLMProvider,
|
|
sessionId: string,
|
|
): string => `${provider}:${sessionId}`;
|
|
|
|
const isProviderSessionActiveModelChangeCacheEntry = (
|
|
value: unknown,
|
|
): value is ProviderSessionActiveModelChangeCacheEntry => {
|
|
const record = readObjectRecord(value);
|
|
return Boolean(
|
|
record
|
|
&& typeof record.provider === 'string'
|
|
&& typeof record.sessionId === 'string'
|
|
&& typeof record.supported === 'boolean'
|
|
&& typeof record.changed === 'boolean'
|
|
&& (typeof record.model === 'string' || record.model === null)
|
|
&& typeof record.updatedAt === 'string',
|
|
);
|
|
};
|
|
|
|
const readProviderSessionActiveModelChangeCacheFile = async (
|
|
filePath: string,
|
|
): Promise<ProviderSessionActiveModelChangeCacheFile> => {
|
|
try {
|
|
const raw = await readFile(filePath, 'utf8');
|
|
const parsed = readObjectRecord(JSON.parse(raw));
|
|
if (
|
|
!parsed
|
|
|| parsed.version !== PROVIDER_SESSION_ACTIVE_MODEL_CHANGE_CACHE_VERSION
|
|
|| !readObjectRecord(parsed.entries)
|
|
) {
|
|
return {
|
|
version: PROVIDER_SESSION_ACTIVE_MODEL_CHANGE_CACHE_VERSION,
|
|
entries: {},
|
|
};
|
|
}
|
|
|
|
const entries = Object.fromEntries(
|
|
Object.entries(parsed.entries).filter((entry): entry is [string, ProviderSessionActiveModelChangeCacheEntry] =>
|
|
isProviderSessionActiveModelChangeCacheEntry(entry[1]),
|
|
),
|
|
);
|
|
|
|
return {
|
|
version: PROVIDER_SESSION_ACTIVE_MODEL_CHANGE_CACHE_VERSION,
|
|
entries,
|
|
};
|
|
} catch {
|
|
return {
|
|
version: PROVIDER_SESSION_ACTIVE_MODEL_CHANGE_CACHE_VERSION,
|
|
entries: {},
|
|
};
|
|
}
|
|
};
|
|
|
|
const writeProviderSessionActiveModelChangeCacheFile = async (
|
|
filePath: string,
|
|
payload: ProviderSessionActiveModelChangeCacheFile,
|
|
): Promise<void> => {
|
|
await mkdir(path.dirname(filePath), { recursive: true });
|
|
await writeFile(filePath, `${JSON.stringify(payload, null, 2)}\n`, 'utf8');
|
|
};
|
|
|
|
const buildUnsupportedProviderSessionActiveModelChange = (
|
|
provider: LLMProvider,
|
|
sessionId: string,
|
|
): ProviderSessionActiveModelChange => ({
|
|
provider,
|
|
sessionId,
|
|
supported: false,
|
|
changed: false,
|
|
model: null,
|
|
});
|
|
|
|
/**
|
|
* Reads the persisted session model-change state for one provider session.
|
|
*
|
|
* Runtime resume paths use this to decide whether they should inject a
|
|
* provider-specific model argument/thread option for the next resumed turn.
|
|
* Missing cache entries are normalized to `{ changed: false }` so callers can
|
|
* treat absence as "use the ordinary model selection flow".
|
|
*/
|
|
export async function readProviderSessionActiveModelChange(
|
|
provider: LLMProvider,
|
|
sessionId: string,
|
|
options: {
|
|
filePath?: string;
|
|
supported?: boolean;
|
|
} = {},
|
|
): Promise<ProviderSessionActiveModelChange> {
|
|
const normalizedSessionId = sessionId.trim();
|
|
if (!normalizedSessionId) {
|
|
return buildUnsupportedProviderSessionActiveModelChange(provider, normalizedSessionId);
|
|
}
|
|
|
|
const supported = options.supported ?? true;
|
|
if (!supported) {
|
|
return buildUnsupportedProviderSessionActiveModelChange(provider, normalizedSessionId);
|
|
}
|
|
|
|
const filePath = options.filePath ?? getProviderSessionActiveModelChangesPath();
|
|
const cacheFile = await readProviderSessionActiveModelChangeCacheFile(filePath);
|
|
const cacheEntry = cacheFile.entries[
|
|
buildProviderSessionActiveModelChangeKey(provider, normalizedSessionId)
|
|
];
|
|
|
|
if (!cacheEntry || !cacheEntry.changed || !cacheEntry.model?.trim()) {
|
|
return {
|
|
provider,
|
|
sessionId: normalizedSessionId,
|
|
supported: true,
|
|
changed: false,
|
|
model: null,
|
|
};
|
|
}
|
|
|
|
return {
|
|
provider,
|
|
sessionId: normalizedSessionId,
|
|
supported: true,
|
|
changed: true,
|
|
model: cacheEntry.model.trim(),
|
|
};
|
|
}
|
|
|
|
/**
|
|
* Persists a session model-change request for one provider.
|
|
*
|
|
* Provider adapters call this when the frontend explicitly selects a different
|
|
* model for an existing session. The stored `changed: true` flag is the single
|
|
* source of truth used later by resume paths to decide whether they should add
|
|
* a provider-native model override on the next invocation.
|
|
*/
|
|
export async function writeProviderSessionActiveModelChange(
|
|
provider: LLMProvider,
|
|
input: ProviderChangeActiveModelInput,
|
|
options: {
|
|
filePath?: string;
|
|
supported?: boolean;
|
|
} = {},
|
|
): Promise<ProviderSessionActiveModelChange> {
|
|
const normalizedSessionId = input.sessionId.trim();
|
|
const normalizedModel = input.model.trim();
|
|
const supported = options.supported ?? true;
|
|
|
|
if (!supported) {
|
|
return buildUnsupportedProviderSessionActiveModelChange(provider, normalizedSessionId);
|
|
}
|
|
|
|
if (!normalizedSessionId || !normalizedModel) {
|
|
return {
|
|
provider,
|
|
sessionId: normalizedSessionId,
|
|
supported: true,
|
|
changed: false,
|
|
model: null,
|
|
};
|
|
}
|
|
|
|
const filePath = options.filePath ?? getProviderSessionActiveModelChangesPath();
|
|
const cacheFile = await readProviderSessionActiveModelChangeCacheFile(filePath);
|
|
cacheFile.entries[buildProviderSessionActiveModelChangeKey(provider, normalizedSessionId)] = {
|
|
provider,
|
|
sessionId: normalizedSessionId,
|
|
supported: true,
|
|
changed: true,
|
|
model: normalizedModel,
|
|
updatedAt: new Date().toISOString(),
|
|
};
|
|
|
|
await writeProviderSessionActiveModelChangeCacheFile(filePath, cacheFile);
|
|
|
|
return {
|
|
provider,
|
|
sessionId: normalizedSessionId,
|
|
supported: true,
|
|
changed: true,
|
|
model: normalizedModel,
|
|
};
|
|
}
|
|
|
|
// ---------------------------
|
|
//----------------- WEBSOCKET PAYLOAD PARSING UTILITIES ------------
|
|
/**
|
|
* Parses one websocket message payload into a plain JSON object record.
|
|
*
|
|
* Use this in realtime handlers that receive raw websocket payloads as `string`,
|
|
* `Buffer`, `ArrayBuffer`, or chunk arrays. The helper converts supported
|
|
* payload formats to UTF-8 text, parses JSON, and returns only object payloads.
|
|
* Primitive/array/invalid payloads return `null` so callers can handle bad input
|
|
* without throwing from deeply nested message handlers.
|
|
*/
|
|
export const parseIncomingJsonObject = (payload: unknown): AnyRecord | null => {
|
|
let text: string | null = null;
|
|
|
|
if (typeof payload === 'string') {
|
|
text = payload;
|
|
} else if (Buffer.isBuffer(payload)) {
|
|
text = payload.toString('utf8');
|
|
} else if (payload instanceof ArrayBuffer) {
|
|
text = Buffer.from(payload).toString('utf8');
|
|
} else if (Array.isArray(payload)) {
|
|
const buffers = payload
|
|
.map((entry) => {
|
|
if (Buffer.isBuffer(entry)) {
|
|
return entry;
|
|
}
|
|
|
|
if (entry instanceof ArrayBuffer) {
|
|
return Buffer.from(entry);
|
|
}
|
|
|
|
if (ArrayBuffer.isView(entry)) {
|
|
return Buffer.from(entry.buffer, entry.byteOffset, entry.byteLength);
|
|
}
|
|
|
|
return null;
|
|
})
|
|
.filter((entry): entry is Buffer => entry !== null);
|
|
|
|
if (buffers.length > 0) {
|
|
text = Buffer.concat(buffers).toString('utf8');
|
|
}
|
|
}
|
|
|
|
if (typeof text !== 'string' || text.trim().length === 0) {
|
|
return null;
|
|
}
|
|
|
|
try {
|
|
const parsed = JSON.parse(text) as unknown;
|
|
return readObjectRecord(parsed);
|
|
} catch {
|
|
return null;
|
|
}
|
|
};
|
|
|
|
/**
|
|
* Reads a JSON config file and guarantees a plain object result.
|
|
*
|
|
* Missing files are treated as an empty config object so provider-specific MCP
|
|
* readers can operate against first-run environments without special-case file
|
|
* existence checks. If the file exists but contains invalid JSON, the parse error
|
|
* is preserved and rethrown.
|
|
*/
|
|
export const readJsonConfig = async (filePath: string): Promise<Record<string, unknown>> => {
|
|
try {
|
|
const content = await readFile(filePath, 'utf8');
|
|
const parsed = JSON.parse(content) as Record<string, unknown>;
|
|
return readObjectRecord(parsed) ?? {};
|
|
} catch (error) {
|
|
const code = (error as NodeJS.ErrnoException).code;
|
|
if (code === 'ENOENT') {
|
|
return {};
|
|
}
|
|
|
|
throw error;
|
|
}
|
|
};
|
|
|
|
/**
|
|
* Writes a JSON config file with stable, human-readable formatting.
|
|
*
|
|
* The parent directory is created automatically so callers can persist config into
|
|
* provider-specific folders without pre-creating the directory tree. Output always
|
|
* ends with a trailing newline to keep the file diff-friendly.
|
|
*/
|
|
export const writeJsonConfig = async (filePath: string, data: Record<string, unknown>): Promise<void> => {
|
|
await mkdir(path.dirname(filePath), { recursive: true });
|
|
await writeFile(filePath, `${JSON.stringify(data, null, 2)}\n`, 'utf8');
|
|
};
|
|
|
|
// ---------------------------
|
|
//----------------- PROVIDER SKILL FILE UTILITIES ------------
|
|
async function hasGitMarker(dirPath: string): Promise<boolean> {
|
|
try {
|
|
const gitMarkerStats = await stat(path.join(dirPath, '.git'));
|
|
return gitMarkerStats.isDirectory() || gitMarkerStats.isFile();
|
|
} catch {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Finds the highest git worktree root visible from a starting directory.
|
|
*
|
|
* Provider skill systems such as Codex and OpenCode walk upward through parent
|
|
* folders when resolving repository/project skills. Use this helper when a
|
|
* provider needs the topmost `.git` marker instead of only the nearest one, so
|
|
* monorepos and nested package folders discover shared root-level skills once.
|
|
*/
|
|
export async function findTopmostGitRoot(startPath: string): Promise<string | null> {
|
|
let currentPath = path.resolve(startPath);
|
|
let topmostGitRoot: string | null = null;
|
|
|
|
while (true) {
|
|
if (await hasGitMarker(currentPath)) {
|
|
topmostGitRoot = currentPath;
|
|
}
|
|
|
|
const parentPath = path.dirname(currentPath);
|
|
if (parentPath === currentPath) {
|
|
break;
|
|
}
|
|
|
|
currentPath = parentPath;
|
|
}
|
|
|
|
return topmostGitRoot;
|
|
}
|
|
|
|
/**
|
|
* Adds one provider skill source after normalizing and de-duplicating its root.
|
|
*
|
|
* Provider skill lookup rules often point at overlapping folders (for example a
|
|
* workspace folder can also be the git root). Use this helper while building a
|
|
* provider's `ProviderSkillSource[]` so the shared skills scanner reads each
|
|
* physical root once and still preserves provider-specific scope/command data.
|
|
*/
|
|
export function addUniqueProviderSkillSource(
|
|
sources: ProviderSkillSource[],
|
|
seenRootDirs: Set<string>,
|
|
source: ProviderSkillSource,
|
|
): void {
|
|
const normalizedRootDir = path.resolve(source.rootDir);
|
|
if (seenRootDirs.has(normalizedRootDir)) {
|
|
return;
|
|
}
|
|
|
|
seenRootDirs.add(normalizedRootDir);
|
|
sources.push({ ...source, rootDir: normalizedRootDir });
|
|
}
|
|
|
|
// ---------------------------
|
|
//----------------- PROVIDER SKILL MARKDOWN UTILITIES ------------
|
|
/**
|
|
* Finds direct child skill markdown files under a provider skill root.
|
|
*
|
|
* Skill systems usually store one skill per child directory, so direct mode
|
|
* scans only `<root>/<skill-name>/SKILL.md`. Recursive mode is reserved for
|
|
* provider sources that can nest skills arbitrarily, and it returns every
|
|
* descendant `SKILL.md`. Missing or unreadable roots return an empty list
|
|
* because users may not have every provider installed or configured.
|
|
*/
|
|
export async function findProviderSkillMarkdownFiles(
|
|
rootDir: string,
|
|
options: { recursive?: boolean } = {},
|
|
): Promise<string[]> {
|
|
const skillFiles: string[] = [];
|
|
|
|
const collectRecursive = async (dirPath: string): Promise<void> => {
|
|
let entries;
|
|
try {
|
|
entries = await readdir(dirPath, { withFileTypes: true });
|
|
} catch {
|
|
return;
|
|
}
|
|
|
|
try {
|
|
const skillPath = path.join(dirPath, 'SKILL.md');
|
|
const skillStats = await stat(skillPath);
|
|
if (skillStats.isFile()) {
|
|
skillFiles.push(skillPath);
|
|
}
|
|
} catch {
|
|
// Directories without SKILL.md are expected while walking plugin trees.
|
|
}
|
|
|
|
for (const entry of entries) {
|
|
if (entry.isDirectory()) {
|
|
await collectRecursive(path.join(dirPath, entry.name));
|
|
}
|
|
}
|
|
};
|
|
|
|
if (options.recursive) {
|
|
await collectRecursive(rootDir);
|
|
return skillFiles.sort((left, right) => left.localeCompare(right));
|
|
}
|
|
|
|
try {
|
|
const entries = await readdir(rootDir, { withFileTypes: true });
|
|
|
|
for (const entry of entries) {
|
|
if (!entry.isDirectory()) {
|
|
continue;
|
|
}
|
|
|
|
const skillPath = path.join(rootDir, entry.name, 'SKILL.md');
|
|
try {
|
|
const skillStats = await stat(skillPath);
|
|
if (skillStats.isFile()) {
|
|
skillFiles.push(skillPath);
|
|
}
|
|
} catch {
|
|
// A partial skill directory should not block discovery of sibling skills.
|
|
}
|
|
}
|
|
|
|
return skillFiles.sort((left, right) => left.localeCompare(right));
|
|
} catch {
|
|
return [];
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Reads the `name` and `description` fields from a provider skill markdown file.
|
|
*
|
|
* The metadata is expected in markdown front matter. If a skill omits `name`, the
|
|
* parent directory name is used as a stable fallback so providers can still
|
|
* expose the skill. Missing descriptions are normalized to an empty string.
|
|
*/
|
|
export async function readProviderSkillMarkdownDefinition(
|
|
skillPath: string,
|
|
): Promise<{ name: string; description: string }> {
|
|
const content = await readFile(skillPath, 'utf8');
|
|
const parsed = parseFrontMatter(content);
|
|
const data = readObjectRecord(parsed.data) ?? {};
|
|
const fallbackName = path.basename(path.dirname(skillPath));
|
|
|
|
return {
|
|
name: readOptionalString(data.name) ?? fallbackName,
|
|
description: readOptionalString(data.description) ?? '',
|
|
};
|
|
}
|
|
|
|
// ---------------------------
|
|
//----------------- SESSION SYNCHRONIZER TITLE HELPERS ------------
|
|
/**
|
|
* Produces a compact session title suitable for UI rendering and DB storage.
|
|
*
|
|
* Use this when converting provider-native names into a consistent title value.
|
|
* The helper collapses repeated whitespace, trims the result, and truncates it
|
|
* to 120 characters so every provider writes stable and bounded metadata.
|
|
* If the normalized input is empty, it returns the supplied fallback title.
|
|
*/
|
|
export function normalizeSessionName(rawValue: string | undefined, fallback: string): string {
|
|
const normalized = (rawValue ?? '').replace(/\s+/g, ' ').trim();
|
|
if (!normalized) {
|
|
return fallback;
|
|
}
|
|
|
|
return normalized.slice(0, 120);
|
|
}
|
|
|
|
// ---------------------------
|
|
//----------------- PROVIDER SESSION VALUE NORMALIZATION UTILITIES ------------
|
|
/**
|
|
* Converts provider-native timestamps into ISO strings.
|
|
*
|
|
* Provider CLIs commonly persist epoch timestamps as milliseconds, seconds, or
|
|
* already-formatted date strings. Use this helper when normalizing session
|
|
* metadata or transcript events so every provider writes the same ISO timestamp
|
|
* shape to API responses and database rows.
|
|
*/
|
|
export function normalizeProviderTimestamp(value: unknown): string {
|
|
if (typeof value === 'number' && Number.isFinite(value) && value > 0) {
|
|
const millis = value < 1_000_000_000_000 ? value * 1000 : value;
|
|
return new Date(millis).toISOString();
|
|
}
|
|
|
|
if (typeof value === 'string' && value.trim()) {
|
|
const parsed = Number(value);
|
|
if (Number.isFinite(parsed)) {
|
|
return normalizeProviderTimestamp(parsed);
|
|
}
|
|
|
|
const date = new Date(value);
|
|
if (!Number.isNaN(date.getTime())) {
|
|
return date.toISOString();
|
|
}
|
|
}
|
|
|
|
return new Date().toISOString();
|
|
}
|
|
|
|
/**
|
|
* Parses a JSON string or narrows an existing object into a plain record.
|
|
*
|
|
* Use this when provider databases store structured JSON inside text columns.
|
|
* Invalid JSON, arrays, and primitive values return `null` so callers can skip
|
|
* malformed optional metadata without hiding the rest of a session transcript.
|
|
*/
|
|
export function readJsonRecord(value: unknown): AnyRecord | null {
|
|
if (typeof value !== 'string') {
|
|
return readObjectRecord(value);
|
|
}
|
|
|
|
try {
|
|
return readObjectRecord(JSON.parse(value));
|
|
} catch {
|
|
return null;
|
|
}
|
|
}
|
|
|
|
// ---------------------------
|
|
//----------------- OPENCODE SESSION STORAGE UTILITIES ------------
|
|
/**
|
|
* Resolves the OpenCode SQLite session database path.
|
|
*
|
|
* OpenCode stores session, message, part, and project metadata in one shared
|
|
* `opencode.db` file under its XDG data directory. Provider readers and
|
|
* synchronizers should use this path for read-only access and should never store
|
|
* it as a deletable transcript path for an individual app session row.
|
|
*/
|
|
export function getOpenCodeDatabasePath(): string {
|
|
return path.join(os.homedir(), '.local', 'share', 'opencode', 'opencode.db');
|
|
}
|
|
|
|
// ---------------------------
|
|
//----------------- SAFE DIRECTORY NAME UTILITIES ------------
|
|
/**
|
|
* Validates that a user or provider supplied identifier can safely be treated
|
|
* as one leaf directory name under an existing root folder.
|
|
*
|
|
* Use this before composing paths like `<root>/<session-id>/file.db>` to block
|
|
* path traversal and accidental nested paths. The returned string is trimmed but
|
|
* otherwise unchanged so callers can still match the provider's on-disk naming.
|
|
*/
|
|
export function sanitizeLeafDirectoryName(inputName: string, label = 'directory name'): string {
|
|
const normalized = inputName.trim();
|
|
if (!normalized) {
|
|
throw new Error(`${label} is required.`);
|
|
}
|
|
|
|
if (
|
|
normalized.includes('..')
|
|
|| normalized.includes(path.posix.sep)
|
|
|| normalized.includes(path.win32.sep)
|
|
|| normalized !== path.basename(normalized)
|
|
) {
|
|
throw new Error(`Invalid ${label} "${inputName}".`);
|
|
}
|
|
|
|
return normalized;
|
|
}
|
|
|
|
// ---------------------------
|
|
//----------------- SESSION SYNCHRONIZER FILESYSTEM HELPERS ------------
|
|
/**
|
|
* Recursively discovers files that match one extension, with optional incremental filtering.
|
|
*
|
|
* Provider synchronizers call this to find transcript artifacts under provider
|
|
* home directories. Pass `lastScanAt` to include only files created after the
|
|
* previous scan, or pass `null` to perform a full rescan. Missing directories
|
|
* are treated as empty because not every provider exists on every machine.
|
|
*/
|
|
export async function findFilesRecursivelyCreatedAfter(
|
|
rootDir: string,
|
|
extension: string,
|
|
lastScanAt: Date | null,
|
|
fileList: string[] = []
|
|
): Promise<string[]> {
|
|
try {
|
|
const entries = await readdir(rootDir, { withFileTypes: true });
|
|
for (const entry of entries) {
|
|
const fullPath = path.join(rootDir, entry.name);
|
|
|
|
if (entry.isDirectory()) {
|
|
await findFilesRecursivelyCreatedAfter(fullPath, extension, lastScanAt, fileList);
|
|
continue;
|
|
}
|
|
|
|
if (!entry.isFile() || !entry.name.endsWith(extension)) {
|
|
continue;
|
|
}
|
|
|
|
if (!lastScanAt) {
|
|
fileList.push(fullPath);
|
|
continue;
|
|
}
|
|
|
|
const fileStat = await stat(fullPath);
|
|
if (fileStat.birthtime > lastScanAt) {
|
|
fileList.push(fullPath);
|
|
}
|
|
}
|
|
} catch {
|
|
// Missing provider folders are expected in first-run or partial setups.
|
|
}
|
|
|
|
return fileList;
|
|
}
|
|
|
|
/**
|
|
* Reads file creation/update timestamps and maps them to DB-friendly ISO strings.
|
|
*
|
|
* Session indexers use this to persist `created_at` and `updated_at` metadata
|
|
* when upserting sessions. If the file cannot be read, an empty object is
|
|
* returned so indexing can continue for other files.
|
|
*/
|
|
export async function readFileTimestamps(
|
|
filePath: string
|
|
): Promise<{ createdAt?: string; updatedAt?: string }> {
|
|
try {
|
|
const fileStat = await stat(filePath);
|
|
return {
|
|
createdAt: fileStat.birthtime.toISOString(),
|
|
updatedAt: fileStat.mtime.toISOString(),
|
|
};
|
|
} catch {
|
|
return {};
|
|
}
|
|
}
|
|
|
|
// ---------------------------
|
|
//----------------- SESSION SYNCHRONIZER JSONL PARSING HELPERS ------------
|
|
/**
|
|
* Builds a first-seen key/value lookup map from a JSONL file.
|
|
*
|
|
* Use this for provider index files where session id -> display name metadata
|
|
* is stored line-by-line. The first value for each key wins, preserving the
|
|
* earliest known label while avoiding repeated map overwrites.
|
|
*/
|
|
export async function buildLookupMap(
|
|
filePath: string,
|
|
keyField: string,
|
|
valueField: string
|
|
): Promise<Map<string, string>> {
|
|
const lookup = new Map<string, string>();
|
|
|
|
try {
|
|
const fileStream = fs.createReadStream(filePath);
|
|
const lineReader = readline.createInterface({ input: fileStream, crlfDelay: Infinity });
|
|
|
|
for await (const line of lineReader) {
|
|
const trimmed = line.trim();
|
|
if (!trimmed) {
|
|
continue;
|
|
}
|
|
|
|
const parsed = JSON.parse(trimmed) as Record<string, unknown>;
|
|
const key = parsed[keyField];
|
|
const value = parsed[valueField];
|
|
|
|
if (typeof key === 'string' && typeof value === 'string' && !lookup.has(key)) {
|
|
lookup.set(key, value);
|
|
}
|
|
}
|
|
} catch {
|
|
// Missing or unreadable lookup files should not block session sync.
|
|
}
|
|
|
|
return lookup;
|
|
}
|
|
|
|
/**
|
|
* Reads a JSONL file and returns the first extracted payload that matches caller criteria.
|
|
*
|
|
* The caller supplies an `extractor` that validates provider-specific row
|
|
* shapes. This helper centralizes line-by-line parsing and lets indexers stop
|
|
* scanning as soon as one valid row is found.
|
|
*/
|
|
export async function extractFirstValidJsonlData<T>(
|
|
filePath: string,
|
|
extractor: (parsedJson: unknown) => T | null | undefined
|
|
): Promise<T | null> {
|
|
try {
|
|
const fileStream = fs.createReadStream(filePath);
|
|
const lineReader = readline.createInterface({ input: fileStream, crlfDelay: Infinity });
|
|
|
|
for await (const line of lineReader) {
|
|
const trimmed = line.trim();
|
|
if (!trimmed) {
|
|
continue;
|
|
}
|
|
|
|
const parsed = JSON.parse(trimmed);
|
|
const extracted = extractor(parsed);
|
|
if (extracted) {
|
|
lineReader.close();
|
|
fileStream.close();
|
|
return extracted;
|
|
}
|
|
}
|
|
} catch {
|
|
// Ignore malformed or missing artifacts so full scans keep progressing.
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|