omdev 0.0.0.dev310__py3-none-any.whl → 0.0.0.dev312__py3-none-any.whl

omdev/.manifests.json

@@ -290,6 +290,18 @@
       }
     }
   },
+  {
+    "module": ".tools.antlr.__main__",
+    "attr": "_CLI_MODULE",
+    "file": "omdev/tools/antlr/__main__.py",
+    "line": 4,
+    "value": {
+      "$.cli.types.CliModule": {
+        "cmd_name": "antlr",
+        "mod_name": "omdev.tools.antlr.__main__"
+      }
+    }
+  },
   {
     "module": ".tools.cloc",
     "attr": "_CLI_MODULE",

omdev/ptk/__init__.py

@@ -1,8 +1,7 @@
 # ruff: noqa: F401
 # flake8: noqa: F401
 
-from prompt_toolkit import Application
-from prompt_toolkit import prompt
+from prompt_toolkit.application import Application
 from prompt_toolkit.application import get_app
 from prompt_toolkit.application import get_app_session
 from prompt_toolkit.application import run_in_terminal
@@ -80,7 +79,9 @@ from prompt_toolkit.output import DummyOutput
 from prompt_toolkit.output import Output
 from prompt_toolkit.search import SearchDirection
 from prompt_toolkit.selection import SelectionType
+from prompt_toolkit.shortcuts import confirm
 from prompt_toolkit.shortcuts import print_formatted_text
+from prompt_toolkit.shortcuts import prompt
 from prompt_toolkit.styles import BaseStyle
 from prompt_toolkit.styles import DynamicStyle
 from prompt_toolkit.styles import Style
@@ -95,3 +96,5 @@ from prompt_toolkit.widgets.toolbars import FormattedTextToolbar
 from prompt_toolkit.widgets.toolbars import SearchToolbar
 from prompt_toolkit.widgets.toolbars import SystemToolbar
 from prompt_toolkit.widgets.toolbars import ValidationToolbar
+
+from .confirm import strict_confirm

omdev/ptk/confirm.py

@@ -0,0 +1,54 @@
+from prompt_toolkit.formatted_text import merge_formatted_text
+from prompt_toolkit.key_binding import KeyBindings
+from prompt_toolkit.key_binding import KeyPressEvent
+from prompt_toolkit.shortcuts import PromptSession
+
+from omlish import check
+
+
+##
+
+
+def create_strict_confirm_session(
+        message: str,
+        suffix: str = ' (y/n) ',
+) -> PromptSession[str | bool]:
+    """Create a `PromptSession` object for the 'confirm' function."""
+
+    bindings = KeyBindings()
+
+    @bindings.add('y')
+    @bindings.add('Y')
+    def yes(event: KeyPressEvent) -> None:
+        session.default_buffer.text = 'y'
+        event.app.exit(result=True)
+
+    @bindings.add('n')
+    @bindings.add('N')
+    def no(event: KeyPressEvent) -> None:
+        session.default_buffer.text = 'n'
+        event.app.exit(result=False)
+
+    complete_message = merge_formatted_text([message, suffix])
+    session: PromptSession[str | bool] = PromptSession(
+        complete_message,
+        key_bindings=bindings,
+    )
+    return session
+
+
+def strict_confirm(message: str = 'Confirm?', suffix: str = ' (y/n) ') -> bool:
+    """Display a confirmation prompt that returns True/False. Requires an explicit answer."""
+
+    while True:
+        session = create_strict_confirm_session(message, suffix)
+        ret = session.prompt()
+
+        if isinstance(ret, str):
+            check.empty(ret)
+
+        elif isinstance(ret, bool):
+            return ret
+
+        else:
+            raise TypeError(ret)

omdev/ptk/markdown/markdown.py

@@ -1,3 +1,8 @@
+"""
+TODO:
+ - accept pre-parsed tokens
+ - provide live view in this pkg
+"""
 import itertools
 import typing as ta
 
@@ -36,6 +41,7 @@ class Markdown:
     def __init__(
             self,
             markup: str,
+            *,
             width: int | None = None,
             strip_trailing_lines: bool = True,
     ) -> None:

omdev/tools/antlr/__main__.py

@@ -0,0 +1,11 @@
+from ...cli import CliModule
+
+
+# @omlish-manifest
+_CLI_MODULE = CliModule('antlr', __name__)
+
+
+if __name__ == '__main__':
+    from .cli import _main  # noqa
+
+    _main()

omdev/tools/antlr/cli.py

@@ -0,0 +1,62 @@
+import logging
+import re
+import subprocess
+import sys
+
+from omlish.argparse import all as ap
+from omlish.logs import all as logs
+
+from .consts import ANTLR_RUNTIME_PACKAGE
+from .gen import GenPy
+from .gen import get_jar_path
+
+
+log = logging.getLogger(__name__)
+
+
+##
+
+
+class Cli(ap.Cli):
+    @ap.cmd()
+    def jar(self) -> None:
+        print(get_jar_path())
+
+    @ap.cmd()
+    def latest(self) -> None:
+        o, _ = subprocess.Popen(
+            [
+                sys.executable,
+                '-m', 'pip',
+                'index', 'versions',
+                ANTLR_RUNTIME_PACKAGE,
+            ],
+            stdout=subprocess.PIPE,
+        ).communicate()
+        tl = o.decode().splitlines()[0]
+        m = re.fullmatch(rf'{ANTLR_RUNTIME_PACKAGE} \((?P<version>[^)]+)\)', tl)
+        if m is None:
+            raise ValueError(f'Failed to parse version: {tl}')
+        v = m.groupdict()['version']
+        print(v)
+
+    #
+
+    @ap.cmd(
+        ap.arg('roots', nargs='+'),
+    )
+    def gen(self) -> None:
+        gp = GenPy(
+            self.args.roots,
+        )
+        gp.run()
+
+
+def _main() -> None:
+    logs.configure_standard_logging(logging.INFO)
+    cli = Cli()
+    cli()
+
+
+if __name__ == '__main__':
+    _main()

omdev/{antlr → tools/antlr}/consts.py

@@ -4,4 +4,4 @@ ANTLR_JAR_URL = f'https://www.antlr.org/download/{ANTLR_JAR_NAME}'
 ANTLR_RUNTIME_PACKAGE = 'antlr4-python3-runtime'
 ANTLR_GITHUB_REPO = 'antlr/antlr4'
 
-ANTLR_RUNTIME_VENDOR = 'omlish.antlr._runtime._all'
+ANTLR_RUNTIME_VENDOR = 'omlish.text.antlr._runtime._all'

omdev/tools/antlr/gen.py

@@ -0,0 +1,187 @@
+"""
+TODO:
+ - mtime cmp
+ - parallelism
+"""
+import logging
+import os.path
+import re
+import shutil
+import subprocess
+import typing as ta
+
+from omlish import check
+from omlish import lang
+from omlish.os.paths import is_path_in_dir
+
+from ...cache import data as dcache
+from .consts import ANTLR_JAR_URL
+from .consts import ANTLR_RUNTIME_VENDOR
+
+
+log = logging.getLogger(__name__)
+
+
+##
+
+
+ANTLR_JAR_CACHE = dcache.UrlSpec(ANTLR_JAR_URL)
+
+
+@lang.cached_function
+def get_jar_path() -> str:
+    return dcache.default().get(ANTLR_JAR_CACHE)
+
+
+##
+
+
+def _find_dirs(*base_paths: str, filter: ta.Callable[[str], bool] = lambda _: True) -> ta.Sequence[str]:  # noqa
+    return sorted(
+        os.path.join(dp, dn)
+        for base_path in base_paths
+        for dp, dns, fns in os.walk(base_path)
+        for dn in dns
+        if filter(dn)
+    )
+
+
+def _find_files(*base_paths: str, filter: ta.Callable[[str], bool] = lambda _: True) -> ta.Sequence[str]:  # noqa
+    return sorted(
+        os.path.join(dp, fn)
+        for base_path in base_paths
+        for dp, dns, fns in os.walk(base_path)
+        for fn in fns
+        if filter(fn)
+    )
+
+
+class GenPy:
+    def __init__(
+            self,
+            root_dirs: str,  # noqa
+            *,
+            out_subdir: str = '_antlr',
+            runtime_import: str = ANTLR_RUNTIME_VENDOR,
+            jar_path: str | None = None,
+            # parallelism: int | None = None,
+    ) -> None:
+        super().__init__()
+
+        check.non_empty_str(out_subdir)
+        check.arg(not os.path.isabs(out_subdir) and '.' not in out_subdir and '/' not in out_subdir)
+
+        self._root_dirs = frozenset(check.non_empty_str(rd) for rd in check.not_isinstance(root_dirs, str))
+        self._out_subdir = out_subdir
+        self._runtime_import = runtime_import
+        self._given_jar_path = jar_path
+
+    #
+
+    def _rmtree(self, tgt: str) -> None:  # noqa
+        if not any(is_path_in_dir(rd, tgt) for rd in self._root_dirs):
+            raise RuntimeError(f'Refusing to delete {tgt!r} outside of {self._root_dirs!r}')
+        shutil.rmtree(tgt)
+
+    #
+
+    @lang.cached_function
+    def jar_path(self) -> str:
+        if (gjp := self._given_jar_path) is not None:
+            return gjp
+        return get_jar_path()
+
+    #
+
+    def process_g4(self, g4_file: str) -> None:
+        ap = os.path.abspath(g4_file)
+        check.state(os.path.isfile(ap))
+
+        od = os.path.join(os.path.dirname(ap), self._out_subdir)
+        os.makedirs(od, exist_ok=True)
+
+        log.info('Compiling grammar %s', g4_file)
+
+        try:
+            subprocess.check_call([
+                'java',
+                '-jar', self.jar_path(),
+                '-Dlanguage=Python3',
+                '-visitor',
+                '-o', self._out_subdir,
+                os.path.basename(g4_file),
+            ], cwd=os.path.dirname(ap))
+
+        except Exception:  # noqa
+            log.exception('Exception in grammar %s', g4_file)
+            raise
+
+    def process_py(self, py_file: str) -> None:
+        ap = os.path.abspath(py_file)
+        with open(ap) as f:
+            in_lines = list(f)
+
+        pfp = py_file.split(os.sep)
+        arp = ANTLR_RUNTIME_VENDOR.split('.')
+        if (cpl := lang.common_prefix_len(pfp, arp)) > 0:
+            pkg_depth = len(os.path.normpath(py_file).split(os.path.sep))
+            antlr_imp = '.'.join([*([''] * (pkg_depth - cpl)), *arp[cpl:]])
+        else:
+            antlr_imp = ANTLR_RUNTIME_VENDOR
+
+        out_lines = [
+            '# type: ignore\n',
+            '# ruff: noqa\n',
+            '# flake8: noqa\n',
+        ]
+
+        for l in in_lines:
+            l = re.sub(r'^(from antlr4)(.*)', rf'from {antlr_imp}\2', l)
+            out_lines.append(l)
+
+        with open(ap, 'w') as f:
+            f.write(''.join(out_lines))
+
+    def process_dir(self, dir: str) -> None:  # noqa
+        log.info('Processing directory %s', dir)
+
+        ad = os.path.join(dir, self._out_subdir)
+        if os.path.exists(ad):
+            self._rmtree(ad)
+
+        for f in os.listdir(dir):
+            fp = os.path.join(dir, f)
+            if not os.path.isfile(fp) or not f.endswith('.g4'):
+                continue
+
+            self.process_g4(fp)
+
+        if not os.path.exists(ad):
+            return
+
+        ip = os.path.join(ad, '__init__.py')
+        check.state(not os.path.exists(ip))
+
+        for f in list(os.listdir(ad)):
+            fp = os.path.join(ad, f)
+            if not os.path.isfile(fp):
+                continue
+
+            if f.split('.')[-1] in ('interp', 'tokens'):
+                os.unlink(fp)
+
+            elif f != '__init__.py' and f.endswith('.py'):
+                self.process_py(fp)
+
+        with open(ip, 'w'):
+            pass
+
+    def run(self) -> None:
+        dns = _find_dirs(*self._root_dirs, filter=lambda dn: os.path.basename(dn) == '_antlr')
+        for dn in dns:
+            self._rmtree(dn)
+
+        fns = _find_files(*self._root_dirs, filter=lambda fn: fn.endswith('.g4'))
+        fds = {os.path.dirname(fn) for fn in fns}
+        for dn in sorted(fds):
+            self.process_dir(dn)

omdev/tools/git/cli.py

@@ -23,7 +23,6 @@ import dataclasses as dc
 import logging
 import os
 import re
-import sys
 import typing as ta
 import urllib.parse
 
@@ -380,6 +379,8 @@ class Cli(ap.Cli):
     )
     def update_submodule_branches(self) -> None:
         def run_submodule(submodule: str, cwd: str | None) -> None:
