codeguard-testgen 1.0.14 → 1.0.16

package/dist/ai.d.ts

@@ -0,0 +1,8 @@
+import { Message, Tool, ToolCall, AIResponse } from './types';
+export declare function callAI(messages: Message[], tools: Tool[], provider?: string): Promise<AIResponse>;
+export declare function addToolResultsToMessages(provider: string, messages: Message[], toolCalls: ToolCall[], toolResults: Array<{
+    id: string;
+    name: string;
+    result: any;
+}>): void;
+//# sourceMappingURL=ai.d.ts.map

package/dist/ai.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai.d.ts","sourceRoot":"","sources":["../src/ai.ts"],"names":[],"mappings":"AACA,OAAO,EAAc,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AAuT1E,wBAAsB,MAAM,CAAC,QAAQ,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,QAAQ,GAAE,MAA4B,GAAG,OAAO,CAAC,UAAU,CAAC,CAW5H;AAOD,wBAAgB,wBAAwB,CACtC,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,OAAO,EAAE,EACnB,SAAS,EAAE,QAAQ,EAAE,EACrB,WAAW,EAAE,KAAK,CAAC;IAAE,EAAE,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,GAAG,CAAA;CAAE,CAAC,GAC5D,IAAI,CA+DN"}

package/dist/ai.js

