@aiready/context-analyzer 0.21.5 → 0.21.7

package/src/classifier.ts

@@ -1,4 +1,16 @@
 import type { DependencyNode, FileClassification } from './types';
+import {
+  isBarrelExport,
+  isTypeDefinition,
+  isNextJsPage,
+  isLambdaHandler,
+  isServiceFile,
+  isEmailTemplate,
+  isParserFile,
+  isSessionFile,
+  isUtilityModule,
+  isConfigFile,
+} from './heuristics';
 
 /**
  * Constants for file classifications to avoid magic strings
@@ -99,334 +111,7 @@ export function classifyFile(
   return Classification.UNKNOWN;
 }
 
-/**
- * Detect if a file is a barrel export (index.ts)
- *
- * @param node The dependency node to check
- * @returns True if the file appears to be a barrel export
- */
-export function isBarrelExport(node: DependencyNode): boolean {
-  const { file, exports } = node;
-  const fileName = file.split('/').pop()?.toLowerCase();
-
-  // Barrel files are typically named index.ts or index.js
-  const isIndexFile = fileName === 'index.ts' || fileName === 'index.js';
-
-  // Small file with many exports is likely a barrel
-  const isSmallAndManyExports =
-    node.tokenCost < 1000 && (exports || []).length > 5;
-
-  // RE-EXPORT HEURISTIC for non-index files
-  const isReexportPattern =
-    (exports || []).length >= 5 &&
-    (exports || []).every(
-      (e) =>
-        e.type === 'const' ||
-        e.type === 'function' ||
-        e.type === 'type' ||
-        e.type === 'interface'
-    );
-
-  return !!isIndexFile || !!isSmallAndManyExports || !!isReexportPattern;
-}
-
-/**
- * Detect if a file is primarily type definitions
- *
- * @param node The dependency node to check
- * @returns True if the file appears to be primarily types
- */
-export function isTypeDefinition(node: DependencyNode): boolean {
-  const { file } = node;
-
-  // Check file extension
-  if (file.endsWith('.d.ts')) return true;
-
-  // Check if all exports are types or interfaces
-  const nodeExports = node.exports || [];
-  const hasExports = nodeExports.length > 0;
-  const areAllTypes =
-    hasExports &&
-    nodeExports.every((e) => e.type === 'type' || e.type === 'interface');
-  const allTypes: boolean = !!areAllTypes;
-
-  // Check if path includes 'types' or 'interfaces'
-  const isTypePath =
-    file.toLowerCase().includes('/types/') ||
-    file.toLowerCase().includes('/interfaces/') ||
-    file.toLowerCase().includes('/models/');
-
-  return allTypes || (isTypePath && hasExports);
-}
-
-/**
- * Detect if a file is a utility module
- *
- * @param node The dependency node to check
- * @returns True if the file appears to be a utility module
- */
-export function isUtilityModule(node: DependencyNode): boolean {
-  const { file } = node;
-
-  // Check if path includes 'utils', 'helpers', etc.
-  const isUtilPath =
-    file.toLowerCase().includes('/utils/') ||
-    file.toLowerCase().includes('/helpers/') ||
-    file.toLowerCase().includes('/util/') ||
-    file.toLowerCase().includes('/helper/');
-
-  const fileName = file.split('/').pop()?.toLowerCase();
-  const isUtilName =
-    fileName?.includes('utils.') ||
-    fileName?.includes('helpers.') ||
-    fileName?.includes('util.') ||
-    fileName?.includes('helper.');
-
-  return !!isUtilPath || !!isUtilName;
-}
-
-/**
- * Detect if a file is a Lambda/API handler
- *
- * @param node The dependency node to check
- * @returns True if the file appears to be a Lambda handler
- */
-export function isLambdaHandler(node: DependencyNode): boolean {
-  const { file, exports } = node;
-  const fileName = file.split('/').pop()?.toLowerCase();
-
-  const handlerPatterns = [
-    'handler',
-    '.handler.',
-    '-handler.',
-    'lambda',
-    '.lambda.',
-    '-lambda.',
-  ];
-  const isHandlerName = handlerPatterns.some((pattern) =>
-    fileName?.includes(pattern)
-  );
-
-  const isHandlerPath =
-    file.toLowerCase().includes('/handlers/') ||
-    file.toLowerCase().includes('/lambdas/') ||
-    file.toLowerCase().includes('/lambda/') ||
-    file.toLowerCase().includes('/functions/');
-
-  const hasHandlerExport = (exports || []).some(
-    (e) =>
-      e.name.toLowerCase() === 'handler' ||
-      e.name.toLowerCase() === 'main' ||
-      e.name.toLowerCase() === 'lambdahandler' ||
-      e.name.toLowerCase().endsWith('handler')
-  );
-
-  return !!isHandlerName || !!isHandlerPath || !!hasHandlerExport;
-}
-
-/**
- * Detect if a file is a service file
- *
- * @param node The dependency node to check
- * @returns True if the file appears to be a service file
- */
-export function isServiceFile(node: DependencyNode): boolean {
-  const { file, exports } = node;
-  const fileName = file.split('/').pop()?.toLowerCase();
-
-  const servicePatterns = ['service', '.service.', '-service.', '_service.'];
-  const isServiceName = servicePatterns.some((pattern) =>
-    fileName?.includes(pattern)
-  );
-  const isServicePath = file.toLowerCase().includes('/services/');
-  const hasServiceNamedExport = (exports || []).some(
-    (e) =>
-      e.name.toLowerCase().includes('service') ||
-      e.name.toLowerCase().endsWith('service')
-  );
-  const hasClassExport = (exports || []).some((e) => e.type === 'class');
-
-  return (
-    !!isServiceName ||
-    !!isServicePath ||
-    (!!hasServiceNamedExport && !!hasClassExport)
-  );
-}
-
-/**
- * Detect if a file is an email template/layout
- *
- * @param node The dependency node to check
- * @returns True if the file appears to be an email template
- */
-export function isEmailTemplate(node: DependencyNode): boolean {
-  const { file, exports } = node;
-  const fileName = file.split('/').pop()?.toLowerCase();
-
-  const emailTemplatePatterns = [
-    '-email-',
-    '.email.',
-    '_email_',
-    '-template',
-    '.template.',
-    '_template',
-    '-mail.',
-    '.mail.',
-  ];
-  const isEmailTemplateName = emailTemplatePatterns.some((pattern) =>
-    fileName?.includes(pattern)
-  );
-  const isEmailPath =
-    file.toLowerCase().includes('/emails/') ||
-    file.toLowerCase().includes('/mail/') ||
-    file.toLowerCase().includes('/notifications/');
-
-  const hasTemplateFunction = (exports || []).some(
-    (e) =>
-      e.type === 'function' &&
-      (e.name.toLowerCase().startsWith('render') ||
-        e.name.toLowerCase().startsWith('generate') ||
-        (e.name.toLowerCase().includes('template') &&
-          e.name.toLowerCase().includes('email')))
-  );
-
-  return !!isEmailPath || !!isEmailTemplateName || !!hasTemplateFunction;
-}
-
-/**
- * Detect if a file is a parser/transformer
- *
- * @param node The dependency node to check
- * @returns True if the file appears to be a parser
- */
-export function isParserFile(node: DependencyNode): boolean {
-  const { file, exports } = node;
-  const fileName = file.split('/').pop()?.toLowerCase();
-
-  const parserPatterns = [
-    'parser',
-    '.parser.',
-    '-parser.',
-    '_parser.',
-    'transform',
-    '.transform.',
-    'converter',
-    'mapper',
-    'serializer',
-  ];
-  const isParserName = parserPatterns.some((pattern) =>
-    fileName?.includes(pattern)
-  );
-  const isParserPath =
-    file.toLowerCase().includes('/parsers/') ||
-    file.toLowerCase().includes('/transformers/');
-
-  const hasParseFunction = (exports || []).some(
-    (e) =>
-      e.type === 'function' &&
-      (e.name.toLowerCase().startsWith('parse') ||
-        e.name.toLowerCase().startsWith('transform') ||
-        e.name.toLowerCase().startsWith('extract'))
-  );
-
-  return !!isParserName || !!isParserPath || !!hasParseFunction;
-}
-
-/**
- * Detect if a file is a session/state management file
- *
- * @param node The dependency node to check
- * @returns True if the file appears to be a session/state file
- */
-export function isSessionFile(node: DependencyNode): boolean {
-  const { file, exports } = node;
-  const fileName = file.split('/').pop()?.toLowerCase();
-
-  const sessionPatterns = ['session', 'state', 'context', 'store'];
-  const isSessionName = sessionPatterns.some((pattern) =>
-    fileName?.includes(pattern)
-  );
-  const isSessionPath =
-    file.toLowerCase().includes('/sessions/') ||
-    file.toLowerCase().includes('/state/');
-
-  const hasSessionExport = (exports || []).some(
-    (e) =>
-      e.name.toLowerCase().includes('session') ||
-      e.name.toLowerCase().includes('state') ||
-      e.name.toLowerCase().includes('store')
-  );
-
-  return !!isSessionName || !!isSessionPath || !!hasSessionExport;
-}
-
-/**
- * Detect if a file is a configuration or schema file
- *
- * @param node The dependency node to check
- * @returns True if the file appears to be a config file
- */
-export function isConfigFile(node: DependencyNode): boolean {
-  const { file, exports } = node;
-  const lowerPath = file.toLowerCase();
-  const fileName = file.split('/').pop()?.toLowerCase();
-
-  const configPatterns = [
-    '.config.',
-    'tsconfig',
-    'jest.config',
-    'package.json',
-    'aiready.json',
-    'next.config',
-    'sst.config',
-  ];
-  const isConfigName = configPatterns.some((p) => fileName?.includes(p));
-  const isConfigPath =
-    lowerPath.includes('/config/') ||
-    lowerPath.includes('/settings/') ||
-    lowerPath.includes('/schemas/');
-
-  const hasSchemaExports = (exports || []).some(
-    (e) =>
-      e.name.toLowerCase().includes('schema') ||
-      e.name.toLowerCase().includes('config') ||
-      e.name.toLowerCase().includes('setting')
-  );
-
-  return !!isConfigName || !!isConfigPath || !!hasSchemaExports;
-}
-
-/**
- * Detect if a file is a Next.js App Router page
- *
- * @param node The dependency node to check
- * @returns True if the file appears to be a Next.js page
- */
-export function isNextJsPage(node: DependencyNode): boolean {
-  const { file, exports } = node;
-  const lowerPath = file.toLowerCase();
-  const fileName = file.split('/').pop()?.toLowerCase();
-
-  const isInAppDir =
-    lowerPath.includes('/app/') || lowerPath.startsWith('app/');
-  const isPageFile = fileName === 'page.tsx' || fileName === 'page.ts';
-
-  if (!isInAppDir || !isPageFile) return false;
-
-  const hasDefaultExport = (exports || []).some((e) => e.type === 'default');
-  const nextJsExports = [
-    'metadata',
-    'generatemetadata',
-    'faqjsonld',
-    'jsonld',
-    'icon',
-  ];
-  const hasNextJsExports = (exports || []).some((e) =>
-    nextJsExports.includes(e.name.toLowerCase())
-  );
-
-  return !!hasDefaultExport || !!hasNextJsExports;
-}
+// [Split Point] Logic below this point handled by heuristics.ts
 
 /**
  * Adjust cohesion score based on file classification

package/src/cli-action.ts

@@ -0,0 +1,110 @@
+import {
+  loadMergedConfig,
+  handleJSONOutput,
+  handleCLIError,
+  getElapsedTime,
+  resolveOutputPath,
+} from '@aiready/core';
+import { analyzeContext } from './analyzer';
+import { generateSummary } from './summary';
+import {
+  displayConsoleReport,
+  generateHTMLReport,
+  runInteractiveSetup,
+} from './utils/output-formatter';
+import chalk from 'chalk';
+import { writeFileSync } from 'fs';
+
+/**
+ * Orchestrates the context analysis CLI command.
+ * Merges configuration, invokes the analyzer, and formats the output (Console/JSON/HTML).
+ *
+ * @param directory - Root directory to analyze
+ * @param options - CLI options including focus area, max depth, and output format
+ */
+export async function contextActionHandler(directory: string, options: any) {
+  console.log(chalk.blue('🔍 Analyzing context window costs...\n'));
+
+  const startTime = Date.now();
+
+  try {
+    // Define defaults
+    const defaults = {
+      maxDepth: 5,
+      maxContextBudget: 10000,
+      minCohesion: 0.6,
+      maxFragmentation: 0.5,
+      focus: 'all',
+      includeNodeModules: false,
+      include: undefined,
+      exclude: undefined,
+      maxResults: 10,
+    };
+
+    // Load and merge config with CLI options
+    let finalOptions = (await loadMergedConfig(directory, defaults, {
+      maxDepth: options.maxDepth ? parseInt(options.maxDepth) : undefined,
+      maxContextBudget: options.maxContext
+        ? parseInt(options.maxContext)
+        : undefined,
+      minCohesion: options.minCohesion
+        ? parseFloat(options.minCohesion)
+        : undefined,
+      maxFragmentation: options.maxFragmentation
+        ? parseFloat(options.maxFragmentation)
+        : undefined,
+      focus:
+        (options.focus as 'fragmentation' | 'cohesion' | 'depth' | 'all') ||
+        undefined,
+      includeNodeModules: options.includeNodeModules,
+      include: options.include?.split(','),
+      exclude: options.exclude?.split(','),
+      maxResults: options.maxResults ? parseInt(options.maxResults) : undefined,
+    })) as any;
+
+    // Interactive setup if requested
+    if (options.interactive) {
+      finalOptions = await runInteractiveSetup(directory, finalOptions);
+    }
+
+    // Run analysis
+    const results = await analyzeContext(finalOptions);
+    const summary = generateSummary(results, finalOptions);
+
+    const duration = getElapsedTime(startTime);
+
+    // Handle output
+    if (options.output === 'json') {
+      handleJSONOutput(
+        {
+          summary: {
+            ...summary,
+            executionTime: duration,
+            config: {
+              scan: { tools: ['context'] },
+              tools: { context: finalOptions },
+            },
+            toolConfigs: { context: finalOptions },
+          },
+          context: { results },
+        },
+        options.outputFile
+      );
+    } else if (options.output === 'html') {
+      const html = generateHTMLReport(summary, results);
+      const outputPath = resolveOutputPath(
+        directory,
+        options.outputFile,
+        'context-report.html'
+      );
+      writeFileSync(outputPath, html, 'utf-8');
+      console.log(chalk.green(`\n✅ HTML report saved to: ${outputPath}`));
+    } else {
+      // Default: Console
+      displayConsoleReport(summary, results, finalOptions.maxResults);
+      console.log(chalk.dim(`\n✨ Analysis completed in ${duration}ms\n`));
+    }
+  } catch (error) {
+    handleCLIError(error, 'context-analyzer');
+  }
+}