driftdetect-mcp 0.6.1 → 0.7.1

package/dist/tools/surgical/similar.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"similar.d.ts","sourceRoot":"","sources":["../../../src/tools/surgical/similar.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAEH,OAAO,KAAK,EAAE,cAAc,EAAgB,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAOnF,MAAM,WAAW,WAAW;IAC1B,yCAAyC;IACzC,MAAM,EAAE,cAAc,GAAG,SAAS,GAAG,WAAW,GAAG,MAAM,GAAG,SAAS,GAAG,MAAM,GAAG,YAAY,CAAC;IAC9F,mCAAmC;IACnC,WAAW,EAAE,MAAM,CAAC;IACpB,4CAA4C;IAC5C,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,+BAA+B;IAC/B,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC9B,KAAK,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC3B,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,MAAM,EAAE,CAAC;CACpB;AAED,MAAM,WAAW,kBAAkB;IACjC,MAAM,EAAE,MAAM,CAAC;IACf,aAAa,EAAE,MAAM,CAAC;IACtB,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,WAAW;IAC1B,OAAO,EAAE,YAAY,EAAE,CAAC;IACxB,WAAW,EAAE,kBAAkB,CAAC;CACjC;AA0BD,wBAAsB,aAAa,CACjC,cAAc,EAAE,cAAc,EAC9B,YAAY,EAAE,YAAY,EAC1B,IAAI,EAAE,WAAW,GAChB,OAAO,CAAC;IAAE,OAAO,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,CAAA;CAAE,CAAC,CA2K7D;AAuFD;;GAEG;AACH,eAAO,MAAM,qBAAqB;;;;;;;;;;;;;;;;;;;;;;;;;;CA0BjC,CAAC"}

package/dist/tools/surgical/similar.js

