@staticpayload/zai-code 1.0.0

package/dist/workspace_model.js

@@ -0,0 +1,253 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WORKSPACE_STATE_FILE = exports.Workspace = void 0;
+exports.getWorkspace = getWorkspace;
+exports.resetWorkspace = resetWorkspace;
+const fs = __importStar(require("fs"));
+const path = __importStar(require("path"));
+const session_1 = require("./session");
+const context_builder_1 = require("./context/context_builder");
+// Workspace state file location
+const WORKSPACE_STATE_FILE = '.zai/workspace.json';
+exports.WORKSPACE_STATE_FILE = WORKSPACE_STATE_FILE;
+// Workspace model - authoritative state
+class Workspace {
+    constructor(rootPath) {
+        this.fileTree = null;
+        this.fileIndex = [];
+        this.rootPath = rootPath || process.cwd();
+    }
+    // Get root path
+    getRoot() {
+        return this.rootPath;
+    }
+    // Index the file tree
+    indexFileTree(maxDepth = 5) {
+        const buildTree = (dirPath, depth) => {
+            const name = path.basename(dirPath) || dirPath;
+            const relativePath = path.relative(this.rootPath, dirPath);
+            const node = {
+                name,
+                path: relativePath || '.',
+                type: 'directory',
+                children: [],
+            };
+            if (depth >= maxDepth)
+                return node;
+            try {
+                const entries = fs.readdirSync(dirPath, { withFileTypes: true });
+                const skipDirs = new Set(['node_modules', '.git', 'dist', 'build', '.next', '__pycache__', 'coverage']);
+                for (const entry of entries) {
+                    if (entry.name.startsWith('.') && entry.name !== '.zai')
+                        continue;
+                    if (skipDirs.has(entry.name))
+                        continue;
+                    const fullPath = path.join(dirPath, entry.name);
+                    const relPath = path.relative(this.rootPath, fullPath);
+                    if (entry.isDirectory()) {
+                        node.children.push(buildTree(fullPath, depth + 1));
+                    }
+                    else if (entry.isFile()) {
+                        try {
+                            const stats = fs.statSync(fullPath);
+                            node.children.push({
+                                name: entry.name,
+                                path: relPath,
+                                type: 'file',
+                                size: stats.size,
+                                extension: path.extname(entry.name),
+                            });
+                        }
+                        catch {
+                            // Skip files we can't stat
+                        }
+                    }
+                }
+                // Sort: directories first, then files
+                node.children.sort((a, b) => {
+                    if (a.type !== b.type)
+                        return a.type === 'directory' ? -1 : 1;
+                    return a.name.localeCompare(b.name);
+                });
+            }
+            catch {
+                // Skip directories we can't read
+            }
+            return node;
+        };
+        this.fileTree = buildTree(this.rootPath, 0);
+        this.fileIndex = (0, context_builder_1.indexWorkspace)(this.rootPath);
+        return this.fileTree;
+    }
+    // Get file tree
+    getFileTree() {
+        return this.fileTree;
+    }
+    // Get file index
+    getFileIndex() {
+        return this.fileIndex;
+    }
+    // Get current state from session
+    getCurrentState() {
+        const session = (0, session_1.getSession)();
+        return {
+            version: 1,
+            workingDirectory: session.workingDirectory,
+            mode: session.mode,
+            openFiles: session.openFiles,
+            currentIntent: session.currentIntent,
+            intentType: session.intentType,
+            lastPlan: session.lastPlan,
+            lastDiff: session.lastDiff,
+            pendingActions: session.pendingActions,
+            dryRun: session.dryRun,
+            lastUpdated: new Date().toISOString(),
+        };
+    }
+    // Save workspace state to disk
+    saveState() {
+        const state = this.getCurrentState();
+        const statePath = path.join(this.rootPath, WORKSPACE_STATE_FILE);
+        const stateDir = path.dirname(statePath);
+        try {
+            if (!fs.existsSync(stateDir)) {
+                fs.mkdirSync(stateDir, { recursive: true });
+            }
+            fs.writeFileSync(statePath, JSON.stringify(state, null, 2), 'utf-8');
+        }
+        catch {
+            // Fail silently - state persistence is optional
+        }
+    }
+    // Load workspace state from disk
+    loadState() {
+        const statePath = path.join(this.rootPath, WORKSPACE_STATE_FILE);
+        try {
+            if (!fs.existsSync(statePath)) {
+                return null;
+            }
+            const content = fs.readFileSync(statePath, 'utf-8');
+            return JSON.parse(content);
+        }
+        catch {
+            return null;
+        }
+    }
+    // Restore session from saved state
+    restoreState() {
+        const state = this.loadState();
+        if (!state)
+            return false;
+        try {
+            // Restore mode
+            (0, session_1.setMode)(state.mode);
+            // Restore open files
+            (0, session_1.clearOpenFiles)();
+            for (const file of state.openFiles) {
+                (0, session_1.addOpenFile)(file);
+            }
+            // Restore intent
+            if (state.currentIntent) {
+                (0, session_1.setIntent)(state.currentIntent);
+            }
+            if (state.intentType) {
+                (0, session_1.setIntentType)(state.intentType);
+            }
+            // Restore plan and diff
+            if (state.lastPlan) {
+                (0, session_1.setLastPlan)(state.lastPlan);
+            }
+            if (state.lastDiff) {
+                (0, session_1.setLastDiff)(state.lastDiff);
+            }
+            if (state.pendingActions) {
+                (0, session_1.setPendingActions)(state.pendingActions);
+            }
+            return true;
+        }
+        catch {
+            return false;
+        }
+    }
+    // Clear saved state
+    clearState() {
+        const statePath = path.join(this.rootPath, WORKSPACE_STATE_FILE);
+        try {
+            if (fs.existsSync(statePath)) {
+                fs.unlinkSync(statePath);
+            }
+        }
+        catch {
+            // Fail silently
+        }
+    }
+    // Print file tree summary
+    printTreeSummary() {
+        if (!this.fileTree) {
+            this.indexFileTree();
+        }
+        const countNodes = (node) => {
+            if (node.type === 'file') {
+                return { files: 1, dirs: 0 };
+            }
+            let files = 0;
+            let dirs = 1;
+            for (const child of node.children || []) {
+                const counts = countNodes(child);
+                files += counts.files;
+                dirs += counts.dirs;
+            }
+            return { files, dirs };
+        };
+        const counts = countNodes(this.fileTree);
+        return `Workspace: ${counts.dirs} directories, ${counts.files} files`;
+    }
+}
+exports.Workspace = Workspace;
+// Global workspace instance
+let currentWorkspace = null;
+// Get or create workspace
+function getWorkspace(rootPath) {
+    if (!currentWorkspace) {
+        currentWorkspace = new Workspace(rootPath);
+    }
+    return currentWorkspace;
+}
+// Reset workspace
+function resetWorkspace() {
+    currentWorkspace = null;
+}
+//# sourceMappingURL=workspace_model.js.map

