#!/usr/bin/env node

import { readdir, readFile } from 'node:fs/promises';
import path from 'node:path';
import { fileURLToPath, pathToFileURL } from 'node:url';

const codeExtensions = new Set(['.ts', '.tsx', '.js', '.jsx', '.mjs', '.cjs', '.py']);

const appImportPatterns = [
  {
    label: 'server source import',
    pattern: /(?:from\s+['"][^'"]*|import\s*\(\s*['"][^'"]*|import\s+['"][^'"]*)(?:@server\/|server\/src|(?:\.\.\/)+server\/src)/,
  },
  {
    label: 'frontend source import',
    pattern: /(?:from\s+['"][^'"]*|import\s*\(\s*['"][^'"]*|import\s+['"][^'"]*)(?:@frontend\/|frontend\/src|(?:\.\.\/)+frontend\/src)/,
  },
  {
    label: 'python service app import',
    pattern: /(?:from\s+['"][^'"]*|import\s*\(\s*['"][^'"]*|import\s+['"][^'"]*|from\s+)(?:python-service\/app|python_service\.app|app\.)/,
  },
];

const llmBoundaryPatterns = [
  {
    label: 'direct Anthropic provider construction',
    pattern: /\bcreateAnthropic\b/,
  },
  {
    label: 'direct Vertex Anthropic provider construction',
    pattern: /\bcreateVertexAnthropic\b/,
  },
  {
    label: 'direct AI SDK gateway construction',
    pattern: /\bcreateGateway\b/,
  },
  {
    label: 'direct AI SDK embedding execution',
    pattern: /\bembedMany\b/,
  },
  {
    label: 'context-owned LLM provider port',
    pattern: /\bLlmProviderPort\b/,
  },
  {
    label: 'scan-owned LLM provider port',
    pattern: /\bKtxScanLlmPort\b/,
  },
  {
    label: 'context-owned gateway LLM provider helper',
    pattern: /\bcreateGatewayLlmProvider\b/,
  },
];

const contextProductionLlmBoundaryPatterns = [
  {
    label: 'context getModelByName call',
    pattern: /\.\s*getModelByName\s*\(/,
  },
];

const concreteDialectImportPatterns = [
  /from\s+['"][^'"]*connectors\/[^'"]+\/dialect\.js['"]/,
  /import\s*\(\s*['"][^'"]*connectors\/[^'"]+\/dialect\.js['"]\s*\)/,
  /from\s+['"]\.\/dialect\.js['"]/,
  /import\s*\(\s*['"]\.\/dialect\.js['"]\s*\)/,
];

function normalizePath(filePath) {
  return filePath.split(path.sep).join('/');
}

function isCodeSource(relativePath) {
  return codeExtensions.has(path.extname(relativePath));
}

function scansForAppImports(relativePath) {
  return isCodeSource(relativePath);
}

function scansForLlmBoundaries(relativePath) {
  return isCodeSource(relativePath) && relativePath.startsWith('packages/cli/src/context/');
}

function isTestSource(relativePath) {
  return (
    /(?:^|\/)[^/]+\.(?:test|spec)\.[cm]?[jt]sx?$/.test(relativePath) ||
    /(?:^|\/)tests\/(?:.+\/)?(?:test_[^/]+|[^/]+_test)\.py$/.test(relativePath)
  );
}

function scansForContextProductionLlmBoundaries(relativePath) {
  return scansForLlmBoundaries(relativePath) && !isTestSource(relativePath);
}

function scansForConcreteDialectImportBoundaries(relativePath) {
  return (
    relativePath.startsWith('packages/cli/src/context/scan/') ||
    /^packages\/cli\/src\/connectors\/[^/]+\/connector\.ts$/.test(relativePath)
  );
}

export function scanFileContent(relativePath, content) {
  const normalizedPath = normalizePath(relativePath);
  const violations = [];

  if (scansForAppImports(normalizedPath)) {
    for (const appImportPattern of appImportPatterns) {
      if (appImportPattern.pattern.test(content)) {
        violations.push({
          file: normalizedPath,
          kind: 'app-import',
          message: `Forbidden ${appImportPattern.label}`,
        });
      }
    }
  }

  if (scansForLlmBoundaries(normalizedPath)) {
    for (const llmBoundaryPattern of llmBoundaryPatterns) {
      if (llmBoundaryPattern.pattern.test(content)) {
        violations.push({
          file: normalizedPath,
          kind: 'llm-boundary',
          message: `Forbidden ${llmBoundaryPattern.label}; use packages/cli/src/llm`,
        });
      }
    }
  }

  if (scansForContextProductionLlmBoundaries(normalizedPath)) {
    for (const llmBoundaryPattern of contextProductionLlmBoundaryPatterns) {
      if (llmBoundaryPattern.pattern.test(content)) {
        violations.push({
          file: normalizedPath,
          kind: 'llm-boundary',
          message: `Forbidden ${llmBoundaryPattern.label}; use getModel(role) inside context modules`,
        });
      }
    }
  }

  if (scansForConcreteDialectImportBoundaries(normalizedPath)) {
    for (const pattern of concreteDialectImportPatterns) {
      if (pattern.test(content)) {
        violations.push({
          file: normalizedPath,
          kind: 'dialect-boundary',
          message:
            'Forbidden concrete connector dialect import; use getDialectForDriver() from context/connections/dialects.ts',
        });
      }
    }
  }

  return violations;
}

async function collectFiles(rootDir, currentDir = rootDir) {
  const entries = await readdir(currentDir, { withFileTypes: true });
  const files = [];

  for (const entry of entries) {
    const fullPath = path.join(currentDir, entry.name);

    if (entry.isDirectory()) {
      if (entry.name === 'node_modules' || entry.name === 'dist' || entry.name === '.venv') {
        continue;
      }

      files.push(...(await collectFiles(rootDir, fullPath)));
      continue;
    }

    if (entry.isFile()) {
      files.push(fullPath);
    }
  }

  return files;
}

export async function collectViolations(rootDir) {
  const files = await collectFiles(rootDir);
  const violations = [];

  for (const file of files) {
    const relativePath = normalizePath(path.relative(rootDir, file));
    const content = await readFile(file, 'utf8');

    violations.push(...scanFileContent(relativePath, content));
  }

  return violations;
}

async function main() {
  const scriptDir = path.dirname(fileURLToPath(import.meta.url));
  const rootDir = path.resolve(scriptDir, '..');
  const violations = await collectViolations(rootDir);

  if (violations.length === 0) {
    process.stdout.write('ktx boundary check passed\n');
    return;
  }

  for (const violation of violations) {
    process.stderr.write(`${violation.file}: ${violation.message}\n`);
  }

  process.exitCode = 1;
}

if (import.meta.url === pathToFileURL(process.argv[1] ?? '').href) {
  await main();
}
