@automaton-labs/aib 0.0.6 → 0.0.8

package/dist/commands/inspect-imports.js

@@ -1,2 +1,339 @@
-"use strict";var C=exports&&exports.__createBinding||(Object.create?(function(t,e,n,r){r===void 0&&(r=n);var i=Object.getOwnPropertyDescriptor(e,n);(!i||("get"in i?!e.__esModule:i.writable||i.configurable))&&(i={enumerable:!0,get:function(){return e[n]}}),Object.defineProperty(t,r,i)}):(function(t,e,n,r){r===void 0&&(r=n),t[r]=e[n]})),M=exports&&exports.__setModuleDefault||(Object.create?(function(t,e){Object.defineProperty(t,"default",{enumerable:!0,value:e})}):function(t,e){t.default=e}),S=exports&&exports.__importStar||(function(){var t=function(e){return t=Object.getOwnPropertyNames||function(n){var r=[];for(var i in n)Object.prototype.hasOwnProperty.call(n,i)&&(r[r.length]=i);return r},t(e)};return function(e){if(e&&e.__esModule)return e;var n={};if(e!=null)for(var r=t(e),i=0;i<r.length;i++)r[i]!=="default"&&C(n,e,r[i]);return M(n,e),n}})(),$=exports&&exports.__importDefault||function(t){return t&&t.__esModule?t:{default:t}};Object.defineProperty(exports,"__esModule",{value:!0}),exports.isInspectImportsRequest=k,exports.runInspectImports=v,exports.buildInspectImportsResult=R,exports.mergeDirectoryFallbackResults=q,exports.formatImportsResponse=P,exports.appendImportsBlock=y;const b=S(require("node:fs")),p=S(require("node:path")),o=$(require("typescript")),d=require("../config/path-display"),L=require("../shared/agent-text"),m=require("../shared/errors"),E=require("./inspect-tree"),O=require("./session-workspace");function k(t,e){if(t[0]==="imports")return!0;if(!t.includes("--stdin")||!a(e))return!1;const n=x(e);return n.length>0&&n.some(r=>c(r.op)==="imports")}function v(t,e,n){const r=(0,d.buildDisplayPathAliases)(t,(0,O.readEffectiveSessionPathAliases)(t)),i=N(t,e,n),s=q(i.map(l=>R(t,l,r))),u=s.length===1?s[0]:{results:s};return(0,L.attachAgentText)(u,P(u))}function R(t,e,n=(0,d.buildDisplayPathAliases)(t,(0,O.readEffectiveSessionPathAliases)(t))){const r={file:g(t,e.file),ranges:e.ranges!==!1,dynamic:e.dynamic===!0},i=(0,d.compactDisplayPath)(t,r.file,n),s=J(r.file);if(s?.isDirectory())return B(t,r.file,n);if(!s?.isFile())return{ok:!1,op:"imports",code:"FILE_NOT_FOUND",error:"file not found",sourceFile:r.file,file:i};const u=b.readFileSync(r.file,"utf8"),l=o.default.createSourceFile(r.file,u,o.default.ScriptTarget.Latest,!0,X(r.file)),_=[],h=[],I=f=>{o.default.isImportDeclaration(f)?_.push(A(l,u,f,r.ranges)):r.dynamic&&o.default.isCallExpression(f)&&f.expression.kind===o.default.SyntaxKind.ImportKeyword&&h.push(A(l,u,f,r.ranges)),o.default.forEachChild(f,I)};return o.default.forEachChild(l,I),{ok:!0,op:"imports",sourceFile:r.file,file:i,imports:_,...h.length>0?{dynamicImports:h}:{}}}function q(t){const e=[],n=new Map;for(const r of t){if(r.op!=="directoryFallback"){e.push(r);continue}const i=c(r.sourceFile)??c(r.directory);if(!i){e.push(r);continue}const s=n.get(i);if(s===void 0){n.set(i,e.length),e.push(r);continue}const u=e[s];u&&(u.skippedOps=z([...T(u.skippedOps),...T(r.skippedOps)]))}return e}function P(t){const e=[];if(Array.isArray(t.results))for(const n of t.results.filter(a))y(e,n);else y(e,t);return e.join(`
-`)}function y(t,e){t.length>0&&t[t.length-1]!==""&&t.push("");const n=c(e.file);if(n){if(e.ok===!1){t.push(`// ${n} - ${c(e.error)??"request failed"}`);return}t.push(`// ${n}`)}const r=F(e.imports),i=F(e.dynamicImports);if(r.length===0&&i.length===0)t.push("no imports");else{if(r.length>0){t.push("imports:");for(const s of r)t.push(...D(s))}if(i.length>0){r.length>0&&t.push(""),t.push("dynamic imports:");for(const s of i)t.push(...D(s))}}}function B(t,e,n){const r=K(t,e),i=(0,d.compactDisplayPath)(t,e,n),s=(0,E.buildInspectTreeResult)(t,{scope:e,displayScope:i,depth:r,limit:40});return{ok:!0,op:"directoryFallback",sourceFile:e,directory:i,skippedOps:["imports"],command:`aib inspect tree ${i} --depth ${r}`,tree:s,lines:Array.isArray(s.lines)?s.lines:[]}}function K(t,e){for(let n=1;n<=4;n+=1){const r=(0,E.buildInspectTreeResult)(t,{scope:e,depth:n,limit:1});if(typeof r.files=="number"&&r.files>0)return n}return 2}function N(t,e,n){if(e[0]==="imports")return[U(t,e.slice(1))];if(!e.includes("--stdin")||!a(n))throw new m.CliError("INSPECT_IMPORTS_FILE_REQUIRED","inspect imports expects a file.");const r=x(n);if(r.some(i=>c(i.op)!=="imports"))throw new m.CliError("INSPECT_IMPORTS_MIXED_BATCH_UNSUPPORTED","inspect imports --stdin expects only imports requests.");return r.map((i,s)=>w(t,i,s))}function U(t,e){const n=e.find(r=>!r.startsWith("--")&&!W(e,r));if(!n)throw new m.CliError("INSPECT_IMPORTS_FILE_REQUIRED","inspect imports expects a file.");return{file:g(t,n),ranges:!e.includes("--no-ranges"),dynamic:!e.includes("--no-dynamic")}}function w(t,e,n){const r=c(e.file)??c(e.fromFile)??c(e.sourceFile);if(!r)throw new m.CliError("INSPECT_IMPORTS_FILE_REQUIRED",`Inspect imports request at index ${n} requires file.`);return{file:g(t,r),ranges:j(e.ranges)!==!1,dynamic:j(e.dynamic)!==!1}}function A(t,e,n,r){return{text:e.slice(n.getStart(t),n.getEnd()).trim(),...r?{range:Q(t,n)}:{}}}function Q(t,e){const n=t.getLineAndCharacterOfPosition(e.getStart(t)).line+1,r=t.getLineAndCharacterOfPosition(e.getEnd()).line+1;return n===r?`@${n}`:`@${n}..${r}`}function D(t){const e=t.text.split(/\r?\n/);return t.range?e.length===0?[t.range]:[`${t.range} ${e[0]??""}`,...e.slice(1)]:e}function F(t){return Array.isArray(t)?t.filter(a).map(e=>({text:c(e.text)??"",...typeof e.range=="string"?{range:e.range}:{}})).filter(e=>e.text!==""):[]}function x(t){return a(t)?Array.isArray(t.requests)?t.requests.filter(a):[t]:[]}function X(t){const e=p.extname(t).toLowerCase();return e===".tsx"?o.default.ScriptKind.TSX:e===".jsx"?o.default.ScriptKind.JSX:e===".js"||e===".cjs"||e===".mjs"?o.default.ScriptKind.JS:o.default.ScriptKind.TS}function g(t,e){return p.isAbsolute(e)?p.normalize(e):p.resolve(t,e)}function J(t){try{return b.statSync(t)}catch{return null}}function W(t,e){const n=t.indexOf(e);return n>0&&t[n-1]?.startsWith("--")===!0}function c(t){return typeof t=="string"&&t.trim()!==""?t:null}function T(t){return Array.isArray(t)?t.filter(e=>typeof e=="string"&&e.trim()!=="").map(e=>e.trim()):[]}function j(t){return typeof t=="boolean"?t:null}function z(t){return Array.from(new Set(t.filter(e=>e!=="")))}function a(t){return!!t&&typeof t=="object"&&!Array.isArray(t)}
+"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;
+    };
+})();
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.isInspectImportsRequest = isInspectImportsRequest;
+exports.runInspectImports = runInspectImports;
+exports.buildInspectImportsResult = buildInspectImportsResult;
+exports.mergeDirectoryFallbackResults = mergeDirectoryFallbackResults;
+exports.formatImportsResponse = formatImportsResponse;
+exports.appendImportsBlock = appendImportsBlock;
+const fs = __importStar(require("node:fs"));
+const path = __importStar(require("node:path"));
+const typescript_1 = __importDefault(require("typescript"));
+const path_display_1 = require("../config/path-display");
+const agent_text_1 = require("../shared/agent-text");
+const errors_1 = require("../shared/errors");
+const inspect_tree_1 = require("./inspect-tree");
+const session_workspace_1 = require("./session-workspace");
+function isInspectImportsRequest(restArgs, stdinPayload) {
+    if (restArgs[0] === "imports") {
+        return true;
+    }
+    if (!restArgs.includes("--stdin") || !isObject(stdinPayload)) {
+        return false;
+    }
+    const requests = expandImportsPayloadRequests(stdinPayload);
+    return requests.length > 0 && requests.some((request) => readString(request.op) === "imports");
+}
+function runInspectImports(cwd, restArgs, stdinPayload) {
+    const aliases = (0, path_display_1.buildDisplayPathAliases)(cwd, (0, session_workspace_1.readEffectiveSessionPathAliases)(cwd));
+    const requests = buildImportsRequests(cwd, restArgs, stdinPayload);
+    const results = mergeDirectoryFallbackResults(requests.map((request) => buildInspectImportsResult(cwd, request, aliases)));
+    const response = results.length === 1 ? results[0] : { results };
+    return (0, agent_text_1.attachAgentText)(response, formatImportsResponse(response));
+}
+function buildInspectImportsResult(cwd, input, aliases = (0, path_display_1.buildDisplayPathAliases)(cwd, (0, session_workspace_1.readEffectiveSessionPathAliases)(cwd))) {
+    const request = {
+        file: resolveCliPath(cwd, input.file),
+        ranges: input.ranges !== false
+    };
+    const displayFile = (0, path_display_1.compactDisplayPath)(cwd, request.file, aliases);
+    const stat = safeStat(request.file);
+    if (stat?.isDirectory()) {
+        return buildImportsDirectoryFallbackResult(cwd, request.file, aliases);
+    }
+    if (!stat?.isFile()) {
+        return {
+            ok: false,
+            op: "imports",
+            code: "FILE_NOT_FOUND",
+            error: "file not found",
+            sourceFile: request.file,
+            file: displayFile
+        };
+    }
+    const sourceText = fs.readFileSync(request.file, "utf8");
+    const sourceFile = typescript_1.default.createSourceFile(request.file, sourceText, typescript_1.default.ScriptTarget.Latest, true, scriptKindForFile(request.file));
+    const staticImports = [];
+    const dynamicImports = [];
+    const visit = (node) => {
+        if (typescript_1.default.isImportDeclaration(node)) {
+            staticImports.push(importEntry(sourceFile, sourceText, node, request.ranges));
+        }
+        else if (typescript_1.default.isCallExpression(node) && node.expression.kind === typescript_1.default.SyntaxKind.ImportKeyword) {
+            dynamicImports.push(importEntry(sourceFile, sourceText, node, request.ranges));
+        }
+        typescript_1.default.forEachChild(node, visit);
+    };
+    typescript_1.default.forEachChild(sourceFile, visit);
+    return {
+        ok: true,
+        op: "imports",
+        sourceFile: request.file,
+        file: displayFile,
+        imports: staticImports,
+        ...(dynamicImports.length > 0 ? { dynamicImports } : {})
+    };
+}
+function mergeDirectoryFallbackResults(results) {
+    const merged = [];
+    const indexBySource = new Map();
+    for (const result of results) {
+        if (result.op !== "directoryFallback") {
+            merged.push(result);
+            continue;
+        }
+        const sourceFile = readString(result.sourceFile) ?? readString(result.directory);
+        if (!sourceFile) {
+            merged.push(result);
+            continue;
+        }
+        const existingIndex = indexBySource.get(sourceFile);
+        if (existingIndex === undefined) {
+            indexBySource.set(sourceFile, merged.length);
+            merged.push(result);
+            continue;
+        }
+        const existing = merged[existingIndex];
+        if (existing) {
+            existing.skippedOps = uniqueStrings([
+                ...readStringArray(existing.skippedOps),
+                ...readStringArray(result.skippedOps)
+            ]);
+        }
+    }
+    return merged;
+}
+function formatImportsResponse(response) {
+    const lines = [];
+    if (Array.isArray(response.results)) {
+        for (const item of response.results.filter(isObject)) {
+            appendImportsBlock(lines, item);
+        }
+    }
+    else {
+        appendImportsBlock(lines, response);
+    }
+    return lines.join("\n");
+}
+function appendImportsBlock(lines, item) {
+    if (lines.length > 0 && lines[lines.length - 1] !== "") {
+        lines.push("");
+    }
+    const file = readString(item.file);
+    if (file) {
+        if (item.ok === false) {
+            lines.push(`// ${file} - ${readString(item.error) ?? "request failed"}`);
+            return;
+        }
+        lines.push(`// ${file}`);
+    }
+    const staticImports = readImportEntries(item.imports);
+    const dynamicImports = readImportEntries(item.dynamicImports);
+    if (staticImports.length === 0 && dynamicImports.length === 0) {
+        lines.push("no imports");
+    }
+    else {
+        if (staticImports.length > 0) {
+            lines.push("imports:");
+            for (const entry of staticImports) {
+                lines.push(...formatImportEntry(entry));
+            }
+        }
+        if (dynamicImports.length > 0) {
+            if (staticImports.length > 0) {
+                lines.push("");
+            }
+            lines.push("dynamic imports:");
+            for (const entry of dynamicImports) {
+                lines.push(...formatImportEntry(entry));
+            }
+        }
+    }
+}
+function buildImportsDirectoryFallbackResult(cwd, sourceFile, aliases) {
+    const depth = detectTreeFallbackDepth(cwd, sourceFile);
+    const directory = (0, path_display_1.compactDisplayPath)(cwd, sourceFile, aliases);
+    const tree = (0, inspect_tree_1.buildInspectTreeResult)(cwd, {
+        scope: sourceFile,
+        displayScope: directory,
+        depth,
+        limit: 40
+    });
+    return {
+        ok: true,
+        op: "directoryFallback",
+        sourceFile,
+        directory,
+        skippedOps: ["imports"],
+        command: `aib inspect tree ${directory} --depth ${depth}`,
+        tree,
+        lines: Array.isArray(tree.lines) ? tree.lines : []
+    };
+}
+function detectTreeFallbackDepth(cwd, sourceFile) {
+    for (let depth = 1; depth <= 4; depth += 1) {
+        const result = (0, inspect_tree_1.buildInspectTreeResult)(cwd, {
+            scope: sourceFile,
+            depth,
+            limit: 1
+        });
+        if (typeof result.files === "number" && result.files > 0) {
+            return depth;
+        }
+    }
+    return 2;
+}
+function buildImportsRequests(cwd, restArgs, stdinPayload) {
+    if (restArgs[0] === "imports") {
+        return importsRequestsFromArgs(cwd, restArgs.slice(1));
+    }
+    if (!restArgs.includes("--stdin") || !isObject(stdinPayload)) {
+        throw new errors_1.CliError("INSPECT_IMPORTS_FILE_REQUIRED", "inspect imports expects a file.");
+    }
+    const rawRequests = expandImportsPayloadRequests(stdinPayload);
+    if (rawRequests.some((request) => readString(request.op) !== "imports")) {
+        throw new errors_1.CliError("INSPECT_IMPORTS_MIXED_BATCH_UNSUPPORTED", "inspect imports --stdin expects only imports requests.");
+    }
+    return rawRequests.map((request, index) => importsRequestFromObject(cwd, request, index));
+}
+function importsRequestsFromArgs(cwd, args) {
+    const files = args.filter((arg) => !arg.startsWith("--") && !isFlagValue(args, arg));
+    if (files.length === 0) {
+        throw new errors_1.CliError("INSPECT_IMPORTS_FILE_REQUIRED", "inspect imports expects a file.");
+    }
+    return files.map((file) => ({
+        file: resolveCliPath(cwd, file),
+        ranges: !args.includes("--no-ranges")
+    }));
+}
+function importsRequestFromObject(cwd, raw, index) {
+    const file = readString(raw.file) ?? readString(raw.fromFile) ?? readString(raw.sourceFile);
+    if (!file) {
+        throw new errors_1.CliError("INSPECT_IMPORTS_FILE_REQUIRED", `Inspect imports request at index ${index} requires file.`);
+    }
+    return {
+        file: resolveCliPath(cwd, file),
+        ranges: readBoolean(raw.ranges) !== false
+    };
+}
+function importEntry(sourceFile, sourceText, node, includeRanges) {
+    const text = sourceText.slice(node.getStart(sourceFile), node.getEnd()).trim();
+    return {
+        text,
+        ...(includeRanges ? { range: rangeLabel(sourceFile, node) } : {})
+    };
+}
+function rangeLabel(sourceFile, node) {
+    const start = sourceFile.getLineAndCharacterOfPosition(node.getStart(sourceFile)).line + 1;
+    const end = sourceFile.getLineAndCharacterOfPosition(node.getEnd()).line + 1;
+    return start === end ? `@${start}` : `@${start}..${end}`;
+}
+function formatImportEntry(entry) {
+    const textLines = entry.text.split(/\r?\n/);
+    if (!entry.range) {
+        return textLines;
+    }
+    if (textLines.length === 0) {
+        return [entry.range];
+    }
+    return [`${entry.range} ${textLines[0] ?? ""}`, ...textLines.slice(1)];
+}
+function readImportEntries(value) {
+    if (!Array.isArray(value)) {
+        return [];
+    }
+    return value.filter(isObject).map((item) => ({
+        text: readString(item.text) ?? "",
+        ...(typeof item.range === "string" ? { range: item.range } : {})
+    })).filter((item) => item.text !== "");
+}
+function expandImportsPayloadRequests(payload) {
+    if (!isObject(payload)) {
+        return [];
+    }
+    if (Array.isArray(payload.requests)) {
+        return payload.requests.filter(isObject);
+    }
+    return [payload];
+}
+function scriptKindForFile(filePath) {
+    const ext = path.extname(filePath).toLowerCase();
+    if (ext === ".tsx") {
+        return typescript_1.default.ScriptKind.TSX;
+    }
+    if (ext === ".jsx") {
+        return typescript_1.default.ScriptKind.JSX;
+    }
+    if (ext === ".js" || ext === ".cjs" || ext === ".mjs") {
+        return typescript_1.default.ScriptKind.JS;
+    }
+    return typescript_1.default.ScriptKind.TS;
+}
+function resolveCliPath(cwd, value) {
+    return path.isAbsolute(value) ? path.normalize(value) : path.resolve(cwd, value);
+}
+function safeStat(filePath) {
+    try {
+        return fs.statSync(filePath);
+    }
+    catch {
+        return null;
+    }
+}
+function isFlagValue(args, value) {
+    const index = args.indexOf(value);
+    return index > 0 && args[index - 1]?.startsWith("--") === true;
+}
+function readString(value) {
+    return typeof value === "string" && value.trim() !== "" ? value : null;
+}
+function readStringArray(value) {
+    return Array.isArray(value)
+        ? value.filter((item) => typeof item === "string" && item.trim() !== "").map((item) => item.trim())
+        : [];
+}
+function readBoolean(value) {
+    return typeof value === "boolean" ? value : null;
+}
+function uniqueStrings(values) {
+    return Array.from(new Set(values.filter((value) => value !== "")));
+}
+function isObject(value) {
+    return Boolean(value) && typeof value === "object" && !Array.isArray(value);
+}