omdev 0.0.0.dev305__py3-none-any.whl → 0.0.0.dev306__py3-none-any.whl

omdev/.manifests.json

@@ -47,6 +47,18 @@
       }
     }
   },
+  {
+    "module": ".cmdlog.__main__",
+    "attr": "_CLI_MODULE",
+    "file": "omdev/cmdlog/__main__.py",
+    "line": 4,
+    "value": {
+      "$.cli.types.CliModule": {
+        "cmd_name": "cmdlog",
+        "mod_name": "omdev.cmdlog.__main__"
+      }
+    }
+  },
   {
     "module": ".imgur",
     "attr": "_FOO_CLI_MODULE",

omdev/cmdlog/__init__.py

@@ -0,0 +1,7 @@
+from ._cmdlog import (  # noqa
+    CmdLogEntry,
+)
+
+from .cmdlog import (  # noqa
+    CmdLog,
+)

omdev/cmdlog/__main__.py

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

omdev/cmdlog/_cmdlog.py

@@ -0,0 +1,73 @@
+# NOTE: not '@omlish-lite' due to root level __init__ imports, but effectively lite.
+import dataclasses as dc
+import fcntl
+import json
+import os.path
+import shutil
+import sys
+import time
+import typing as ta
+
+
+##
+
+
+@dc.dataclass(frozen=True)
+class CmdLogEntry:
+    cmd: str
+    pid: int
+    ppid: int
+    time: float
+    cwd: str
+
+    argv: ta.Sequence[str]
+    env: ta.Mapping[str, str]
+
+
+##
+
+
+LOG_FILE_ENV_VAR: str = '_CMD_LOG_FILE'
+
+
+def _main() -> None:
+    cmd = os.path.basename(sys.argv[0])
+
+    entry = CmdLogEntry(
+        cmd=cmd,
+        pid=os.getpid(),
+        ppid=os.getppid(),
+        time=time.time(),
+        cwd=os.getcwd(),
+
+        argv=sys.argv,
+        env=dict(os.environ),
+    )
+
+    entry_json = json.dumps(dc.asdict(entry), separators=(',', ':'), indent=None)
+
+    log_file = os.environ[LOG_FILE_ENV_VAR]
+    fd = os.open(log_file, os.O_WRONLY | os.O_CREAT | os.O_APPEND)
+    fcntl.flock(fd, fcntl.LOCK_EX)
+    os.write(fd, entry_json.encode() + b'\n')
+    os.close(fd)
+
+    u_ap = None  # type: str | None
+    self_ap = os.path.abspath(os.path.realpath(__file__))
+    for p in os.environ.get('PATH', '').split(os.pathsep):
+        if (p_cmd := shutil.which(cmd, path=p)) is None:
+            continue
+        p_ap = os.path.abspath(os.path.realpath(p_cmd))
+        if p_ap == self_ap:
+            continue
+        u_ap = p_ap
+        break
+
+    if u_ap is None:
+        raise FileNotFoundError(cmd)
+
+    os.execl(u_ap, cmd, *sys.argv[1:])
+
+
+if __name__ == '__main__':
+    _main()

omdev/cmdlog/cli.py

@@ -0,0 +1,67 @@
+import argparse
+import os.path
+import subprocess
+import sys
+import typing as ta
+
+from omlish.formats import json
+
+from .cmdlog import CmdLog
+
+
+##
+
+
+DEFAULT_PROXIED_CMDS: ta.Collection[str] = [
+    'ar',
+    'as',
+    'clang',
+    'clang++',
+    'g++',
+    'gcc',
+    'ld',
+    'lld',
+    'make',
+]
+
+
+def _main() -> None:
+    parser = argparse.ArgumentParser()
+    parser.add_argument('cmd', nargs=argparse.REMAINDER)
+    parser.add_argument('-l', '--log-file')
+    parser.add_argument('-P', '--print', action='store_true')
+    parser.add_argument('-p', '--proxy', action='append')
+    args = parser.parse_args()
+
+    if not args.cmd:
+        parser.error('Must specify cmd')
+        raise RuntimeError  # noqa
+
+    exec_cmd, *exec_argv = args.cmd
+
+    #
+
+    cl = CmdLog(
+        args.proxy if args.proxy is not None else DEFAULT_PROXIED_CMDS,
+        log_file=os.path.abspath(args.log_file) if args.log_file is not None else None,
+    )
+    cl.exe()
+    cl.proxy_cmds()
+
+    rc = subprocess.call(
+        [exec_cmd, *exec_argv],
+        env=cl.child_env(),
+    )
+
+    if args.log_file is None or args.print:
+        if os.path.exists(cl.log_file()):
+            with open(cl.log_file()) as f:
+                log_lines = f.readlines()
+            entry_dcts = [json.loads(sl) for l in log_lines if (sl := l.strip())]
+            print(json.dumps_compact(entry_dcts))
+
+    sys.exit(rc)
+
+
+if __name__ == '__main__':
+    _main()

omdev/cmdlog/cmdlog.py

@@ -0,0 +1,86 @@
+import inspect
+import os.path
+import sys
+import tempfile
+import typing as ta
+
+from omlish import cached
+from omlish import check
+
+from . import _cmdlog
+
+
+##
+
+
+class CmdLog:
+    def __init__(
+            self,
+            proxied_cmds: ta.Iterable[str],
+            *,
+            tmp_dir: str | None = None,
+            log_file: str | None = None,
+            exe_name: str | None = None,
+    ) -> None:
+        super().__init__()
+
+        self._proxied_cmds = {
+            check.non_empty_str(c)
+            for c in check.not_isinstance(proxied_cmds, str)
+        }
+
+        self._given_tmp_dir = tmp_dir
+        self._given_log_file = log_file
+        self._exe_name = check.non_empty_str(exe_name) if exe_name is not None else self.DEFAULT_EXE_NAME
+
+    #
+
+    @cached.function
+    def tmp_dir(self) -> str:
+        if (gtd := self._given_tmp_dir) is not None:
+            return gtd
+        return tempfile.mkdtemp()
+
+    #
+
+    DEFAULT_LOG_FILE_NAME: ta.ClassVar[str] = '_cmdlog.jsonl'
+
+    @cached.function
+    def log_file(self) -> str:
+        if (glf := self._given_log_file) is not None:
+            return glf
+        return os.path.join(self.tmp_dir(), self.DEFAULT_LOG_FILE_NAME)
+
+    #
+
+    DEFAULT_EXE_NAME: ta.ClassVar[str] = '_cmdlog.py'
+
+    @cached.function
+    def exe(self) -> str:
+        src = '\n'.join([
+            f'#!{os.path.abspath(sys.executable)}',
+            inspect.getsource(_cmdlog),
+        ])
+
+        exe = os.path.join(self.tmp_dir(), self._exe_name)
+
+        with open(exe, 'w') as f:
+            f.write(src)
+
+        os.chmod(exe, 0o550)  # noqa
+
+        return exe
+
+    #
+
+    def proxy_cmds(self) -> None:
+        for p_c in self._proxied_cmds:
+            os.symlink(self._exe_name, os.path.join(self.tmp_dir(), p_c))
+
+    #
+
+    def child_env(self) -> ta.Mapping[str, str]:
+        return {
+            _cmdlog.LOG_FILE_ENV_VAR: self.log_file(),
+            'PATH': os.pathsep.join([self.tmp_dir(), os.environ['PATH']]),
+        }

{omdev-0.0.0.dev305.dist-info → omdev-0.0.0.dev306.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: omdev
-Version: 0.0.0.dev305
+Version: 0.0.0.dev306
 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.dev305
+Requires-Dist: omlish==0.0.0.dev306
 Provides-Extra: all
 Requires-Dist: black~=25.1; extra == "all"
 Requires-Dist: pycparser~=2.22; extra == "all"

{omdev-0.0.0.dev305.dist-info → omdev-0.0.0.dev306.dist-info}/RECORD

@@ -1,4 +1,4 @@
-omdev/.manifests.json,sha256=ouzCsEt6dHvLXr-DyZ-gLxbnizOFe0BE6u6xKH3RKEI,10721
+omdev/.manifests.json,sha256=M4fAkk-dlyvEzkya4HhuU9Hb9QpVYgp1AgJB6eXgmuI,10982
 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
@@ -114,6 +114,11 @@ omdev/clipboard/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 omdev/clipboard/clipboard.py,sha256=9HFpcijpn0XDTI89ZRm2WA1G7O4HsTdVXZHqMULu3N0,1630
 omdev/clipboard/darwin_cf.py,sha256=1gFkxEN6w9HTcA0eiWw14oL7xjC1wSmY_hUrRr5OU4Y,7687
 omdev/clipboard/linux_x11.py,sha256=oa-mxMRNaZJOdBAZ8Nki-CAGIb63X8OFUTXKjmiwfSo,6718
+omdev/cmdlog/__init__.py,sha256=026o1leEXU5bsNZPpEOeLf_Ml1wQXzbiFDmlcKHX-Nw,95
+omdev/cmdlog/__main__.py,sha256=m31h6AcI9rjRNVeBGoLcR-5pWp-yS8LXCorf4iBhX9w,162
+omdev/cmdlog/_cmdlog.py,sha256=9VSuUKXBMBHAH3OfBCeqp16YPddPT9Man2zDHgzzCtI,1507
+omdev/cmdlog/cli.py,sha256=9AJC3xeQA-2pdmTnsjo__9WRAML1jpEJUJMn6qQQpJA,1427
+omdev/cmdlog/cmdlog.py,sha256=MJqfCG7sVWjSK_i1shD7cgWpFZXZkPvGhGEh-yd6iwM,1982
 omdev/dataserver/__init__.py,sha256=Y3l4WY4JRi2uLG6kgbGp93fuGfkxkKwZDvhsa0Rwgtk,15
 omdev/dataserver/handlers.py,sha256=rCptrmV2RnutmGE5MAgjLDYW1QncqsSHhyRBL4H5bsg,5440
 omdev/dataserver/http.py,sha256=SozI4233RP40m2EX45HtBvdIj-CBFlhHH7ekIbMir-Q,1724
@@ -277,9 +282,9 @@ omdev/tools/json/rendering.py,sha256=tMcjOW5edfozcMSTxxvF7WVTsbYLoe9bCKFh50qyaGw
 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.dev305.dist-info/licenses/LICENSE,sha256=B_hVtavaA8zCYDW99DYdcpDLKz1n3BBRjZrcbv8uG8c,1451
-omdev-0.0.0.dev305.dist-info/METADATA,sha256=Dp_xG1ayIA3fP9VGM69G_Xa_oVLFp24uVQFOSMaNQew,1674
-omdev-0.0.0.dev305.dist-info/WHEEL,sha256=wXxTzcEDnjrTwFYjLPcsW_7_XihufBwmpiBeiXNBGEA,91
-omdev-0.0.0.dev305.dist-info/entry_points.txt,sha256=dHLXFmq5D9B8qUyhRtFqTGWGxlbx3t5ejedjrnXNYLU,33
-omdev-0.0.0.dev305.dist-info/top_level.txt,sha256=1nr7j30fEWgLYHW3lGR9pkdHkb7knv1U1ES1XRNVQ6k,6
-omdev-0.0.0.dev305.dist-info/RECORD,,
+omdev-0.0.0.dev306.dist-info/licenses/LICENSE,sha256=B_hVtavaA8zCYDW99DYdcpDLKz1n3BBRjZrcbv8uG8c,1451
+omdev-0.0.0.dev306.dist-info/METADATA,sha256=fiHu3utIwUEtAURhnU-PYrXupHEa1PloV_2uGgqozvk,1674
+omdev-0.0.0.dev306.dist-info/WHEEL,sha256=wXxTzcEDnjrTwFYjLPcsW_7_XihufBwmpiBeiXNBGEA,91
+omdev-0.0.0.dev306.dist-info/entry_points.txt,sha256=dHLXFmq5D9B8qUyhRtFqTGWGxlbx3t5ejedjrnXNYLU,33
+omdev-0.0.0.dev306.dist-info/top_level.txt,sha256=1nr7j30fEWgLYHW3lGR9pkdHkb7knv1U1ES1XRNVQ6k,6
+omdev-0.0.0.dev306.dist-info/RECORD,,