@staticpayload/zai-code 1.0.0

package/README.md

@@ -0,0 +1,115 @@
+# Z.ai Code
+
+A Z.ai-native AI code editor — CLI-first, interactive editor shell.
+
+**This tool uses Z.ai API exclusively.** No other AI providers.
+
+## Features
+
+- **Interactive editor shell** — not a chatbot, a code editor
+- **Plan → Diff → Apply workflow** — no auto-execution
+- **Keyboard-driven settings menu** — BIOS-style configuration
+- **Model switching** — select from available Z.ai models
+- **Execution profiles** — safe, balanced, fast presets
+- **Git awareness** — branch, dirty state, warnings
+- **Dry-run mode** — preview without applying
+- **Undo/rollback** — revert last operation
+- **Task history** — audit log of all operations
+- **Safe file operations** — binary blocked, large files warned
+
+## Requirements
+
+- Node.js 18 or higher
+- Z.ai API key
+
+## Installation
+
+```bash
+npm install -g @staticpayload/zai-code
+```
+
+Or run directly:
+
+```bash
+npx @staticpayload/zai-code
+```
+
+## Usage
+
+```bash
+zcode
+```
+
+This opens the interactive editor shell. The prompt shows current state:
+
+```
+[edit][clean]> 
+```
+
+### Commands
+
+**Navigation:**
+- `/help` — show commands
+- `/context` — show current context
+- `/files` — list open files
+- `/open <path>` — add file to context
+
+**Execution:**
+- `/plan` — generate execution plan
+- `/generate` — create file changes
+- `/diff` — review pending changes
+- `/apply` — apply changes
+- `/undo` — rollback last operation
+
+**Modes:**
+- `/mode <edit|explain|review|debug>` — set mode
+- `/dry-run on|off` — toggle dry-run
+- `/profile list|set <name>` — manage profiles
+
+**Tasks:**
+- `/decompose` — break task into steps
+- `/step` — plan current step
+- `/next` — complete step and advance
+
+**System:**
+- `/settings` — open interactive settings menu
+- `/git` — show repository status
+- `/exec <cmd>` — run allowed shell command
+- `/history` — view task history
+- `/doctor` — system health check
+
+### Workflow
+
+1. Enter a task: `add error handling to auth.ts`
+2. System detects intent: `Intent: CODE_EDIT`
+3. Run `/plan` to generate a plan
+4. Run `/generate` to create changes
+5. Run `/diff` to review
+6. Run `/apply` to execute
+
+No changes are made without explicit `/apply`.
+
+## Settings
+
+Run `/settings` to open the interactive menu:
+
+- **Model** — select AI model
+- **UI** — ASCII logo, colors, prompt style
+- **Execution** — confirmation mode, iteration limits
+- **Context** — file scope, token limits
+- **Debug** — logging options
+
+Navigate with arrow keys, Tab between sections, Esc to exit.
+
+## Configuration
+
+Global settings: `~/.zai/settings.json`
+Project settings: `.zai/settings.json` (overrides global)
+
+## License
+
+GPL-3.0
+
+## Author
+
+StaticPayload <chaitanyamishra.ai@gmail.com>

package/bin/zcode

@@ -0,0 +1,3 @@
+#!/usr/bin/env node
+
+require('../dist/cli.js');

package/dist/apply.d.ts

@@ -0,0 +1,44 @@
+import { ResponseSchema } from './runtime';
+export interface ApplyResult {
+    success: boolean;
+    applied: string[];
+    failed: Array<{
+        path: string;
+        error: string;
+    }>;
+}
+export interface ApplyOptions {
+    dryRun?: boolean;
+    basePath?: string;
+}
+export interface PathValidationResult {
+    valid: boolean;
+    resolved: string;
+    error?: string;
+}
+export interface FileOperationResult {
+    success: boolean;
+    error?: string;
+}
+export interface DiffHunk {
+    start: number;
+    end: number;
+    content: string;
+}
+/**
+ * Validate a file path is safe (no path traversal, absolute or within basePath)
+ */
+export declare function validatePath(filePath: string, basePath?: string): PathValidationResult;
+/**
+ * Apply a single file operation atomically
+ */
+export declare function applyFileOperation(operation: 'create' | 'modify' | 'delete', filePath: string, content?: string, options?: ApplyOptions): FileOperationResult;
+/**
+ * Apply diff hunks to a file
+ */
+export declare function applyDiff(filePath: string, hunks: DiffHunk[], options?: ApplyOptions): FileOperationResult;
+/**
+ * Main entry: apply all changes from a ResponseSchema
+ */
+export declare function applyResponse(response: ResponseSchema, options?: ApplyOptions): ApplyResult;
+//# sourceMappingURL=apply.d.ts.map