+            log.info('Updating submodule %s', submodule)
+
             submodule_path = submodule if cwd is None else os.path.join(cwd, submodule)
 
             # Get the HEAD branch from origin
@@ -401,21 +402,32 @@ class Cli(ap.Cli):
             subprocesses.check_call('git', 'checkout', head_branch, cwd=submodule_path)
             subprocesses.check_call('git', 'pull', cwd=submodule_path)
 
+        failed: set[str] = set()
+
         def run(cwd: str | None) -> None:
             submodules = subprocesses.check_output(
                 'git', 'submodule', 'foreach', '-q', 'echo $name',
                 cwd=cwd,
             ).decode().strip().splitlines()
 
-            for submodule in submodules:
+            for submodule in sorted(submodules):
                 try:
                     run_submodule(submodule, cwd)
-                except Exception as e:
+
+                except Exception:  # noqa
+                    failed.add(submodule)
+
                     if self.args.on_error_resume_next:
-                        print(e, file=sys.stderr)
+                        log.exception('Failed to update submodule %s', submodule)
                     else:
                         raise
 
+            if failed:
+                log.error(
+                    'The following submodules failed to update:\n%s',
+                    '\n'.join([f'  {f}' for f in failed]),
+                )
+
         if not self.args.dir:
             run(None)
         else:

