Function parsePythonMountsToGraph

src/detectors/routes.ts:1914–1969

Python/FastAPI: scan for APIRouter(prefix=...) + include_router() chains

(
  content: string,
  _entryDir: string,
  files: string[],
  project: ProjectInfo,
  mountEdges: Map<string, { prefix: string; mountedBy: string }>,
  sourceFile: string
)

  mountEdges: Map<string, { prefix: string; mountedBy: string }>,
  sourceFile: string
): void {
  // Step 1: Build alias map: "auth_router" → "backend/routes/auth.py"
  //   from routes.auth import router as auth_router
  const aliasRe = /from\s+([\w.]+)\s+import\s+router\s+as\s+(\w+)/g;
  const aliasMap = new Map<string, string>(); // alias → source file
  let m;
  while ((m = aliasRe.exec(content)) !== null) {
    const moduleDots = m[1];
    const alias = m[2];
    const modPath = moduleDots.replace(/\./g, "/");
    const hit = files.find(f => {
      const rel = f.replace(/\\/g, "/");
      return rel.endsWith(`/${modPath}.py`) || rel.endsWith(`${modPath}.py`);
    });
    if (hit) aliasMap.set(alias, relative(project.root, hit).replace(/\\/g, "/"));
  }

  // Also handle: from routes.auth import router (no alias)
  const noAliasRe = /from\s+([\w.]+)\s+import\s+router(?!\s+as)\b/g;
  while ((m = noAliasRe.exec(content)) !== null) {
    const modPath = m[1].replace(/\./g, "/");
    const hit = files.find(f => f.replace(/\\/g, "/").endsWith(`${modPath}.py`));
    if (hit) aliasMap.set("router", relative(project.root, hit).replace(/\\/g, "/"));
  }

  // Step 2: Find APIRouter with prefix: api_router = APIRouter(prefix="/api")
  const prefixRouterRe = /(\w+)\s*=\s*APIRouter\s*\([^)]*prefix\s*=\s*['"]([^'"]+)['"]/g;
  const routerPrefixes = new Map<string, string>(); // varName → prefix
  while ((m = prefixRouterRe.exec(content)) !== null) {
    routerPrefixes.set(m[1], m[2]);
  }

  // Step 3: Chain include_router calls:
  // api_router.include_router(auth_router)
  // api_router.include_router(cv_router, prefix="/cv")
  const includeRe = /(\w+)\s*\.\s*include_router\s*\(\s*(\w+)(?:[^)]*prefix\s*=\s*['"]([^'"]+)['"])?\s*\)/g;
  while ((m = includeRe.exec(content)) !== null) {
    const parentVar = m[1];
    const childVar = m[2];
    const extraPrefix = m[3] || "";
    const parentPrefix = routerPrefixes.get(parentVar) || "";
    const fullPrefix = parentPrefix + extraPrefix;

    const targetFile = aliasMap.get(childVar);
    if (targetFile && fullPrefix && !mountEdges.has(targetFile)) {
      mountEdges.set(targetFile, { prefix: fullPrefix, mountedBy: sourceFile });
    }
  }
}