@@ -0,0 +1,285 @@
+/**
+ * drift_similar - Find Semantically Similar Code
+ *
+ * Layer: Surgical
+ * Token Budget: 500 target, 1000 max
+ * Cache TTL: 5 minutes
+ * Invalidation Keys: patterns, callgraph
+ *
+ * Finds code semantically similar to what the AI is about to write.
+ * Solves: AI needs to see an example but there are 50 options. Which is most relevant?
+ */
+import { createResponseBuilder, Errors, metrics } from '../../infrastructure/index.js';
+// ============================================================================
+// Intent to Pattern Category Mapping
+// ============================================================================
+const INTENT_CATEGORIES = {
+    api_endpoint: ['api', 'errors', 'auth'],
+    service: ['data-access', 'errors', 'logging'],
+    component: ['components', 'styling', 'accessibility'],
+    hook: ['components', 'data-access'],
+    utility: ['errors', 'logging'],
+    test: ['testing'],
+    middleware: ['api', 'auth', 'errors'],
+};
+const INTENT_DECORATORS = {
+    api_endpoint: ['@app.route', '@router', 'HttpGet', 'HttpPost', '@Get', '@Post', 'Route'],
+    middleware: ['@middleware', 'Middleware'],
+    test: ['@test', 'describe', 'it', 'test'],
+};
+// ============================================================================
+// Handler
+// ============================================================================
+export async function handleSimilar(callGraphStore, patternStore, args) {
+    const startTime = Date.now();
+    const builder = createResponseBuilder();
+    // Validate input
+    if (!args.intent) {
+        throw Errors.missingParameter('intent');
+    }
+    if (!args.description || args.description.trim() === '') {
+        throw Errors.missingParameter('description');
+    }
+    const intent = args.intent;
+    const description = args.description.toLowerCase();
+    const scope = args.scope;
+    const limit = Math.min(args.limit ?? 3, 10);
+    // Load stores
+    await callGraphStore.initialize();
+    await patternStore.initialize();
+    const graph = callGraphStore.getGraph();
+    if (!graph) {
+        throw Errors.custom('CALLGRAPH_NOT_BUILT', 'Call graph has not been built. Run "drift callgraph build" first.', ['drift_status']);
+    }
+    // Get patterns for scoring
+    const allPatterns = patternStore.getAll();
+    const patternsByFile = new Map();
+    for (const pattern of allPatterns) {
+        for (const loc of pattern.locations) {
+            const existing = patternsByFile.get(loc.file) ?? [];
+            existing.push(pattern.name);
+            patternsByFile.set(loc.file, existing);
+        }
+    }
+    // Score all functions
+    const scored = [];
+    const targetCategories = INTENT_CATEGORIES[intent] ?? [];
+    const targetDecorators = INTENT_DECORATORS[intent] ?? [];
+    for (const [, func] of graph.functions) {
+        // Skip if scope specified and doesn't match
+        if (scope && !func.file.includes(scope)) {
+            continue;
+        }
+        const reasons = [];
+        let score = 0;
+        // 1. Decorator match (strong signal)
+        for (const dec of func.decorators) {
+            if (targetDecorators.some(td => dec.includes(td))) {
+                score += 0.3;
+                reasons.push(`Has ${intent} decorator`);
+                break;
+            }
+        }
+        // 2. Pattern category match
+        const filePatterns = patternsByFile.get(func.file) ?? [];
+        const categoryMatches = filePatterns.filter(p => targetCategories.some(cat => p.toLowerCase().includes(cat)));
+        if (categoryMatches.length > 0) {
+            score += 0.2 * Math.min(categoryMatches.length, 3);
+            reasons.push(`Matches ${categoryMatches.length} ${intent} patterns`);
+        }
+        // 3. Name similarity (fuzzy match on description keywords)
+        const keywords = description.split(/\s+/).filter(w => w.length > 2);
+        const nameWords = func.name.toLowerCase().split(/(?=[A-Z])|_|-/).map(w => w.toLowerCase());
+        const qualifiedWords = func.qualifiedName.toLowerCase().split(/[._-]/);
+        let keywordMatches = 0;
+        for (const kw of keywords) {
+            if (nameWords.some(nw => nw.includes(kw) || kw.includes(nw))) {
+                keywordMatches++;
+            }
+            else if (qualifiedWords.some(qw => qw.includes(kw) || kw.includes(qw))) {
+                keywordMatches += 0.5;
+            }
+        }
+        if (keywordMatches > 0 && keywords.length > 0) {
+            const keywordScore = (keywordMatches / keywords.length) * 0.3;
+            score += keywordScore;
+            reasons.push(`Name matches ${Math.round(keywordMatches)} keywords`);
+        }
+        // 4. Exported functions are more likely to be good examples
+        if (func.isExported) {
+            score += 0.1;
+        }
+        // 5. Has parameters (more complete example)
+        if (func.parameters.length > 0) {
+            score += 0.05;
+        }
+        // 6. Has return type (better documented)
+        if (func.returnType) {
+            score += 0.05;
+        }
+        if (score > 0.1) {
+            scored.push({ func, score, reasons });
+        }
+    }
+    // Sort by score and take top N
+    scored.sort((a, b) => b.score - a.score);
+    const topMatches = scored.slice(0, limit);
+    // Build matches
+    const matches = topMatches.map(({ func, score, reasons }) => ({
+        file: func.file,
+        function: func.name,
+        class: func.className,
+        similarity: Math.round(score * 100) / 100,
+        reason: reasons.slice(0, 2).join('. '),
+        preview: buildPreview(func),
+        patterns: patternsByFile.get(func.file)?.slice(0, 3) ?? [],
+    }));
+    // Analyze conventions from matches
+    const conventions = analyzeConventions(topMatches.map(m => m.func));
+    const data = {
+        matches,
+        conventions,
+    };
+    // Build summary
+    let summary;
+    if (matches.length === 0) {
+        summary = `No similar ${intent} found matching "${args.description}"`;
+    }
+    else {
+        summary = `Found ${matches.length} similar ${intent} example${matches.length !== 1 ? 's' : ''}. Top: ${matches[0].file}`;
+    }
+    // Build hints
+    const hints = {
+        nextActions: matches.length > 0
+            ? [
+                `Use drift_signature to get full signature for "${matches[0].function}"`,
+                `Use drift_imports to get correct imports for your new file`,
+            ]
+            : [
+                'Try a different intent or broader description',
+                'Use drift_patterns_list to see available patterns',
+            ],
+        relatedTools: ['drift_signature', 'drift_imports', 'drift_code_examples'],
+    };
+    if (matches.some(m => m.similarity < 0.3)) {
+        hints.warnings = ['Some matches have low similarity - review carefully'];
+    }
+    // Record metrics
+    metrics.recordRequest('drift_similar', Date.now() - startTime, true, false);
+    return builder
+        .withSummary(summary)
+        .withData(data)
+        .withHints(hints)
+        .buildContent();
+}
+// ============================================================================
+// Helpers
+// ============================================================================
+/**
+ * Build a preview of the function signature
+ */
+function buildPreview(func) {
+    const parts = [];
+    // Decorators (first one)
+    if (func.decorators.length > 0) {
+        parts.push(`@${func.decorators[0]}`);
+    }
+    // Export + async
+    if (func.isExported)
+        parts.push('export');
+    if (func.isAsync)
+        parts.push('async');
+    parts.push('function');
+    parts.push(func.name);
+    // Parameters (abbreviated)
+    const params = func.parameters.slice(0, 3).map(p => {
+        if (p.type)
+            return `${p.name}: ${p.type}`;
+        return p.name;
+    });
+    if (func.parameters.length > 3) {
+        params.push('...');
+    }
+    let preview = parts.join(' ') + `(${params.join(', ')})`;
+    if (func.returnType) {
+        preview += `: ${func.returnType}`;
+    }
+    return preview;
+}
+/**
+ * Analyze conventions from matched functions
+ */
+function analyzeConventions(funcs) {
+    if (funcs.length === 0) {
+        return {
+            naming: 'Unknown',
+            errorHandling: 'Unknown',
+            imports: 'Unknown',
+        };
+    }
+    // Naming convention
+    const names = funcs.map(f => f.name);
+    const hasCamelCase = names.some(n => /^[a-z][a-zA-Z]*$/.test(n));
+    const hasSnakeCase = names.some(n => /^[a-z][a-z_]*$/.test(n));
+    const hasPascalCase = names.some(n => /^[A-Z][a-zA-Z]*$/.test(n));
+    let naming = 'mixed';
+    if (hasCamelCase && !hasSnakeCase)
+        naming = 'camelCase';
+    else if (hasSnakeCase && !hasCamelCase)
+        naming = 'snake_case';
+    else if (hasPascalCase)
+        naming = 'PascalCase';
+    // Error handling (check return types)
+    const returnTypes = funcs.map(f => f.returnType).filter(Boolean);
+    let errorHandling = 'Unknown';
+    if (returnTypes.some(rt => rt?.includes('Result'))) {
+        errorHandling = 'Result<T> pattern';
+    }
+    else if (returnTypes.some(rt => rt?.includes('Promise'))) {
+        errorHandling = 'Async with try/catch';
+    }
+    else if (funcs.some(f => f.isAsync)) {
+        errorHandling = 'Async functions';
+    }
+    // Import style (inferred from file paths)
+    const files = funcs.map(f => f.file);
+    let imports = 'relative paths';
+    if (files.some(f => f.includes('@/'))) {
+        imports = 'Path alias (@/)';
+    }
+    else if (files.some(f => f.includes('~/'))) {
+        imports = 'Path alias (~/)';
+    }
+    return { naming, errorHandling, imports };
+}
+/**
+ * Tool definition for MCP registration
+ */
+export const similarToolDefinition = {
+    name: 'drift_similar',
+    description: 'Find code semantically similar to what you\'re about to write. Returns relevant examples with patterns and conventions. Use before writing new code to find the right template.',
+    inputSchema: {
+        type: 'object',
+        properties: {
+            intent: {
+                type: 'string',
+                enum: ['api_endpoint', 'service', 'component', 'hook', 'utility', 'test', 'middleware'],
+                description: 'What kind of code are you writing?',
+            },
+            description: {
+                type: 'string',
+                description: 'Natural language description (e.g., "user preferences CRUD")',
+            },
+            scope: {
+                type: 'string',
+                description: 'Optional: limit to specific directory',
+            },
+            limit: {
+                type: 'number',
+                description: 'Max results (default: 3, max: 10)',
+            },
+        },
+        required: ['intent', 'description'],
+    },
+};
+//# sourceMappingURL=similar.js.map