@@ -0,0 +1,332 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.callAI = callAI;
+exports.addToolResultsToMessages = addToolResultsToMessages;
+const globals_1 = require("./globals");
+const DEFAULT_RETRY_OPTIONS = {
+    maxRetries: 3,
+    initialDelayMs: 1000,
+    maxDelayMs: 30000,
+    timeoutMs: 120000,
+    retryableStatuses: [429, 500, 502, 503, 504],
+};
+async function fetchWithRetry(url, options, retryOptions) {
+    const opts = { ...DEFAULT_RETRY_OPTIONS, ...retryOptions };
+    let lastError = null;
+    for (let attempt = 0; attempt <= opts.maxRetries; attempt++) {
+        const controller = new AbortController();
+        const timeoutId = setTimeout(() => controller.abort(), opts.timeoutMs);
+        try {
+            const response = await fetch(url, { ...options, signal: controller.signal });
+            clearTimeout(timeoutId);
+            // Non-retryable error: return immediately
+            if (!opts.retryableStatuses.includes(response.status)) {
+                return response;
+            }
+            // Retryable HTTP status
+            if (attempt === opts.maxRetries) {
+                return response; // Return on last attempt so caller can read the body
+            }
+            // Respect Retry-After header for 429s
+            let delayMs = opts.initialDelayMs * Math.pow(2, attempt);
+            if (response.status === 429) {
+                const retryAfter = response.headers.get('Retry-After');
+                if (retryAfter) {
+                    const retryAfterSec = parseInt(retryAfter, 10);
+                    if (!isNaN(retryAfterSec)) {
+                        delayMs = retryAfterSec * 1000;
+                    }
+                }
+            }
+            // Add jitter (±20%)
+            delayMs = delayMs * (0.8 + Math.random() * 0.4);
+            delayMs = Math.min(delayMs, opts.maxDelayMs);
+            const delaySec = (delayMs / 1000).toFixed(1);
+            if (globals_1.g.globalSpinner && globals_1.g.globalSpinner.isSpinning) {
+                globals_1.g.globalSpinner.text = `⏳ Retrying API call (attempt ${attempt + 1}/${opts.maxRetries}, waiting ${delaySec}s)...`;
+            }
+            await new Promise(resolve => setTimeout(resolve, delayMs));
+            lastError = new Error(`HTTP ${response.status}`);
+        }
+        catch (error) {
+            clearTimeout(timeoutId);
+            if (error.name === 'AbortError') {
+                lastError = new Error(`Request timed out after ${opts.timeoutMs / 1000}s`);
+            }
+            else {
+                lastError = error;
+            }
+            if (attempt === opts.maxRetries) {
+                break;
+            }
+            // Network errors are retryable
+            const delayMs = Math.min(opts.initialDelayMs * Math.pow(2, attempt) * (0.8 + Math.random() * 0.4), opts.maxDelayMs);
+            const delaySec = (delayMs / 1000).toFixed(1);
+            if (globals_1.g.globalSpinner && globals_1.g.globalSpinner.isSpinning) {
+                globals_1.g.globalSpinner.text = `⏳ Retrying API call (attempt ${attempt + 1}/${opts.maxRetries}, waiting ${delaySec}s)...`;
+            }
+            await new Promise(resolve => setTimeout(resolve, delayMs));
+        }
+    }
+    throw lastError || new Error('Request failed after all retries');
+}
+// ---------------------------------------------------------------------------
+// Provider configuration
+// ---------------------------------------------------------------------------
+function getAIProviders() {
+    return {
+        openai: {
+            url: 'https://api.openai.com/v1/chat/completions',
+            model: globals_1.g.CONFIG.models.openai,
+            headers: (apiKey) => ({
+                'Content-Type': 'application/json',
+                'Authorization': `Bearer ${apiKey}`
+            })
+        },
+        gemini: {
+            url: (apiKey) => `https://generativelanguage.googleapis.com/v1beta/models/${globals_1.g.CONFIG.models.gemini}:generateContent?key=${apiKey}`,
+            model: globals_1.g.CONFIG.models.gemini,
+            headers: () => ({
+                'Content-Type': 'application/json'
+            })
+        },
+        claude: {
+            url: 'https://api.anthropic.com/v1/messages',
+            model: globals_1.g.CONFIG.models.claude,
+            headers: (apiKey) => ({
+                'Content-Type': 'application/json',
+                'x-api-key': apiKey || '',
+                'anthropic-version': '2023-06-01'
+            })
+        }
+    };
+}
+// ---------------------------------------------------------------------------
+// Provider call implementations
+// ---------------------------------------------------------------------------
+async function callOpenAI(messages, tools) {
+    const apiKey = globals_1.g.CONFIG.apiKeys.openai;
+    if (!apiKey)
+        throw new Error('OpenAI API key not configured in codeguard.json');
+    const AI_PROVIDERS = getAIProviders();
+    const response = await fetchWithRetry(AI_PROVIDERS.openai.url, {
+        method: 'POST',
+        headers: AI_PROVIDERS.openai.headers(apiKey),
+        body: JSON.stringify({
+            model: AI_PROVIDERS.openai.model,
+            messages,
+            tools: tools.map(t => ({
+                type: 'function',
+                function: {
+                    name: t.name,
+                    description: t.description,
+                    parameters: t.input_schema
+                }
+            })),
+            tool_choice: 'auto'
+        })
+    });
+    const data = await response.json();
+    if (data.error) {
+        throw new Error(`OpenAI API error: ${data.error.message || JSON.stringify(data.error)}`);
+    }
+    if (!data.choices || data.choices.length === 0) {
+        throw new Error(`OpenAI API returned no choices. Response: ${JSON.stringify(data)}`);
+    }
+    const message = data.choices[0].message;
+    return {
+        content: message.content,
+        toolCalls: message.tool_calls?.map((tc) => ({
+            id: tc.id,
+            name: tc.function.name,
+            input: JSON.parse(tc.function.arguments)
+        }))
+    };
+}
+async function callGemini(messages, tools) {
+    const apiKey = globals_1.g.CONFIG.apiKeys.gemini;
+    if (!apiKey)
+        throw new Error('Gemini API key not configured in codeguard.json');
+    const contents = messages.map(m => {
+        const role = m.role === 'assistant' || m.role === 'model' ? 'model' : 'user';
+        if (m.functionCall) {
+            return {
+                role: 'model',
+                parts: [{
+                        functionCall: {
+                            name: m.functionCall.name,
+                            args: m.functionCall.args
+                        }
+                    }]
+            };
+        }
+        if (m.functionResponse) {
+            return {
+                role: 'user',
+                parts: [{
+                        functionResponse: {
+                            name: m.functionResponse.name,
+                            response: m.functionResponse.response
+                        }
+                    }]
+            };
+        }
+        return {
+            role,
+            parts: [{ text: m.content || '' }]
+        };
+    });
+    const AI_PROVIDERS = getAIProviders();
+    const url = typeof AI_PROVIDERS.gemini.url === 'function' ? AI_PROVIDERS.gemini.url(apiKey) : AI_PROVIDERS.gemini.url;
+    const response = await fetchWithRetry(url, {
+        method: 'POST',
+        headers: AI_PROVIDERS.gemini.headers(),
+        body: JSON.stringify({
+            contents,
+            tools: [{
+                    functionDeclarations: tools.map(t => ({
+                        name: t.name,
+                        description: t.description,
+                        parameters: t.input_schema
+                    }))
+                }]
+        })
+    });
+    const data = await response.json();
+    if (data.error) {
+        const errorMsg = data.error.message || JSON.stringify(data.error);
+        if (errorMsg.includes('quota') || errorMsg.includes('rate limit')) {
+            throw new Error(`Gemini API quota exceeded. Please wait or use a different provider (Claude or OpenAI). Error: ${errorMsg}`);
+        }
+        throw new Error(`Gemini API error: ${errorMsg}`);
+    }
+    if (!data.candidates || data.candidates.length === 0) {
+        throw new Error(`Gemini API returned no candidates. Response: ${JSON.stringify(data)}`);
+    }
+    const candidate = data.candidates[0];
+    const parts = candidate.content.parts;
+    const functionCalls = parts.filter((p) => p.functionCall);
+    if (functionCalls.length > 0) {
+        return {
+            toolCalls: functionCalls.map((fc) => ({
+                id: `gemini-${fc.functionCall.name}-${Date.now()}`,
+                name: fc.functionCall.name,
+                input: fc.functionCall.args
+            }))
+        };
+    }
+    const textPart = parts.find((p) => p.text);
+    return { content: textPart?.text || '' };
+}
+async function callClaude(messages, tools) {
+    const apiKey = globals_1.g.CONFIG.apiKeys.claude;
+    if (!apiKey)
+        throw new Error('Claude API key not configured in codeguard.json');
+    const AI_PROVIDERS = getAIProviders();
+    const response = await fetchWithRetry(AI_PROVIDERS.claude.url, {
+        method: 'POST',
+        headers: AI_PROVIDERS.claude.headers(apiKey),
+        body: JSON.stringify({
+            model: AI_PROVIDERS.claude.model,
+            max_tokens: 4096,
+            messages,
+            tools
+        })
+    });
+    const data = await response.json();
+    if (data.error) {
+        throw new Error(`Claude API error: ${data.error.message || JSON.stringify(data.error)}`);
+    }
+    if (!data.content || !Array.isArray(data.content)) {
+        throw new Error(`Claude API returned invalid content. Response: ${JSON.stringify(data)}`);
+    }
+    const textContent = data.content.find((c) => c.type === 'text');
+    const toolUse = data.content.filter((c) => c.type === 'tool_use');
+    return {
+        content: textContent?.text,
+        toolCalls: toolUse.map((tu) => ({
+            id: tu.id,
+            name: tu.name,
+            input: tu.input
+        })),
+        stopReason: data.stop_reason
+    };
+}
+async function callAI(messages, tools, provider = globals_1.g.CONFIG.aiProvider) {
+    switch (provider) {
+        case 'openai':
+            return await callOpenAI(messages, tools);
+        case 'gemini':
+            return await callGemini(messages, tools);
+        case 'claude':
+            return await callClaude(messages, tools);
+        default:
+            throw new Error(`Unknown AI provider: ${provider}`);
+    }
+}
+// ---------------------------------------------------------------------------
+// Helper: append tool call + result messages in the correct provider format.
+// Extracted from the 4 identical blocks scattered across conversation loops.
+// ---------------------------------------------------------------------------
+function addToolResultsToMessages(provider, messages, toolCalls, toolResults) {
+    if (provider === 'claude') {
+        messages.push({
+            role: 'assistant',
+            content: toolCalls.map(tc => ({
+                type: 'tool_use',
+                id: tc.id,
+                name: tc.name,
+                input: tc.input
+            }))
+        });
+        messages.push({
+            role: 'user',
+            content: toolResults.map(tr => ({
+                type: 'tool_result',
+                tool_use_id: tr.id,
+                content: JSON.stringify(tr.result)
+            }))
+        });
+    }
+    else if (provider === 'openai') {
+        messages.push({
+            role: 'assistant',
+            tool_calls: toolCalls.map(tc => ({
+                id: tc.id,
+                type: 'function',
+                function: {
+                    name: tc.name,
+                    arguments: JSON.stringify(tc.input)
+                }
+            }))
+        });
+        for (const tr of toolResults) {
+            messages.push({
+                role: 'tool',
+                tool_call_id: tr.id,
+                content: JSON.stringify(tr.result)
+            });
+        }
+    }
+    else {
+        // Gemini
+        for (const toolCall of toolCalls) {
+            messages.push({
+                role: 'model',
+                functionCall: {
+                    name: toolCall.name,
+                    args: toolCall.input
+                }
+            });
+            const result = toolResults.find(tr => tr.name === toolCall.name);
+            messages.push({
+                role: 'user',
+                functionResponse: {
+                    name: toolCall.name,
+                    response: result?.result
+                }
+            });
+        }
+    }
+}
+//# sourceMappingURL=ai.js.map