package/dist/apply.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"apply.d.ts","sourceRoot":"","sources":["../src/apply.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAG3C,MAAM,WAAW,WAAW;IAC1B,OAAO,EAAE,OAAO,CAAC;IACjB,OAAO,EAAE,MAAM,EAAE,CAAC;IAClB,MAAM,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;CAChD;AAED,MAAM,WAAW,YAAY;IAC3B,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,oBAAoB;IACnC,KAAK,EAAE,OAAO,CAAC;IACf,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,mBAAmB;IAClC,OAAO,EAAE,OAAO,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,QAAQ;IACvB,KAAK,EAAE,MAAM,CAAC;IACd,GAAG,EAAE,MAAM,CAAC;IACZ,OAAO,EAAE,MAAM,CAAC;CACjB;AAED;;GAEG;AACH,wBAAgB,YAAY,CAC1B,QAAQ,EAAE,MAAM,EAChB,QAAQ,CAAC,EAAE,MAAM,GAChB,oBAAoB,CA+CtB;AAkCD;;GAEG;AACH,wBAAgB,kBAAkB,CAChC,SAAS,EAAE,QAAQ,GAAG,QAAQ,GAAG,QAAQ,EACzC,QAAQ,EAAE,MAAM,EAChB,OAAO,CAAC,EAAE,MAAM,EAChB,OAAO,CAAC,EAAE,YAAY,GACrB,mBAAmB,CAqFrB;AAED;;GAEG;AACH,wBAAgB,SAAS,CACvB,QAAQ,EAAE,MAAM,EAChB,KAAK,EAAE,QAAQ,EAAE,EACjB,OAAO,CAAC,EAAE,YAAY,GACrB,mBAAmB,CAqErB;AAED;;GAEG;AACH,wBAAgB,aAAa,CAC3B,QAAQ,EAAE,cAAc,EACxB,OAAO,CAAC,EAAE,YAAY,GACrB,WAAW,CAqDb"}

package/dist/apply.js