package/dist/workspace_model.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"workspace_model.js","sourceRoot":"","sources":["../src/workspace_model.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA0PA,oCAKC;AAGD,wCAEC;AApQD,uCAAyB;AACzB,2CAA6B;AAC7B,uCAAqM;AAErM,+DAAsE;AAEtE,gCAAgC;AAChC,MAAM,oBAAoB,GAAG,qBAAqB,CAAC;AA+P1C,oDAAoB;AApO7B,wCAAwC;AACxC,MAAa,SAAS;IAKpB,YAAY,QAAiB;QAHrB,aAAQ,GAAwB,IAAI,CAAC;QACrC,cAAS,GAAgB,EAAE,CAAC;QAGlC,IAAI,CAAC,QAAQ,GAAG,QAAQ,IAAI,OAAO,CAAC,GAAG,EAAE,CAAC;IAC5C,CAAC;IAED,gBAAgB;IAChB,OAAO;QACL,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAED,sBAAsB;IACtB,aAAa,CAAC,WAAmB,CAAC;QAChC,MAAM,SAAS,GAAG,CAAC,OAAe,EAAE,KAAa,EAAgB,EAAE;YACjE,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,OAAO,CAAC;YAC/C,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YAE3D,MAAM,IAAI,GAAiB;gBACzB,IAAI;gBACJ,IAAI,EAAE,YAAY,IAAI,GAAG;gBACzB,IAAI,EAAE,WAAW;gBACjB,QAAQ,EAAE,EAAE;aACb,CAAC;YAEF,IAAI,KAAK,IAAI,QAAQ;gBAAE,OAAO,IAAI,CAAC;YAEnC,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,EAAE,CAAC,WAAW,CAAC,OAAO,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC;gBACjE,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,CAAC,cAAc,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,UAAU,CAAC,CAAC,CAAC;gBAExG,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;oBAC5B,IAAI,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM;wBAAE,SAAS;oBAClE,IAAI,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC;wBAAE,SAAS;oBAEvC,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;oBAChD,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;oBAEvD,IAAI,KAAK,CAAC,WAAW,EAAE,EAAE,CAAC;wBACxB,IAAI,CAAC,QAAS,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC;oBACtD,CAAC;yBAAM,IAAI,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC;wBAC1B,IAAI,CAAC;4BACH,MAAM,KAAK,GAAG,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;4BACpC,IAAI,CAAC,QAAS,CAAC,IAAI,CAAC;gCAClB,IAAI,EAAE,KAAK,CAAC,IAAI;gCAChB,IAAI,EAAE,OAAO;gCACb,IAAI,EAAE,MAAM;gCACZ,IAAI,EAAE,KAAK,CAAC,IAAI;gCAChB,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC;6BACpC,CAAC,CAAC;wBACL,CAAC;wBAAC,MAAM,CAAC;4BACP,2BAA2B;wBAC7B,CAAC;oBACH,CAAC;gBACH,CAAC;gBAED,sCAAsC;gBACtC,IAAI,CAAC,QAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;oBAC3B,IAAI,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,IAAI;wBAAE,OAAO,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;oBAC9D,OAAO,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;gBACtC,CAAC,CAAC,CAAC;YACL,CAAC;YAAC,MAAM,CAAC;gBACP,iCAAiC;YACnC,CAAC;YAED,OAAO,IAAI,CAAC;QACd,CAAC,CAAC;QAEF,IAAI,CAAC,QAAQ,GAAG,SAAS,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;QAC5C,IAAI,CAAC,SAAS,GAAG,IAAA,gCAAc,EAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC/C,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAED,gBAAgB;IAChB,WAAW;QACT,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAED,iBAAiB;IACjB,YAAY;QACV,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED,iCAAiC;IACjC,eAAe;QACb,MAAM,OAAO,GAAG,IAAA,oBAAU,GAAE,CAAC;QAC7B,OAAO;YACL,OAAO,EAAE,CAAC;YACV,gBAAgB,EAAE,OAAO,CAAC,gBAAgB;YAC1C,IAAI,EAAE,OAAO,CAAC,IAAI;YAClB,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,aAAa,EAAE,OAAO,CAAC,aAAa;YACpC,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,QAAQ,EAAE,OAAO,CAAC,QAAQ;YAC1B,QAAQ,EAAE,OAAO,CAAC,QAAQ;YAC1B,cAAc,EAAE,OAAO,CAAC,cAAc;YACtC,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,WAAW,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACtC,CAAC;IACJ,CAAC;IAED,+BAA+B;IAC/B,SAAS;QACP,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,EAAE,CAAC;QACrC,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,oBAAoB,CAAC,CAAC;QACjE,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAEzC,IAAI,CAAC;YACH,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC7B,EAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAC9C,CAAC;YACD,EAAE,CAAC,aAAa,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;QACvE,CAAC;QAAC,MAAM,CAAC;YACP,gDAAgD;QAClD,CAAC;IACH,CAAC;IAED,iCAAiC;IACjC,SAAS;QACP,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,oBAAoB,CAAC,CAAC;QAEjE,IAAI,CAAC;YACH,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;gBAC9B,OAAO,IAAI,CAAC;YACd,CAAC;YACD,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;YACpD,OAAO,IAAI,CAAC,KAAK,CAAC,OAAO,CAAmB,CAAC;QAC/C,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAED,mCAAmC;IACnC,YAAY;QACV,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;QAC/B,IAAI,CAAC,KAAK;YAAE,OAAO,KAAK,CAAC;QAEzB,IAAI,CAAC;YACH,eAAe;YACf,IAAA,iBAAO,EAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAEpB,qBAAqB;YACrB,IAAA,wBAAc,GAAE,CAAC;YACjB,KAAK,MAAM,IAAI,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;gBACnC,IAAA,qBAAW,EAAC,IAAI,CAAC,CAAC;YACpB,CAAC;YAED,iBAAiB;YACjB,IAAI,KAAK,CAAC,aAAa,EAAE,CAAC;gBACxB,IAAA,mBAAS,EAAC,KAAK,CAAC,aAAa,CAAC,CAAC;YACjC,CAAC;YACD,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;gBACrB,IAAA,uBAAa,EAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAClC,CAAC;YAED,wBAAwB;YACxB,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;gBACnB,IAAA,qBAAW,EAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YAC9B,CAAC;YACD,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;gBACnB,IAAA,qBAAW,EAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YAC9B,CAAC;YACD,IAAI,KAAK,CAAC,cAAc,EAAE,CAAC;gBACzB,IAAA,2BAAiB,EAAC,KAAK,CAAC,cAAc,CAAC,CAAC;YAC1C,CAAC;YAED,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED,oBAAoB;IACpB,UAAU;QACR,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,oBAAoB,CAAC,CAAC;QACjE,IAAI,CAAC;YACH,IAAI,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;gBAC7B,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;YAC3B,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,gBAAgB;QAClB,CAAC;IACH,CAAC;IAED,0BAA0B;IAC1B,gBAAgB;QACd,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACnB,IAAI,CAAC,aAAa,EAAE,CAAC;QACvB,CAAC;QAED,MAAM,UAAU,GAAG,CAAC,IAAkB,EAAmC,EAAE;YACzE,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBACzB,OAAO,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;YAC/B,CAAC;YACD,IAAI,KAAK,GAAG,CAAC,CAAC;YACd,IAAI,IAAI,GAAG,CAAC,CAAC;YACb,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,QAAQ,IAAI,EAAE,EAAE,CAAC;gBACxC,MAAM,MAAM,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC;gBACjC,KAAK,IAAI,MAAM,CAAC,KAAK,CAAC;gBACtB,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC;YACtB,CAAC;YACD,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;QACzB,CAAC,CAAC;QAEF,MAAM,MAAM,GAAG,UAAU,CAAC,IAAI,CAAC,QAAS,CAAC,CAAC;QAC1C,OAAO,cAAc,MAAM,CAAC,IAAI,iBAAiB,MAAM,CAAC,KAAK,QAAQ,CAAC;IACxE,CAAC;CACF;AAjND,8BAiNC;AAED,4BAA4B;AAC5B,IAAI,gBAAgB,GAAqB,IAAI,CAAC;AAE9C,0BAA0B;AAC1B,SAAgB,YAAY,CAAC,QAAiB;IAC5C,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACtB,gBAAgB,GAAG,IAAI,SAAS,CAAC,QAAQ,CAAC,CAAC;IAC7C,CAAC;IACD,OAAO,gBAAgB,CAAC;AAC1B,CAAC;AAED,kBAAkB;AAClB,SAAgB,cAAc;IAC5B,gBAAgB,GAAG,IAAI,CAAC;AAC1B,CAAC"}