package/dist/ai.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai.js","sourceRoot":"","sources":["../src/ai.ts"],"names":[],"mappings":";;AAwTA,wBAWC;AAOD,4DAoEC;AA9YD,uCAA8B;AAe9B,MAAM,qBAAqB,GAAiB;IAC1C,UAAU,EAAE,CAAC;IACb,cAAc,EAAE,IAAI;IACpB,UAAU,EAAE,KAAK;IACjB,SAAS,EAAE,MAAM;IACjB,iBAAiB,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC;CAC7C,CAAC;AAEF,KAAK,UAAU,cAAc,CAC3B,GAAW,EACX,OAAoB,EACpB,YAAoC;IAEpC,MAAM,IAAI,GAAiB,EAAE,GAAG,qBAAqB,EAAE,GAAG,YAAY,EAAE,CAAC;IACzE,IAAI,SAAS,GAAiB,IAAI,CAAC;IAEnC,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,IAAI,CAAC,UAAU,EAAE,OAAO,EAAE,EAAE,CAAC;QAC5D,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;QACzC,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;QAEvE,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE,EAAE,GAAG,OAAO,EAAE,MAAM,EAAE,UAAU,CAAC,MAAM,EAAE,CAAC,CAAC;YAC7E,YAAY,CAAC,SAAS,CAAC,CAAC;YAExB,0CAA0C;YAC1C,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;gBACtD,OAAO,QAAQ,CAAC;YAClB,CAAC;YAED,wBAAwB;YACxB,IAAI,OAAO,KAAK,IAAI,CAAC,UAAU,EAAE,CAAC;gBAChC,OAAO,QAAQ,CAAC,CAAC,qDAAqD;YACxE,CAAC;YAED,sCAAsC;YACtC,IAAI,OAAO,GAAG,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;YACzD,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBAC5B,MAAM,UAAU,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;gBACvD,IAAI,UAAU,EAAE,CAAC;oBACf,MAAM,aAAa,GAAG,QAAQ,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;oBAC/C,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,EAAE,CAAC;wBAC1B,OAAO,GAAG,aAAa,GAAG,IAAI,CAAC;oBACjC,CAAC;gBACH,CAAC;YACH,CAAC;YAED,oBAAoB;YACpB,OAAO,GAAG,OAAO,GAAG,CAAC,GAAG,GAAG,IAAI,CAAC,MAAM,EAAE,GAAG,GAAG,CAAC,CAAC;YAChD,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;YAE7C,MAAM,QAAQ,GAAG,CAAC,OAAO,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;YAC7C,IAAI,WAAC,CAAC,aAAa,IAAI,WAAC,CAAC,aAAa,CAAC,UAAU,EAAE,CAAC;gBAClD,WAAC,CAAC,aAAa,CAAC,IAAI,GAAG,gCAAgC,OAAO,GAAG,CAAC,IAAI,IAAI,CAAC,UAAU,aAAa,QAAQ,OAAO,CAAC;YACpH,CAAC;YAED,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;YAC3D,SAAS,GAAG,IAAI,KAAK,CAAC,QAAQ,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QAEnD,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,YAAY,CAAC,SAAS,CAAC,CAAC;YAExB,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBAChC,SAAS,GAAG,IAAI,KAAK,CAAC,2BAA2B,IAAI,CAAC,SAAS,GAAG,IAAI,GAAG,CAAC,CAAC;YAC7E,CAAC;iBAAM,CAAC;gBACN,SAAS,GAAG,KAAK,CAAC;YACpB,CAAC;YAED,IAAI,OAAO,KAAK,IAAI,CAAC,UAAU,EAAE,CAAC;gBAChC,MAAM;YACR,CAAC;YAED,+BAA+B;YAC/B,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CACtB,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,GAAG,CAAC,GAAG,GAAG,IAAI,CAAC,MAAM,EAAE,GAAG,GAAG,CAAC,EACxE,IAAI,CAAC,UAAU,CAChB,CAAC;YACF,MAAM,QAAQ,GAAG,CAAC,OAAO,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;YAC7C,IAAI,WAAC,CAAC,aAAa,IAAI,WAAC,CAAC,aAAa,CAAC,UAAU,EAAE,CAAC;gBAClD,WAAC,CAAC,aAAa,CAAC,IAAI,GAAG,gCAAgC,OAAO,GAAG,CAAC,IAAI,IAAI,CAAC,UAAU,aAAa,QAAQ,OAAO,CAAC;YACpH,CAAC;YACD,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;QAC7D,CAAC;IACH,CAAC;IAED,MAAM,SAAS,IAAI,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;AACnE,CAAC;AAED,8EAA8E;AAC9E,yBAAyB;AACzB,8EAA8E;AAE9E,SAAS,cAAc;IACrB,OAAO;QACL,MAAM,EAAE;YACN,GAAG,EAAE,4CAA4C;YACjD,KAAK,EAAE,WAAC,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM;YAC7B,OAAO,EAAE,CAAC,MAAe,EAAE,EAAE,CAAC,CAAC;gBAC7B,cAAc,EAAE,kBAAkB;gBAClC,eAAe,EAAE,UAAU,MAAM,EAAE;aACpC,CAAC;SACH;QACD,MAAM,EAAE;YACN,GAAG,EAAE,CAAC,MAAc,EAAE,EAAE,CAAC,2DAA2D,WAAC,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,wBAAwB,MAAM,EAAE;YAC1I,KAAK,EAAE,WAAC,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM;YAC7B,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC;gBACd,cAAc,EAAE,kBAAkB;aACnC,CAAC;SACH;QACD,MAAM,EAAE;YACN,GAAG,EAAE,uCAAuC;YAC5C,KAAK,EAAE,WAAC,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM;YAC7B,OAAO,EAAE,CAAC,MAAe,EAAE,EAAE,CAAC,CAAC;gBAC7B,cAAc,EAAE,kBAAkB;gBAClC,WAAW,EAAE,MAAM,IAAI,EAAE;gBACzB,mBAAmB,EAAE,YAAY;aAClC,CAAC;SACH;KACF,CAAC;AACJ,CAAC;AAED,8EAA8E;AAC9E,gCAAgC;AAChC,8EAA8E;AAE9E,KAAK,UAAU,UAAU,CAAC,QAAmB,EAAE,KAAa;IAC1D,MAAM,MAAM,GAAG,WAAC,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;IACvC,IAAI,CAAC,MAAM;QAAE,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;IAEhF,MAAM,YAAY,GAAG,cAAc,EAAE,CAAC;IACtC,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,YAAY,CAAC,MAAM,CAAC,GAAa,EAAE;QACvE,MAAM,EAAE,MAAM;QACd,OAAO,EAAE,YAAY,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;QAC5C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;YACnB,KAAK,EAAE,YAAY,CAAC,MAAM,CAAC,KAAK;YAChC,QAAQ;YACR,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBACrB,IAAI,EAAE,UAAU;gBAChB,QAAQ,EAAE;oBACR,IAAI,EAAE,CAAC,CAAC,IAAI;oBACZ,WAAW,EAAE,CAAC,CAAC,WAAW;oBAC1B,UAAU,EAAE,CAAC,CAAC,YAAY;iBAC3B;aACF,CAAC,CAAC;YACH,WAAW,EAAE,MAAM;SACpB,CAAC;KACH,CAAC,CAAC;IAEH,MAAM,IAAI,GAAQ,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;IAExC,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CAAC,qBAAqB,IAAI,CAAC,KAAK,CAAC,OAAO,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IAC3F,CAAC;IAED,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC/C,MAAM,IAAI,KAAK,CAAC,6CAA6C,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACvF,CAAC;IAED,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;IAExC,OAAO;QACL,OAAO,EAAE,OAAO,CAAC,OAAO;QACxB,SAAS,EAAE,OAAO,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC,EAAO,EAAE,EAAE,CAAC,CAAC;YAC/C,EAAE,EAAE,EAAE,CAAC,EAAE;YACT,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI;YACtB,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC;SACzC,CAAC,CAAC;KACJ,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,UAAU,CAAC,QAAmB,EAAE,KAAa;IAC1D,MAAM,MAAM,GAAG,WAAC,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;IACvC,IAAI,CAAC,MAAM;QAAE,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;IAEhF,MAAM,QAAQ,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE;QAChC,MAAM,IAAI,GAAG,CAAC,CAAC,IAAI,KAAK,WAAW,IAAI,CAAC,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC;QAE7E,IAAI,CAAC,CAAC,YAAY,EAAE,CAAC;YACnB,OAAO;gBACL,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,CAAC;wBACN,YAAY,EAAE;4BACZ,IAAI,EAAE,CAAC,CAAC,YAAY,CAAC,IAAI;4BACzB,IAAI,EAAE,CAAC,CAAC,YAAY,CAAC,IAAI;yBAC1B;qBACF,CAAC;aACH,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,CAAC,gBAAgB,EAAE,CAAC;YACvB,OAAO;gBACL,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE,CAAC;wBACN,gBAAgB,EAAE;4BAChB,IAAI,EAAE,CAAC,CAAC,gBAAgB,CAAC,IAAI;4BAC7B,QAAQ,EAAE,CAAC,CAAC,gBAAgB,CAAC,QAAQ;yBACtC;qBACF,CAAC;aACH,CAAC;QACJ,CAAC;QAED,OAAO;YACL,IAAI;YACJ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,IAAI,EAAE,EAAE,CAAC;SACnC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,MAAM,YAAY,GAAG,cAAc,EAAE,CAAC;IACtC,MAAM,GAAG,GAAG,OAAO,YAAY,CAAC,MAAM,CAAC,GAAG,KAAK,UAAU,CAAC,CAAC,CAAC,YAAY,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,MAAM,CAAC,GAAG,CAAC;IACtH,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,GAAG,EAAE;QACzC,MAAM,EAAE,MAAM;QACd,OAAO,EAAE,YAAY,CAAC,MAAM,CAAC,OAAO,EAAE;QACtC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;YACnB,QAAQ;YACR,KAAK,EAAE,CAAC;oBACN,oBAAoB,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;wBACpC,IAAI,EAAE,CAAC,CAAC,IAAI;wBACZ,WAAW,EAAE,CAAC,CAAC,WAAW;wBAC1B,UAAU,EAAE,CAAC,CAAC,YAAY;qBAC3B,CAAC,CAAC;iBACJ,CAAC;SACH,CAAC;KACH,CAAC,CAAC;IAEH,MAAM,IAAI,GAAQ,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;IAExC,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;QACf,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAElE,IAAI,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,QAAQ,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;YAClE,MAAM,IAAI,KAAK,CAAC,iGAAiG,QAAQ,EAAE,CAAC,CAAC;QAC/H,CAAC;QAED,MAAM,IAAI,KAAK,CAAC,qBAAqB,QAAQ,EAAE,CAAC,CAAC;IACnD,CAAC;IAED,IAAI,CAAC,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACrD,MAAM,IAAI,KAAK,CAAC,gDAAgD,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC1F,CAAC;IAED,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;IACrC,MAAM,KAAK,GAAG,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC;IAEtC,MAAM,aAAa,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC;IAC/D,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC7B,OAAO;YACL,SAAS,EAAE,aAAa,CAAC,GAAG,CAAC,CAAC,EAAO,EAAE,EAAE,CAAC,CAAC;gBACzC,EAAE,EAAE,UAAU,EAAE,CAAC,YAAY,CAAC,IAAI,IAAI,IAAI,CAAC,GAAG,EAAE,EAAE;gBAClD,IAAI,EAAE,EAAE,CAAC,YAAY,CAAC,IAAI;gBAC1B,KAAK,EAAE,EAAE,CAAC,YAAY,CAAC,IAAI;aAC5B,CAAC,CAAC;SACJ,CAAC;IACJ,CAAC;IAED,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IAChD,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,IAAI,EAAE,EAAE,CAAC;AAC3C,CAAC;AAED,KAAK,UAAU,UAAU,CAAC,QAAmB,EAAE,KAAa;IAC1D,MAAM,MAAM,GAAG,WAAC,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;IACvC,IAAI,CAAC,MAAM;QAAE,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;IAEhF,MAAM,YAAY,GAAG,cAAc,EAAE,CAAC;IACtC,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,YAAY,CAAC,MAAM,CAAC,GAAa,EAAE;QACvE,MAAM,EAAE,MAAM;QACd,OAAO,EAAE,YAAY,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;QAC5C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;YACnB,KAAK,EAAE,YAAY,CAAC,MAAM,CAAC,KAAK;YAChC,UAAU,EAAE,IAAI;YAChB,QAAQ;YACR,KAAK;SACN,CAAC;KACH,CAAC,CAAC;IAEH,MAAM,IAAI,GAAQ,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;IAExC,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CAAC,qBAAqB,IAAI,CAAC,KAAK,CAAC,OAAO,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IAC3F,CAAC;IAED,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;QAClD,MAAM,IAAI,KAAK,CAAC,kDAAkD,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC5F,CAAC;IAED,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;IACrE,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC;IAEvE,OAAO;QACL,OAAO,EAAE,WAAW,EAAE,IAAI;QAC1B,SAAS,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC,EAAO,EAAE,EAAE,CAAC,CAAC;YACnC,EAAE,EAAE,EAAE,CAAC,EAAE;YACT,IAAI,EAAE,EAAE,CAAC,IAAI;YACb,KAAK,EAAE,EAAE,CAAC,KAAK;SAChB,CAAC,CAAC;QACH,UAAU,EAAE,IAAI,CAAC,WAAW;KAC7B,CAAC;AACJ,CAAC;AAEM,KAAK,UAAU,MAAM,CAAC,QAAmB,EAAE,KAAa,EAAE,WAAmB,WAAC,CAAC,MAAM,CAAC,UAAU;IACrG,QAAQ,QAAQ,EAAE,CAAC;QACjB,KAAK,QAAQ;YACX,OAAO,MAAM,UAAU,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAC3C,KAAK,QAAQ;YACX,OAAO,MAAM,UAAU,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAC3C,KAAK,QAAQ;YACX,OAAO,MAAM,UAAU,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAC3C;YACE,MAAM,IAAI,KAAK,CAAC,wBAAwB,QAAQ,EAAE,CAAC,CAAC;IACxD,CAAC;AACH,CAAC;AAED,8EAA8E;AAC9E,6EAA6E;AAC7E,6EAA6E;AAC7E,8EAA8E;AAE9E,SAAgB,wBAAwB,CACtC,QAAgB,EAChB,QAAmB,EACnB,SAAqB,EACrB,WAA6D;IAE7D,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;QAC1B,QAAQ,CAAC,IAAI,CAAC;YACZ,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;gBAC5B,IAAI,EAAE,UAAU;gBAChB,EAAE,EAAE,EAAE,CAAC,EAAE;gBACT,IAAI,EAAE,EAAE,CAAC,IAAI;gBACb,KAAK,EAAE,EAAE,CAAC,KAAK;aAChB,CAAC,CAAC;SACO,CAAC,CAAC;QAEd,QAAQ,CAAC,IAAI,CAAC;YACZ,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;gBAC9B,IAAI,EAAE,aAAa;gBACnB,WAAW,EAAE,EAAE,CAAC,EAAE;gBAClB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,MAAM,CAAC;aACnC,CAAC,CAAC;SACO,CAAC,CAAC;IAEhB,CAAC;SAAM,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;QACjC,QAAQ,CAAC,IAAI,CAAC;YACZ,IAAI,EAAE,WAAW;YACjB,UAAU,EAAE,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;gBAC/B,EAAE,EAAE,EAAE,CAAC,EAAE;gBACT,IAAI,EAAE,UAAU;gBAChB,QAAQ,EAAE;oBACR,IAAI,EAAE,EAAE,CAAC,IAAI;oBACb,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,KAAK,CAAC;iBACpC;aACF,CAAC,CAAC;SACO,CAAC,CAAC;QAEd,KAAK,MAAM,EAAE,IAAI,WAAW,EAAE,CAAC;YAC7B,QAAQ,CAAC,IAAI,CAAC;gBACZ,IAAI,EAAE,MAAM;gBACZ,YAAY,EAAE,EAAE,CAAC,EAAE;gBACnB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,MAAM,CAAC;aACxB,CAAC,CAAC;QAChB,CAAC;IAEH,CAAC;SAAM,CAAC;QACN,SAAS;QACT,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;YACjC,QAAQ,CAAC,IAAI,CAAC;gBACZ,IAAI,EAAE,OAAO;gBACb,YAAY,EAAE;oBACZ,IAAI,EAAE,QAAQ,CAAC,IAAI;oBACnB,IAAI,EAAE,QAAQ,CAAC,KAAK;iBACrB;aACS,CAAC,CAAC;YAEd,MAAM,MAAM,GAAG,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,KAAK,QAAQ,CAAC,IAAI,CAAC,CAAC;YACjE,QAAQ,CAAC,IAAI,CAAC;gBACZ,IAAI,EAAE,MAAM;gBACZ,gBAAgB,EAAE;oBAChB,IAAI,EAAE,QAAQ,CAAC,IAAI;oBACnB,QAAQ,EAAE,MAAM,EAAE,MAAM;iBACzB;aACS,CAAC,CAAC;QAChB,CAAC;IACH,CAAC;AACH,CAAC"}

