infrawise 0.5.0 → 0.6.0

package/dist/server/index.js

@@ -1,31 +1,20 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.currentFindings = exports.currentGraph = void 0;
-exports.setGraphState = setGraphState;
-exports.createMcpServer = createMcpServer;
-exports.createServer = createServer;
-const fs_1 = require("fs");
-const path_1 = require("path");
-const fastify_1 = __importDefault(require("fastify"));
-const cors_1 = __importDefault(require("@fastify/cors"));
-const mcp_js_1 = require("@modelcontextprotocol/sdk/server/mcp.js");
-const streamableHttp_js_1 = require("@modelcontextprotocol/sdk/server/streamableHttp.js");
-const zod_1 = require("zod");
-const core_1 = require("../core");
-const { version } = JSON.parse((0, fs_1.readFileSync)((0, path_1.join)(__dirname, '../../package.json'), 'utf8'));
-const analyzers_1 = require("../analyzers");
-const graph_1 = require("../graph");
+import { readFileSync } from 'fs';
+import { join } from 'path';
+import Fastify from 'fastify';
+import cors from '@fastify/cors';
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { StreamableHTTPServerTransport } from '@modelcontextprotocol/sdk/server/streamableHttp.js';
+import { z } from 'zod';
+import { logger } from '../core/index.js';
+const { version } = JSON.parse(readFileSync(join(import.meta.dirname, '../../package.json'), 'utf8'));
+import { summarizeFindings } from '../analyzers/index.js';
+import { getTableNodes, getFunctionNodes, getQueueNodes, getTopicNodes, getSecretNodes, getParameterNodes, getLogGroupNodes, getLambdaNodes, getScanEdges, getOutgoingEdges, } from '../graph/index.js';
 // ── State ────────────────────────────────────────────────────────────────────
 let currentGraph = { nodes: [], edges: [] };
-exports.currentGraph = currentGraph;
 let currentFindings = [];
