@staticpayload/zai-code 1.0.0

package/dist/task_runner.js

@@ -0,0 +1,230 @@
+"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.decomposeTask = decomposeTask;
+exports.planCurrentStep = planCurrentStep;
+exports.printProgress = printProgress;
+exports.completeCurrentStep = completeCurrentStep;
+exports.skipCurrentStep = skipCurrentStep;
+const session_1 = require("./session");
+const runtime_1 = require("./runtime");
+const auth_1 = require("./auth");
+const context_builder_1 = require("./context/context_builder");
+const ui_1 = require("./ui");
+const path = __importStar(require("path"));
+// Task decomposition schema
+const TASK_DECOMPOSITION_SCHEMA = {
+    type: 'object',
+    required: ['status', 'steps'],
+    properties: {
+        status: { type: 'string', enum: ['success', 'error'] },
+        steps: {
+            type: 'array',
+            items: {
+                type: 'object',
+                required: ['id', 'description'],
+                properties: {
+                    id: { type: 'string' },
+                    description: { type: 'string' },
+                    dependencies: { type: 'array', items: { type: 'string' } },
+                },
+            },
+        },
+        error: { type: 'string' },
+    },
+};
+// Plan response schema
+const PLAN_RESPONSE_SCHEMA = {
+    type: 'object',
+    required: ['status', 'plan'],
+    properties: {
+        status: { type: 'string', enum: ['success', 'error'] },
+        plan: {
+            type: 'array',
+            items: {
+                type: 'object',
+                required: ['id', 'description'],
+                properties: {
+                    id: { type: 'string' },
+                    description: { type: 'string' },
+                    files: { type: 'array', items: { type: 'string' } },
+                },
+            },
+        },
+        error: { type: 'string' },
+    },
+};
+// Decompose a large task into steps
+async function decomposeTask() {
+    const session = (0, session_1.getSession)();
+    const intent = (0, session_1.getIntent)();
+    if (!intent) {
+        return { success: false, steps: [], message: 'No intent set.' };
+    }
+    const apiKey = await (0, auth_1.ensureAuthenticated)();
+    // Build context
+    const context = (0, context_builder_1.buildContext)(session.workingDirectory, intent, (0, session_1.getIntentType)() || 'COMMAND', session.openFiles.map(f => path.isAbsolute(f) ? f : path.join(session.workingDirectory, f)));
+    const filesContext = (0, context_builder_1.formatContextForModel)(context);
+    const instruction = `Break down this task into discrete, sequential steps.
+Each step should be a single, focused unit of work.
+
+Task: ${intent}
+Mode: ${session.mode}
+Working directory: ${session.workingDirectory}
+
+${filesContext}
+
+Output 2-5 steps. Each step should have an id (step1, step2, ...) and a clear description.
+Order steps by dependency - prerequisites first.`;
+    const result = await (0, runtime_1.execute)({
+        instruction,
+        schema: TASK_DECOMPOSITION_SCHEMA,
+        enforceSchema: true,
+    }, apiKey);
+    if (!result.success) {
+        return { success: false, steps: [], message: `Decomposition failed: ${result.error}` };
+    }
+    const response = result.output;
+    if (response.status === 'error' || !response.steps) {
+        return { success: false, steps: [], message: `Decomposition failed: ${response.error}` };
+    }
+    const taskSteps = response.steps.map(s => ({
+        id: s.id,
+        description: s.description,
+        status: 'pending',
+    }));
+    (0, session_1.setTaskSteps)(taskSteps);
+    return { success: true, steps: taskSteps, message: `Task decomposed into ${taskSteps.length} steps.` };
+}
+// Plan current step
+async function planCurrentStep() {
+    const step = (0, session_1.getCurrentStep)();
+    if (!step) {
+        return { success: false, message: 'No current step. Use /decompose first.' };
+    }
+    const session = (0, session_1.getSession)();
+    const apiKey = await (0, auth_1.ensureAuthenticated)();
+    const context = (0, context_builder_1.buildContext)(session.workingDirectory, step.description, (0, session_1.getIntentType)() || 'COMMAND', session.openFiles.map(f => path.isAbsolute(f) ? f : path.join(session.workingDirectory, f)));
+    const filesContext = (0, context_builder_1.formatContextForModel)(context);
+    const instruction = `Create a plan for this step:
+
+Step: ${step.description}
+Overall task: ${(0, session_1.getIntent)()}
+Mode: ${session.mode}
+
+${filesContext}
+
+Output a detailed plan for this step only.`;
+    const result = await (0, runtime_1.execute)({
+        instruction,
+        schema: PLAN_RESPONSE_SCHEMA,
+        enforceSchema: true,
+    }, apiKey);
+    if (!result.success) {
+        return { success: false, message: `Planning failed: ${result.error}` };
+    }
+    const response = result.output;
+    if (response.status === 'error' || !response.plan) {
+        return { success: false, message: `Planning failed: ${response.error}` };
+    }
+    step.plan = response.plan.map(p => ({
+        id: p.id,
+        description: p.description,
+        status: 'pending',
+    }));
+    (0, session_1.setLastPlan)(step.plan);
+    (0, session_1.updateStepStatus)(step.id, 'planned');
+    return { success: true, message: `Step ${step.id} planned.` };
+}
+// Print step progress
+function printProgress() {
+    const session = (0, session_1.getSession)();
+    const progress = (0, session_1.getStepProgress)();
+    if (session.taskSteps.length === 0) {
+        console.log('No task decomposed.');
+        return;
+    }
+    console.log(`Progress: Step ${progress.current}/${progress.total} (${progress.completed} completed)`);
+    console.log('');
+    for (let i = 0; i < session.taskSteps.length; i++) {
+        const step = session.taskSteps[i];
+        const isCurrent = i === session.currentStepIndex;
+        const marker = step.status === 'applied' ? '[x]'
+            : step.status === 'skipped' ? '[-]'
+                : isCurrent ? '[>]'
+                    : '[ ]';
+        const prefix = step.status === 'applied' || step.status === 'skipped'
+            ? (0, ui_1.dim)(marker)
+            : isCurrent
+                ? (0, ui_1.info)(marker)
+                : (0, ui_1.dim)(marker);
+        console.log(`${prefix} ${step.id}: ${step.description}`);
+    }
+}
+// Mark current step complete and advance
+function completeCurrentStep() {
+    const step = (0, session_1.getCurrentStep)();
+    if (!step) {
+        return { success: false, hasMore: false, message: 'No current step.' };
+    }
+    (0, session_1.updateStepStatus)(step.id, 'applied');
+    const nextStep = (0, session_1.advanceStep)();
+    if (nextStep) {
+        return {
+            success: true,
+            hasMore: true,
+            message: `Step ${step.id} complete. Next: ${nextStep.description}`,
+        };
+    }
+    return { success: true, hasMore: false, message: 'All steps complete.' };
+}
+// Skip current step
+function skipCurrentStep() {
+    const step = (0, session_1.getCurrentStep)();
+    if (!step) {
+        return { success: false, hasMore: false, message: 'No current step.' };
+    }
+    (0, session_1.updateStepStatus)(step.id, 'skipped');
+    const nextStep = (0, session_1.advanceStep)();
+    if (nextStep) {
+        return {
+            success: true,
+            hasMore: true,
+            message: `Step ${step.id} skipped. Next: ${nextStep.description}`,
+        };
+    }
+    return { success: true, hasMore: false, message: 'All steps complete.' };
+}
+//# sourceMappingURL=task_runner.js.map