package/dist/ast.d.ts

@@ -0,0 +1,8 @@
+export declare function parseFileToAST(filePath: string, content: string): any;
+export declare function analyzeFileAST(filePath: string, functionName?: string): any;
+export declare function getFunctionAST(filePath: string, functionName: string): any;
+export declare function getImportsAST(filePath: string): any;
+export declare function getTypeDefinitions(filePath: string): any;
+export declare function getFilePreamble(filePath: string): any;
+export declare function getClassMethods(filePath: string, className: string): any;
+//# sourceMappingURL=ast.d.ts.map

package/dist/ast.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ast.d.ts","sourceRoot":"","sources":["../src/ast.ts"],"names":[],"mappings":"AASA,wBAAgB,cAAc,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,GAAG,CAqBrE;AAED,wBAAgB,cAAc,CAAC,QAAQ,EAAE,MAAM,EAAE,YAAY,CAAC,EAAE,MAAM,GAAG,GAAG,CAmN3E;AA2LD,wBAAgB,cAAc,CAAC,QAAQ,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,GAAG,GAAG,CAoF1E;AAgKD,wBAAgB,aAAa,CAAC,QAAQ,EAAE,MAAM,GAAG,GAAG,CA0EnD;AAED,wBAAgB,kBAAkB,CAAC,QAAQ,EAAE,MAAM,GAAG,GAAG,CAuFxD;AAED,wBAAgB,eAAe,CAAC,QAAQ,EAAE,MAAM,GAAG,GAAG,CA2JrD;AAED,wBAAgB,eAAe,CAAC,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,GAAG,CA2FxE"}