-exports.currentFindings = currentFindings;
-function setGraphState(graph, findings) {
-    exports.currentGraph = currentGraph = graph;
-    exports.currentFindings = currentFindings = findings;
+export function setGraphState(graph, findings) {
+    currentGraph = graph;
+    currentFindings = findings;
 }
 // ── Helpers ──────────────────────────────────────────────────────────────────
 function toText(data) {
@@ -34,25 +23,25 @@ function toText(data) {
 function logged(name, fn) {
     return async (args) => {
         const hasArgs = Object.keys(args).length > 0;
-        core_1.logger.info(`→ ${name}${hasArgs ? `  ${JSON.stringify(args)}` : ''}`);
+        logger.info(`→ ${name}${hasArgs ? `  ${JSON.stringify(args)}` : ''}`);
         return fn(args);
     };
 }
 // ── MCP Server ────────────────────────────────────────────────────────────────
-function createMcpServer() {
-    const mcp = new mcp_js_1.McpServer({ name: 'infrawise', version });
+export function createMcpServer() {
+    const mcp = new McpServer({ name: 'infrawise', version });
     mcp.registerTool('get_infra_overview', {
         description: 'Returns a complete snapshot of all infrastructure: databases, queues, topics, secrets, parameters, log groups, lambdas, and all findings. Start here for a full picture.',
-        inputSchema: zod_1.z.object({}),
+        inputSchema: z.object({}),
     }, logged('get_infra_overview', async () => {
-        const tables = (0, graph_1.getTableNodes)(currentGraph);
-        const queues = (0, graph_1.getQueueNodes)(currentGraph);
-        const topics = (0, graph_1.getTopicNodes)(currentGraph);
-        const secrets = (0, graph_1.getSecretNodes)(currentGraph);
-        const parameters = (0, graph_1.getParameterNodes)(currentGraph);
-        const logGroups = (0, graph_1.getLogGroupNodes)(currentGraph);
-        const lambdas = (0, graph_1.getLambdaNodes)(currentGraph);
-        const functions = (0, graph_1.getFunctionNodes)(currentGraph);
+        const tables = getTableNodes(currentGraph);
+        const queues = getQueueNodes(currentGraph);
+        const topics = getTopicNodes(currentGraph);
+        const secrets = getSecretNodes(currentGraph);
+        const parameters = getParameterNodes(currentGraph);
+        const logGroups = getLogGroupNodes(currentGraph);
+        const lambdas = getLambdaNodes(currentGraph);
+        const functions = getFunctionNodes(currentGraph);
         return toText({
             summary: {
                 tables: tables.length, functions: functions.length,
@@ -60,7 +49,7 @@ function createMcpServer() {
                 secrets: secrets.length, parameters: parameters.length,
                 logGroups: logGroups.length, lambdas: lambdas.length,
                 totalNodes: currentGraph.nodes.length, totalEdges: currentGraph.edges.length,
-                findings: (0, analyzers_1.summarizeFindings)(currentFindings),
+                findings: summarizeFindings(currentFindings),
             },
             databases: tables.map((t) => ({ name: t.name, type: t.databaseType })),
             queues: queues.map((q) => ({ name: q.name, hasDLQ: q.hasDLQ, encrypted: q.encrypted, approximateMessages: q.approximateMessages })),
@@ -74,27 +63,27 @@ function createMcpServer() {
     }));
     mcp.registerTool('get_graph_summary', {
         description: 'Returns the full infrastructure graph (all nodes and edges) plus findings summary.',
-        inputSchema: zod_1.z.object({}),
+        inputSchema: z.object({}),
     }, logged('get_graph_summary', async () => toText({
         nodes: currentGraph.nodes,
         edges: currentGraph.edges,
         findings: currentFindings,
         summary: {
             totalNodes: currentGraph.nodes.length, totalEdges: currentGraph.edges.length,
-            tables: (0, graph_1.getTableNodes)(currentGraph).length, functions: (0, graph_1.getFunctionNodes)(currentGraph).length,
-            queues: (0, graph_1.getQueueNodes)(currentGraph).length, scans: (0, graph_1.getScanEdges)(currentGraph).length,
-            ...(0, analyzers_1.summarizeFindings)(currentFindings),
+            tables: getTableNodes(currentGraph).length, functions: getFunctionNodes(currentGraph).length,
+            queues: getQueueNodes(currentGraph).length, scans: getScanEdges(currentGraph).length,
+            ...summarizeFindings(currentFindings),
         },
     })));
     mcp.registerTool('analyze_function', {
         description: 'Analyze a specific function for all infrastructure issues: DB queries, queue publishing, secret access, etc.',
-        inputSchema: zod_1.z.object({ function: zod_1.z.string().describe('Function name to analyze') }),
+        inputSchema: z.object({ function: z.string().describe('Function name to analyze') }),
     }, logged('analyze_function', async ({ function: functionName }) => {
         const funcNode = currentGraph.nodes.find((n) => n.type === 'function' && n.name === functionName);
         if (!funcNode) {
             return toText({ function: functionName, found: false, issues: [], recommendations: [`Function "${functionName}" not found in the analyzed codebase.`] });
         }
-        const outEdges = (0, graph_1.getOutgoingEdges)(currentGraph, funcNode.id);
+        const outEdges = getOutgoingEdges(currentGraph, funcNode.id);
         const relatedFindings = currentFindings.filter((f) => {
             const meta = f.metadata;
             return meta?.functionName === functionName || String(meta?.callerFunctions ?? '').includes(functionName);
@@ -112,13 +101,14 @@ function createMcpServer() {
     }));
     mcp.registerTool('suggest_gsi', {
         description: 'Get GSI suggestions for a DynamoDB table and attribute',
-        inputSchema: zod_1.z.object({
-            table: zod_1.z.string().describe('DynamoDB table name'),
-            attribute: zod_1.z.string().describe('Attribute to create the GSI on'),
+        inputSchema: z.object({
+            table: z.string().describe('DynamoDB table name'),
+            attribute: z.string().describe('Attribute to create the GSI on'),
         }),
     }, logged('suggest_gsi', async ({ table: tableName, attribute }) => {
         const sanitizedAttr = attribute.replace(/[^a-zA-Z0-9_]/g, '_');
-        const indexName = `${tableName}-${sanitizedAttr}-index`;
+        const sanitizedTable = tableName.replace(/[^a-zA-Z0-9_-]/g, '_');
+        const indexName = `${sanitizedTable}-${sanitizedAttr}-index`;
         const tableNode = currentGraph.nodes.find((n) => n.type === 'table' && n.databaseType === 'dynamodb' && 'name' in n && n.name === tableName);
         return toText({
             table: tableName, attribute, found: !!tableNode,
@@ -129,9 +119,9 @@ function createMcpServer() {
     }));
     mcp.registerTool('postgres_index_suggestions', {
         description: 'Get PostgreSQL index suggestions for a table column',
-        inputSchema: zod_1.z.object({
-            table: zod_1.z.string().describe('PostgreSQL table name'),
-            column: zod_1.z.string().describe('Column name to index'),
+        inputSchema: z.object({
+            table: z.string().describe('PostgreSQL table name'),
+            column: z.string().describe('Column name to index'),
         }),
     }, logged('postgres_index_suggestions', async ({ table: tableName, column }) => {
         const sanitizedCol = column.replace(/[^a-zA-Z0-9_]/g, '_');
@@ -139,33 +129,37 @@ function createMcpServer() {
         const indexName = `idx_${sanitizedTable}_${sanitizedCol}`;
         return toText({
             table: tableName, column,
-            recommendation: `CREATE INDEX CONCURRENTLY ${indexName} ON ${tableName} (${column});`,
+            recommendation: `CREATE INDEX CONCURRENTLY ${indexName} ON ${sanitizedTable} (${sanitizedCol});`,
             rationale: `An index on "${column}" eliminates sequential scans when filtering on this column.`,
             notes: ['Use CONCURRENTLY to avoid locking the table', 'Run ANALYZE after creation',
-                `Partial index: CREATE INDEX CONCURRENTLY ${indexName}_partial ON ${tableName} (${column}) WHERE ${column} IS NOT NULL;`],
+                `Partial index: CREATE INDEX CONCURRENTLY ${indexName}_partial ON ${sanitizedTable} (${sanitizedCol}) WHERE ${sanitizedCol} IS NOT NULL;`],
         });
     }));
     mcp.registerTool('suggest_mongo_index', {
         description: 'Get index suggestions for a MongoDB collection field',
-        inputSchema: zod_1.z.object({
-            collection: zod_1.z.string().describe('MongoDB collection name'),
-            field: zod_1.z.string().describe('Field name to index'),
+        inputSchema: z.object({
+            collection: z.string().describe('MongoDB collection name'),
+            field: z.string().describe('Field name to index'),
         }),
-    }, logged('suggest_mongo_index', async ({ collection, field }) => toText({
-        collection, field,
-        recommendation: `db.${collection}.createIndex({ ${field}: 1 })`,
-        rationale: `An index on "${field}" eliminates full collection scans when filtering on this field.`,
-        notes: [
-            `Compound: db.${collection}.createIndex({ ${field}: 1, otherField: 1 })`,
-            `Text: db.${collection}.createIndex({ ${field}: "text" })`,
-            `Verify: db.${collection}.explain("executionStats").find({ ${field}: value })`,
-        ],
-    })));
+    }, logged('suggest_mongo_index', async ({ collection, field }) => {
+        const sanitizedCollection = collection.replace(/[^a-zA-Z0-9_]/g, '_');
+        const sanitizedField = field.replace(/[^a-zA-Z0-9_.]/g, '_');
+        return toText({
+            collection, field,
+            recommendation: `db.${sanitizedCollection}.createIndex({ ${sanitizedField}: 1 })`,
+            rationale: `An index on "${field}" eliminates full collection scans when filtering on this field.`,
+            notes: [
+                `Compound: db.${sanitizedCollection}.createIndex({ ${sanitizedField}: 1, otherField: 1 })`,
+                `Text: db.${sanitizedCollection}.createIndex({ ${sanitizedField}: "text" })`,
+                `Verify: db.${sanitizedCollection}.explain("executionStats").find({ ${sanitizedField}: value })`,
+            ],
+        });
+    }));
     mcp.registerTool('mysql_index_suggestions', {
         description: 'Get MySQL index suggestions for a table column',
-        inputSchema: zod_1.z.object({
-            table: zod_1.z.string().describe('MySQL table name'),
-            column: zod_1.z.string().describe('Column name to index'),
+        inputSchema: z.object({
+            table: z.string().describe('MySQL table name'),
+            column: z.string().describe('Column name to index'),
         }),
     }, logged('mysql_index_suggestions', async ({ table: tableName, column }) => {
         const sanitizedCol = column.replace(/[^a-zA-Z0-9_]/g, '_');
@@ -173,17 +167,17 @@ function createMcpServer() {
         const indexName = `idx_${sanitizedTable}_${sanitizedCol}`;
         return toText({
             table: tableName, column,
-            recommendation: `ALTER TABLE ${tableName} ADD INDEX ${indexName} (${column});`,
+            recommendation: `ALTER TABLE ${sanitizedTable} ADD INDEX ${indexName} (${sanitizedCol});`,
             rationale: `An index on "${column}" eliminates full table scans when filtering on this column.`,
             notes: ['MySQL InnoDB adds indexes online (no full lock for 5.6+)', 'EXPLAIN SELECT ... to verify after adding',
-                `Composite: ALTER TABLE ${tableName} ADD INDEX idx_composite (${column}, other_column);`],
+                `Composite: ALTER TABLE ${sanitizedTable} ADD INDEX idx_composite (${sanitizedCol}, other_column);`],
         });
     }));
     mcp.registerTool('get_queue_details', {
         description: 'Returns all SQS queues with DLQ status, encryption, message counts, and retention.',
-        inputSchema: zod_1.z.object({}),
+        inputSchema: z.object({}),
     }, logged('get_queue_details', async () => {
-        const queues = (0, graph_1.getQueueNodes)(currentGraph);
+        const queues = getQueueNodes(currentGraph);
         const queueFindings = currentFindings.filter((f) => f.metadata?.queueName);
         return toText({
             total: queues.length,
@@ -196,16 +190,16 @@ function createMcpServer() {
     }));
     mcp.registerTool('get_topic_details', {
         description: 'Returns all SNS topics with subscription counts and protocols.',
-        inputSchema: zod_1.z.object({}),
+        inputSchema: z.object({}),
     }, logged('get_topic_details', async () => {
-        const topics = (0, graph_1.getTopicNodes)(currentGraph);
+        const topics = getTopicNodes(currentGraph);
         return toText({ total: topics.length, topics: topics.map((t) => ({ name: t.name, provider: t.provider, subscriptionCount: t.subscriptionCount, encrypted: t.encrypted })) });
     }));
     mcp.registerTool('get_secrets_overview', {
         description: 'Returns all Secrets Manager secrets: names, rotation status. Secret VALUES are never included.',
-        inputSchema: zod_1.z.object({}),
+        inputSchema: z.object({}),
     }, logged('get_secrets_overview', async () => {
-        const secrets = (0, graph_1.getSecretNodes)(currentGraph);
+        const secrets = getSecretNodes(currentGraph);
         const secretFindings = currentFindings.filter((f) => f.metadata?.secretName);
         return toText({
             total: secrets.length, note: 'Secret values are never included in this response.',
@@ -217,9 +211,9 @@ function createMcpServer() {
     }));
     mcp.registerTool('get_parameter_overview', {
         description: 'Returns all SSM Parameter Store parameters: names, types, tiers. Parameter VALUES are never included.',
-        inputSchema: zod_1.z.object({}),
+        inputSchema: z.object({}),
     }, logged('get_parameter_overview', async () => {
-        const parameters = (0, graph_1.getParameterNodes)(currentGraph);
+        const parameters = getParameterNodes(currentGraph);
         return toText({
             total: parameters.length, note: 'Parameter values are never included in this response.',
             parameters: parameters.map((p) => ({ name: p.name, provider: p.provider, type: p.paramType, tier: p.tier })),
@@ -227,9 +221,9 @@ function createMcpServer() {
     }));
     mcp.registerTool('get_lambda_overview', {
         description: 'Returns all Lambda functions: runtime, memory, timeout, env var key names (values never included).',
-        inputSchema: zod_1.z.object({}),
+        inputSchema: z.object({}),
     }, logged('get_lambda_overview', async () => {
-        const lambdas = (0, graph_1.getLambdaNodes)(currentGraph);
+        const lambdas = getLambdaNodes(currentGraph);
         const lambdaFindings = currentFindings.filter((f) => f.metadata?.functionName);
         return toText({
             total: lambdas.length, note: 'Environment variable values are never included.',
@@ -242,9 +236,9 @@ function createMcpServer() {
     }));
     mcp.registerTool('get_log_errors', {
         description: 'Returns recent error patterns from CloudWatch log groups. Returns pattern counts and frequencies — never raw log messages.',
-        inputSchema: zod_1.z.object({ logGroup: zod_1.z.string().describe('Filter to a specific log group name (optional)').optional() }),
+        inputSchema: z.object({ logGroup: z.string().describe('Filter to a specific log group name (optional)').optional() }),
     }, logged('get_log_errors', async ({ logGroup: filterName }) => {
-        const logGroups = (0, graph_1.getLogGroupNodes)(currentGraph).filter((lg) => !filterName || lg.name.includes(filterName));
+        const logGroups = getLogGroupNodes(currentGraph).filter((lg) => !filterName || lg.name.includes(filterName));
         return toText({
             note: 'Only error patterns and counts are returned — no raw log messages.',
             windowHours: 24,
@@ -254,9 +248,9 @@ function createMcpServer() {
     return mcp;
 }
 // ── Fastify server ────────────────────────────────────────────────────────────
-function createServer(port = 3000) {
-    const fastify = (0, fastify_1.default)({ logger: false });
-    fastify.register(cors_1.default, { origin: true });
+export function createServer(port = 3000) {
+    const fastify = Fastify({ logger: false });
+    fastify.register(cors, { origin: true });
     const mcp = createMcpServer();
     fastify.get('/health', async () => ({
         status: 'ok', version,
@@ -265,7 +259,7 @@ function createServer(port = 3000) {
         findings: currentFindings.length,
     }));
     fastify.post('/mcp', async (request, reply) => {
-        const transport = new streamableHttp_js_1.StreamableHTTPServerTransport({ sessionIdGenerator: undefined });
+        const transport = new StreamableHTTPServerTransport({ sessionIdGenerator: undefined });
         reply.raw.on('close', () => transport.close());
         await mcp.connect(transport);
         await transport.handleRequest(request.raw, reply.raw, request.body);
@@ -276,12 +270,13 @@ function createServer(port = 3000) {
         start: async () => {
             try {
                 await fastify.listen({ port, host: '0.0.0.0' });
-                core_1.logger.info(`Infrawise MCP server running at http://localhost:${port}`);
+                logger.info(`Infrawise MCP server running at http://localhost:${port}`);
             }
             catch (e) {
-                core_1.logger.error(`Failed to start server: ${e instanceof Error ? e.message : String(e)}`);
+                logger.error(`Failed to start server: ${e instanceof Error ? e.message : String(e)}`);
                 process.exit(1);
             }
         },
     };
 }
+export { currentGraph, currentFindings };

package/dist/types.js

@@ -1,2 +1 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
+export {};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "infrawise",
-  "version": "0.5.0",
+  "version": "0.6.0",
   "mcpName": "io.github.Sidd27/infrawise",
   "description": "CLI-first infrastructure intelligence platform — analyzes DynamoDB, PostgreSQL, MySQL, MongoDB, SQS, SNS, SSM, Secrets Manager, Lambda, CloudWatch Logs and exposes findings as an MCP server for Claude Code",
   "keywords": [
@@ -32,6 +32,7 @@
     "url": "git+https://github.com/Sidd27/infrawise.git"
   },
   "license": "MIT",
+  "type": "module",
   "bin": {
     "infrawise": "dist/cli/index.js"
   },
@@ -48,46 +49,46 @@
     "lint": "eslint src",
     "typecheck": "tsc",
     "dev": "node dist/cli/index.js dev",
-    "release": "node scripts/release.mjs"
+    "release": "node scripts/release.js"
   },
   "dependencies": {
-    "@aws-sdk/client-cloudwatch-logs": "^3.0.0",
-    "@aws-sdk/client-dynamodb": "^3.0.0",
-    "@aws-sdk/client-lambda": "^3.0.0",
-    "@aws-sdk/client-rds": "^3.0.0",
-    "@aws-sdk/client-secrets-manager": "^3.0.0",
-    "@aws-sdk/client-sns": "^3.0.0",
-    "@aws-sdk/client-sqs": "^3.0.0",
-    "@aws-sdk/client-ssm": "^3.0.0",
-    "@aws-sdk/credential-providers": "^3.0.0",
-    "@fastify/cors": "^9.0.0",
+    "@aws-sdk/client-cloudwatch-logs": "^3.1048.0",
+    "@aws-sdk/client-dynamodb": "^3.1048.0",
+    "@aws-sdk/client-lambda": "^3.1048.0",
+    "@aws-sdk/client-rds": "^3.1048.0",
+    "@aws-sdk/client-secrets-manager": "^3.1048.0",
+    "@aws-sdk/client-sns": "^3.1048.0",
+    "@aws-sdk/client-sqs": "^3.1048.0",
+    "@aws-sdk/client-ssm": "^3.1048.0",
+    "@aws-sdk/credential-providers": "^3.1048.0",
     "@modelcontextprotocol/sdk": "^1.29.0",
-    "chalk": "^4.1.2",
+    "chalk": "^5.6.2",
     "commander": "^14.0.3",
-    "fastify": "^4.26.0",
-    "inquirer": "^8.2.7",
+    "fastify": "^5.8.5",
+    "inquirer": "^13.4.3",
     "js-yaml": "^4.1.0",
-    "mongodb": "^6.5.0",
+    "mongodb": "^7.2.0",
     "mysql2": "^3.9.0",
-    "ora": "^5.4.1",
+    "ora": "^9.4.0",
     "pg": "^8.11.0",
-    "pino": "^8.19.0",
-    "pino-pretty": "^11.0.0",
-    "ts-morph": "^22.0.0",
-    "typescript": "^5.9.3",
-    "zod": "^3.25.76"
+    "pino": "^10.3.1",
+    "pino-pretty": "^13.1.3",
+    "ts-morph": "^28.0.0",
+    "zod": "^4.4.3"
   },
   "devDependencies": {
-    "@types/inquirer": "^8.2.12",
+    "@fastify/cors": "^11.2.0",
+    "@types/inquirer": "^9.0.9",
     "@types/js-yaml": "^4.0.9",
-    "@types/node": "^22.0.0",
+    "@types/node": "^25.8.0",
     "@types/pg": "^8.11.0",
-    "@typescript-eslint/eslint-plugin": "^7.0.0",
-    "@typescript-eslint/parser": "^7.0.0",
-    "@vitest/coverage-v8": "^1.6.1",
-    "eslint": "^8.57.0",
-    "typescript": "^5.9.3",
-    "vitest": "^1.5.0"
+    "@typescript-eslint/eslint-plugin": "^8.59.3",
+    "@typescript-eslint/parser": "^8.59.3",
+    "@vitest/coverage-v8": "^4.1.6",
+    "eslint": "^10.4.0",
+    "typescript": "^6.0.3",
+    "vite": "^8.0.13",
+    "vitest": "^4.1.6"
   },
   "files": [
     "dist",