lat.md 0.5.0 → 0.7.0

package/dist/src/cli/hook.js

@@ -0,0 +1,147 @@
+import { dirname } from 'node:path';
+import { findLatticeDir } from '../lattice.js';
+import { plainStyler } from '../context.js';
+import { expandPrompt } from './expand.js';
+import { runSearch } from './search.js';
+import { getSection, formatSectionOutput } from './section.js';
+import { getLlmKey } from '../config.js';
+function outputPromptSubmit(context) {
+    process.stdout.write(JSON.stringify({
+        hookSpecificOutput: {
+            hookEventName: 'UserPromptSubmit',
+            additionalContext: context,
+        },
+    }));
+}
+function outputStop(reason) {
+    process.stdout.write(JSON.stringify({
+        decision: 'block',
+        reason,
+    }));
+}
+async function readStdin() {
+    const chunks = [];
+    for await (const chunk of process.stdin) {
+        chunks.push(chunk);
+    }
+    return Buffer.concat(chunks).toString('utf-8');
+}
+function hasWikiLinks(text) {
+    return /\[\[[^\]]+\]\]/.test(text);
+}
+function makeHookCtx(latDir) {
+    return {
+        latDir,
+        projectRoot: dirname(latDir),
+        styler: plainStyler,
+        mode: 'cli',
+    };
+}
+async function searchAndExpand(ctx, userPrompt) {
+    let key;
+    try {
+        key = getLlmKey();
+    }
+    catch {
+        return null;
+    }
+    if (!key)
+        return null;
+    const result = await runSearch(ctx.latDir, userPrompt, key, 5);
+    if (result.matches.length === 0)
+        return null;
+    const parts = [
+        `Search results for the user prompt (${result.matches.length} matches):`,
+        '',
+    ];
+    for (const match of result.matches) {
+        const sectionResult = await getSection(ctx, match.section.id);
+        if (sectionResult.kind === 'found') {
+            parts.push(formatSectionOutput(ctx, sectionResult));
+            parts.push('');
+        }
+    }
+    return parts.join('\n');
+}
+async function handleUserPromptSubmit() {
+    let userPrompt = '';
+    try {
+        const raw = await readStdin();
+        const input = JSON.parse(raw);
+        userPrompt = input.user_prompt ?? '';
+    }
+    catch {
+        // If we can't parse stdin, still emit the reminder
+    }
+    const parts = [];
+    parts.push("Before starting work, run `lat search` with one or more queries describing the user's intent.", 'ALWAYS do this, even when the task seems straightforward — search results may reveal critical design details, protocols, or constraints.', 'Use `lat section` to read the full content of relevant matches.', 'Do not read files, write code, or run commands until you have searched.');
+    const latDir = findLatticeDir();
+    if (latDir && userPrompt) {
+        const ctx = makeHookCtx(latDir);
+        // If the user prompt contains [[refs]], resolve them inline
+        if (hasWikiLinks(userPrompt)) {
+            try {
+                const expanded = await expandPrompt(ctx, userPrompt);
+                if (expanded) {
+                    parts.push('', 'Expanded user prompt with resolved [[refs]]:', expanded);
+                }
+                else {
+                    parts.push('', 'NOTE: The user prompt contains [[refs]] but they could not be resolved. Ask the user to correct them.');
+                }
+            }
+            catch {
+                parts.push('', 'NOTE: The user prompt contains [[refs]] but resolution failed. Run `lat expand` on the prompt text manually.');
+            }
+        }
+        // Search for relevant sections and include their full content
+        try {
+            const searchContext = await searchAndExpand(ctx, userPrompt);
+            if (searchContext) {
+                parts.push('', searchContext);
+            }
+        }
+        catch {
+            // Search failed (no key, index error, etc.) — agent can search manually
+        }
+    }
+    outputPromptSubmit(parts.join('\n'));
+}
+async function handleStop() {
+    // Only emit the reminder if we're in a project with lat.md
+    const latDir = findLatticeDir();
+    if (!latDir)
+        return;
+    // Read stdin to check if we already blocked once
+    let stopHookActive = false;
+    try {
+        const raw = await readStdin();
+        const input = JSON.parse(raw);
+        stopHookActive = input.stop_hook_active ?? false;
+    }
+    catch {
+        // If we can't parse stdin, treat as first attempt
+    }
+    // Don't block twice — avoids infinite loop
+    if (stopHookActive)
+        return;
+    const parts = [];
+    parts.push('Before finishing, verify:', '- Did you update `lat.md/`? Run `lat search` with a query describing what you changed to find relevant sections that may need updating.', '- Did you run `lat check` and confirm all links and code refs pass?', 'If you made code changes but did not update lat.md/, do that now.');
+    outputStop(parts.join('\n'));
+}
+export async function hookCmd(agent, event) {
+    if (agent !== 'claude') {
+        console.error(`Unknown agent: ${agent}. Supported: claude`);
+        process.exit(1);
+    }
+    switch (event) {
+        case 'UserPromptSubmit':
+            await handleUserPromptSubmit();
+            break;
+        case 'Stop':
+            await handleStop();
+            break;
+        default:
+            console.error(`Unknown hook event: ${event}. Supported: UserPromptSubmit, Stop`);
+            process.exit(1);
+    }
+}