package/package.json

@@ -0,0 +1,45 @@
+{
+  "name": "@staticpayload/zai-code",
+  "version": "1.0.0",
+  "description": "Z.ai-native AI code editor - CLI-first, interactive editor shell with keyboard-driven settings, plan-diff-apply workflow, and safety-first design",
+  "main": "dist/cli.js",
+  "bin": {
+    "zcode": "./bin/zcode"
+  },
+  "files": [
+    "bin/",
+    "dist/"
+  ],
+  "engines": {
+    "node": ">=18.0.0"
+  },
+  "scripts": {
+    "build": "tsc",
+    "prepublishOnly": "npm run build"
+  },
+  "keywords": [
+    "cli",
+    "zai",
+    "code-editor",
+    "ai",
+    "interactive"
+  ],
+  "author": "StaticPayload <chaitanyamishra.ai@gmail.com>",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/staticpayload/zai-code.git"
+  },
+  "homepage": "https://github.com/staticpayload/zai-code",
+  "bugs": {
+    "url": "https://github.com/staticpayload/zai-code/issues"
+  },
+  "license": "GPL-3.0",
+  "devDependencies": {
+    "@types/keytar": "^4.4.0",
+    "@types/node": "^20.10.6",
+    "typescript": "^5.3.3"
+  },
+  "dependencies": {
+    "keytar": "^7.9.0"
+  }
+}