"""parse_warn v3 (2026-05-03)
Reads PyInstaller warn file, extracts missing module names, applies translations.
Usage: python ~parse_warn.py <envname>
Prints one conda package name per line to stdout.
Skips internal modules (starting with _) and known-safe/unix-only names.
Supports PyInstaller 5.x format (W: no module named 'foo') and
PyInstaller 6.x format (missing module named foo - imported by ... (delayed|top-level|conditional)).
"""
__version__ = "parse_warn v3 (2026-05-03)"
__all__ = ["main", "parse_warn_file", "TRANSLATIONS", "SKIP"]

import os
import re
import sys

TRANSLATIONS = {
    "cv2": "opencv",
    "PIL": "pillow",
    "Image": "pillow",
    "sklearn": "scikit-learn",
    "bs4": "beautifulsoup4",
    "serial": "pyserial",
    "yaml": "pyyaml",
    "git": "gitpython",
    "wx": "wxpython",
    "dateutil": "python-dateutil",
    "dotenv": "python-dotenv",
    "Crypto": "pycryptodome",
    "OpenSSL": "pyopenssl",
    "jwt": "pyjwt",
    "usb": "pyusb",
    "attr": "attrs",
    "win32api": "pywin32",
    "win32con": "pywin32",
    "win32com": "pywin32",
    "win32gui": "pywin32",
    "win32file": "pywin32",
    "win32process": "pywin32",
    "win32event": "pywin32",
    "pywintypes": "pywin32",
    "pythoncom": "pywin32",
    "winerror": "pywin32",
    "fitz": "pymupdf",
    "docx": "python-docx",
    "pptx": "python-pptx",
    "pydantic_core": "pydantic-core",
    "skimage": "scikit-image",
    "Cryptodome": "pycryptodome",
    "zmq": "pyzmq",
}

SKIP = frozenset([
    "pkg_resources",
    "distutils",
    "setuptools",
    "importlib",
    "importlib.metadata",
    "importlib.resources",
    "importlib.abc",
    # stdlib packages whose submodules surface as "missing" in the warn file
    # (e.g. collections.abc -> collections). They are never conda packages, so
    # forwarding them to "conda install" only produces a noisy PackagesNotFound.
    "collections",
    # Unix-only platform modules absent on Windows; these appear as optional/delayed
    # stdlib imports in the warn file and are safe to ignore.
    "grp",
    "pwd",
    "posix",
    "resource",
    "fcntl",
    "readline",
    "termios",
    "tty",
    "pty",
    "crypt",
    "spwd",
    "nis",
    "syslog",
    "ossaudiodev",
    # Python-2-only stdlib modules (removed entirely in Python 3, never a real PyPI/conda
    # package) that still appear as conditional imports inside real packages' own Python
    # 2/3 compatibility shims -- e.g. xlrd/timemachine.py does
    # "try: from cStringIO import StringIO except ImportError: from io import StringIO",
    # a dead code path under any Python 3 interpreter, but PyInstaller's warn file flags it
    # regardless. Confirmed via a real, unflagged CI run of the layered-dependency-chain
    # E2E test (self.layered_e2e.chain): "missing module named cStringIO - imported by
    # xlrd.timemachine (conditional)" -- warnfix genuinely attempted and failed to install
    # a package literally named "cStringIO", which can never exist, forcing an unnecessary
    # extra provider cascade past conda. Different axis from the Unix-only list above
    # (Python-version-only, not platform-only), but the same "guaranteed to never be a
    # real installable package" property that makes it safe to filter unconditionally.
    "cStringIO",
    "StringIO",
])


def parse_warn_file(warn_path):
    """Parse a PyInstaller warn file; return list of conda package names (deduplicated, ordered).

    Returns an empty list if warn_path does not exist.
    Each entry in TRANSLATIONS maps an import-name to a conda package name.
    Unknown import names pass through unchanged.
    """
    if not os.path.exists(warn_path):
        return []
    seen = set()
    result = []
    with open(warn_path, "r", encoding="utf-8", errors="ignore") as fh:
        for line in fh:
            line = line.strip()
            # PyInstaller 5.x format: W: no module named 'foo'
            m = re.match(r"W: no module named '([^']+)'", line)
            if m:
                mod = m.group(1).split(".")[0]
            else:
                # PyInstaller 6.x format:
                # missing module named foo - imported by bar (top-level)
                # missing module named foo - imported by bar (delayed)
                # missing module named foo - imported by bar (conditional)
                # Skip entries that are ONLY optional (try-except guards with no
                # other qualifier) -- those are intentionally resilient to missing
                # modules. top-level, delayed, and conditional imports are required
                # at runtime. Unix-only stdlib shims (grp, pwd, posix, etc.) land
                # in those categories too but are already filtered by SKIP.
                m = re.match(r"missing module named (\S+)", line)
                if not m:
                    continue
                if not re.search(r'\([^)]*(?:top-level|delayed|conditional)[^)]*\)', line):
                    continue
                mod = m.group(1).strip("'\"").split(".")[0]
            if mod.startswith("_"):
                continue
            if mod in SKIP:
                continue
            pkg = TRANSLATIONS.get(mod, mod)
            if pkg not in seen:
                seen.add(pkg)
                result.append(pkg)
    return result


def main():
    if len(sys.argv) < 2:
        sys.exit(1)
    envname = sys.argv[1]
    warn_path = os.path.join("build", envname, "warn-" + envname + ".txt")
    for pkg in parse_warn_file(warn_path):
        sys.stdout.write(pkg + "\n")


if __name__ == "__main__":
    main()