package/dist/task_runner.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"task_runner.js","sourceRoot":"","sources":["../src/task_runner.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAiEA,sCAwDC;AAGD,0CAsDC;AAGD,sCA0BC;AAGD,kDAmBC;AAGD,0CAmBC;AA3PD,uCAamB;AACnB,uCAAoD;AACpD,iCAA6C;AAC7C,+DAAgF;AAChF,6BAAiD;AACjD,2CAA6B;AAE7B,4BAA4B;AAC5B,MAAM,yBAAyB,GAAG;IAChC,IAAI,EAAE,QAAQ;IACd,QAAQ,EAAE,CAAC,QAAQ,EAAE,OAAO,CAAC;IAC7B,UAAU,EAAE;QACV,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,SAAS,EAAE,OAAO,CAAC,EAAE;QACtD,KAAK,EAAE;YACL,IAAI,EAAE,OAAO;YACb,KAAK,EAAE;gBACL,IAAI,EAAE,QAAQ;gBACd,QAAQ,EAAE,CAAC,IAAI,EAAE,aAAa,CAAC;gBAC/B,UAAU,EAAE;oBACV,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;oBACtB,WAAW,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;oBAC/B,YAAY,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;iBAC3D;aACF;SACF;QACD,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;KAC1B;CACO,CAAC;AAEX,uBAAuB;AACvB,MAAM,oBAAoB,GAAG;IAC3B,IAAI,EAAE,QAAQ;IACd,QAAQ,EAAE,CAAC,QAAQ,EAAE,MAAM,CAAC;IAC5B,UAAU,EAAE;QACV,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,SAAS,EAAE,OAAO,CAAC,EAAE;QACtD,IAAI,EAAE;YACJ,IAAI,EAAE,OAAO;YACb,KAAK,EAAE;gBACL,IAAI,EAAE,QAAQ;gBACd,QAAQ,EAAE,CAAC,IAAI,EAAE,aAAa,CAAC;gBAC/B,UAAU,EAAE;oBACV,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;oBACtB,WAAW,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;oBAC/B,KAAK,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;iBACpD;aACF;SACF;QACD,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;KAC1B;CACO,CAAC;AAEX,oCAAoC;AAC7B,KAAK,UAAU,aAAa;IACjC,MAAM,OAAO,GAAG,IAAA,oBAAU,GAAE,CAAC;IAC7B,MAAM,MAAM,GAAG,IAAA,mBAAS,GAAE,CAAC;IAE3B,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC;IAClE,CAAC;IAED,MAAM,MAAM,GAAG,MAAM,IAAA,0BAAmB,GAAE,CAAC;IAE3C,gBAAgB;IAChB,MAAM,OAAO,GAAG,IAAA,8BAAY,EAC1B,OAAO,CAAC,gBAAgB,EACxB,MAAM,EACN,IAAA,uBAAa,GAAE,IAAI,SAAS,EAC5B,OAAO,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC,CAC5F,CAAC;IACF,MAAM,YAAY,GAAG,IAAA,uCAAqB,EAAC,OAAO,CAAC,CAAC;IAEpD,MAAM,WAAW,GAAG;;;QAGd,MAAM;QACN,OAAO,CAAC,IAAI;qBACC,OAAO,CAAC,gBAAgB;;EAE3C,YAAY;;;iDAGmC,CAAC;IAEhD,MAAM,MAAM,GAAG,MAAM,IAAA,iBAAO,EAAC;QAC3B,WAAW;QACX,MAAM,EAAE,yBAAyB;QACjC,aAAa,EAAE,IAAI;KACpB,EAAE,MAAM,CAAC,CAAC;IAEX,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;QACpB,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,OAAO,EAAE,yBAAyB,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;IACzF,CAAC;IAED,MAAM,QAAQ,GAAG,MAAM,CAAC,MAAgG,CAAC;IAEzH,IAAI,QAAQ,CAAC,MAAM,KAAK,OAAO,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;QACnD,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,OAAO,EAAE,yBAAyB,QAAQ,CAAC,KAAK,EAAE,EAAE,CAAC;IAC3F,CAAC;IAED,MAAM,SAAS,GAAe,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACrD,EAAE,EAAE,CAAC,CAAC,EAAE;QACR,WAAW,EAAE,CAAC,CAAC,WAAW;QAC1B,MAAM,EAAE,SAAkB;KAC3B,CAAC,CAAC,CAAC;IAEJ,IAAA,sBAAY,EAAC,SAAS,CAAC,CAAC;IAExB,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,SAAS,EAAE,OAAO,EAAE,wBAAwB,SAAS,CAAC,MAAM,SAAS,EAAE,CAAC;AACzG,CAAC;AAED,oBAAoB;AACb,KAAK,UAAU,eAAe;IACnC,MAAM,IAAI,GAAG,IAAA,wBAAc,GAAE,CAAC;IAE9B,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,wCAAwC,EAAE,CAAC;IAC/E,CAAC;IAED,MAAM,OAAO,GAAG,IAAA,oBAAU,GAAE,CAAC;IAC7B,MAAM,MAAM,GAAG,MAAM,IAAA,0BAAmB,GAAE,CAAC;IAE3C,MAAM,OAAO,GAAG,IAAA,8BAAY,EAC1B,OAAO,CAAC,gBAAgB,EACxB,IAAI,CAAC,WAAW,EAChB,IAAA,uBAAa,GAAE,IAAI,SAAS,EAC5B,OAAO,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC,CAC5F,CAAC;IACF,MAAM,YAAY,GAAG,IAAA,uCAAqB,EAAC,OAAO,CAAC,CAAC;IAEpD,MAAM,WAAW,GAAG;;QAEd,IAAI,CAAC,WAAW;gBACR,IAAA,mBAAS,GAAE;QACnB,OAAO,CAAC,IAAI;;EAElB,YAAY;;2CAE6B,CAAC;IAE1C,MAAM,MAAM,GAAG,MAAM,IAAA,iBAAO,EAAC;QAC3B,WAAW;QACX,MAAM,EAAE,oBAAoB;QAC5B,aAAa,EAAE,IAAI;KACpB,EAAE,MAAM,CAAC,CAAC;IAEX,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;QACpB,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,oBAAoB,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;IACzE,CAAC;IAED,MAAM,QAAQ,GAAG,MAAM,CAAC,MAA+F,CAAC;IAExH,IAAI,QAAQ,CAAC,MAAM,KAAK,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QAClD,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,oBAAoB,QAAQ,CAAC,KAAK,EAAE,EAAE,CAAC;IAC3E,CAAC;IAED,IAAI,CAAC,IAAI,GAAG,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAClC,EAAE,EAAE,CAAC,CAAC,EAAE;QACR,WAAW,EAAE,CAAC,CAAC,WAAW;QAC1B,MAAM,EAAE,SAAkB;KAC3B,CAAC,CAAC,CAAC;IAEJ,IAAA,qBAAW,EAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACvB,IAAA,0BAAgB,EAAC,IAAI,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;IAErC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,QAAQ,IAAI,CAAC,EAAE,WAAW,EAAE,CAAC;AAChE,CAAC;AAED,sBAAsB;AACtB,SAAgB,aAAa;IAC3B,MAAM,OAAO,GAAG,IAAA,oBAAU,GAAE,CAAC;IAC7B,MAAM,QAAQ,GAAG,IAAA,yBAAe,GAAE,CAAC;IAEnC,IAAI,OAAO,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACnC,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC;QACnC,OAAO;IACT,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,kBAAkB,QAAQ,CAAC,OAAO,IAAI,QAAQ,CAAC,KAAK,KAAK,QAAQ,CAAC,SAAS,aAAa,CAAC,CAAC;IACtG,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAEhB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAClD,MAAM,IAAI,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;QAClC,MAAM,SAAS,GAAG,CAAC,KAAK,OAAO,CAAC,gBAAgB,CAAC;QACjD,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,KAAK;YAC9C,CAAC,CAAC,IAAI,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,KAAK;gBACnC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK;oBACnB,CAAC,CAAC,KAAK,CAAC;QACV,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,KAAK,SAAS,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS;YACnE,CAAC,CAAC,IAAA,QAAG,EAAC,MAAM,CAAC;YACb,CAAC,CAAC,SAAS;gBACX,CAAC,CAAC,IAAA,SAAI,EAAC,MAAM,CAAC;gBACd,CAAC,CAAC,IAAA,QAAG,EAAC,MAAM,CAAC,CAAC;QAChB,OAAO,CAAC,GAAG,CAAC,GAAG,MAAM,IAAI,IAAI,CAAC,EAAE,KAAK,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;IAC3D,CAAC;AACH,CAAC;AAED,yCAAyC;AACzC,SAAgB,mBAAmB;IACjC,MAAM,IAAI,GAAG,IAAA,wBAAc,GAAE,CAAC;IAE9B,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,kBAAkB,EAAE,CAAC;IACzE,CAAC;IAED,IAAA,0BAAgB,EAAC,IAAI,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;IACrC,MAAM,QAAQ,GAAG,IAAA,qBAAW,GAAE,CAAC;IAE/B,IAAI,QAAQ,EAAE,CAAC;QACb,OAAO;YACL,OAAO,EAAE,IAAI;YACb,OAAO,EAAE,IAAI;YACb,OAAO,EAAE,QAAQ,IAAI,CAAC,EAAE,oBAAoB,QAAQ,CAAC,WAAW,EAAE;SACnE,CAAC;IACJ,CAAC;IAED,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,qBAAqB,EAAE,CAAC;AAC3E,CAAC;AAED,oBAAoB;AACpB,SAAgB,eAAe;IAC7B,MAAM,IAAI,GAAG,IAAA,wBAAc,GAAE,CAAC;IAE9B,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,kBAAkB,EAAE,CAAC;IACzE,CAAC;IAED,IAAA,0BAAgB,EAAC,IAAI,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;IACrC,MAAM,QAAQ,GAAG,IAAA,qBAAW,GAAE,CAAC;IAE/B,IAAI,QAAQ,EAAE,CAAC;QACb,OAAO;YACL,OAAO,EAAE,IAAI;YACb,OAAO,EAAE,IAAI;YACb,OAAO,EAAE,QAAQ,IAAI,CAAC,EAAE,mBAAmB,QAAQ,CAAC,WAAW,EAAE;SAClE,CAAC;IACJ,CAAC;IAED,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,qBAAqB,EAAE,CAAC;AAC3E,CAAC"}