package/dist/src/cli/index.js

@@ -8,8 +8,6 @@ import { dirname, join } from 'node:path';
 import { fileURLToPath } from 'node:url';
 import { Command } from 'commander';
 import { resolveContext } from './context.js';
-import { locateCmd } from './locate.js';
-import { refsCmd } from './refs.js';
 function findPackageJson() {
     let dir = dirname(fileURLToPath(import.meta.url));
     while (true) {
@@ -24,6 +22,14 @@ function findPackageJson() {
         dir = parent;
     }
 }
+function handleResult(result) {
+    if (result.isError) {
+        console.error(result.output);
+        process.exit(1);
+    }
+    if (result.output)
+        console.log(result.output);
+}
 const version = findPackageJson();
 const program = new Command();
 program
@@ -39,7 +45,17 @@ program
     .argument('<query>', 'section id to search for')
     .action(async (query) => {
     const ctx = resolveContext(program.opts());
-    await locateCmd(ctx, query);
+    const { locateCommand } = await import('./locate.js');
+    handleResult(await locateCommand(ctx, query));
+});
+program
+    .command('section')
+    .description('Show a section with its content, outgoing refs, and incoming refs')
+    .argument('<query>', 'section id to look up')
+    .action(async (query) => {
+    const ctx = resolveContext(program.opts());
+    const { sectionCommand } = await import('./section.js');
+    handleResult(await sectionCommand(ctx, query));
 });
 program
     .command('refs')
@@ -53,46 +69,50 @@ program
         process.exit(1);
     }
     const ctx = resolveContext(program.opts());
-    await refsCmd(ctx, query, scope);
+    const { refsCommand } = await import('./refs.js');
+    handleResult(await refsCommand(ctx, query, scope));
 });
 const check = program
     .command('check')
     .description('Validate links and code references')
     .action(async () => {
     const ctx = resolveContext(program.opts());
-    const { checkAllCmd } = await import('./check.js');
-    await checkAllCmd(ctx);
+    const { checkAllCommand } = await import('./check.js');
+    handleResult(await checkAllCommand(ctx));
 });
 check
     .command('md')
     .description('Validate wiki links in lat.md markdown files')
     .action(async () => {
     const ctx = resolveContext(program.opts());
-    const { checkMdCmd } = await import('./check.js');
-    await checkMdCmd(ctx);
+    const { checkMdCommand } = await import('./check.js');
+    handleResult(await checkMdCommand(ctx));
 });
 check
     .command('code-refs')
     .description('Validate @lat code references and coverage')
     .action(async () => {
     const ctx = resolveContext(program.opts());
-    const { checkCodeRefsCmd } = await import('./check.js');
-    await checkCodeRefsCmd(ctx);
+    const { checkCodeRefsCommand } = await import('./check.js');
+    handleResult(await checkCodeRefsCommand(ctx));
 });
 check
     .command('index')
     .description('Validate directory index files in lat.md')
     .action(async () => {
     const ctx = resolveContext(program.opts());
-    const { checkIndexCmd } = await import('./check.js');
-    await checkIndexCmd(ctx);
+    const { checkIndexCommand } = await import('./check.js');
+    handleResult(await checkIndexCommand(ctx));
 });