@@ -0,0 +1,307 @@
+"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.validatePath = validatePath;
+exports.applyFileOperation = applyFileOperation;
+exports.applyDiff = applyDiff;
+exports.applyResponse = applyResponse;
+const fs = __importStar(require("fs"));
+const path = __importStar(require("path"));
+const rollback_1 = require("./rollback");
+/**
+ * Validate a file path is safe (no path traversal, absolute or within basePath)
+ */
+function validatePath(filePath, basePath) {
+    try {
+        let resolvedPath = path.resolve(filePath);
+        // Normalize the path to remove any redundant components
+        resolvedPath = path.normalize(resolvedPath);
+        // If basePath is provided, ensure the resolved path is within it
+        if (basePath) {
+            const resolvedBase = path.resolve(basePath);
+            const resolvedBaseNormalized = path.normalize(resolvedBase);
+            // Check if the resolved path is within the base path
+            const relativePath = path.relative(resolvedBaseNormalized, resolvedPath);
+            // If the relative path starts with '..', it's outside the base path
+            if (relativePath.startsWith('..')) {
+                return {
+                    valid: false,
+                    resolved: resolvedPath,
+                    error: `Path '${filePath}' is outside base path '${basePath}'`,
+                };
+            }
+        }
+        // Check for path traversal attempts in the original input
+        const normalizedInput = path.normalize(filePath);
+        if (normalizedInput.includes('..')) {
+            // If basePath is set, allow relative paths with .. as long as they resolve within basePath
+            if (!basePath) {
+                return {
+                    valid: false,
+                    resolved: resolvedPath,
+                    error: `Path '${filePath}' contains path traversal`,
+                };
+            }
+            // Re-check with basePath logic above
+        }
+        return { valid: true, resolved: resolvedPath };
+    }
+    catch (error) {
+        return {
+            valid: false,
+            resolved: filePath,
+            error: error instanceof Error ? error.message : String(error),
+        };
+    }
+}
+/**
+ * Write content to a file atomically using temp file + rename
+ */
+function atomicWrite(filePath, content) {
+    try {
+        // Ensure parent directory exists
+        const parentDir = path.dirname(filePath);
+        if (!fs.existsSync(parentDir)) {
+            fs.mkdirSync(parentDir, { recursive: true });
+        }
+        // Create temp file in same directory to ensure same filesystem
+        const tempPath = path.join(parentDir, `.tmp_${process.pid}_${Date.now()}_${Math.random().toString(36).slice(2)}`);
+        // Write to temp file
+        fs.writeFileSync(tempPath, content, 'utf-8');
+        // Atomic rename
+        fs.renameSync(tempPath, filePath);
+        return { success: true };
+    }
+    catch (error) {
+        return {
+            success: false,
+            error: error instanceof Error ? error.message : String(error),
+        };
+    }
+}
+/**
+ * Apply a single file operation atomically
+ */
+function applyFileOperation(operation, filePath, content, options) {
+    // Validate the path
+    const validation = validatePath(filePath, options?.basePath);
+    if (!validation.valid) {
+        return { success: false, error: validation.error };
+    }
+    const resolvedPath = validation.resolved;
+    // Block binary files
+    const binaryExtensions = ['.exe', '.dll', '.so', '.dylib', '.bin', '.o', '.a', '.lib', '.png', '.jpg', '.jpeg', '.gif', '.ico', '.pdf', '.zip', '.tar', '.gz', '.rar'];
+    const ext = path.extname(filePath).toLowerCase();
+    if (binaryExtensions.includes(ext)) {
+        return { success: false, error: 'Binary file modification blocked' };
+    }
+    // Warn on large content
+    if (content && content.length > 50000) {
+        console.log(`Warning: Large file (${Math.round(content.length / 1000)}KB)`);
+    }
+    // Dry run - just validate
+    if (options?.dryRun) {
+        switch (operation) {
+            case 'create':
+                if (fs.existsSync(resolvedPath)) {
+                    return { success: false, error: `File already exists: ${resolvedPath}` };
+                }
+                break;
+            case 'modify':
+                if (!fs.existsSync(resolvedPath)) {
+                    return { success: false, error: `File does not exist: ${resolvedPath}` };
+                }
+                if (content === undefined) {
+                    return { success: false, error: 'Content required for modify operation' };
+                }
+                break;
+            case 'delete':
+                if (!fs.existsSync(resolvedPath)) {
+                    return { success: false, error: `File does not exist: ${resolvedPath}` };
+                }
+                break;
+        }
+        return { success: true };
+    }
+    // Actual operation
+    switch (operation) {
+        case 'create':
+            if (fs.existsSync(resolvedPath)) {
+                return { success: false, error: `File already exists: ${resolvedPath}` };
+            }
+            if (content === undefined) {
+                return { success: false, error: 'Content required for create operation' };
+            }
+            (0, rollback_1.createBackup)(resolvedPath, 'create');
+            return atomicWrite(resolvedPath, content);
+        case 'modify':
+            if (!fs.existsSync(resolvedPath)) {
+                return { success: false, error: `File does not exist: ${resolvedPath}` };
+            }
+            if (content === undefined) {
+                return { success: false, error: 'Content required for modify operation' };
+            }
+            (0, rollback_1.createBackup)(resolvedPath, 'modify');
+            return atomicWrite(resolvedPath, content);
+        case 'delete':
+            if (!fs.existsSync(resolvedPath)) {
+                return { success: false, error: `File does not exist: ${resolvedPath}` };
+            }
+            (0, rollback_1.createBackup)(resolvedPath, 'delete');
+            try {
+                fs.unlinkSync(resolvedPath);
+                return { success: true };
+            }
+            catch (error) {
+                return {
+                    success: false,
+                    error: error instanceof Error ? error.message : String(error),
+                };
+            }
+        default:
+            return { success: false, error: `Unknown operation: ${operation}` };
+    }
+}
+/**
+ * Apply diff hunks to a file
+ */
+function applyDiff(filePath, hunks, options) {
+    // Validate the path
+    const validation = validatePath(filePath, options?.basePath);
+    if (!validation.valid) {
+        return { success: false, error: validation.error };
+    }
+    const resolvedPath = validation.resolved;
+    // Check file exists
+    if (!fs.existsSync(resolvedPath)) {
+        return { success: false, error: `File does not exist: ${resolvedPath}` };
+    }
+    // Read existing file
+    let lines;
+    try {
+        const content = fs.readFileSync(resolvedPath, 'utf-8');
+        lines = content.split('\n');
+    }
+    catch (error) {
+        return {
+            success: false,
+            error: error instanceof Error ? error.message : String(error),
+        };
+    }
+    // Sort hunks in reverse order by start line to apply from bottom to top
+    // This prevents line number shifts from affecting earlier hunks
+    const sortedHunks = [...hunks].sort((a, b) => b.start - a.start);
+    // Apply each hunk
+    for (const hunk of sortedHunks) {
+        const { start, end, content } = hunk;
+        // Convert from 1-indexed to 0-indexed
+        const startIndex = start - 1;
+        const endIndex = end - 1;
+        // Validate line numbers
+        if (startIndex < 0 || startIndex >= lines.length) {
+            return {
+                success: false,
+                error: `Invalid hunk start line: ${start} (file has ${lines.length} lines)`,
+            };
+        }
+        if (endIndex < startIndex || endIndex >= lines.length) {
+            return {
+                success: false,
+                error: `Invalid hunk end line: ${end} (file has ${lines.length} lines)`,
+            };
+        }
+        // Split content into lines
+        const newLines = content.split('\n');
+        // Replace lines from start to end with new content
+        lines.splice(startIndex, endIndex - startIndex + 1, ...newLines);
+    }
+    // Dry run - don't write
+    if (options?.dryRun) {
+        return { success: true };
+    }
+    // Create backup before applying diff
+    (0, rollback_1.createBackup)(resolvedPath, 'modify');
+    // Write atomically
+    const newContent = lines.join('\n');
+    return atomicWrite(resolvedPath, newContent);
+}
+/**
+ * Main entry: apply all changes from a ResponseSchema
+ */
+function applyResponse(response, options) {
+    const result = {
+        success: true,
+        applied: [],
+        failed: [],
+    };
+    // Track if any operation failed
+    let hasFailures = false;
+    // Apply file operations if present
+    if (response.files && response.files.length > 0) {
+        for (const fileOp of response.files) {
+            const opResult = applyFileOperation(fileOp.operation, fileOp.path, fileOp.content, options);
+            if (opResult.success) {
+                result.applied.push(`${fileOp.operation}:${fileOp.path}`);
+            }
+            else {
+                result.failed.push({
+                    path: fileOp.path,
+                    error: opResult.error || 'Unknown error',
+                });
+                hasFailures = true;
+            }
+        }
+    }
+    // Apply diff operations if present
+    if (response.diffs && response.diffs.length > 0) {
+        for (const diff of response.diffs) {
+            const diffResult = applyDiff(diff.file, diff.hunks, options);
+            if (diffResult.success) {
+                result.applied.push(`diff:${diff.file}`);
+            }
+            else {
+                result.failed.push({
+                    path: diff.file,
+                    error: diffResult.error || 'Unknown error',
+                });
+                hasFailures = true;
+            }
+        }
+    }
+    // Set overall success status
+    result.success = !hasFailures && result.failed.length === 0;
+    return result;
+}
+//# sourceMappingURL=apply.js.map

