diff --git a/src/docgen/detectors/javascript.py b/src/docgen/detectors/javascript.py index 2caf895..ed03028 100644 --- a/src/docgen/detectors/javascript.py +++ b/src/docgen/detectors/javascript.py @@ -1,10 +1,12 @@ +#!/usr/bin/env python3 """JavaScript endpoint detector for Express and Fastify.""" import re from pathlib import Path from typing import Optional -from docgen.models import Endpoint, HTTPMethod + from docgen.detectors.base import BaseDetector +from docgen.models import Endpoint, HTTPMethod class JavaScriptDetector(BaseDetector): @@ -59,7 +61,12 @@ class JavaScriptDetector(BaseDetector): def _detect_framework(self, content: str) -> Optional[str]: """Auto-detect the JavaScript framework.""" - if "from '@fastify/" in content or "from 'fastify'" in content or "import fastify" in content: + fastify_patterns = [ + "from '@fastify/", + "from 'fastify'", + "import fastify", + ] + if any(p in content for p in fastify_patterns): return "fastify" if 'require("fastify")' in content or "require('fastify')" in content: return "fastify"