package/dist/tools/surgical/similar.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"similar.js","sourceRoot":"","sources":["../../../src/tools/surgical/similar.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAGH,OAAO,EAAE,qBAAqB,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,+BAA+B,CAAC;AAsCvF,+EAA+E;AAC/E,qCAAqC;AACrC,+EAA+E;AAE/E,MAAM,iBAAiB,GAA6B;IAClD,YAAY,EAAE,CAAC,KAAK,EAAE,QAAQ,EAAE,MAAM,CAAC;IACvC,OAAO,EAAE,CAAC,aAAa,EAAE,QAAQ,EAAE,SAAS,CAAC;IAC7C,SAAS,EAAE,CAAC,YAAY,EAAE,SAAS,EAAE,eAAe,CAAC;IACrD,IAAI,EAAE,CAAC,YAAY,EAAE,aAAa,CAAC;IACnC,OAAO,EAAE,CAAC,QAAQ,EAAE,SAAS,CAAC;IAC9B,IAAI,EAAE,CAAC,SAAS,CAAC;IACjB,UAAU,EAAE,CAAC,KAAK,EAAE,MAAM,EAAE,QAAQ,CAAC;CACtC,CAAC;AAEF,MAAM,iBAAiB,GAA6B;IAClD,YAAY,EAAE,CAAC,YAAY,EAAE,SAAS,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC;IACxF,UAAU,EAAE,CAAC,aAAa,EAAE,YAAY,CAAC;IACzC,IAAI,EAAE,CAAC,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE,MAAM,CAAC;CAC1C,CAAC;AAEF,+EAA+E;AAC/E,UAAU;AACV,+EAA+E;AAE/E,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,cAA8B,EAC9B,YAA0B,EAC1B,IAAiB;IAEjB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7B,MAAM,OAAO,GAAG,qBAAqB,EAAe,CAAC;IAErD,iBAAiB;IACjB,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;QACjB,MAAM,MAAM,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;IAC1C,CAAC;IACD,IAAI,CAAC,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;QACxD,MAAM,MAAM,CAAC,gBAAgB,CAAC,aAAa,CAAC,CAAC;IAC/C,CAAC;IAED,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;IAC3B,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC,WAAW,EAAE,CAAC;IACnD,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;IACzB,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;IAE5C,cAAc;IACd,MAAM,cAAc,CAAC,UAAU,EAAE,CAAC;IAClC,MAAM,YAAY,CAAC,UAAU,EAAE,CAAC;IAEhC,MAAM,KAAK,GAAG,cAAc,CAAC,QAAQ,EAAE,CAAC;IACxC,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,MAAM,MAAM,CAAC,MAAM,CACjB,qBAAqB,EACrB,mEAAmE,EACnE,CAAC,cAAc,CAAC,CACjB,CAAC;IACJ,CAAC;IAED,2BAA2B;IAC3B,MAAM,WAAW,GAAG,YAAY,CAAC,MAAM,EAAE,CAAC;IAC1C,MAAM,cAAc,GAAG,IAAI,GAAG,EAAoB,CAAC;IAEnD,KAAK,MAAM,OAAO,IAAI,WAAW,EAAE,CAAC;QAClC,KAAK,MAAM,GAAG,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;YACpC,MAAM,QAAQ,GAAG,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;YACpD,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YAC5B,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;QACzC,CAAC;IACH,CAAC;IAED,sBAAsB;IACtB,MAAM,MAAM,GAAoE,EAAE,CAAC;IACnF,MAAM,gBAAgB,GAAG,iBAAiB,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;IACzD,MAAM,gBAAgB,GAAG,iBAAiB,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;IAEzD,KAAK,MAAM,CAAC,EAAE,IAAI,CAAC,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;QACvC,4CAA4C;QAC5C,IAAI,KAAK,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YACxC,SAAS;QACX,CAAC;QAED,MAAM,OAAO,GAAa,EAAE,CAAC;QAC7B,IAAI,KAAK,GAAG,CAAC,CAAC;QAEd,qCAAqC;QACrC,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YAClC,IAAI,gBAAgB,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;gBAClD,KAAK,IAAI,GAAG,CAAC;gBACb,OAAO,CAAC,IAAI,CAAC,OAAO,MAAM,YAAY,CAAC,CAAC;gBACxC,MAAM;YACR,CAAC;QACH,CAAC;QAED,4BAA4B;QAC5B,MAAM,YAAY,GAAG,cAAc,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;QACzD,MAAM,eAAe,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAC9C,gBAAgB,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAC5D,CAAC;QACF,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC/B,KAAK,IAAI,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,eAAe,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YACnD,OAAO,CAAC,IAAI,CAAC,WAAW,eAAe,CAAC,MAAM,IAAI,MAAM,WAAW,CAAC,CAAC;QACvE,CAAC;QAED,2DAA2D;QAC3D,MAAM,QAAQ,GAAG,WAAW,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACpE,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC;QAC3F,MAAM,cAAc,GAAG,IAAI,CAAC,aAAa,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAEvE,IAAI,cAAc,GAAG,CAAC,CAAC;QACvB,KAAK,MAAM,EAAE,IAAI,QAAQ,EAAE,CAAC;YAC1B,IAAI,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;gBAC7D,cAAc,EAAE,CAAC;YACnB,CAAC;iBAAM,IAAI,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;gBACzE,cAAc,IAAI,GAAG,CAAC;YACxB,CAAC;QACH,CAAC;QACD,IAAI,cAAc,GAAG,CAAC,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9C,MAAM,YAAY,GAAG,CAAC,cAAc,GAAG,QAAQ,CAAC,MAAM,CAAC,GAAG,GAAG,CAAC;YAC9D,KAAK,IAAI,YAAY,CAAC;YACtB,OAAO,CAAC,IAAI,CAAC,gBAAgB,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;QACtE,CAAC;QAED,4DAA4D;QAC5D,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,KAAK,IAAI,GAAG,CAAC;QACf,CAAC;QAED,4CAA4C;QAC5C,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC/B,KAAK,IAAI,IAAI,CAAC;QAChB,CAAC;QAED,yCAAyC;QACzC,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,KAAK,IAAI,IAAI,CAAC;QAChB,CAAC;QAED,IAAI,KAAK,GAAG,GAAG,EAAE,CAAC;YAChB,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;IAED,+BAA+B;IAC/B,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;IACzC,MAAM,UAAU,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IAE1C,gBAAgB;IAChB,MAAM,OAAO,GAAmB,UAAU,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;QAC5E,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,QAAQ,EAAE,IAAI,CAAC,IAAI;QACnB,KAAK,EAAE,IAAI,CAAC,SAAS;QACrB,UAAU,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,GAAG,CAAC,GAAG,GAAG;QACzC,MAAM,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;QACtC,OAAO,EAAE,YAAY,CAAC,IAAI,CAAC;QAC3B,QAAQ,EAAE,cAAc,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,EAAE;KAC3D,CAAC,CAAC,CAAC;IAEJ,mCAAmC;IACnC,MAAM,WAAW,GAAG,kBAAkB,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;IAEpE,MAAM,IAAI,GAAgB;QACxB,OAAO;QACP,WAAW;KACZ,CAAC;IAEF,gBAAgB;IAChB,IAAI,OAAe,CAAC;IACpB,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,GAAG,cAAc,MAAM,oBAAoB,IAAI,CAAC,WAAW,GAAG,CAAC;IACxE,CAAC;SAAM,CAAC;QACN,OAAO,GAAG,SAAS,OAAO,CAAC,MAAM,YAAY,MAAM,WAAW,OAAO,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,UAAU,OAAO,CAAC,CAAC,CAAE,CAAC,IAAI,EAAE,CAAC;IAC5H,CAAC;IAED,cAAc;IACd,MAAM,KAAK,GAA2E;QACpF,WAAW,EAAE,OAAO,CAAC,MAAM,GAAG,CAAC;YAC7B,CAAC,CAAC;gBACE,kDAAkD,OAAO,CAAC,CAAC,CAAE,CAAC,QAAQ,GAAG;gBACzE,4DAA4D;aAC7D;YACH,CAAC,CAAC;gBACE,+CAA+C;gBAC/C,mDAAmD;aACpD;QACL,YAAY,EAAE,CAAC,iBAAiB,EAAE,eAAe,EAAE,qBAAqB,CAAC;KAC1E,CAAC;IAEF,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,UAAU,GAAG,GAAG,CAAC,EAAE,CAAC;QAC1C,KAAK,CAAC,QAAQ,GAAG,CAAC,qDAAqD,CAAC,CAAC;IAC3E,CAAC;IAED,iBAAiB;IACjB,OAAO,CAAC,aAAa,CAAC,eAAe,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;IAE5E,OAAO,OAAO;SACX,WAAW,CAAC,OAAO,CAAC;SACpB,QAAQ,CAAC,IAAI,CAAC;SACd,SAAS,CAAC,KAAK,CAAC;SAChB,YAAY,EAAE,CAAC;AACpB,CAAC;AAED,+EAA+E;AAC/E,UAAU;AACV,+EAA+E;AAE/E;;GAEG;AACH,SAAS,YAAY,CAAC,IAAkB;IACtC,MAAM,KAAK,GAAa,EAAE,CAAC;IAE3B,yBAAyB;IACzB,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC/B,KAAK,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IACvC,CAAC;IAED,iBAAiB;IACjB,IAAI,IAAI,CAAC,UAAU;QAAE,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC1C,IAAI,IAAI,CAAC,OAAO;QAAE,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACtC,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IACvB,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAEtB,2BAA2B;IAC3B,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE;QACjD,IAAI,CAAC,CAAC,IAAI;YAAE,OAAO,GAAG,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC;QAC1C,OAAO,CAAC,CAAC,IAAI,CAAC;IAChB,CAAC,CAAC,CAAC;IACH,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC/B,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACrB,CAAC;IAED,IAAI,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC;IAEzD,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;QACpB,OAAO,IAAI,KAAK,IAAI,CAAC,UAAU,EAAE,CAAC;IACpC,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAED;;GAEG;AACH,SAAS,kBAAkB,CAAC,KAAqB;IAC/C,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACvB,OAAO;YACL,MAAM,EAAE,SAAS;YACjB,aAAa,EAAE,SAAS;YACxB,OAAO,EAAE,SAAS;SACnB,CAAC;IACJ,CAAC;IAED,oBAAoB;IACpB,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IACrC,MAAM,YAAY,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;IACjE,MAAM,YAAY,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;IAC/D,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;IAElE,IAAI,MAAM,GAAG,OAAO,CAAC;IACrB,IAAI,YAAY,IAAI,CAAC,YAAY;QAAE,MAAM,GAAG,WAAW,CAAC;SACnD,IAAI,YAAY,IAAI,CAAC,YAAY;QAAE,MAAM,GAAG,YAAY,CAAC;SACzD,IAAI,aAAa;QAAE,MAAM,GAAG,YAAY,CAAC;IAE9C,sCAAsC;IACtC,MAAM,WAAW,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IACjE,IAAI,aAAa,GAAG,SAAS,CAAC;IAC9B,IAAI,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,QAAQ,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC;QACnD,aAAa,GAAG,mBAAmB,CAAC;IACtC,CAAC;SAAM,IAAI,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,QAAQ,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC;QAC3D,aAAa,GAAG,sBAAsB,CAAC;IACzC,CAAC;SAAM,IAAI,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC;QACtC,aAAa,GAAG,iBAAiB,CAAC;IACpC,CAAC;IAED,0CAA0C;IAC1C,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IACrC,IAAI,OAAO,GAAG,gBAAgB,CAAC;IAC/B,IAAI,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC;QACtC,OAAO,GAAG,iBAAiB,CAAC;IAC9B,CAAC;SAAM,IAAI,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC;QAC7C,OAAO,GAAG,iBAAiB,CAAC;IAC9B,CAAC;IAED,OAAO,EAAE,MAAM,EAAE,aAAa,EAAE,OAAO,EAAE,CAAC;AAC5C,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,MAAM,qBAAqB,GAAG;IACnC,IAAI,EAAE,eAAe;IACrB,WAAW,EAAE,iLAAiL;IAC9L,WAAW,EAAE;QACX,IAAI,EAAE,QAAiB;QACvB,UAAU,EAAE;YACV,MAAM,EAAE;gBACN,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,CAAC,cAAc,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,YAAY,CAAC;gBACvF,WAAW,EAAE,oCAAoC;aAClD;YACD,WAAW,EAAE;gBACX,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,8DAA8D;aAC5E;YACD,KAAK,EAAE;gBACL,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,uCAAuC;aACrD;YACD,KAAK,EAAE;gBACL,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,mCAAmC;aACjD;SACF;QACD,QAAQ,EAAE,CAAC,QAAQ,EAAE,aAAa,CAAC;KACpC;CACF,CAAC"}