package/dist/apply.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"apply.js","sourceRoot":"","sources":["../src/apply.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAqCA,oCAkDC;AAqCD,gDA0FC;AAKD,8BAyEC;AAKD,sCAwDC;AAjWD,uCAAyB;AACzB,2CAA6B;AAG7B,yCAA0C;AA8B1C;;GAEG;AACH,SAAgB,YAAY,CAC1B,QAAgB,EAChB,QAAiB;IAEjB,IAAI,CAAC;QACH,IAAI,YAAY,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAE1C,wDAAwD;QACxD,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QAE5C,iEAAiE;QACjE,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,YAAY,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;YAC5C,MAAM,sBAAsB,GAAG,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;YAE5D,qDAAqD;YACrD,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,sBAAsB,EAAE,YAAY,CAAC,CAAC;YAEzE,oEAAoE;YACpE,IAAI,YAAY,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;gBAClC,OAAO;oBACL,KAAK,EAAE,KAAK;oBACZ,QAAQ,EAAE,YAAY;oBACtB,KAAK,EAAE,SAAS,QAAQ,2BAA2B,QAAQ,GAAG;iBAC/D,CAAC;YACJ,CAAC;QACH,CAAC;QAED,0DAA0D;QAC1D,MAAM,eAAe,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QACjD,IAAI,eAAe,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;YACnC,2FAA2F;YAC3F,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,OAAO;oBACL,KAAK,EAAE,KAAK;oBACZ,QAAQ,EAAE,YAAY;oBACtB,KAAK,EAAE,SAAS,QAAQ,2BAA2B;iBACpD,CAAC;YACJ,CAAC;YACD,qCAAqC;QACvC,CAAC;QAED,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,YAAY,EAAE,CAAC;IACjD,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO;YACL,KAAK,EAAE,KAAK;YACZ,QAAQ,EAAE,QAAQ;YAClB,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;SAC9D,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAS,WAAW,CAAC,QAAgB,EAAE,OAAe;IACpD,IAAI,CAAC;QACH,iCAAiC;QACjC,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACzC,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;YAC9B,EAAE,CAAC,SAAS,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC/C,CAAC;QAED,+DAA+D;QAC/D,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CACxB,SAAS,EACT,QAAQ,OAAO,CAAC,GAAG,IAAI,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAC3E,CAAC;QAEF,qBAAqB;QACrB,EAAE,CAAC,aAAa,CAAC,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;QAE7C,gBAAgB;QAChB,EAAE,CAAC,UAAU,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QAElC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAC3B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO;YACL,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;SAC9D,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAgB,kBAAkB,CAChC,SAAyC,EACzC,QAAgB,EAChB,OAAgB,EAChB,OAAsB;IAEtB,oBAAoB;IACpB,MAAM,UAAU,GAAG,YAAY,CAAC,QAAQ,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAC;IAC7D,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;QACtB,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,UAAU,CAAC,KAAK,EAAE,CAAC;IACrD,CAAC;IACD,MAAM,YAAY,GAAG,UAAU,CAAC,QAAQ,CAAC;IAEzC,qBAAqB;IACrB,MAAM,gBAAgB,GAAG,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC;IACvK,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;IACjD,IAAI,gBAAgB,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;QACnC,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,kCAAkC,EAAE,CAAC;IACvE,CAAC;IAED,wBAAwB;IACxB,IAAI,OAAO,IAAI,OAAO,CAAC,MAAM,GAAG,KAAK,EAAE,CAAC;QACtC,OAAO,CAAC,GAAG,CAAC,wBAAwB,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;IAC9E,CAAC;IAED,0BAA0B;IAC1B,IAAI,OAAO,EAAE,MAAM,EAAE,CAAC;QACpB,QAAQ,SAAS,EAAE,CAAC;YAClB,KAAK,QAAQ;gBACX,IAAI,EAAE,CAAC,UAAU,CAAC,YAAY,CAAC,EAAE,CAAC;oBAChC,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,wBAAwB,YAAY,EAAE,EAAE,CAAC;gBAC3E,CAAC;gBACD,MAAM;YACR,KAAK,QAAQ;gBACX,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,YAAY,CAAC,EAAE,CAAC;oBACjC,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,wBAAwB,YAAY,EAAE,EAAE,CAAC;gBAC3E,CAAC;gBACD,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;oBAC1B,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,uCAAuC,EAAE,CAAC;gBAC5E,CAAC;gBACD,MAAM;YACR,KAAK,QAAQ;gBACX,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,YAAY,CAAC,EAAE,CAAC;oBACjC,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,wBAAwB,YAAY,EAAE,EAAE,CAAC;gBAC3E,CAAC;gBACD,MAAM;QACV,CAAC;QACD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAC3B,CAAC;IAED,mBAAmB;IACnB,QAAQ,SAAS,EAAE,CAAC;QAClB,KAAK,QAAQ;YACX,IAAI,EAAE,CAAC,UAAU,CAAC,YAAY,CAAC,EAAE,CAAC;gBAChC,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,wBAAwB,YAAY,EAAE,EAAE,CAAC;YAC3E,CAAC;YACD,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;gBAC1B,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,uCAAuC,EAAE,CAAC;YAC5E,CAAC;YACD,IAAA,uBAAY,EAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;YACrC,OAAO,WAAW,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;QAE5C,KAAK,QAAQ;YACX,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,YAAY,CAAC,EAAE,CAAC;gBACjC,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,wBAAwB,YAAY,EAAE,EAAE,CAAC;YAC3E,CAAC;YACD,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;gBAC1B,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,uCAAuC,EAAE,CAAC;YAC5E,CAAC;YACD,IAAA,uBAAY,EAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;YACrC,OAAO,WAAW,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;QAE5C,KAAK,QAAQ;YACX,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,YAAY,CAAC,EAAE,CAAC;gBACjC,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,wBAAwB,YAAY,EAAE,EAAE,CAAC;YAC3E,CAAC;YACD,IAAA,uBAAY,EAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;YACrC,IAAI,CAAC;gBACH,EAAE,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;gBAC5B,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;YAC3B,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;iBAC9D,CAAC;YACJ,CAAC;QAEH;YACE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,sBAAsB,SAAS,EAAE,EAAE,CAAC;IACxE,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAgB,SAAS,CACvB,QAAgB,EAChB,KAAiB,EACjB,OAAsB;IAEtB,oBAAoB;IACpB,MAAM,UAAU,GAAG,YAAY,CAAC,QAAQ,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAC;IAC7D,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;QACtB,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,UAAU,CAAC,KAAK,EAAE,CAAC;IACrD,CAAC;IACD,MAAM,YAAY,GAAG,UAAU,CAAC,QAAQ,CAAC;IAEzC,oBAAoB;IACpB,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,YAAY,CAAC,EAAE,CAAC;QACjC,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,wBAAwB,YAAY,EAAE,EAAE,CAAC;IAC3E,CAAC;IAED,qBAAqB;IACrB,IAAI,KAAe,CAAC;IACpB,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;QACvD,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC9B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO;YACL,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;SAC9D,CAAC;IACJ,CAAC;IAED,wEAAwE;IACxE,gEAAgE;IAChE,MAAM,WAAW,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;IAEjE,kBAAkB;IAClB,KAAK,MAAM,IAAI,IAAI,WAAW,EAAE,CAAC;QAC/B,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC;QAErC,sCAAsC;QACtC,MAAM,UAAU,GAAG,KAAK,GAAG,CAAC,CAAC;QAC7B,MAAM,QAAQ,GAAG,GAAG,GAAG,CAAC,CAAC;QAEzB,wBAAwB;QACxB,IAAI,UAAU,GAAG,CAAC,IAAI,UAAU,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;YACjD,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,4BAA4B,KAAK,cAAc,KAAK,CAAC,MAAM,SAAS;aAC5E,CAAC;QACJ,CAAC;QACD,IAAI,QAAQ,GAAG,UAAU,IAAI,QAAQ,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;YACtD,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,0BAA0B,GAAG,cAAc,KAAK,CAAC,MAAM,SAAS;aACxE,CAAC;QACJ,CAAC;QAED,2BAA2B;QAC3B,MAAM,QAAQ,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAErC,mDAAmD;QACnD,KAAK,CAAC,MAAM,CAAC,UAAU,EAAE,QAAQ,GAAG,UAAU,GAAG,CAAC,EAAE,GAAG,QAAQ,CAAC,CAAC;IACnE,CAAC;IAED,wBAAwB;IACxB,IAAI,OAAO,EAAE,MAAM,EAAE,CAAC;QACpB,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAC3B,CAAC;IAED,qCAAqC;IACrC,IAAA,uBAAY,EAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;IAErC,mBAAmB;IACnB,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACpC,OAAO,WAAW,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;AAC/C,CAAC;AAED;;GAEG;AACH,SAAgB,aAAa,CAC3B,QAAwB,EACxB,OAAsB;IAEtB,MAAM,MAAM,GAAgB;QAC1B,OAAO,EAAE,IAAI;QACb,OAAO,EAAE,EAAE;QACX,MAAM,EAAE,EAAE;KACX,CAAC;IAEF,gCAAgC;IAChC,IAAI,WAAW,GAAG,KAAK,CAAC;IAExB,mCAAmC;IACnC,IAAI,QAAQ,CAAC,KAAK,IAAI,QAAQ,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAChD,KAAK,MAAM,MAAM,IAAI,QAAQ,CAAC,KAAK,EAAE,CAAC;YACpC,MAAM,QAAQ,GAAG,kBAAkB,CACjC,MAAM,CAAC,SAAS,EAChB,MAAM,CAAC,IAAI,EACX,MAAM,CAAC,OAAO,EACd,OAAO,CACR,CAAC;YAEF,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;gBACrB,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;YAC5D,CAAC;iBAAM,CAAC;gBACN,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC;oBACjB,IAAI,EAAE,MAAM,CAAC,IAAI;oBACjB,KAAK,EAAE,QAAQ,CAAC,KAAK,IAAI,eAAe;iBACzC,CAAC,CAAC;gBACH,WAAW,GAAG,IAAI,CAAC;YACrB,CAAC;QACH,CAAC;IACH,CAAC;IAED,mCAAmC;IACnC,IAAI,QAAQ,CAAC,KAAK,IAAI,QAAQ,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAChD,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,KAAK,EAAE,CAAC;YAClC,MAAM,UAAU,GAAG,SAAS,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;YAE7D,IAAI,UAAU,CAAC,OAAO,EAAE,CAAC;gBACvB,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;YAC3C,CAAC;iBAAM,CAAC;gBACN,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC;oBACjB,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,KAAK,EAAE,UAAU,CAAC,KAAK,IAAI,eAAe;iBAC3C,CAAC,CAAC;gBACH,WAAW,GAAG,IAAI,CAAC;YACrB,CAAC;QACH,CAAC;IACH,CAAC;IAED,6BAA6B;IAC7B,MAAM,CAAC,OAAO,GAAG,CAAC,WAAW,IAAI,MAAM,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,CAAC;IAE5D,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/auth.d.ts

@@ -0,0 +1,9 @@
+export declare function getApiKey(): Promise<string | null>;
+export declare function setApiKey(key: string): Promise<void>;
+export declare function deleteApiKey(): Promise<void>;
+export declare function hasValidCredentials(): Promise<boolean>;
+export declare function validateApiKey(key: string): Promise<boolean>;
+export declare function promptForApiKey(): Promise<string>;
+export declare function runOnboarding(): Promise<void>;
+export declare function ensureAuthenticated(): Promise<string>;
+//# sourceMappingURL=auth.d.ts.map

package/dist/auth.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"auth.d.ts","sourceRoot":"","sources":["../src/auth.ts"],"names":[],"mappings":"AAqCA,wBAAsB,SAAS,IAAI,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,CAMxD;AAED,wBAAsB,SAAS,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAE1D;AAED,wBAAsB,YAAY,IAAI,OAAO,CAAC,IAAI,CAAC,CAElD;AAED,wBAAsB,mBAAmB,IAAI,OAAO,CAAC,OAAO,CAAC,CAG5D;AAED,wBAAsB,cAAc,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAUlE;AAED,wBAAsB,eAAe,IAAI,OAAO,CAAC,MAAM,CAAC,CA4CvD;AAED,wBAAsB,aAAa,IAAI,OAAO,CAAC,IAAI,CAAC,CAanD;AAED,wBAAsB,mBAAmB,IAAI,OAAO,CAAC,MAAM,CAAC,CAU3D"}

