Merge pull request #1 from Adwaith-Rajesh/target

Target

Author: Adwaith-Rajesh

.pre-commit-config.yaml

@@ -18,6 +18,7 @@ repos:
     rev: v2.0.4
     hooks:
       - id: autopep8
+        args: ["--max-line-length", "120", "--in-place"]
   - repo: https://github.com/asottile/reorder_python_imports
     rev: v3.10.0
     hooks:

buildme/__init__.py

@@ -1,2 +1,4 @@
 from buildme.core import CommandRunner  # noqa: F401
+from buildme.core import target  # noqa: F401
+from buildme.core import TargetData  # noqa: F401
 from buildme.hfuncs import *  # noqa: F401 F403

buildme/cli.py

@@ -1,4 +1,10 @@
 import argparse
+import sys
+from itertools import compress
+from typing import Any
+
+from buildme.core import _check_target_exists
+from buildme.core import _exec_target
 
 
 def _get_buildme_file_contents(filepath: str) -> str:
@@ -10,6 +16,10 @@ def _create_build_script_code(old_code: str, targets: list[str], usr_opt_var_nam
     return old_code + ''.join(f'\n{t}({usr_opt_var_name})' for t in targets)
 
 
+def _check_target_exists_map(targets: list[str]) -> list[bool]:
+    return [_check_target_exists(name) for name in targets]
+
+
 def main() -> int:
     parser = argparse.ArgumentParser()
     user_arg_parser = argparse.ArgumentParser()
@@ -25,18 +35,19 @@ def main() -> int:
             user_arg_parser.add_argument(a.split('=')[0], type=str)
 
     usr_known_args, _ = user_arg_parser.parse_known_args()
-    usr_known_args_var_name = f'{usr_known_args=}'.split(
-        '=')[0]  # gets the var name as string
-
-    new_buildme_code = _create_build_script_code(
-        _get_buildme_file_contents(args.path),
-        args.targets,
-        usr_known_args_var_name
-    )
-
-    # the dangerous part
-    exec(new_buildme_code, {
-         usr_known_args_var_name: usr_known_args, **globals()})
+    # gets the var name as string
+    usr_known_args_var_name = f'{usr_known_args=}'.split('=')[0]  # noqa: F841
+
+    target_globals: dict[str, Any] = {}
+    exec(_get_buildme_file_contents(args.path), target_globals)
+
+    if not all(invalid_map := _check_target_exists_map(targets=args.targets)):
+        print('unknown target(s): ' + ' '.join(compress(args.targets, [not i for i in invalid_map])),
+              file=sys.stderr)
+        return 1
+
+    for t in args.targets:
+        _exec_target(t, usr_known_args, target_globals)
 
     return 0
 

buildme/core.py

@@ -1,5 +1,11 @@
 import shlex
+import sys
+from argparse import Namespace
+from functools import wraps
 from subprocess import Popen
+from typing import Any
+from typing import Callable
+from typing import NamedTuple
 
 
 class CommandRunner:
@@ -11,7 +17,7 @@ def __init__(self, shell: bool = False, print_cmd: bool = True,
         self.print_cmd_sep = print_cmd_sep
 
     def run(self, cmd: str) -> int:
-        if self.print_cmd:
+        if self.print_cmd_sep:
             print('=' * 80)
 
         if self.print_cmd:
@@ -20,15 +26,56 @@ def run(self, cmd: str) -> int:
         p = Popen(args=shlex.split(cmd), shell=self.shell)
         p.wait()
 
-        if self.print_cmd:
+        if self.print_cmd_sep:
             print('=' * 80)
 
         if self.exit_non_zero and p.returncode != 0:
-            print(
-                f'Last command exited with non zero exit code. [{p.returncode}]')
+            print(f'Last command exited with non zero exit code. [{p.returncode}]')
             exit(1)
 
         return p.returncode
 
     def set_exit_non_zero(self, val: bool) -> None:
         self.exit_non_zero = val
+
+
+class TargetData(NamedTuple):
+    name: str
+    depends: list[str] = []
+
+
+TargetFuncType = Callable[[Namespace, TargetData], None]
+WrapTargetFuncType = Callable[[Namespace], None]
+
+_targets = {}
+
+
+def target(depends: list[str] = []) -> Callable[[TargetFuncType], WrapTargetFuncType]:
+    def target_dec(fn: TargetFuncType) -> WrapTargetFuncType:
+        _targets[fn.__name__] = TargetData(name=fn.__name__, depends=depends)
+
+        @wraps(fn)
+        def target_wrap(opts: Namespace) -> None:
+            fn(opts, _targets[fn.__name__])
+        return target_wrap
+    return target_dec
+
+
+def _get_target_data(name: str) -> TargetData | None:
+    return _targets.get(name, None)
+
+
+def _check_target_exists(name: str) -> bool: return name in _targets
+
+
+def _exec_target(name: str, opts: Namespace, target_globals: dict[str, Any]) -> None:
+    if name not in _targets:
+        print(f'unknown target: {name}', file=sys.stderr)
+        exit(1)
+
+    if callable(fn := target_globals[name]):
+        t_data = _get_target_data(name)
+        if t_data:
+            for d in t_data.depends:
+                _exec_target(d, opts, target_globals)
+        fn(opts)

buildme/hfuncs.py

@@ -1,4 +1,6 @@
 # helper funcs
+import glob
+import shutil
 from pathlib import Path
 
 
@@ -10,4 +12,18 @@ def touch(filepath: str) -> None:
     Path(filepath).touch()
 
 
+def rmdir(path: str) -> None:
+    if Path(path).is_dir():
+        shutil.rmtree(path)
+
+
+def rm(path: str) -> None:
+    Path(path).unlink(missing_ok=True)
+
+
+def get_files_in_dir(dir_pattern: str, recurse: bool = False,
+                     include_hidden: bool = False) -> list[str]:
+    return glob.glob(dir_pattern, recursive=recurse, include_hidden=include_hidden)
+
+
 create_file = touch

setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = buildme
-version = 0.1.0
+version = 0.2.0
 author = Adwaith Rajesh
 author_email = adwaithrajesh3180@gmail.com
 description = A simple command runner disguised as a build system.