package/dist/ui.d.ts

@@ -0,0 +1,15 @@
+import { SessionState } from './session';
+declare const ASCII_LOGO = "\n _____ ___  ___\n|__  // _ |/  /\n  / // __ | __|\n /___\\___|___|  code\n";
+export declare function renderStartup(projectName: string): string;
+export declare function getPrompt(session: SessionState): string;
+export declare function renderStatus(session: SessionState): string;
+export declare function renderStatusBar(session: SessionState): string;
+export declare function getWarnings(session: SessionState): string[];
+export declare function success(msg: string): string;
+export declare function warning(msg: string): string;
+export declare function error(msg: string): string;
+export declare function dim(msg: string): string;
+export declare function info(msg: string): string;
+export declare function hint(msg: string): string;
+export { ASCII_LOGO };
+//# sourceMappingURL=ui.d.ts.map

package/dist/ui.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ui.d.ts","sourceRoot":"","sources":["../src/ui.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AAoBzC,QAAA,MAAM,UAAU,iFAKf,CAAC;AAGF,wBAAgB,aAAa,CAAC,WAAW,EAAE,MAAM,GAAG,MAAM,CAczD;AAwBD,wBAAgB,SAAS,CAAC,OAAO,EAAE,YAAY,GAAG,MAAM,CAevD;AAGD,wBAAgB,YAAY,CAAC,OAAO,EAAE,YAAY,GAAG,MAAM,CAI1D;AAGD,wBAAgB,eAAe,CAAC,OAAO,EAAE,YAAY,GAAG,MAAM,CAmB7D;AAGD,wBAAgB,WAAW,CAAC,OAAO,EAAE,YAAY,GAAG,MAAM,EAAE,CAuB3D;AAGD,wBAAgB,OAAO,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CAG3C;AAED,wBAAgB,OAAO,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CAG3C;AAED,wBAAgB,KAAK,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CAGzC;AAED,wBAAgB,GAAG,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CAGvC;AAED,wBAAgB,IAAI,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CAGxC;AAGD,wBAAgB,IAAI,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CAGxC;AAGD,OAAO,EAAE,UAAU,EAAE,CAAC"}