package/dist/tools/surgical/test-template.d.ts

@@ -0,0 +1,70 @@
+/**
+ * drift_test_template - Generate Test Scaffolding
+ *
+ * Layer: Surgical
+ * Token Budget: 800 target, 1500 max
+ * Cache TTL: 5 minutes
+ * Invalidation Keys: test-topology, file:{targetFile}
+ *
+ * Generates test scaffolding based on existing test patterns.
+ * Solves: Tests are the most convention-heavy code. Every codebase is different.
+ */
+import type { CallGraphStore } from 'driftdetect-core';
+export interface TestTemplateArgs {
+    /** File being tested */
+    targetFile: string;
+    /** Specific function to test (optional) */
+    function?: string;
+    /** Test type */
+    type?: 'unit' | 'integration' | 'e2e';
+}
+export interface TestConventions {
+    framework: string;
+    style: string;
+    mockStyle: string;
+    assertionStyle: string;
+    filePattern: string;
+}
+export interface ExampleTest {
+    file: string;
+    preview: string;
+}
+export interface TestTemplateData {
+    testFile: string;
+    template: string;
+    conventions: TestConventions;
+    exampleTest?: ExampleTest | undefined;
+}
+export declare function handleTestTemplate(store: CallGraphStore, args: TestTemplateArgs, projectRoot: string): Promise<{
+    content: Array<{
+        type: string;
+        text: string;
+    }>;
+}>;
+/**
+ * Tool definition for MCP registration
+ */
+export declare const testTemplateToolDefinition: {
+    name: string;
+    description: string;
+    inputSchema: {
+        type: "object";
+        properties: {
+            targetFile: {
+                type: string;
+                description: string;
+            };
+            function: {
+                type: string;
+                description: string;
+            };
+            type: {
+                type: string;
+                enum: string[];
+                description: string;
+            };
+        };
+        required: string[];
+    };
+};
+//# sourceMappingURL=test-template.d.ts.map