{omdev-0.0.0.dev310.dist-info → omdev-0.0.0.dev312.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: omdev
-Version: 0.0.0.dev310
+Version: 0.0.0.dev312
 Summary: omdev
 Author: wrmsr
 License: BSD-3-Clause
@@ -12,7 +12,7 @@ Classifier: Operating System :: OS Independent
 Classifier: Operating System :: POSIX
 Requires-Python: >=3.12
 License-File: LICENSE
-Requires-Dist: omlish==0.0.0.dev310
+Requires-Dist: omlish==0.0.0.dev312
 Provides-Extra: all
 Requires-Dist: black~=25.1; extra == "all"
 Requires-Dist: pycparser~=2.22; extra == "all"

{omdev-0.0.0.dev310.dist-info → omdev-0.0.0.dev312.dist-info}/RECORD

@@ -1,4 +1,4 @@
-omdev/.manifests.json,sha256=M4fAkk-dlyvEzkya4HhuU9Hb9QpVYgp1AgJB6eXgmuI,10982
+omdev/.manifests.json,sha256=upvKAe9ZFn6SkZusjneWtBKlCoEdy7WJSRP2Kq_ION0,11257
 omdev/__about__.py,sha256=16xa_1BdZanTpZbkjAOQ11_x5kJcb1m1tKdvb06J7VI,1202
 omdev/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 omdev/cmake.py,sha256=9rfSvFHPmKDj9ngvfDB2vK8O-xO_ZwUm7hMKLWA-yOw,4578
@@ -16,9 +16,6 @@ omdev/amalg/srcfiles.py,sha256=TKSYSfv78Wb8bcsQS3V76dV_H_MICPwzv06Ar09ew9U,3079
 omdev/amalg/strip.py,sha256=dWQQ5WbtcebLi_PGjPzVYey2mJOnrRES8rkoUS8L52w,1444
 omdev/amalg/types.py,sha256=BXXJI0VctKTsZv_wXiyMMq3-xShxZ1ak0wxXUK8n9_g,89
 omdev/amalg/typing.py,sha256=oLlkCnnZQkyKM_xxqm9uxECjn9xj0c4MIwYxxItBdPY,2191
-omdev/antlr/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-omdev/antlr/consts.py,sha256=6iKuncbRXnGAqemNAceaR2mdFSs-8VbBV-pluM-8Bu4,284
-omdev/antlr/gen.py,sha256=ZuDoh8ksIGlTSufEnJXogpk7gRVfmPirERdLx2nMWDo,3020
 omdev/cache/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 omdev/cache/compute/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 omdev/cache/compute/cache.py,sha256=FA6tZWoQpE8D8Os3abdtfT90juzkqBmESjJE3WQXZtQ,3623
@@ -204,7 +201,8 @@ omdev/precheck/lite.py,sha256=rG6NYg0X85eN5UQ8-S7fagP5Ng2Nxwuf5Uh7uI8nQsE,4716
 omdev/precheck/main.py,sha256=VEPdq9n8Yo_HiDqfCdHH1pj3xuJgOzrXie_vWoNbKqw,4303
 omdev/precheck/manifests.py,sha256=4V4_mMh5ptW0jWBmq0nG3bsx3dyjejVs1ReEUoOz-JA,784
 omdev/precheck/scripts.py,sha256=Br163NGbLY6Mjkxr9htdzqqsYWiEJbg9Gq-Glz6YQUI,1342
-omdev/ptk/__init__.py,sha256=snOGUkwTyItkQ-iWNNkFO-ZXDJKt3LtcCcX6-CmzEuw,5038
+omdev/ptk/__init__.py,sha256=IemfBhxxqZe1GeZ2_IMQzqWAB5j0XyARp_aajnonyKY,5142
+omdev/ptk/confirm.py,sha256=kObMUAu-EZC0vdT9LLwtbNA6YLLNmn-Uy18SQTWBTb8,1471
 omdev/ptk/apps/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 omdev/ptk/apps/ncdu.py,sha256=KTbAgwhzcROOvq20VGN92UwMWcgkMMVOeFfpjZAsKUk,4561
 omdev/ptk/markdown/LICENSE,sha256=oSmc9j-n23wTJUO0TbY8sIHrf9pFZUovFWfDL7D53IA,1489
@@ -212,7 +210,7 @@ omdev/ptk/markdown/__init__.py,sha256=Mj0KTyzdGVhkp79QoUiS6iJjnDz8c9chygcbVIgm1d
 omdev/ptk/markdown/__main__.py,sha256=zaZuKy3-llEztkpoIAIzvspOyCsTJQI33ZuadvSHJKM,181
 omdev/ptk/markdown/border.py,sha256=4gWpwTimurgJ_MjE-UKMCDMb4DnVcgFPfXutObQWOxk,1726
 omdev/ptk/markdown/cli.py,sha256=FlV12qpyVWgpfeOzyFg9DlgLkss3rgCet_Jebs6_Xqo,537
-omdev/ptk/markdown/markdown.py,sha256=B45hRwgX46z5fJFUsJ2jCVn6kc_vpwPjRjLDdILKlfA,12278
+omdev/ptk/markdown/markdown.py,sha256=wRxCWSV2u71NP_MTi8A8vVLjZ1TmmaWUZMnvW468Ejc,12368
 omdev/ptk/markdown/parser.py,sha256=UppwouvAYh3qzQMKL-BjcyqBIl2KHcocXlWiKFZ7cBA,990
 omdev/ptk/markdown/styles.py,sha256=lc17zooXhff5_2tkqLsCmdq2b_rfSAehmHVR-8Lo2qk,777
 omdev/ptk/markdown/tags.py,sha256=askGU252Zu8KxG2menVHZHXZ4fGbItgy0G4UW9F3mFI,6983
@@ -266,9 +264,14 @@ omdev/tools/prof.py,sha256=hQakAsViJD4gLJpLLZnTkOqmTDAwM48Nx5q-O_aFlYM,1467
 omdev/tools/qr.py,sha256=tm68lPwEAkEwIL2sUKPKBYfwwPtjVWG1DBZwur8_jY8,1737
 omdev/tools/shadow.py,sha256=4E2ilxa16liIvQxvgU37ITkOMrP6ufShRQfeW7wwtRc,1697
 omdev/tools/sqlrepl.py,sha256=wAjrfXNrRV63-NJCC2HlGQnFh7lUH0bHMnOjYotQqFs,5753
+omdev/tools/antlr/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+omdev/tools/antlr/__main__.py,sha256=7db8YIfCp4GoU-lW7jJJDiLQ6dDrd25M7gUv1AHGj4w,170
+omdev/tools/antlr/cli.py,sha256=QeO5vSUJDL43lAa2EYsLnN-jZBcNOTwB1C9Ua_h7qoA,1228
+omdev/tools/antlr/consts.py,sha256=4xH4jyNE5czXRWCn65jFF0MrAodMPe_kYMWpgMxWmpo,289
+omdev/tools/antlr/gen.py,sha256=0k2o96gye00iMWlC7l6mVjUoYo9lfouGPBmA7_li23U,5085
 omdev/tools/git/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 omdev/tools/git/__main__.py,sha256=gI87SBUgTkKUcUM-RtZWnei-UUDDqzbr5aPztb-gvbE,168
-omdev/tools/git/cli.py,sha256=HOQX9VDvlzHrsoC8Jh-sVBBPP1IZit-Us0s_imkbuNc,14484
+omdev/tools/git/cli.py,sha256=NL30UkmWWoepNCClzeJTHwka3uGgPzGlsao_RDqVOek,14850
 omdev/tools/git/consts.py,sha256=JuXivUNDkNhM4pe97icjRVAKM8cNRbrODquHINNKqOE,40
 omdev/tools/git/messages.py,sha256=NWztIK0nAKJIOVzuVQcR_5LHZUgqyVkrOlpl7dFLMdU,2424
 omdev/tools/json/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -282,9 +285,9 @@ omdev/tools/json/rendering.py,sha256=3HhdlKSetS6iK1tjF2aILzsl8Mb3D8wW92vYwGpRdVA
 omdev/tools/pawk/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 omdev/tools/pawk/__main__.py,sha256=VCqeRVnqT1RPEoIrqHFSu4PXVMg4YEgF4qCQm90-eRI,66
 omdev/tools/pawk/pawk.py,sha256=zsEkfQX0jF5bn712uqPAyBSdJt2dno1LH2oeSMNfXQI,11424
-omdev-0.0.0.dev310.dist-info/licenses/LICENSE,sha256=B_hVtavaA8zCYDW99DYdcpDLKz1n3BBRjZrcbv8uG8c,1451
-omdev-0.0.0.dev310.dist-info/METADATA,sha256=1vFppFYXizN650neIH42bBGsgdTgQSxyoNI9c8-D738,1674
-omdev-0.0.0.dev310.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
-omdev-0.0.0.dev310.dist-info/entry_points.txt,sha256=dHLXFmq5D9B8qUyhRtFqTGWGxlbx3t5ejedjrnXNYLU,33
-omdev-0.0.0.dev310.dist-info/top_level.txt,sha256=1nr7j30fEWgLYHW3lGR9pkdHkb7knv1U1ES1XRNVQ6k,6
-omdev-0.0.0.dev310.dist-info/RECORD,,
+omdev-0.0.0.dev312.dist-info/licenses/LICENSE,sha256=B_hVtavaA8zCYDW99DYdcpDLKz1n3BBRjZrcbv8uG8c,1451
+omdev-0.0.0.dev312.dist-info/METADATA,sha256=d0-7zTLSAmgIA9Rori0icQcE8n6xnia9N8IrOXkdVKw,1674
+omdev-0.0.0.dev312.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
+omdev-0.0.0.dev312.dist-info/entry_points.txt,sha256=dHLXFmq5D9B8qUyhRtFqTGWGxlbx3t5ejedjrnXNYLU,33
+omdev-0.0.0.dev312.dist-info/top_level.txt,sha256=1nr7j30fEWgLYHW3lGR9pkdHkb7knv1U1ES1XRNVQ6k,6
+omdev-0.0.0.dev312.dist-info/RECORD,,

omdev/antlr/gen.py

@@ -1,114 +0,0 @@
-"""
-TODO:
- - fix relpath in omlish/
- - sem-bounded parallelism
-"""
-import argparse
-import os.path
-import re
-import shutil
-import subprocess
-
-from omlish import check
-from omlish import lang
-
-from ..cache import data as dcache
-from .consts import ANTLR_JAR_URL
-from .consts import ANTLR_RUNTIME_VENDOR
-
-
-ANTLR_JAR_CACHE = dcache.UrlSpec(ANTLR_JAR_URL)
-
-
-class GenPy:
-    def __init__(
-            self,
-            dir: str,  # noqa
-            *,
-            out_subdir: str = '_antlr',
-            runtime_import: str = ANTLR_RUNTIME_VENDOR,
-    ) -> None:
-        super().__init__()
-        check.arg(not os.path.isabs(out_subdir) and '..' not in out_subdir)
-        self._dir = dir
-        self._out_subdir = out_subdir
-        self._runtime_import = runtime_import
-        self._out_dir = os.path.join(dir, out_subdir)
-
-    @lang.cached_function
-    def jar(self) -> str:
-        return dcache.default().get(ANTLR_JAR_CACHE)
-
-    def process_g4(self, g4_file: str) -> None:
-        subprocess.check_call([
-            'java',
-            '-jar', self.jar(),
-            '-Dlanguage=Python3',
-            '-visitor',
-            '-o', self._out_subdir,
-            g4_file,
-        ], cwd=self._dir)
-
-    def process_py(self, py_file: str) -> None:
-        ap = os.path.join(self._out_dir, py_file)
-        with open(ap) as f:
-            in_lines = list(f)
-
-        out_lines = [
-            '# type: ignore\n',
-            '# ruff: noqa\n',
-            '# flake8: noqa\n',
-        ]
-        for l in in_lines:
-            l = re.sub(r'^(from antlr4)(.*)', rf'from {self._runtime_import}\2', l)
-            out_lines.append(l)
-
-        with open(ap, 'w') as f:
-            f.write(''.join(out_lines))
-
-    def run(self) -> None:
-        if os.path.exists(self._out_dir):
-            shutil.rmtree(self._out_dir)
-        os.mkdir(self._out_dir)
-        with open(os.path.join(self._out_dir, '__init__.py'), 'w'):
-            pass
-
-        for f in os.listdir(self._dir):
-            fp = os.path.join(self._dir, f)
-            if not os.path.isfile(fp):
-                continue
-            if f.endswith('.g4'):
-                self.process_g4(f)
-
-        for f in list(os.listdir(self._out_dir)):
-            fp = os.path.join(self._out_dir, f)
-            if not os.path.isfile(fp):
-                continue
-            if f.split('.')[-1] in ('interp', 'tokens'):
-                os.unlink(fp)
-            elif f != '__init__.py' and f.endswith('.py'):
-                self.process_py(f)
-
-
-def _main() -> None:
-    parser = argparse.ArgumentParser()
-    parser.add_argument('roots', nargs='*')
-    args = parser.parse_args()
-
-    if not args.roots:
-        parser.print_help()
-        return
-
-    base_dir = os.getcwd()
-    if not os.path.isfile(os.path.join(base_dir, 'pyproject.toml')):
-        raise RuntimeError('Must run from project root')
-
-    for root_dir in args.roots:
-        print(f'Processing {root_dir}')
-        GenPy(
-            root_dir,
-        ).run()
-
-
-if __name__ == '__main__':
-    _main()