package/dist/ui.js

@@ -0,0 +1,160 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ASCII_LOGO = void 0;
+exports.renderStartup = renderStartup;
+exports.getPrompt = getPrompt;
+exports.renderStatus = renderStatus;
+exports.renderStatusBar = renderStatusBar;
+exports.getWarnings = getWarnings;
+exports.success = success;
+exports.warning = warning;
+exports.error = error;
+exports.dim = dim;
+exports.info = info;
+exports.hint = hint;
+const settings_1 = require("./settings");
+const git_1 = require("./git");
+const profiles_1 = require("./profiles");
+// Colors - only if enabled
+function getColors() {
+    const enabled = (0, settings_1.shouldShowColor)();
+    return {
+        reset: enabled ? '\x1b[0m' : '',
+        dim: enabled ? '\x1b[2m' : '',
+        bold: enabled ? '\x1b[1m' : '',
+        green: enabled ? '\x1b[32m' : '',
+        yellow: enabled ? '\x1b[33m' : '',
+        red: enabled ? '\x1b[31m' : '',
+        cyan: enabled ? '\x1b[36m' : '',
+    };
+}
+// ASCII logo - minimal, 3 lines
+const ASCII_LOGO = `
+ _____ ___  ___
+|__  // _ |/  /
+  / // __ | __|
+ /___\\___|___|  code
+`;
+exports.ASCII_LOGO = ASCII_LOGO;
+// Show logo on startup (only if enabled and first run or setting is on)
+function renderStartup(projectName) {
+    const c = getColors();
+    const lines = [];
+    // Logo (optional)
+    if ((0, settings_1.shouldShowLogo)()) {
+        lines.push(c.dim + ASCII_LOGO.trim() + c.reset);
+        lines.push('');
+    }
+    // Header
+    lines.push(`Z.ai Code ${c.dim}— project: ${projectName}${c.reset}`);
+    return lines.join('\n');
+}
+// Get state label
+function getStateLabel(session) {
+    if (session.pendingActions)
+        return 'dirty';
+    if (session.lastDiff)
+        return 'dirty';
+    if (session.lastPlan && session.lastPlan.length > 0)
+        return 'pending';
+    return 'clean';
+}
+// Get focus (current file or step)
+function getFocus(session) {
+    if (session.openFiles.length === 1) {
+        const file = session.openFiles[0];
+        // Return just filename
+        return file.split('/').pop() || null;
+    }
+    if (session.taskSteps.length > 0 && session.currentStepIndex < session.taskSteps.length) {
+        return `step${session.currentStepIndex + 1}`;
+    }
+    return null;
+}
+// Build prompt: [mode][dry-run?][state][focus]>
+function getPrompt(session) {
+    const state = getStateLabel(session);
+    const focus = getFocus(session);
+    let prompt = `[${session.mode}]`;
+    if (session.dryRun) {
+        prompt += '[dry-run]';
+    }
+    prompt += `[${state}]`;
+    if (focus) {
+        prompt += `[${focus}]`;
+    }
+    prompt += '> ';
+    return prompt;
+}
+// Status line: Mode: <mode> | State: <state>
+function renderStatus(session) {
+    const c = getColors();
+    const state = getStateLabel(session);
+    return `${c.dim}Mode: ${session.mode} | State: ${state}${c.reset}`;
+}
+// Status bar: git | model | profile | pending
+function renderStatusBar(session) {
+    const c = getColors();
+    const gitInfo = (0, git_1.getGitInfo)(session.workingDirectory);
+    const gitStatus = (0, git_1.formatGitStatus)(gitInfo);
+    const model = (0, settings_1.getModel)().split('-').slice(0, 2).join('-'); // Shorten model name
+    const profile = (0, profiles_1.getActiveProfileName)();
+    const pending = session.pendingActions ? (session.pendingActions.files?.length || 0) + (session.pendingActions.diffs?.length || 0) : 0;
+    const parts = [
+        gitStatus,
+        `model:${model}`,
+        `profile:${profile}`,
+    ];
+    if (pending > 0) {
+        parts.push(`pending:${pending}`);
+    }
+    return c.dim + parts.join(' | ') + c.reset;
+}
+// Contextual warnings
+function getWarnings(session) {
+    const warnings = [];
+    const fs = require('fs');
+    const path = require('path');
+    const os = require('os');
+    // Check: running in home directory
+    if (session.workingDirectory === os.homedir()) {
+        warnings.push('Warning: running in home directory');
+    }
+    // Check: not in a git repo
+    const gitDir = path.join(session.workingDirectory, '.git');
+    if (!fs.existsSync(gitDir)) {
+        warnings.push('Note: not a git repository');
+    }
+    // Check: pending changes
+    if (session.pendingActions) {
+        warnings.push('Pending changes exist. Use /diff to review.');
+    }
+    return warnings;
+}
+// Output helpers
+function success(msg) {
+    const c = getColors();
+    return `${c.green}${msg}${c.reset}`;
+}
+function warning(msg) {
+    const c = getColors();
+    return `${c.yellow}${msg}${c.reset}`;
+}
+function error(msg) {
+    const c = getColors();
+    return `${c.red}${msg}${c.reset}`;
+}
+function dim(msg) {
+    const c = getColors();
+    return `${c.dim}${msg}${c.reset}`;
+}
+function info(msg) {
+    const c = getColors();
+    return `${c.cyan}${msg}${c.reset}`;
+}
+// Hint for next action
+function hint(msg) {
+    const c = getColors();
+    return `${c.dim}next: ${msg}${c.reset}`;
+}
+//# sourceMappingURL=ui.js.map