package/dist/tools/surgical/test-template.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-template.d.ts","sourceRoot":"","sources":["../../../src/tools/surgical/test-template.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAEH,OAAO,KAAK,EAAE,cAAc,EAAgB,MAAM,kBAAkB,CAAC;AASrE,MAAM,WAAW,gBAAgB;IAC/B,wBAAwB;IACxB,UAAU,EAAE,MAAM,CAAC;IACnB,2CAA2C;IAC3C,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,gBAAgB;IAChB,IAAI,CAAC,EAAE,MAAM,GAAG,aAAa,GAAG,KAAK,CAAC;CACvC;AAED,MAAM,WAAW,eAAe;IAC9B,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,MAAM,CAAC;IAClB,cAAc,EAAE,MAAM,CAAC;IACvB,WAAW,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,gBAAgB;IAC/B,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,WAAW,EAAE,eAAe,CAAC;IAC7B,WAAW,CAAC,EAAE,WAAW,GAAG,SAAS,CAAC;CACvC;AAMD,wBAAsB,kBAAkB,CACtC,KAAK,EAAE,cAAc,EACrB,IAAI,EAAE,gBAAgB,EACtB,WAAW,EAAE,MAAM,GAClB,OAAO,CAAC;IAAE,OAAO,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,CAAA;CAAE,CAAC,CAsF7D;AAiOD;;GAEG;AACH,eAAO,MAAM,0BAA0B;;;;;;;;;;;;;;;;;;;;;;CAsBtC,CAAC"}

