jaclang 0.4.7__py3-none-any.whl → 0.5.0__py3-none-any.whl

jaclang/__init__.py

@@ -1,20 +1,23 @@
 """The Jac Programming Language."""
+
 import os
 import sys
 
 sys.path.insert(0, os.path.join(os.path.dirname(os.path.abspath(__file__)), "vendor"))
 
-from jaclang.compiler.importer import jac_import  # noqa: E402
+from jaclang.plugin.default import JacFeatureDefaults  # noqa: E402
 from jaclang.plugin.feature import JacFeature  # noqa: E402
 from jaclang.vendor import lark  # noqa: E402
 from jaclang.vendor import mypy  # noqa: E402
 from jaclang.vendor import pluggy  # noqa: E402
 
+jac_import = JacFeature.jac_import
+
 __all__ = [
     "jac_import",
     "lark",
     "mypy",
     "pluggy",
 ]
-
+JacFeature.pm.register(JacFeatureDefaults)
 JacFeature.pm.load_setuptools_entrypoints("jac")

jaclang/cli/cli.py

@@ -1,17 +1,19 @@
 """Command line interface tool for the Jac language."""
+
 import os
+import pickle
 import shutil
 from typing import Optional
 
-from jaclang import jac_import as __jac_import__
+from jaclang import jac_import
 from jaclang.cli.cmdreg import CommandRegistry, CommandShell
+from jaclang.compiler.compile import jac_file_to_pass
 from jaclang.compiler.constant import Constants
+from jaclang.compiler.passes.main.schedules import py_code_gen_typed
 from jaclang.compiler.passes.tool.schedules import format_pass
-from jaclang.compiler.transpiler import jac_file_to_pass
 from jaclang.plugin.feature import JacFeature as Jac
 from jaclang.utils.lang_tools import AstTool
 
-
 cmd_registry = CommandRegistry()
 
 
@@ -45,13 +47,59 @@ def run(filename: str, main: bool = True) -> None:
     :param filename: The path to the .jac file.
     :param main: If True, use '__main__' as the module name, else use the actual module name.
     """
+    base, mod = os.path.split(filename)
+    base = base if base else "./"
+    mod = mod[:-4]
     if filename.endswith(".jac"):
-        base, mod = os.path.split(filename)
-        base = base if base else "./"
-        mod = mod[:-4]
-        __jac_import__(
+        jac_import(
             target=mod, base_path=base, override_name="__main__" if main else None
         )
+    elif filename.endswith(".jir"):
+        with open(filename, "rb") as f:
+            ir = pickle.load(f)
+            jac_import(
+                target=mod,
+                base_path=base,
+                override_name="__main__" if main else None,
+                mod_bundle=ir,
+            )
+    else:
+        print("Not a .jac file.")
+
+
+@cmd_registry.register
+def build(filename: str) -> None:
+    """Build the specified .jac file."""
+    if filename.endswith(".jac"):
+        out = jac_file_to_pass(file_path=filename, schedule=py_code_gen_typed)
+        errs = len(out.errors_had)
+        warnings = len(out.warnings_had)
+        print(f"Errors: {errs}, Warnings: {warnings}")
+        for i in out.ir.flatten():
+            i.gen.mypy_ast = []
+            i.gen.py = ""
+        with open(filename[:-4] + ".jir", "wb") as f:
+            pickle.dump(out.ir, f)
+    else:
+        print("Not a .jac file.")
+
+
+@cmd_registry.register
+def check(filename: str) -> None:
+    """Run type checker for a specified .jac file.
+
+    :param filename: The path to the .jac file.
+    """
+    if filename.endswith(".jac"):
+        out = jac_file_to_pass(
+            file_path=filename,
+            schedule=py_code_gen_typed,
+        )
+
+        errs = len(out.errors_had)
+        warnings = len(out.warnings_had)
+
+        print(f"Errors: {errs}, Warnings: {warnings}")
     else:
         print("Not a .jac file.")
 
@@ -68,7 +116,7 @@ def enter(filename: str, entrypoint: str, args: list) -> None:
         base, mod_name = os.path.split(filename)
         base = base if base else "./"
         mod_name = mod_name[:-4]
-        mod = __jac_import__(target=mod_name, base_path=base)
+        mod = jac_import(target=mod_name, base_path=base)
         if not mod:
             print("Errors occurred while importing the module.")
             return

jaclang/cli/cmdreg.py

@@ -1,4 +1,5 @@
 """Common code for command line interface tool for the Jac language."""
+
 from __future__ import annotations
 
 import argparse
@@ -81,18 +82,22 @@ class CommandRegistry:
                     first = False
                     cmd_parser.add_argument(
                         f"{param_name}",
-                        type=eval(param.annotation)
-                        if isinstance(param.annotation, str)
-                        else param.annotation,
+                        type=(
+                            eval(param.annotation)
+                            if isinstance(param.annotation, str)
+                            else param.annotation
+                        ),
                     )
                 else:
                     cmd_parser.add_argument(
                         f"-{param_name[:1]}",
                         f"--{param_name}",
                         required=True,
-                        type=eval(param.annotation)
-                        if isinstance(param.annotation, str)
-                        else param.annotation,
+                        type=(
+                            eval(param.annotation)
+                            if isinstance(param.annotation, str)
+                            else param.annotation
+                        ),
                     )
             elif first:
                 first = False
@@ -104,9 +109,11 @@ class CommandRegistry:
                     f"-{param_name[:1]}",
                     f"--{param_name}",
                     default=param.default,
-                    type=eval(param.annotation)
-                    if isinstance(param.annotation, str)
-                    else param.annotation,
+                    type=(
+                        eval(param.annotation)
+                        if isinstance(param.annotation, str)
+                        else param.annotation
+                    ),
                 )
         return func