package/dist/ui.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ui.js","sourceRoot":"","sources":["../src/ui.ts"],"names":[],"mappings":";;;AA4BA,sCAcC;AAwBD,8BAeC;AAGD,oCAIC;AAGD,0CAmBC;AAGD,kCAuBC;AAGD,0BAGC;AAED,0BAGC;AAED,sBAGC;AAED,kBAGC;AAED,oBAGC;AAGD,oBAGC;AAvKD,yCAAuE;AACvE,+BAAoD;AACpD,yCAAkD;AAElD,2BAA2B;AAC3B,SAAS,SAAS;IAChB,MAAM,OAAO,GAAG,IAAA,0BAAe,GAAE,CAAC;IAClC,OAAO;QACL,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE;QAC/B,GAAG,EAAE,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE;QAC7B,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE;QAC9B,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE;QAChC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE;QACjC,GAAG,EAAE,OAAO,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE;QAC9B,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE;KAChC,CAAC;AACJ,CAAC;AAED,gCAAgC;AAChC,MAAM,UAAU,GAAG;;;;;CAKlB,CAAC;AAkJO,gCAAU;AAhJnB,wEAAwE;AACxE,SAAgB,aAAa,CAAC,WAAmB;IAC/C,MAAM,CAAC,GAAG,SAAS,EAAE,CAAC;IACtB,MAAM,KAAK,GAAa,EAAE,CAAC;IAE3B,kBAAkB;IAClB,IAAI,IAAA,yBAAc,GAAE,EAAE,CAAC;QACrB,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,GAAG,UAAU,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;QAChD,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACjB,CAAC;IAED,SAAS;IACT,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,GAAG,cAAc,WAAW,GAAG,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC;IAEpE,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAED,kBAAkB;AAClB,SAAS,aAAa,CAAC,OAAqB;IAC1C,IAAI,OAAO,CAAC,cAAc;QAAE,OAAO,OAAO,CAAC;IAC3C,IAAI,OAAO,CAAC,QAAQ;QAAE,OAAO,OAAO,CAAC;IACrC,IAAI,OAAO,CAAC,QAAQ,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC;QAAE,OAAO,SAAS,CAAC;IACtE,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,mCAAmC;AACnC,SAAS,QAAQ,CAAC,OAAqB;IACrC,IAAI,OAAO,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACnC,MAAM,IAAI,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;QAClC,uBAAuB;QACvB,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC;IACvC,CAAC;IACD,IAAI,OAAO,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,IAAI,OAAO,CAAC,gBAAgB,GAAG,OAAO,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC;QACxF,OAAO,OAAO,OAAO,CAAC,gBAAgB,GAAG,CAAC,EAAE,CAAC;IAC/C,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,gDAAgD;AAChD,SAAgB,SAAS,CAAC,OAAqB;IAC7C,MAAM,KAAK,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC;IACrC,MAAM,KAAK,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC;IAEhC,IAAI,MAAM,GAAG,IAAI,OAAO,CAAC,IAAI,GAAG,CAAC;IACjC,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;QACnB,MAAM,IAAI,WAAW,CAAC;IACxB,CAAC;IACD,MAAM,IAAI,IAAI,KAAK,GAAG,CAAC;IACvB,IAAI,KAAK,EAAE,CAAC;QACV,MAAM,IAAI,IAAI,KAAK,GAAG,CAAC;IACzB,CAAC;IACD,MAAM,IAAI,IAAI,CAAC;IAEf,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,6CAA6C;AAC7C,SAAgB,YAAY,CAAC,OAAqB;IAChD,MAAM,CAAC,GAAG,SAAS,EAAE,CAAC;IACtB,MAAM,KAAK,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC;IACrC,OAAO,GAAG,CAAC,CAAC,GAAG,SAAS,OAAO,CAAC,IAAI,aAAa,KAAK,GAAG,CAAC,CAAC,KAAK,EAAE,CAAC;AACrE,CAAC;AAED,8CAA8C;AAC9C,SAAgB,eAAe,CAAC,OAAqB;IACnD,MAAM,CAAC,GAAG,SAAS,EAAE,CAAC;IACtB,MAAM,OAAO,GAAG,IAAA,gBAAU,EAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;IACrD,MAAM,SAAS,GAAG,IAAA,qBAAe,EAAC,OAAO,CAAC,CAAC;IAC3C,MAAM,KAAK,GAAG,IAAA,mBAAQ,GAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,qBAAqB;IAChF,MAAM,OAAO,GAAG,IAAA,+BAAoB,GAAE,CAAC;IACvC,MAAM,OAAO,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,KAAK,EAAE,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,cAAc,CAAC,KAAK,EAAE,MAAM,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAEvI,MAAM,KAAK,GAAG;QACZ,SAAS;QACT,SAAS,KAAK,EAAE;QAChB,WAAW,OAAO,EAAE;KACrB,CAAC;IAEF,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;QAChB,KAAK,CAAC,IAAI,CAAC,WAAW,OAAO,EAAE,CAAC,CAAC;IACnC,CAAC;IAED,OAAO,CAAC,CAAC,GAAG,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC;AAC7C,CAAC;AAED,sBAAsB;AACtB,SAAgB,WAAW,CAAC,OAAqB;IAC/C,MAAM,QAAQ,GAAa,EAAE,CAAC;IAC9B,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACzB,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;IAC7B,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAEzB,mCAAmC;IACnC,IAAI,OAAO,CAAC,gBAAgB,KAAK,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC;QAC9C,QAAQ,CAAC,IAAI,CAAC,oCAAoC,CAAC,CAAC;IACtD,CAAC;IAED,2BAA2B;IAC3B,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,gBAAgB,EAAE,MAAM,CAAC,CAAC;IAC3D,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;QAC3B,QAAQ,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;IAC9C,CAAC;IAED,yBAAyB;IACzB,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC;QAC3B,QAAQ,CAAC,IAAI,CAAC,6CAA6C,CAAC,CAAC;IAC/D,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,iBAAiB;AACjB,SAAgB,OAAO,CAAC,GAAW;IACjC,MAAM,CAAC,GAAG,SAAS,EAAE,CAAC;IACtB,OAAO,GAAG,CAAC,CAAC,KAAK,GAAG,GAAG,GAAG,CAAC,CAAC,KAAK,EAAE,CAAC;AACtC,CAAC;AAED,SAAgB,OAAO,CAAC,GAAW;IACjC,MAAM,CAAC,GAAG,SAAS,EAAE,CAAC;IACtB,OAAO,GAAG,CAAC,CAAC,MAAM,GAAG,GAAG,GAAG,CAAC,CAAC,KAAK,EAAE,CAAC;AACvC,CAAC;AAED,SAAgB,KAAK,CAAC,GAAW;IAC/B,MAAM,CAAC,GAAG,SAAS,EAAE,CAAC;IACtB,OAAO,GAAG,CAAC,CAAC,GAAG,GAAG,GAAG,GAAG,CAAC,CAAC,KAAK,EAAE,CAAC;AACpC,CAAC;AAED,SAAgB,GAAG,CAAC,GAAW;IAC7B,MAAM,CAAC,GAAG,SAAS,EAAE,CAAC;IACtB,OAAO,GAAG,CAAC,CAAC,GAAG,GAAG,GAAG,GAAG,CAAC,CAAC,KAAK,EAAE,CAAC;AACpC,CAAC;AAED,SAAgB,IAAI,CAAC,GAAW;IAC9B,MAAM,CAAC,GAAG,SAAS,EAAE,CAAC;IACtB,OAAO,GAAG,CAAC,CAAC,IAAI,GAAG,GAAG,GAAG,CAAC,CAAC,KAAK,EAAE,CAAC;AACrC,CAAC;AAED,uBAAuB;AACvB,SAAgB,IAAI,CAAC,GAAW;IAC9B,MAAM,CAAC,GAAG,SAAS,EAAE,CAAC;IACtB,OAAO,GAAG,CAAC,CAAC,GAAG,SAAS,GAAG,GAAG,CAAC,CAAC,KAAK,EAAE,CAAC;AAC1C,CAAC"}