package/dist/auth.js

@@ -0,0 +1,170 @@
+"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.getApiKey = getApiKey;
+exports.setApiKey = setApiKey;
+exports.deleteApiKey = deleteApiKey;
+exports.hasValidCredentials = hasValidCredentials;
+exports.validateApiKey = validateApiKey;
+exports.promptForApiKey = promptForApiKey;
+exports.runOnboarding = runOnboarding;
+exports.ensureAuthenticated = ensureAuthenticated;
+const keytar = __importStar(require("keytar"));
+const readline = __importStar(require("readline"));
+const config_1 = require("./config");
+const SERVICE_NAME = 'zai-code';
+const ACCOUNT_NAME = 'api-key';
+async function httpsGet(url, headers) {
+    return new Promise((resolve, reject) => {
+        const parsedUrl = new URL(url);
+        const options = {
+            hostname: parsedUrl.hostname,
+            port: parsedUrl.port || (parsedUrl.protocol === 'https:' ? 443 : 80),
+            path: parsedUrl.pathname + parsedUrl.search,
+            method: 'GET',
+            headers: {
+                ...headers,
+                'User-Agent': 'zai-cli',
+            },
+        };
+        const req = require('https').request(options, (res) => {
+            let data = '';
+            res.on('data', (chunk) => {
+                data += chunk;
+            });
+            res.on('end', () => {
+                resolve({ statusCode: res.statusCode, data });
+            });
+        });
+        req.on('error', reject);
+        req.end();
+    });
+}
+async function getApiKey() {
+    try {
+        return await keytar.getPassword(SERVICE_NAME, ACCOUNT_NAME);
+    }
+    catch (error) {
+        return null;
+    }
+}
+async function setApiKey(key) {
+    await keytar.setPassword(SERVICE_NAME, ACCOUNT_NAME, key);
+}
+async function deleteApiKey() {
+    await keytar.deletePassword(SERVICE_NAME, ACCOUNT_NAME);
+}
+async function hasValidCredentials() {
+    const key = await getApiKey();
+    return key !== null && key.length > 0;
+}
+async function validateApiKey(key) {
+    try {
+        const config = (0, config_1.loadConfig)();
+        const response = await httpsGet(`${config.api.baseUrl}/models`, {
+            Authorization: `Bearer ${key}`,
+        });
+        return response.statusCode >= 200 && response.statusCode < 300;
+    }
+    catch {
+        return false;
+    }
+}
+async function promptForApiKey() {
+    return new Promise((resolve) => {
+        const rl = readline.createInterface({
+            input: process.stdin,
+            output: process.stdout,
+        });
+        // Use readline with muted input for hidden password
+        const stdin = process.stdin;
+        const stdout = process.stdout;
+        stdout.write('Enter API key: ');
+        stdin.setRawMode(true);
+        stdin.resume();
+        stdin.setEncoding('utf8');
+        let key = '';
+        const onData = (char) => {
+            if (char === '\r' || char === '\n' || char === '\u0004') {
+                // Enter or Ctrl-D
+                stdin.setRawMode(false);
+                stdin.pause();
+                stdin.removeListener('data', onData);
+                stdout.write('\n');
+                rl.close();
+                resolve(key);
+            }
+            else if (char === '\u0003') {
+                // Ctrl-C
+                stdout.write('\n');
+                process.exit(0);
+            }
+            else if (char === '\u007f') {
+                // Backspace
+                if (key.length > 0) {
+                    key = key.slice(0, -1);
+                }
+            }
+            else {
+                key += char;
+            }
+        };
+        stdin.on('data', onData);
+    });
+}
+async function runOnboarding() {
+    while (true) {
+        console.log('Z.ai API key required');
+        const key = await promptForApiKey();
+        if (await validateApiKey(key)) {
+            await setApiKey(key);
+            console.log('Authentication successful');
+            return;
+        }
+        else {
+            console.log('Invalid API key');
+        }
+    }
+}
+async function ensureAuthenticated() {
+    if (await hasValidCredentials()) {
+        const key = await getApiKey();
+        return key;
+    }
+    await runOnboarding();
+    const key = await getApiKey();
+    return key;
+}
+//# sourceMappingURL=auth.js.map