-program
-    .command('prompt')
-    .description('Expand [[refs]] in a prompt to lat.md section locations')
-    .argument('[text]', 'prompt text')
-    .option('--stdin', 'read prompt from stdin')
-    .action(async (text, opts) => {
+check
+    .command('sections')
+    .description('Validate section leading paragraphs in lat.md')
+    .action(async () => {
+    const ctx = resolveContext(program.opts());
+    const { checkSectionsCommand } = await import('./check.js');
+    handleResult(await checkSectionsCommand(ctx));
+});
+async function runExpand(text, opts) {
     if (opts.stdin) {
         const chunks = [];
         for await (const chunk of process.stdin) {
@@ -101,12 +121,33 @@ program
         text = Buffer.concat(chunks).toString('utf-8');
     }
     if (!text) {
-        console.error('Provide prompt text as an argument or use --stdin');
+        console.error('Provide text as an argument or use --stdin');
         process.exit(1);
     }
     const ctx = resolveContext(program.opts());
-    const { promptCmd } = await import('./prompt.js');
-    await promptCmd(ctx, text);
+    const { expandCommand } = await import('./expand.js');
+    const result = await expandCommand(ctx, text);
+    if (result.isError) {
+        console.error(result.output);
+        process.exit(1);
+    }
+    // Use stdout.write (no trailing newline) for piping
+    process.stdout.write(result.output);
+}
+program
+    .command('expand')
+    .description('Expand [[refs]] in text to lat.md section locations')
+    .argument('[text]', 'text containing [[refs]]')
+    .option('--stdin', 'read text from stdin')
+    .action(runExpand);
+// Deprecated alias — hidden from --help
+program
+    .command('prompt', { hidden: true })
+    .argument('[text]')
+    .option('--stdin')
+    .action(async (text, opts) => {
+    console.error('Warning: `lat prompt` is deprecated, use `lat expand` instead.');
+    await runExpand(text, opts);
 });
 program
     .command('search')
@@ -116,16 +157,15 @@ program
     .option('--reindex', 'force full re-indexing')
     .action(async (query, opts) => {
     const ctx = resolveContext(program.opts());
-    const { searchCmd } = await import('./search.js');
-    await searchCmd(ctx, query, {
-        limit: parseInt(opts.limit),
-        reindex: opts.reindex,
-    });
+    const { searchCommand, cliProgress } = await import('./search.js');
+    const progress = cliProgress(!!opts.reindex, ctx.styler);
+    const result = await searchCommand(ctx, query, { limit: parseInt(opts.limit), reindex: opts.reindex }, progress);
+    handleResult(result);
 });
 program
     .command('gen')
-    .description('Generate a file to stdout (agents.md, claude.md)')
-    .argument('<target>', 'file to generate: agents.md or claude.md')
+    .description('Generate a file to stdout (agents.md, claude.md, cursor-rules.md)')
+    .argument('<target>', 'file to generate: agents.md, claude.md, cursor-rules.md')
     .action(async (target) => {
     const { genCmd } = await import('./gen.js');
     await genCmd(target);
@@ -138,6 +178,15 @@ program
     const { initCmd } = await import('./init.js');
     await initCmd(dir);
 });
+program
+    .command('hook')
+    .description('Handle agent hook events (called by agent hooks, not directly)')
+    .argument('<agent>', 'agent name (claude)')
+    .argument('<event>', 'hook event (UserPromptSubmit, Stop)')
+    .action(async (agent, event) => {
+    const { hookCmd } = await import('./hook.js');
+    await hookCmd(agent, event);
+});
 program
     .command('mcp')
     .description('Start the MCP server (stdio transport)')