package/dist/workspace.d.ts

@@ -0,0 +1,17 @@
+export interface WorkspaceContext {
+    root: string;
+    files: Array<{
+        path: string;
+        size: number;
+    }>;
+    gitStatus?: string;
+}
+export interface CollectOptions {
+    maxFiles?: number;
+    maxFileSize?: number;
+    extensions?: string[];
+}
+export declare function collectWorkspace(rootPath: string, options?: CollectOptions): WorkspaceContext;
+export declare function getFileContent(filePath: string, maxSize?: number): string | null;
+export declare function buildContextString(workspace: WorkspaceContext, includeContent?: boolean): string;
+//# sourceMappingURL=workspace.d.ts.map

package/dist/workspace.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"workspace.d.ts","sourceRoot":"","sources":["../src/workspace.ts"],"names":[],"mappings":"AAGA,MAAM,WAAW,gBAAgB;IAC/B,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAC7C,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,cAAc;IAC7B,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC;CACvB;AAiBD,wBAAgB,gBAAgB,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,cAAc,GAAG,gBAAgB,CA8D7F;AAED,wBAAgB,cAAc,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAYhF;AAED,wBAAgB,kBAAkB,CAAC,SAAS,EAAE,gBAAgB,EAAE,cAAc,CAAC,EAAE,OAAO,GAAG,MAAM,CAahG"}

package/dist/workspace.js

