backend: initial commit

bincio/extract/parsers/factory.py

@@ -0,0 +1,46 @@
+"""Parser factory — selects the right parser based on file extension."""
+
+from pathlib import Path
+
+from bincio.extract.models import ParsedActivity
+from bincio.extract.parsers.base import BaseParser
+from bincio.extract.parsers.fit import FitParser
+from bincio.extract.parsers.gpx import GpxParser
+from bincio.extract.parsers.tcx import TcxParser
+
+# Supported extensions (including .gz variants)
+SUPPORTED = {".fit", ".gpx", ".tcx", ".fit.gz", ".gpx.gz", ".tcx.gz"}
+
+_PARSERS: dict[str, type[BaseParser]] = {
+    ".fit": FitParser,
+    ".gpx": GpxParser,
+    ".tcx": TcxParser,
+}
+
+
+def _base_ext(path: Path) -> str:
+    """Return the meaningful extension, stripping .gz if present."""
+    if path.suffix == ".gz":
+        return Path(path.stem).suffix  # e.g. ".fit" from "ride.fit.gz"
+    return path.suffix
+
+
+def is_supported(path: Path) -> bool:
+    suffix = "".join(path.suffixes[-2:]) if path.suffix == ".gz" else path.suffix
+    return suffix in SUPPORTED
+
+
+def parse_file(path: Path) -> ParsedActivity:
+    """Parse an activity file, handling .gz transparently."""
+    ext = _base_ext(path)
+    parser_cls = _PARSERS.get(ext)
+    if parser_cls is None:
+        raise ValueError(f"Unsupported file type: {path.name!r}")
+
+    raw_bytes, content_bytes = BaseParser._read_file(path)
+    parser = parser_cls()
+    activity = parser.parse(path, content_bytes)
+    # Attach hash of the *original* bytes (compressed if .gz) for dedup
+    activity.source_hash = BaseParser._sha256(raw_bytes)
+    activity.source_file = path.name
+    return activity