johankit 0.0.3 → 0.0.4

package/dist/cli/commands/paste.js

@@ -8,29 +8,34 @@ exports.paste = void 0;
 const write_1 = require("../../core/write");
 const clipboard_1 = require("../../core/clipboard");
 const cleanCodeBlock_1 = __importDefault(require("../../utils/cleanCodeBlock"));
+const diff_1 = require("../../core/diff");
+const validation_1 = require("../../core/validation");
 async function paste(dir) {
     try {
         const content = await (0, clipboard_1.readClipboard)();
         if (!content) {
             throw new Error("Clipboard empty or inaccessible");
         }
-        let files;
+        let parsed;
         try {
-            const { lang, cleaned } = (0, cleanCodeBlock_1.default)(content);
-            files = JSON.parse(cleaned);
+            const { cleaned } = (0, cleanCodeBlock_1.default)(content);
+            parsed = JSON.parse(cleaned);
         }
         catch (e) {
             throw new Error("Clipboard content is not valid JSON");
         }
-        if (!Array.isArray(files)) {
-            throw new Error("Clipboard content is not a JSON array");
+        if (Array.isArray(parsed) && parsed.every(f => f.path && f.content)) {
+            // Caso seja snapshot
+            (0, write_1.writeFiles)(dir, parsed, true);
         }
-        // Validação simples do snapshot
-        const isValidSnapshot = files.every(f => typeof f.path === 'string' && typeof f.content === 'string');
-        if (!isValidSnapshot) {
-            throw new Error("JSON does not match FileSnapshot structure {path, content}");
+        else if (Array.isArray(parsed) && parsed.every(f => f.type && f.path)) {
+            // Caso seja DiffPatch
+            const validated = (0, validation_1.validatePatches)(parsed);
+            (0, diff_1.applyDiff)(dir, validated);
+        }
+        else {
+            throw new Error("JSON is neither FileSnapshot array nor DiffPatch array");
         }
-        (0, write_1.writeFiles)(dir, files, true);
         process.stdout.write("✔ Pasted from clipboard\n");
     }
     catch (error) {

package/dist/cli/commands/prompt.js

@@ -4,14 +4,43 @@ exports.prompt = void 0;
 // src/cli/commands/prompt.ts
 const scan_1 = require("../../core/scan");
 const clipboard_1 = require("../../core/clipboard");
-async function prompt(dir, userPrompt) {
+async function prompt(dir, userPrompt, diff = false) {
     const snapshot = (0, scan_1.scanDir)(dir);
+    const llmExamples = [
+        {
+            type: 'FileSnapshot',
+            example: [
+                {
+                    path: 'src/example.ts',
+                    content: 'export const x = 42;'
+                }
+            ]
+        },
+        {
+            type: 'DiffPatch',
+            example: [
+                {
+                    type: 'modify',
+                    path: 'src/example.ts',
+                    content: 'export const x = 43;'
+                },
+                {
+                    type: 'create',
+                    path: 'src/newFile.ts',
+                    content: 'export const newFile = true;'
+                },
+                {
+                    type: 'delete',
+                    path: 'src/oldFile.ts'
+                }
+            ]
+        }
+    ];
     const template = `
 You are an AI software engineer.
 
 You will receive a JSON array representing a snapshot of a codebase.
 Each item has the following structure:
-
 {
   "path": "relative/path/to/file.ext",
   "content": "full file content"
@@ -27,19 +56,22 @@ ${JSON.stringify(snapshot, null, 2)}
 YOUR TASK
 Propose changes according to the user request.
 
-Return ONLY a JSON array of patches.
+Return ONLY a JSON array of ${diff ? 'DiffPatch' : 'FileSnapshot'}.
 
 PATCH FORMAT (STRICT)
 {
-  "path": "relative/path/to/file.ext",
-  "content": "FULL updated file content (omit for delete)"
+  \"path\": \"relative/path/to/file.ext\",
+  \"content\": \"FULL updated file content (omit for delete)\"
 }
 
+EXAMPLE RESPONSE FROM LLM:
+${JSON.stringify(diff ? llmExamples.find(e => e.type === 'DiffPatch')?.example : llmExamples.find(e => e.type === 'FileSnapshot')?.example, null, 2)}
+
 IMPORTANT RULES
 - Do NOT return explanations
 - Do NOT return markdown
 - Return ONLY valid JSON inside the \"\`\`\`\"
-- Always return within a Markdown Code Block (with \"\`\`\`json\" syntax highlighting)\")
+- Always return within a Markdown Code Block.
 
 USER REQUEST
 ${userPrompt}
@@ -47,7 +79,7 @@ ${userPrompt}
     try {
         await (0, clipboard_1.copyToClipboard)(template.trim());
         process.stdout.write(template.trim());
-        process.stdout.write("\n\n✔ Prompt + Snapshot copied to clipboard\n");
+        process.stdout.write("\n\n✔ Prompt + Snapshot + Example copied to clipboard\n");
     }
     catch (e) {
         process.stdout.write(template.trim());

package/dist/cli/commands/sync.js

@@ -6,10 +6,10 @@ const scan_1 = require("../../core/scan");
 const diff_1 = require("../../core/diff");
 const clipboard_1 = require("../../core/clipboard");
 const validation_1 = require("../../core/validation");
-async function sync(dir) {
+const write_1 = require("../../core/write");
+async function sync(dir, diff = false) {
     try {
         const snapshotBefore = (0, scan_1.scanDir)(dir);
-        // Gera prompt genérico para AI com snapshot atual
         const template = `
 You are an AI software engineer.
 
@@ -33,7 +33,7 @@ ${JSON.stringify(snapshotBefore, null, 2)}
 YOUR TASK
 Propose changes according to the user request.
 
-Return ONLY a JSON array of patches.
+Return ONLY a JSON array of ${diff ? 'DiffPatch' : 'FileSnapshot'}.
 
 PATCH FORMAT (STRICT)
 {
@@ -51,7 +51,6 @@ USER REQUEST
 `;
         await (0, clipboard_1.copyToClipboard)(template.trim());
         process.stdout.write("✔ Prompt with snapshot copied to clipboard\n");
-        // Aguarda entrada do usuário (resposta da AI) via stdin
         const input = await readStdin();
         let patches;
         try {
@@ -60,8 +59,13 @@ USER REQUEST
         catch {
             throw new Error("Invalid JSON input");
         }
-        const validated = (0, validation_1.validatePatches)(patches);
-        (0, diff_1.applyDiff)(dir, validated);
+        if (diff) {
+            const validated = (0, validation_1.validatePatches)(patches);
+            (0, diff_1.applyDiff)(dir, validated);
+        }
+        else {
+            (0, write_1.writeFiles)(dir, patches, true);
+        }
         const snapshotAfter = (0, scan_1.scanDir)(dir);
         await (0, clipboard_1.copyToClipboard)(JSON.stringify(snapshotAfter, null, 2));
         process.stdout.write("✔ Sync applied and new snapshot copied to clipboard\n");

package/dist/cli/commands/three.js

@@ -0,0 +1,106 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.three = void 0;
+const path_1 = __importDefault(require("path"));
+const ts_morph_1 = require("ts-morph");
+/**
+ * Heurística simples para classificar arquivos
+ */
+function detectFileKind(filePath, exportsCount) {
+    if (filePath.includes("/cli/"))
+        return "cli";
+    if (filePath.endsWith("index.ts") || filePath.endsWith("main.ts"))
+        return "entry";
+    if (exportsCount === 0)
+        return "util";
+    if (exportsCount > 3)
+        return "domain";
+    return "unknown";
+}
+async function three(dir) {
+    const project = new ts_morph_1.Project({});
+    project.addSourceFilesAtPaths([
+        `${dir}/**/*.{ts,tsx,js,jsx}`,
+        `!${dir}/**/node_modules/**/*`,
+        `!${dir}/**/dist/**/*`,
+    ]);
+    const tree = [];
+    for (const file of project.getSourceFiles()) {
+        const absolutePath = file.getFilePath();
+        const filePath = path_1.default.relative(dir, absolutePath);
+        if (filePath.includes("node_modules") || filePath.includes("/dist/")) {
+            continue;
+        }
+        /* ---------------- IMPORTS ---------------- */
+        const imports = file
+            .getImportDeclarations()
+            .map(i => i.getModuleSpecifierValue());
+        /* ---------------- CLASSES ---------------- */
+        const classes = file.getClasses().map(cls => ({
+            name: cls.getName() || "<anonymous>",
+            methods: cls.getMethods().map(m => ({
+                name: m.getName(),
+                params: m.getParameters().map(p => p.getName()),
+                returnType: safeType(() => m.getReturnType().getText()),
+                scope: "class",
+            }))
+        }));
+        /* ---------------- FUNCTIONS ---------------- */
+        const functions = file.getFunctions().map(fn => ({
+            name: fn.getName() || "<anonymous>",
+            params: fn.getParameters().map(p => p.getName()),
+            returnType: safeType(() => fn.getReturnType().getText()),
+            scope: "global",
+        }));
+        /* ---------------- VARIABLES ---------------- */
+        const variables = file.getVariableDeclarations().map(v => ({
+            name: v.getName(),
+            type: safeType(() => v.getType().getText()),
+            scope: v.getParent() instanceof ts_morph_1.SourceFile ? "global" : "local",
+        }));
+        /* ---------------- EXPORTS ---------------- */
+        const exports = [];
+        let mainExport;
+        file.getExportedDeclarations().forEach((decls, name) => {
+            decls.forEach(d => {
+                let kind = "unknown";
+                if (d.getKind() === ts_morph_1.SyntaxKind.ClassDeclaration)
+                    kind = "class";
+                if (d.getKind() === ts_morph_1.SyntaxKind.FunctionDeclaration)
+                    kind = "function";
+                if (d.getKind() === ts_morph_1.SyntaxKind.VariableDeclaration)
+                    kind = "variable";
+                if (d.getKind() === ts_morph_1.SyntaxKind.TypeAliasDeclaration)
+                    kind = "type";
+                exports.push({ name, kind });
+            });
+        });
+        if (exports.length === 1) {
+            mainExport = exports[0].name;
+        }
+        const kind = detectFileKind(filePath, exports.length);
+        tree.push({
+            path: filePath,
+            kind,
+            imports,
+            classes,
+            functions,
+            variables,
+            exports,
+            mainExport,
+        });
+    }
+    return tree;
+}
+exports.three = three;
+function safeType(fn) {
+    try {
+        return fn();
+    }
+    catch {
+        return "unknown";
+    }
+}

package/dist/index.js

@@ -1,54 +1,72 @@
 #!/usr/bin/env node
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-const copy_1 = require("./cli/commands/copy");
-const paste_1 = require("./cli/commands/paste");
-const prompt_1 = require("./cli/commands/prompt");
-const sync_1 = require("./cli/commands/sync");
+exports.JohankitService = void 0;
+const JohankitService_1 = require("./services/JohankitService");
+Object.defineProperty(exports, "JohankitService", { enumerable: true, get: function () { return JohankitService_1.JohankitService; } });
 const [, , command, ...args] = process.argv;
+const service = new JohankitService_1.JohankitService();
 async function main() {
-    switch (command) {
-        case "copy": {
-            const dir = args[0] ?? ".";
-            const exts = args[1]?.split(",");
-            await (0, copy_1.copy)(dir);
-            break;
-        }
-        case "paste": {
-            const dir = args[0] ?? ".";
-            await (0, paste_1.paste)(dir);
-            break;
-        }
-        case "prompt": {
-            const dir = args[0] ?? ".";
-            const userPrompt = args.slice(1).join(" ");
-            if (!userPrompt) {
-                console.error("Missing user prompt");
-                process.exit(1);
+    try {
+        switch (command) {
+            case "copy": {
+                const dir = args[0] ?? ".";
+                await service.copy(dir);
+                break;
+            }
+            case "paste": {
+                const dir = args[0] ?? ".";
+                await service.paste(dir);
+                break;
+            }
+            case "prompt": {
+                const dir = args[0] ?? ".";
+                const diff = args.includes("--diff");
+                const userPrompt = args.filter(a => a !== "--diff").slice(1).join(" ");
+                await service.prompt(dir, userPrompt, diff);
+                break;
             }
-            await (0, prompt_1.prompt)(dir, userPrompt);
-            break;
+            case "sync": {
+                const dir = args[0] ?? ".";
+                await service.sync(dir);
+                break;
+            }
+            case "three": {
+                const dir = args[0] ?? ".";
+                const output = await service.three(dir);
+                console.log(output);
+                break;
+            }
+            default:
+                showHelp();
+                break;
+        }
+    }
+    catch (error) {
+        if (error instanceof Error) {
+            console.error("Error:", error.message);
         }
-        case "sync": {
-            const dir = args[0] ?? ".";
-            await (0, sync_1.sync)(dir);
-            break;
+        else {
+            console.error("Unexpected error:", error);
         }
-        default:
-            help();
+        process.exit(1);
     }
 }
-function help() {
+function showHelp() {
     console.log(`
 Usage:
-  johankit copy <dir> [exts]
+  johankit copy <dir>
   johankit paste <dir>
-  johankit prompt <dir> "<user request>"
+  johankit prompt <dir> "<user request>" [--diff]
   johankit sync <dir>
+  johankit three <dir>
 
 Examples:
-  johankit prompt src "refatorar para async/await"
+  johankit copy src
+  johankit paste src
+  johankit prompt src "refactor to async/await"
   johankit sync src
+  johankit three src
 `);
 }
 main();

package/dist/services/JohankitService.js

@@ -0,0 +1,59 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.JohankitService = void 0;
+const createAsciiTree_1 = __importDefault(require("../utils/createAsciiTree"));
+const copy_1 = require("../cli/commands/copy");
+const paste_1 = require("../cli/commands/paste");
+const prompt_1 = require("../cli/commands/prompt");
+const three_1 = require("../cli/commands/three");
+const scan_1 = require("../core/scan");
+const diff_1 = require("../core/diff");
+const validation_1 = require("../core/validation");
+const write_1 = require("../core/write");
+class JohankitService {
+    constructor(options = {}) {
+        this.isolated = options.isolated ?? false;
+    }
+    setInput(input) {
+        this.internalInput = input;
+    }
+    async copy(dir = ".") {
+        return (0, copy_1.copy)(dir);
+    }
+    async paste(dir = ".") {
+        return (0, paste_1.paste)(dir);
+    }
+    async prompt(dir, userPrompt, diff = false) {
+        if (!userPrompt) {
+            throw new Error("Missing user prompt");
+        }
+        return (0, prompt_1.prompt)(dir, userPrompt, diff);
+    }
+    async sync(dir = ".", diff = false) {
+        const snapshotBefore = (0, scan_1.scanDir)(dir);
+        const input = this.isolated ? this.internalInput : undefined;
+        if (!this.isolated && input === undefined) {
+            throw new Error("sync() without isolation must be used via CLI (stdin)");
+        }
+        if (!input) {
+            throw new Error("No input provided for isolated sync");
+        }
+        if (diff) {
+            const validated = (0, validation_1.validatePatches)(input);
+            (0, diff_1.applyDiff)(dir, validated);
+        }
+        else {
+            (0, write_1.writeFiles)(dir, input, true);
+        }
+        const snapshotAfter = (0, scan_1.scanDir)(dir);
+        return { before: snapshotBefore, after: snapshotAfter };
+    }
+    async three(dir = ".") {
+        const tree = await (0, three_1.three)(dir);
+        return (0, createAsciiTree_1.default)(tree);
+    }
+}
+exports.JohankitService = JohankitService;

package/dist/utils/createAsciiTree.js

@@ -0,0 +1,46 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+function createAsciiTree(files) {
+    let out = "";
+    files.forEach((file, i) => {
+        const lastFile = i === files.length - 1;
+        const fPrefix = lastFile ? "└── " : "├── ";
+        const fIndent = lastFile ? "    " : "│   ";
+        out += `${fPrefix}📄 ${file.path}  (${file.kind})\n`;
+        /* -------- Imports -------- */
+        if (file.imports.length) {
+            out += `${fIndent}├── 📦 Imports [${file.imports.length}]\n`;
+            file.imports.forEach((imp, idx) => {
+                const last = idx === file.imports.length - 1;
+                out += `${fIndent}${last ? "└── " : "├── "}${imp}\n`;
+            });
+        }
+        /* -------- Exports -------- */
+        if (file.exports.length) {
+            out += `${fIndent}├── 🔑 Exports\n`;
+            file.exports.forEach((e, idx) => {
+                const last = idx === file.exports.length - 1;
+                const main = e.name === file.mainExport ? " ⭐" : "";
+                out += `${fIndent}${last ? "└── " : "├── "}${e.name} (${e.kind})${main}\n`;
+            });
+        }
+        /* -------- Variables -------- */
+        file.variables.forEach(v => {
+            out += `${fIndent}├── 💡 ${v.name}: ${v.type} (${v.scope})\n`;
+        });
+        /* -------- Functions -------- */
+        file.functions.forEach(fn => {
+            out += `${fIndent}├── ⚡ ${fn.name}(${fn.params.join(", ")}): ${fn.returnType}\n`;
+        });
+        /* -------- Classes -------- */
+        file.classes.forEach(cls => {
+            out += `${fIndent}├── ⚙️ Class ${cls.name}\n`;
+            cls.methods.forEach((m, mi) => {
+                const last = mi === cls.methods.length - 1;
+                out += `${fIndent}│   ${last ? "└── " : "├── "}➡️ ${m.name}(${m.params.join(", ")}): ${m.returnType}\n`;
+            });
+        });
+    });
+    return out;
+}
+exports.default = createAsciiTree;

package/johankit.yaml

@@ -0,0 +1,2 @@
+ignore:
+  - dist

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "johankit",
-  "version": "0.0.3",
+  "version": "0.0.4",
   "description": "",
   "main": "index.js",
   "bin": {
@@ -27,6 +27,7 @@
     "clipboardy": "^5.0.2",
     "commander": "^14.0.2",
     "js-yaml": "^4.1.1",
+    "ts-morph": "^27.0.2",
     "vm2": "^3.10.0"
   }
 }

package/src/cli/commands/paste.ts

@@ -2,6 +2,8 @@
 import { writeFiles } from "../../core/write";
 import { readClipboard } from "../../core/clipboard";
 import cleanCodeBlock from "../../utils/cleanCodeBlock";
+import { applyDiff } from "../../core/diff";
+import { validatePatches } from "../../core/validation";
 
 export async function paste(dir: string) {
     try {
@@ -11,28 +13,25 @@ export async function paste(dir: string) {
             throw new Error("Clipboard empty or inaccessible");
         }
 
-        let files;
+        let parsed;
         try {
-            const { lang, cleaned } = cleanCodeBlock(content) 
-            files = JSON.parse(cleaned);
+            const { cleaned } = cleanCodeBlock(content);
+            parsed = JSON.parse(cleaned);
         } catch (e) {
             throw new Error("Clipboard content is not valid JSON");
         }
 
-        if (!Array.isArray(files)) {
-            throw new Error("Clipboard content is not a JSON array");
+        if (Array.isArray(parsed) && parsed.every(f => f.path && f.content)) {
+            // Caso seja snapshot
+            writeFiles(dir, parsed, true);
+        } else if (Array.isArray(parsed) && parsed.every(f => f.type && f.path)) {
+            // Caso seja DiffPatch
+            const validated = validatePatches(parsed);
+            applyDiff(dir, validated);
+        } else {
+            throw new Error("JSON is neither FileSnapshot array nor DiffPatch array");
         }
 
-        // Validação simples do snapshot
-        const isValidSnapshot = files.every(f =>
-            typeof f.path === 'string' && typeof f.content === 'string'
-        );
-
-        if (!isValidSnapshot) {
-            throw new Error("JSON does not match FileSnapshot structure {path, content}");
-        }
-
-        writeFiles(dir, files, true);
         process.stdout.write("✔ Pasted from clipboard\n");
     } catch (error) {
         process.stderr.write("✖ Paste failed\n");
@@ -41,4 +40,4 @@ export async function paste(dir: string) {
         }
         process.exit(1);
     }
-}
+}

package/src/cli/commands/prompt.ts

@@ -2,15 +2,50 @@
 import { scanDir } from "../../core/scan";
 import { copyToClipboard } from "../../core/clipboard";
 
-export async function prompt(dir: string, userPrompt: string) {
+interface LLMResponseExample {
+  type: 'FileSnapshot' | 'DiffPatch';
+  example: any;
+}
+
+export async function prompt(dir: string, userPrompt: string, diff = false) {
   const snapshot = scanDir(dir);
 
+  const llmExamples: LLMResponseExample[] = [
+    {
+      type: 'FileSnapshot',
+      example: [
+        {
+          path: 'src/example.ts',
+          content: 'export const x = 42;'
+        }
+      ]
+    },
+    {
+      type: 'DiffPatch',
+      example: [
+        {
+          type: 'modify',
+          path: 'src/example.ts',
+          content: 'export const x = 43;'
+        },
+        {
+          type: 'create',
+          path: 'src/newFile.ts',
+          content: 'export const newFile = true;'
+        },
+        {
+          type: 'delete',
+          path: 'src/oldFile.ts'
+        }
+      ]
+    }
+  ];
+
   const template = `
 You are an AI software engineer.
 
 You will receive a JSON array representing a snapshot of a codebase.
 Each item has the following structure:
-
 {
   "path": "relative/path/to/file.ext",
   "content": "full file content"
@@ -26,19 +61,22 @@ ${JSON.stringify(snapshot, null, 2)}
 YOUR TASK
 Propose changes according to the user request.
 
-Return ONLY a JSON array of patches.
+Return ONLY a JSON array of ${diff ? 'DiffPatch' : 'FileSnapshot'}.
 
 PATCH FORMAT (STRICT)
 {
-  "path": "relative/path/to/file.ext",
-  "content": "FULL updated file content (omit for delete)"
+  \"path\": \"relative/path/to/file.ext\",
+  \"content\": \"FULL updated file content (omit for delete)\"
 }
 
+EXAMPLE RESPONSE FROM LLM:
+${JSON.stringify(diff ? llmExamples.find(e => e.type === 'DiffPatch')?.example : llmExamples.find(e => e.type === 'FileSnapshot')?.example, null, 2)}
+
 IMPORTANT RULES
 - Do NOT return explanations
 - Do NOT return markdown
 - Return ONLY valid JSON inside the \"\`\`\`\"
-- Always return within a Markdown Code Block (with \"\`\`\`json\" syntax highlighting)\")
+- Always return within a Markdown Code Block.
 
 USER REQUEST
 ${userPrompt}
@@ -47,7 +85,7 @@ ${userPrompt}
   try {
     await copyToClipboard(template.trim());
     process.stdout.write(template.trim());
-    process.stdout.write("\n\n✔ Prompt + Snapshot copied to clipboard\n");
+    process.stdout.write("\n\n✔ Prompt + Snapshot + Example copied to clipboard\n");
   } catch (e) {
     process.stdout.write(template.trim());
     process.stderr.write("\n✖ Failed to copy to clipboard (output only)\n");