@@ -0,0 +1,134 @@
+"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.collectWorkspace = collectWorkspace;
+exports.getFileContent = getFileContent;
+exports.buildContextString = buildContextString;
+const fs = __importStar(require("fs"));
+const path = __importStar(require("path"));
+const DEFAULT_EXTENSIONS = ['.ts', '.js', '.tsx', '.jsx', '.json', '.md', '.py', '.go', '.rs', '.java', '.c', '.cpp', '.h', '.css', '.html'];
+const DEFAULT_MAX_FILES = 100;
+const DEFAULT_MAX_FILE_SIZE = 100 * 1024;
+const IGNORED_DIRS = ['node_modules', '.git', '.svn', 'dist', 'build', 'out', 'target', 'vendor', '.venv', 'venv', '__pycache__', '.next', '.nuxt'];
+function shouldIgnore(dirName) {
+    return IGNORED_DIRS.includes(dirName);
+}
+function hasAllowedExtension(filePath, extensions) {
+    const ext = path.extname(filePath).toLowerCase();
+    return extensions.includes(ext);
+}
+function collectWorkspace(rootPath, options) {
+    const maxFiles = options?.maxFiles ?? DEFAULT_MAX_FILES;
+    const maxFileSize = options?.maxFileSize ?? DEFAULT_MAX_FILE_SIZE;
+    const extensions = options?.extensions ?? DEFAULT_EXTENSIONS;
+    const files = [];
+    function walkDirectory(dir) {
+        if (files.length >= maxFiles) {
+            return;
+        }
+        try {
+            const entries = fs.readdirSync(dir, { withFileTypes: true });
+            for (const entry of entries) {
+                if (files.length >= maxFiles) {
+                    break;
+                }
+                const fullPath = path.join(dir, entry.name);
+                if (entry.isDirectory()) {
+                    if (!shouldIgnore(entry.name)) {
+                        walkDirectory(fullPath);
+                    }
+                }
+                else if (entry.isFile()) {
+                    if (hasAllowedExtension(entry.name, extensions)) {
+                        try {
+                            const stats = fs.statSync(fullPath);
+                            if (stats.size <= maxFileSize) {
+                                const relativePath = path.relative(rootPath, fullPath);
+                                files.push({ path: relativePath, size: stats.size });
+                            }
+                        }
+                        catch {
+                            // Skip files that can't be read
+                        }
+                    }
+                }
+            }
+        }
+        catch {
+            // Skip directories that can't be read
+        }
+    }
+    walkDirectory(rootPath);
+    let gitStatus;
+    try {
+        const gitDir = path.join(rootPath, '.git');
+        if (fs.existsSync(gitDir)) {
+            gitStatus = 'git';
+        }
+    }
+    catch {
+        // Not a git repo or can't check
+    }
+    return {
+        root: rootPath,
+        files,
+        gitStatus,
+    };
+}
+function getFileContent(filePath, maxSize) {
+    const max = maxSize ?? DEFAULT_MAX_FILE_SIZE;
+    try {
+        const stats = fs.statSync(filePath);
+        if (stats.size > max) {
+            return null;
+        }
+        return fs.readFileSync(filePath, 'utf-8');
+    }
+    catch {
+        return null;
+    }
+}
+function buildContextString(workspace, includeContent) {
+    let result = `Working directory: ${workspace.root}\n`;
+    result += `Files (${workspace.files.length}):\n`;
+    for (const file of workspace.files) {
+        result += `  - ${file.path}\n`;
+    }
+    if (workspace.gitStatus) {
+        result += `\nGit repository: ${workspace.gitStatus}\n`;
+    }
+    return result;
+}
+//# sourceMappingURL=workspace.js.map