package/dist/tools/surgical/test-template.js

@@ -0,0 +1,298 @@
+/**
+ * drift_test_template - Generate Test Scaffolding
+ *
+ * Layer: Surgical
+ * Token Budget: 800 target, 1500 max
+ * Cache TTL: 5 minutes
+ * Invalidation Keys: test-topology, file:{targetFile}
+ *
+ * Generates test scaffolding based on existing test patterns.
+ * Solves: Tests are the most convention-heavy code. Every codebase is different.
+ */
+import { createResponseBuilder, Errors, metrics } from '../../infrastructure/index.js';
+import * as fs from 'node:fs/promises';
+import * as path from 'node:path';
+// ============================================================================
+// Handler
+// ============================================================================
+export async function handleTestTemplate(store, args, projectRoot) {
+    const startTime = Date.now();
+    const builder = createResponseBuilder();
+    // Validate input
+    if (!args.targetFile || args.targetFile.trim() === '') {
+        throw Errors.missingParameter('targetFile');
+    }
+    const targetFile = args.targetFile.trim();
+    const targetFunction = args.function?.trim();
+    const testType = args.type ?? 'unit';
+    // Load call graph
+    await store.initialize();
+    const graph = store.getGraph();
+    if (!graph) {
+        throw Errors.custom('CALLGRAPH_NOT_BUILT', 'Call graph has not been built. Run "drift callgraph build" first.', ['drift_status']);
+    }
+    // Find the target function if specified
+    let targetFunc;
+    if (targetFunction) {
+        for (const [, func] of graph.functions) {
+            if (func.file.endsWith(targetFile) && func.name === targetFunction) {
+                targetFunc = func;
+                break;
+            }
+        }
+    }
+    // Detect test conventions from existing tests
+    const detectedConventions = await detectTestConventions(projectRoot, targetFile);
+    // Find example test in same directory
+    const exampleTest = await findExampleTest(projectRoot, targetFile, detectedConventions);
+    // Generate test file path
+    const testFile = generateTestFilePath(targetFile, detectedConventions);
+    // Generate template
+    const template = generateTestTemplate(targetFile, targetFunc, detectedConventions, testType);
+    const data = {
+        testFile,
+        template,
+        conventions: detectedConventions,
+        exampleTest,
+    };
+    // Build summary
+    const funcName = targetFunc?.name ?? path.basename(targetFile, path.extname(targetFile));
+    const summary = `Generated ${testType} test template for "${funcName}" using ${detectedConventions.framework}/${detectedConventions.style}`;
+    // Build hints
+    const hints = {
+        nextActions: [
+            `Create ${testFile} with the template`,
+            'Fill in test cases based on function behavior',
+            exampleTest ? `Reference ${exampleTest.file} for patterns` : 'Add assertions matching codebase style',
+        ],
+        relatedTools: ['drift_signature', 'drift_callers', 'drift_similar'],
+    };
+    if (!exampleTest) {
+        hints.warnings = ['No existing tests found in directory - conventions inferred from project'];
+    }
+    // Record metrics
+    metrics.recordRequest('drift_test_template', Date.now() - startTime, true, false);
+    return builder
+        .withSummary(summary)
+        .withData(data)
+        .withHints(hints)
+        .buildContent();
+}
+// ============================================================================
+// Helpers
+// ============================================================================
+/**
+ * Detect test conventions from existing tests
+ */
+async function detectTestConventions(projectRoot, targetFile) {
+    const conventions = {
+        framework: 'vitest',
+        style: 'describe/it',
+        mockStyle: 'vi.mock',
+        assertionStyle: 'expect',
+        filePattern: '*.test.ts',
+    };
+    // Check package.json for test framework
+    try {
+        const pkgPath = path.join(projectRoot, 'package.json');
+        const pkgContent = await fs.readFile(pkgPath, 'utf-8');
+        const pkg = JSON.parse(pkgContent);
+        const deps = { ...pkg.dependencies, ...pkg.devDependencies };
+        if (deps['vitest']) {
+            conventions.framework = 'vitest';
+            conventions.mockStyle = 'vi.mock';
+        }
+        else if (deps['jest']) {
+            conventions.framework = 'jest';
+            conventions.mockStyle = 'jest.mock';
+        }
+        else if (deps['mocha']) {
+            conventions.framework = 'mocha';
+            conventions.mockStyle = 'sinon';
+            conventions.assertionStyle = 'chai';
+        }
+    }
+    catch {
+        // Use defaults
+    }
+    // Check for test file patterns in the target directory
+    const targetDir = path.dirname(targetFile);
+    const fullTargetDir = path.join(projectRoot, targetDir);
+    try {
+        const files = await fs.readdir(fullTargetDir);
+        // Check for __tests__ directory
+        if (files.includes('__tests__')) {
+            conventions.filePattern = '__tests__/*.test.ts';
+        }
+        // Check for .spec.ts files
+        if (files.some(f => f.endsWith('.spec.ts'))) {
+            conventions.filePattern = '*.spec.ts';
+        }
+    }
+    catch {
+        // Directory doesn't exist or can't be read
+    }
+    return conventions;
+}
+/**
+ * Find an example test in the same directory
+ */
+async function findExampleTest(projectRoot, targetFile, _conventions) {
+    const targetDir = path.dirname(targetFile);
+    // Possible test locations
+    const testDirs = [
+        path.join(projectRoot, targetDir, '__tests__'),
+        path.join(projectRoot, targetDir),
+        path.join(projectRoot, targetDir.replace('/src/', '/test/')),
+    ];
+    for (const testDir of testDirs) {
+        try {
+            const files = await fs.readdir(testDir);
+            const testFiles = files.filter(f => f.endsWith('.test.ts') ||
+                f.endsWith('.spec.ts') ||
+                f.endsWith('.test.js') ||
+                f.endsWith('.spec.js'));
+            if (testFiles.length > 0) {
+                const testFile = testFiles[0];
+                const fullPath = path.join(testDir, testFile);
+                const content = await fs.readFile(fullPath, 'utf-8');
+                // Get first 20 lines as preview
+                const preview = content.split('\n').slice(0, 20).join('\n');
+                return {
+                    file: path.relative(projectRoot, fullPath),
+                    preview,
+                };
+            }
+        }
+        catch {
+            // Directory doesn't exist
+        }
+    }
+    return undefined;
+}
+/**
+ * Generate test file path based on conventions
+ */
+function generateTestFilePath(targetFile, conventions) {
+    const dir = path.dirname(targetFile);
+    const base = path.basename(targetFile, path.extname(targetFile));
+    const ext = path.extname(targetFile);
+    if (conventions.filePattern.includes('__tests__')) {
+        return path.join(dir, '__tests__', `${base}.test${ext}`);
+    }
+    if (conventions.filePattern.includes('.spec.')) {
+        return path.join(dir, `${base}.spec${ext}`);
+    }
+    return path.join(dir, `${base}.test${ext}`);
+}
+/**
+ * Generate test template
+ */
+function generateTestTemplate(targetFile, targetFunc, conventions, _testType) {
+    const moduleName = path.basename(targetFile, path.extname(targetFile));
+    const funcName = targetFunc?.name ?? moduleName;
+    // Import statement based on framework
+    let imports;
+    if (conventions.framework === 'vitest') {
+        imports = `import { describe, it, expect, beforeEach, vi } from 'vitest';`;
+    }
+    else if (conventions.framework === 'jest') {
+        imports = `// Jest globals are available automatically`;
+    }
+    else {
+        imports = `import { describe, it } from 'mocha';\nimport { expect } from 'chai';`;
+    }
+    // Import the module under test
+    const relativePath = './' + moduleName;
+    const moduleImport = targetFunc
+        ? `import { ${funcName} } from '${relativePath}';`
+        : `import * as ${moduleName} from '${relativePath}';`;
+    // Mock setup based on framework
+    let mockSetup = '';
+    if (conventions.framework === 'vitest') {
+        mockSetup = `
+beforeEach(() => {
+  vi.clearAllMocks();
+});`;
+    }
+    else if (conventions.framework === 'jest') {
+        mockSetup = `
+beforeEach(() => {
+  jest.clearAllMocks();
+});`;
+    }
+    // Generate test cases
+    let testCases;
+    if (targetFunc) {
+        const params = targetFunc.parameters.map(p => p.name).join(', ');
+        const hasParams = targetFunc.parameters.length > 0;
+        testCases = `
+  it('should ${funcName} successfully', async () => {
+    // Arrange
+    ${hasParams ? `const ${targetFunc.parameters[0]?.name ?? 'input'} = {}; // TODO: Add test data` : '// No input needed'}
+    
+    // Act
+    const result = await ${funcName}(${params});
+    
+    // Assert
+    expect(result).toBeDefined();
+    // TODO: Add specific assertions
+  });
+
+  it('should handle errors gracefully', async () => {
+    // Arrange
+    // TODO: Set up error condition
+    
+    // Act & Assert
+    // TODO: Add error handling test
+  });`;
+    }
+    else {
+        testCases = `
+  it('should work correctly', () => {
+    // Arrange
+    // TODO: Set up test data
+    
+    // Act
+    // TODO: Call function under test
+    
+    // Assert
+    // TODO: Add assertions
+  });`;
+    }
+    // Combine template
+    return `${imports}
+${moduleImport}
+
+describe('${funcName}', () => {${mockSetup}
+${testCases}
+});
+`;
+}
+/**
+ * Tool definition for MCP registration
+ */
+export const testTemplateToolDefinition = {
+    name: 'drift_test_template',
+    description: 'Generate test scaffolding based on existing test patterns. Returns ready-to-use template matching codebase conventions (framework, mocking style, file location).',
+    inputSchema: {
+        type: 'object',
+        properties: {
+            targetFile: {
+                type: 'string',
+                description: 'File being tested (relative path)',
+            },
+            function: {
+                type: 'string',
+                description: 'Optional: specific function to test',
+            },
+            type: {
+                type: 'string',
+                enum: ['unit', 'integration', 'e2e'],
+                description: 'Test type (default: unit)',
+            },
+        },
+        required: ['targetFile'],
+    },
+};
+//# sourceMappingURL=test-template.js.map