@imdigitalashish/zpi 0.1.1

package/dist/core/settings-manager.js

@@ -0,0 +1,656 @@
+import { existsSync, mkdirSync, readFileSync, writeFileSync } from "fs";
+import { dirname, join } from "path";
+import lockfile from "proper-lockfile";
+import { CONFIG_DIR_NAME, getAgentDir } from "../config.js";
+/** Deep merge settings: project/overrides take precedence, nested objects merge recursively */
+function deepMergeSettings(base, overrides) {
+    const result = { ...base };
+    for (const key of Object.keys(overrides)) {
+        const overrideValue = overrides[key];
+        const baseValue = base[key];
+        if (overrideValue === undefined) {
+            continue;
+        }
+        // For nested objects, merge recursively
+        if (typeof overrideValue === "object" &&
+            overrideValue !== null &&
+            !Array.isArray(overrideValue) &&
+            typeof baseValue === "object" &&
+            baseValue !== null &&
+            !Array.isArray(baseValue)) {
+            result[key] = { ...baseValue, ...overrideValue };
+        }
+        else {
+            // For primitives and arrays, override value wins
+            result[key] = overrideValue;
+        }
+    }
+    return result;
+}
+export class FileSettingsStorage {
+    globalSettingsPath;
+    projectSettingsPath;
+    constructor(cwd = process.cwd(), agentDir = getAgentDir()) {
+        this.globalSettingsPath = join(agentDir, "settings.json");
+        this.projectSettingsPath = join(cwd, CONFIG_DIR_NAME, "settings.json");
+    }
+    withLock(scope, fn) {
+        const path = scope === "global" ? this.globalSettingsPath : this.projectSettingsPath;
+        const dir = dirname(path);
+        if (!existsSync(dir)) {
+            mkdirSync(dir, { recursive: true });
+        }
+        let release;
+        try {
+            release = lockfile.lockSync(path, { realpath: false });
+            const current = existsSync(path) ? readFileSync(path, "utf-8") : undefined;
+            const next = fn(current);
+            if (next !== undefined) {
+                writeFileSync(path, next, "utf-8");
+            }
+        }
+        finally {
+            if (release) {
+                release();
+            }
+        }
+    }
+}
+export class InMemorySettingsStorage {
+    global;
+    project;
+    withLock(scope, fn) {
+        const current = scope === "global" ? this.global : this.project;
+        const next = fn(current);
+        if (next !== undefined) {
+            if (scope === "global") {
+                this.global = next;
+            }
+            else {
+                this.project = next;
+            }
+        }
+    }
+}
+export class SettingsManager {
+    storage;
+    globalSettings;
+    projectSettings;
+    settings;
+    modifiedFields = new Set(); // Track global fields modified during session
+    modifiedNestedFields = new Map(); // Track global nested field modifications
+    modifiedProjectFields = new Set(); // Track project fields modified during session
+    modifiedProjectNestedFields = new Map(); // Track project nested field modifications
+    globalSettingsLoadError = null; // Track if global settings file had parse errors
+    projectSettingsLoadError = null; // Track if project settings file had parse errors
+    writeQueue = Promise.resolve();
+    errors;
+    constructor(storage, initialGlobal, initialProject, globalLoadError = null, projectLoadError = null, initialErrors = []) {
+        this.storage = storage;
+        this.globalSettings = initialGlobal;
+        this.projectSettings = initialProject;
+        this.globalSettingsLoadError = globalLoadError;
+        this.projectSettingsLoadError = projectLoadError;
+        this.errors = [...initialErrors];
+        this.settings = deepMergeSettings(this.globalSettings, this.projectSettings);
+    }
+    /** Create a SettingsManager that loads from files */
+    static create(cwd = process.cwd(), agentDir = getAgentDir()) {
+        const storage = new FileSettingsStorage(cwd, agentDir);
+        return SettingsManager.fromStorage(storage);
+    }
+    /** Create a SettingsManager from an arbitrary storage backend */
+    static fromStorage(storage) {
+        const globalLoad = SettingsManager.tryLoadFromStorage(storage, "global");
+        const projectLoad = SettingsManager.tryLoadFromStorage(storage, "project");
+        const initialErrors = [];
+        if (globalLoad.error) {
+            initialErrors.push({ scope: "global", error: globalLoad.error });
+        }
+        if (projectLoad.error) {
+            initialErrors.push({ scope: "project", error: projectLoad.error });
+        }
+        return new SettingsManager(storage, globalLoad.settings, projectLoad.settings, globalLoad.error, projectLoad.error, initialErrors);
+    }
+    /** Create an in-memory SettingsManager (no file I/O) */
+    static inMemory(settings = {}) {
+        const storage = new InMemorySettingsStorage();
+        return new SettingsManager(storage, settings, {});
+    }
+    static loadFromStorage(storage, scope) {
+        let content;
+        storage.withLock(scope, (current) => {
+            content = current;
+            return undefined;
+        });
+        if (!content) {
+            return {};
+        }
+        const settings = JSON.parse(content);
+        return SettingsManager.migrateSettings(settings);
+    }
+    static tryLoadFromStorage(storage, scope) {
+        try {
+            return { settings: SettingsManager.loadFromStorage(storage, scope), error: null };
+        }
+        catch (error) {
+            return { settings: {}, error: error };
+        }
+    }
+    /** Migrate old settings format to new format */
+    static migrateSettings(settings) {
+        // Migrate queueMode -> steeringMode
+        if ("queueMode" in settings && !("steeringMode" in settings)) {
+            settings.steeringMode = settings.queueMode;
+            delete settings.queueMode;
+        }
+        // Migrate legacy websockets boolean -> transport enum
+        if (!("transport" in settings) && typeof settings.websockets === "boolean") {
+            settings.transport = settings.websockets ? "websocket" : "sse";
+            delete settings.websockets;
+        }
+        // Migrate old skills object format to new array format
+        if ("skills" in settings &&
+            typeof settings.skills === "object" &&
+            settings.skills !== null &&
+            !Array.isArray(settings.skills)) {
+            const skillsSettings = settings.skills;
+            if (skillsSettings.enableSkillCommands !== undefined && settings.enableSkillCommands === undefined) {
+                settings.enableSkillCommands = skillsSettings.enableSkillCommands;
+            }
+            if (Array.isArray(skillsSettings.customDirectories) && skillsSettings.customDirectories.length > 0) {
+                settings.skills = skillsSettings.customDirectories;
+            }
+            else {
+                delete settings.skills;
+            }
+        }
+        return settings;
+    }
+    getGlobalSettings() {
+        return structuredClone(this.globalSettings);
+    }
+    getProjectSettings() {
+        return structuredClone(this.projectSettings);
+    }
+    reload() {
+        const globalLoad = SettingsManager.tryLoadFromStorage(this.storage, "global");
+        if (!globalLoad.error) {
+            this.globalSettings = globalLoad.settings;
+            this.globalSettingsLoadError = null;
+        }
+        else {
+            this.globalSettingsLoadError = globalLoad.error;
+            this.recordError("global", globalLoad.error);
+        }
+        this.modifiedFields.clear();
+        this.modifiedNestedFields.clear();
+        this.modifiedProjectFields.clear();
+        this.modifiedProjectNestedFields.clear();
+        const projectLoad = SettingsManager.tryLoadFromStorage(this.storage, "project");
+        if (!projectLoad.error) {
+            this.projectSettings = projectLoad.settings;
+            this.projectSettingsLoadError = null;
+        }
+        else {
+            this.projectSettingsLoadError = projectLoad.error;
+            this.recordError("project", projectLoad.error);
+        }
+        this.settings = deepMergeSettings(this.globalSettings, this.projectSettings);
+    }
+    /** Apply additional overrides on top of current settings */
+    applyOverrides(overrides) {
+        this.settings = deepMergeSettings(this.settings, overrides);
+    }
+    /** Mark a global field as modified during this session */
+    markModified(field, nestedKey) {
+        this.modifiedFields.add(field);
+        if (nestedKey) {
+            if (!this.modifiedNestedFields.has(field)) {
+                this.modifiedNestedFields.set(field, new Set());
+            }
+            this.modifiedNestedFields.get(field).add(nestedKey);
+        }
+    }
+    /** Mark a project field as modified during this session */
+    markProjectModified(field, nestedKey) {
+        this.modifiedProjectFields.add(field);
+        if (nestedKey) {
+            if (!this.modifiedProjectNestedFields.has(field)) {
+                this.modifiedProjectNestedFields.set(field, new Set());
+            }
+            this.modifiedProjectNestedFields.get(field).add(nestedKey);
+        }
+    }
+    recordError(scope, error) {
+        const normalizedError = error instanceof Error ? error : new Error(String(error));
+        this.errors.push({ scope, error: normalizedError });
+    }
+    clearModifiedScope(scope) {
+        if (scope === "global") {
+            this.modifiedFields.clear();
+            this.modifiedNestedFields.clear();
+            return;
+        }
+        this.modifiedProjectFields.clear();
+        this.modifiedProjectNestedFields.clear();
+    }
+    enqueueWrite(scope, task) {
+        this.writeQueue = this.writeQueue
+            .then(() => {
+            task();
+            this.clearModifiedScope(scope);
+        })
+            .catch((error) => {
+            this.recordError(scope, error);
+        });
+    }
+    cloneModifiedNestedFields(source) {
+        const snapshot = new Map();
+        for (const [key, value] of source.entries()) {
+            snapshot.set(key, new Set(value));
+        }
+        return snapshot;
+    }
+    persistScopedSettings(scope, snapshotSettings, modifiedFields, modifiedNestedFields) {
+        this.storage.withLock(scope, (current) => {
+            const currentFileSettings = current
+                ? SettingsManager.migrateSettings(JSON.parse(current))
+                : {};
+            const mergedSettings = { ...currentFileSettings };
+            for (const field of modifiedFields) {
+                const value = snapshotSettings[field];
+                if (modifiedNestedFields.has(field) && typeof value === "object" && value !== null) {
+                    const nestedModified = modifiedNestedFields.get(field);
+                    const baseNested = currentFileSettings[field] ?? {};
+                    const inMemoryNested = value;
+                    const mergedNested = { ...baseNested };
+                    for (const nestedKey of nestedModified) {
+                        mergedNested[nestedKey] = inMemoryNested[nestedKey];
+                    }
+                    mergedSettings[field] = mergedNested;
+                }
+                else {
+                    mergedSettings[field] = value;
+                }
+            }
+            return JSON.stringify(mergedSettings, null, 2);
+        });
+    }
+    save() {
+        this.settings = deepMergeSettings(this.globalSettings, this.projectSettings);
+        if (this.globalSettingsLoadError) {
+            return;
+        }
+        const snapshotGlobalSettings = structuredClone(this.globalSettings);
+        const modifiedFields = new Set(this.modifiedFields);
+        const modifiedNestedFields = this.cloneModifiedNestedFields(this.modifiedNestedFields);
+        this.enqueueWrite("global", () => {
+            this.persistScopedSettings("global", snapshotGlobalSettings, modifiedFields, modifiedNestedFields);
+        });
+    }
+    saveProjectSettings(settings) {
+        this.projectSettings = structuredClone(settings);
+        this.settings = deepMergeSettings(this.globalSettings, this.projectSettings);
+        if (this.projectSettingsLoadError) {
+            return;
+        }
+        const snapshotProjectSettings = structuredClone(this.projectSettings);
+        const modifiedFields = new Set(this.modifiedProjectFields);
+        const modifiedNestedFields = this.cloneModifiedNestedFields(this.modifiedProjectNestedFields);
+        this.enqueueWrite("project", () => {
+            this.persistScopedSettings("project", snapshotProjectSettings, modifiedFields, modifiedNestedFields);
+        });
+    }
+    async flush() {
+        await this.writeQueue;
+    }
+    drainErrors() {
+        const drained = [...this.errors];
+        this.errors = [];
+        return drained;
+    }
+    getLastChangelogVersion() {
+        return this.settings.lastChangelogVersion;
+    }
+    setLastChangelogVersion(version) {
+        this.globalSettings.lastChangelogVersion = version;
+        this.markModified("lastChangelogVersion");
+        this.save();
+    }
+    getDefaultProvider() {
+        return this.settings.defaultProvider;
+    }
+    getDefaultModel() {
+        return this.settings.defaultModel;
+    }
+    setDefaultProvider(provider) {
+        this.globalSettings.defaultProvider = provider;
+        this.markModified("defaultProvider");
+        this.save();
+    }
+    setDefaultModel(modelId) {
+        this.globalSettings.defaultModel = modelId;
+        this.markModified("defaultModel");
+        this.save();
+    }
+    setDefaultModelAndProvider(provider, modelId) {
+        this.globalSettings.defaultProvider = provider;
+        this.globalSettings.defaultModel = modelId;
+        this.markModified("defaultProvider");
+        this.markModified("defaultModel");
+        this.save();
+    }
+    getSteeringMode() {
+        return this.settings.steeringMode || "one-at-a-time";
+    }
+    setSteeringMode(mode) {
+        this.globalSettings.steeringMode = mode;
+        this.markModified("steeringMode");
+        this.save();
+    }
+    getFollowUpMode() {
+        return this.settings.followUpMode || "one-at-a-time";
+    }
+    setFollowUpMode(mode) {
+        this.globalSettings.followUpMode = mode;
+        this.markModified("followUpMode");
+        this.save();
+    }
+    getTheme() {
+        return this.settings.theme;
+    }
+    setTheme(theme) {
+        this.globalSettings.theme = theme;
+        this.markModified("theme");
+        this.save();
+    }
+    getDefaultThinkingLevel() {
+        return this.settings.defaultThinkingLevel;
+    }
+    setDefaultThinkingLevel(level) {
+        this.globalSettings.defaultThinkingLevel = level;
+        this.markModified("defaultThinkingLevel");
+        this.save();
+    }
+    getTransport() {
+        return this.settings.transport ?? "sse";
+    }
+    setTransport(transport) {
+        this.globalSettings.transport = transport;
+        this.markModified("transport");
+        this.save();
+    }
+    getCompactionEnabled() {
+        return this.settings.compaction?.enabled ?? true;
+    }
+    setCompactionEnabled(enabled) {
+        if (!this.globalSettings.compaction) {
+            this.globalSettings.compaction = {};
+        }
+        this.globalSettings.compaction.enabled = enabled;
+        this.markModified("compaction", "enabled");
+        this.save();
+    }
+    getCompactionReserveTokens() {
+        return this.settings.compaction?.reserveTokens ?? 16384;
+    }
+    getCompactionKeepRecentTokens() {
+        return this.settings.compaction?.keepRecentTokens ?? 20000;
+    }
+    getCompactionSettings() {
+        return {
+            enabled: this.getCompactionEnabled(),
+            reserveTokens: this.getCompactionReserveTokens(),
+            keepRecentTokens: this.getCompactionKeepRecentTokens(),
+        };
+    }
+    getBranchSummarySettings() {
+        return {
+            reserveTokens: this.settings.branchSummary?.reserveTokens ?? 16384,
+        };
+    }
+    getRetryEnabled() {
+        return this.settings.retry?.enabled ?? true;
+    }
+    setRetryEnabled(enabled) {
+        if (!this.globalSettings.retry) {
+            this.globalSettings.retry = {};
+        }
+        this.globalSettings.retry.enabled = enabled;
+        this.markModified("retry", "enabled");
+        this.save();
+    }
+    getRetrySettings() {
+        return {
+            enabled: this.getRetryEnabled(),
+            maxRetries: this.settings.retry?.maxRetries ?? 3,
+            baseDelayMs: this.settings.retry?.baseDelayMs ?? 2000,
+            maxDelayMs: this.settings.retry?.maxDelayMs ?? 60000,
+        };
+    }
+    getHideThinkingBlock() {
+        return this.settings.hideThinkingBlock ?? false;
+    }
+    setHideThinkingBlock(hide) {
+        this.globalSettings.hideThinkingBlock = hide;
+        this.markModified("hideThinkingBlock");
+        this.save();
+    }
+    getShellPath() {
+        return this.settings.shellPath;
+    }
+    setShellPath(path) {
+        this.globalSettings.shellPath = path;
+        this.markModified("shellPath");
+        this.save();
+    }
+    getQuietStartup() {
+        return this.settings.quietStartup ?? false;
+    }
+    setQuietStartup(quiet) {
+        this.globalSettings.quietStartup = quiet;
+        this.markModified("quietStartup");
+        this.save();
+    }
+    getShellCommandPrefix() {
+        return this.settings.shellCommandPrefix;
+    }
+    setShellCommandPrefix(prefix) {
+        this.globalSettings.shellCommandPrefix = prefix;
+        this.markModified("shellCommandPrefix");
+        this.save();
+    }
+    getCollapseChangelog() {
+        return this.settings.collapseChangelog ?? false;
+    }
+    setCollapseChangelog(collapse) {
+        this.globalSettings.collapseChangelog = collapse;
+        this.markModified("collapseChangelog");
+        this.save();
+    }
+    getPackages() {
+        return [...(this.settings.packages ?? [])];
+    }
+    setPackages(packages) {
+        this.globalSettings.packages = packages;
+        this.markModified("packages");
+        this.save();
+    }
+    setProjectPackages(packages) {
+        const projectSettings = structuredClone(this.projectSettings);
+        projectSettings.packages = packages;
+        this.markProjectModified("packages");
+        this.saveProjectSettings(projectSettings);
+    }
+    getExtensionPaths() {
+        return [...(this.settings.extensions ?? [])];
+    }
+    setExtensionPaths(paths) {
+        this.globalSettings.extensions = paths;
+        this.markModified("extensions");
+        this.save();
+    }
+    setProjectExtensionPaths(paths) {
+        const projectSettings = structuredClone(this.projectSettings);
+        projectSettings.extensions = paths;
+        this.markProjectModified("extensions");
+        this.saveProjectSettings(projectSettings);
+    }
+    getSkillPaths() {
+        return [...(this.settings.skills ?? [])];
+    }
+    setSkillPaths(paths) {
+        this.globalSettings.skills = paths;
+        this.markModified("skills");
+        this.save();
+    }
+    setProjectSkillPaths(paths) {
+        const projectSettings = structuredClone(this.projectSettings);
+        projectSettings.skills = paths;
+        this.markProjectModified("skills");
+        this.saveProjectSettings(projectSettings);
+    }
+    getPromptTemplatePaths() {
+        return [...(this.settings.prompts ?? [])];
+    }
+    setPromptTemplatePaths(paths) {
+        this.globalSettings.prompts = paths;
+        this.markModified("prompts");
+        this.save();
+    }
+    setProjectPromptTemplatePaths(paths) {
+        const projectSettings = structuredClone(this.projectSettings);
+        projectSettings.prompts = paths;
+        this.markProjectModified("prompts");
+        this.saveProjectSettings(projectSettings);
+    }
+    getThemePaths() {
+        return [...(this.settings.themes ?? [])];
+    }
+    setThemePaths(paths) {
+        this.globalSettings.themes = paths;
+        this.markModified("themes");
+        this.save();
+    }
+    setProjectThemePaths(paths) {
+        const projectSettings = structuredClone(this.projectSettings);
+        projectSettings.themes = paths;
+        this.markProjectModified("themes");
+        this.saveProjectSettings(projectSettings);
+    }
+    getEnableSkillCommands() {
+        return this.settings.enableSkillCommands ?? true;
+    }
+    setEnableSkillCommands(enabled) {
+        this.globalSettings.enableSkillCommands = enabled;
+        this.markModified("enableSkillCommands");
+        this.save();
+    }
+    getThinkingBudgets() {
+        return this.settings.thinkingBudgets;
+    }
+    getShowImages() {
+        return this.settings.terminal?.showImages ?? true;
+    }
+    setShowImages(show) {
+        if (!this.globalSettings.terminal) {
+            this.globalSettings.terminal = {};
+        }
+        this.globalSettings.terminal.showImages = show;
+        this.markModified("terminal", "showImages");
+        this.save();
+    }
+    getClearOnShrink() {
+        // Settings takes precedence, then env var, then default false
+        if (this.settings.terminal?.clearOnShrink !== undefined) {
+            return this.settings.terminal.clearOnShrink;
+        }
+        return process.env.PI_CLEAR_ON_SHRINK === "1";
+    }
+    setClearOnShrink(enabled) {
+        if (!this.globalSettings.terminal) {
+            this.globalSettings.terminal = {};
+        }
+        this.globalSettings.terminal.clearOnShrink = enabled;
+        this.markModified("terminal", "clearOnShrink");
+        this.save();
+    }
+    getImageAutoResize() {
+        return this.settings.images?.autoResize ?? true;
+    }
+    setImageAutoResize(enabled) {
+        if (!this.globalSettings.images) {
+            this.globalSettings.images = {};
+        }
+        this.globalSettings.images.autoResize = enabled;
+        this.markModified("images", "autoResize");
+        this.save();
+    }
+    getBlockImages() {
+        return this.settings.images?.blockImages ?? false;
+    }
+    setBlockImages(blocked) {
+        if (!this.globalSettings.images) {
+            this.globalSettings.images = {};
+        }
+        this.globalSettings.images.blockImages = blocked;
+        this.markModified("images", "blockImages");
+        this.save();
+    }
+    getEnabledModels() {
+        return this.settings.enabledModels;
+    }
+    setEnabledModels(patterns) {
+        this.globalSettings.enabledModels = patterns;
+        this.markModified("enabledModels");
+        this.save();
+    }
+    getDoubleEscapeAction() {
+        return this.settings.doubleEscapeAction ?? "tree";
+    }
+    setDoubleEscapeAction(action) {
+        this.globalSettings.doubleEscapeAction = action;
+        this.markModified("doubleEscapeAction");
+        this.save();
+    }
+    getShowHardwareCursor() {
+        return this.settings.showHardwareCursor ?? process.env.PI_HARDWARE_CURSOR === "1";
+    }
+    setShowHardwareCursor(enabled) {
+        this.globalSettings.showHardwareCursor = enabled;
+        this.markModified("showHardwareCursor");
+        this.save();
+    }
+    getEditorPaddingX() {
+        return this.settings.editorPaddingX ?? 0;
+    }
+    setEditorPaddingX(padding) {
+        this.globalSettings.editorPaddingX = Math.max(0, Math.min(3, Math.floor(padding)));
+        this.markModified("editorPaddingX");
+        this.save();
+    }
+    getAutocompleteMaxVisible() {
+        return this.settings.autocompleteMaxVisible ?? 5;
+    }
+    setAutocompleteMaxVisible(maxVisible) {
+        this.globalSettings.autocompleteMaxVisible = Math.max(3, Math.min(20, Math.floor(maxVisible)));
+        this.markModified("autocompleteMaxVisible");
+        this.save();
+    }
+    getCodeBlockIndent() {
+        return this.settings.markdown?.codeBlockIndent ?? "  ";
+    }
+    getMemoryEnabled() {
+        return this.settings.memory?.enabled ?? true;
+    }
+    setMemoryEnabled(enabled) {
+        if (!this.globalSettings.memory) {
+            this.globalSettings.memory = {};
+        }
+        this.globalSettings.memory.enabled = enabled;
+        this.markModified("memory", "enabled");
+        this.save();
+    }
+}
+//# sourceMappingURL=settings-manager.js.map