package/dist/workspace.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"workspace.js","sourceRoot":"","sources":["../src/workspace.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA8BA,4CA8DC;AAED,wCAYC;AAED,gDAaC;AAzHD,uCAAyB;AACzB,2CAA6B;AAc7B,MAAM,kBAAkB,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;AAE7I,MAAM,iBAAiB,GAAG,GAAG,CAAC;AAC9B,MAAM,qBAAqB,GAAG,GAAG,GAAG,IAAI,CAAC;AACzC,MAAM,YAAY,GAAG,CAAC,cAAc,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,aAAa,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;AAEpJ,SAAS,YAAY,CAAC,OAAe;IACnC,OAAO,YAAY,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;AACxC,CAAC;AAED,SAAS,mBAAmB,CAAC,QAAgB,EAAE,UAAoB;IACjE,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;IACjD,OAAO,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;AAClC,CAAC;AAED,SAAgB,gBAAgB,CAAC,QAAgB,EAAE,OAAwB;IACzE,MAAM,QAAQ,GAAG,OAAO,EAAE,QAAQ,IAAI,iBAAiB,CAAC;IACxD,MAAM,WAAW,GAAG,OAAO,EAAE,WAAW,IAAI,qBAAqB,CAAC;IAClE,MAAM,UAAU,GAAG,OAAO,EAAE,UAAU,IAAI,kBAAkB,CAAC;IAE7D,MAAM,KAAK,GAA0C,EAAE,CAAC;IAExD,SAAS,aAAa,CAAC,GAAW;QAChC,IAAI,KAAK,CAAC,MAAM,IAAI,QAAQ,EAAE,CAAC;YAC7B,OAAO;QACT,CAAC;QAED,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,EAAE,CAAC,WAAW,CAAC,GAAG,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC;YAE7D,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;gBAC5B,IAAI,KAAK,CAAC,MAAM,IAAI,QAAQ,EAAE,CAAC;oBAC7B,MAAM;gBACR,CAAC;gBAED,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;gBAE5C,IAAI,KAAK,CAAC,WAAW,EAAE,EAAE,CAAC;oBACxB,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;wBAC9B,aAAa,CAAC,QAAQ,CAAC,CAAC;oBAC1B,CAAC;gBACH,CAAC;qBAAM,IAAI,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC;oBAC1B,IAAI,mBAAmB,CAAC,KAAK,CAAC,IAAI,EAAE,UAAU,CAAC,EAAE,CAAC;wBAChD,IAAI,CAAC;4BACH,MAAM,KAAK,GAAG,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;4BACpC,IAAI,KAAK,CAAC,IAAI,IAAI,WAAW,EAAE,CAAC;gCAC9B,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;gCACvD,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;4BACvD,CAAC;wBACH,CAAC;wBAAC,MAAM,CAAC;4BACP,gCAAgC;wBAClC,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,sCAAsC;QACxC,CAAC;IACH,CAAC;IAED,aAAa,CAAC,QAAQ,CAAC,CAAC;IAExB,IAAI,SAA6B,CAAC;IAClC,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAC3C,IAAI,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;YAC1B,SAAS,GAAG,KAAK,CAAC;QACpB,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,gCAAgC;IAClC,CAAC;IAED,OAAO;QACL,IAAI,EAAE,QAAQ;QACd,KAAK;QACL,SAAS;KACV,CAAC;AACJ,CAAC;AAED,SAAgB,cAAc,CAAC,QAAgB,EAAE,OAAgB;IAC/D,MAAM,GAAG,GAAG,OAAO,IAAI,qBAAqB,CAAC;IAE7C,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACpC,IAAI,KAAK,CAAC,IAAI,GAAG,GAAG,EAAE,CAAC;YACrB,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IAC5C,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,SAAgB,kBAAkB,CAAC,SAA2B,EAAE,cAAwB;IACtF,IAAI,MAAM,GAAG,sBAAsB,SAAS,CAAC,IAAI,IAAI,CAAC;IACtD,MAAM,IAAI,UAAU,SAAS,CAAC,KAAK,CAAC,MAAM,MAAM,CAAC;IAEjD,KAAK,MAAM,IAAI,IAAI,SAAS,CAAC,KAAK,EAAE,CAAC;QACnC,MAAM,IAAI,OAAO,IAAI,CAAC,IAAI,IAAI,CAAC;IACjC,CAAC;IAED,IAAI,SAAS,CAAC,SAAS,EAAE,CAAC;QACxB,MAAM,IAAI,qBAAqB,SAAS,CAAC,SAAS,IAAI,CAAC;IACzD,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/workspace_model.d.ts

@@ -0,0 +1,45 @@
+import { SessionMode, PlanStep, IntentType } from './session';
+import { ResponseSchema } from './runtime';
+import { FileScore } from './context/context_builder';
+declare const WORKSPACE_STATE_FILE = ".zai/workspace.json";
+export interface WorkspaceState {
+    version: number;
+    workingDirectory: string;
+    mode: SessionMode;
+    openFiles: string[];
+    currentIntent: string | null;
+    intentType: IntentType | null;
+    lastPlan: PlanStep[] | null;
+    lastDiff: ResponseSchema | null;
+    pendingActions: ResponseSchema | null;
+    dryRun: boolean;
+    lastUpdated: string;
+}
+export interface FileTreeNode {
+    name: string;
+    path: string;
+    type: 'file' | 'directory';
+    size?: number;
+    children?: FileTreeNode[];
+    extension?: string;
+}
+export declare class Workspace {
+    private rootPath;
+    private fileTree;
+    private fileIndex;
+    constructor(rootPath?: string);
+    getRoot(): string;
+    indexFileTree(maxDepth?: number): FileTreeNode;
+    getFileTree(): FileTreeNode | null;
+    getFileIndex(): FileScore[];
+    getCurrentState(): WorkspaceState;
+    saveState(): void;
+    loadState(): WorkspaceState | null;
+    restoreState(): boolean;
+    clearState(): void;
+    printTreeSummary(): string;
+}
+export declare function getWorkspace(rootPath?: string): Workspace;
+export declare function resetWorkspace(): void;
+export { WORKSPACE_STATE_FILE };
+//# sourceMappingURL=workspace_model.d.ts.map

package/dist/workspace_model.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"workspace_model.d.ts","sourceRoot":"","sources":["../src/workspace_model.ts"],"names":[],"mappings":"AAEA,OAAO,EAA4B,WAAW,EAAE,QAAQ,EAA+G,UAAU,EAAE,MAAM,WAAW,CAAC;AACrM,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAC3C,OAAO,EAAkB,SAAS,EAAE,MAAM,2BAA2B,CAAC;AAGtE,QAAA,MAAM,oBAAoB,wBAAwB,CAAC;AAGnD,MAAM,WAAW,cAAc;IAC7B,OAAO,EAAE,MAAM,CAAC;IAChB,gBAAgB,EAAE,MAAM,CAAC;IACzB,IAAI,EAAE,WAAW,CAAC;IAClB,SAAS,EAAE,MAAM,EAAE,CAAC;IACpB,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,UAAU,EAAE,UAAU,GAAG,IAAI,CAAC;IAC9B,QAAQ,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC;IAC5B,QAAQ,EAAE,cAAc,GAAG,IAAI,CAAC;IAChC,cAAc,EAAE,cAAc,GAAG,IAAI,CAAC;IACtC,MAAM,EAAE,OAAO,CAAC;IAChB,WAAW,EAAE,MAAM,CAAC;CACrB;AAGD,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,GAAG,WAAW,CAAC;IAC3B,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,YAAY,EAAE,CAAC;IAC1B,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAGD,qBAAa,SAAS;IACpB,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,QAAQ,CAA6B;IAC7C,OAAO,CAAC,SAAS,CAAmB;gBAExB,QAAQ,CAAC,EAAE,MAAM;IAK7B,OAAO,IAAI,MAAM;IAKjB,aAAa,CAAC,QAAQ,GAAE,MAAU,GAAG,YAAY;IA6DjD,WAAW,IAAI,YAAY,GAAG,IAAI;IAKlC,YAAY,IAAI,SAAS,EAAE;IAK3B,eAAe,IAAI,cAAc;IAkBjC,SAAS,IAAI,IAAI;IAgBjB,SAAS,IAAI,cAAc,GAAG,IAAI;IAelC,YAAY,IAAI,OAAO;IAwCvB,UAAU,IAAI,IAAI;IAYlB,gBAAgB,IAAI,MAAM;CAsB3B;AAMD,wBAAgB,YAAY,CAAC,QAAQ,CAAC,EAAE,MAAM,GAAG,SAAS,CAKzD;AAGD,wBAAgB,cAAc,IAAI,IAAI,CAErC;AAED,OAAO,EAAE,oBAAoB,EAAE,CAAC"}