jaclang 0.7.2__py3-none-any.whl → 0.7.8__py3-none-any.whl

jaclang/compiler/codeloc.py

@@ -55,7 +55,7 @@ class CodeLocInfo:
     @property
     def last_line(self) -> int:
         """Get line number."""
-        return self.last_tok.line_no
+        return self.last_tok.end_line
 
     @property
     def col_start(self) -> int:
@@ -107,4 +107,4 @@ class CodeLocInfo:
 
     def __str__(self) -> str:
         """Stringify."""
-        return f"{self.first_tok.line_no}:{self.first_tok.c_start} - {self.last_tok.line_no}:{self.last_tok.c_end}"
+        return f"{self.first_line}:{self.col_start} - {self.last_line}:{self.col_end}"

jaclang/compiler/constant.py

@@ -1,9 +1,93 @@
 """Constants across the project."""
 
-from enum import Enum
+from enum import Enum, IntEnum, IntFlag, StrEnum
 
 
-class Constants(str, Enum):
+class SymbolType(Enum):
+    """Symbol types."""
+
+    MODULE = "module"  # LSP: Module
+    MOD_VAR = "mod_var"  # LSP: Variable
+    VAR = "variable"  # LSP: Variable
+    IMM_VAR = "immutable"  # LSP: Constant
+    ABILITY = "ability"  # LSP: Function
+    OBJECT_ARCH = "object"  # LSP: Class
+    NODE_ARCH = "node"  # LSP: Class
+    EDGE_ARCH = "edge"  # LSP: Class
+    WALKER_ARCH = "walker"  # LSP: Class
+    ENUM_ARCH = "enum"  # LSP: Enum
+    TEST = "test"  # LSP: Function
+    TYPE = "type"  # LSP: TypeParameter
+    IMPL = "impl"  # LSP: Interface or Property
+    HAS_VAR = "field"  # LSP: Field
+    METHOD = "method"  # LSP: Method
+    CONSTRUCTOR = "constructor"  # LSP: Constructor
+    ENUM_MEMBER = "enum_member"  # LSP: EnumMember
+    NUMBER = "number"  # LSP: Number
+    STRING = "string"  # LSP: String
+    BOOL = "bool"  # LSP: Boolean
+    SEQUENCE = "sequence"  # LSP: Array
+    NULL = "null"  # LSP: Null
+    UNKNOWN = "unknown"  # LSP: Unknown
+
+    def __str__(self) -> str:
+        """Stringify."""
+        return self.value
+
+
+class JacSemTokenType(IntEnum):
+    """LSP Token types for Jac."""
+
+    NAMESPACE = 0
+    TYPE = 1
+    CLASS = 2
+    ENUM = 3
+    INTERFACE = 4
+    STRUCT = 5
+    TYPE_PARAMETER = 6
+    PARAMETER = 7
+    VARIABLE = 8
+    PROPERTY = 9
+    ENUM_MEMBER = 10
+    EVENT = 11
+    FUNCTION = 12
+    METHOD = 13
+    MACRO = 14
+    KEYWORD = 15
+    MODIFIER = 16
+    COMMENT = 17
+    STRING = 18
+    NUMBER = 19
+    REGEXP = 20
+    OPERATOR = 21
+
+    @staticmethod
+    def as_str_list() -> list[str]:
+        """Return the string representation of the token."""
+        return [i.name.lower() for i in JacSemTokenType]
+
+
+class JacSemTokenModifier(IntFlag):
+    """LSP Token modifiers for Jac."""
+
+    DECLARATION = 1 << 0
+    DEFINITION = 1 << 1
+    READONLY = 1 << 2
+    STATIC = 1 << 3
+    DEPRECATED = 1 << 4
+    ABSTRACT = 1 << 5
+    ASYNC = 1 << 6
+    MODIFICATION = 1 << 7
+    DOCUMENTATION = 1 << 8
+    DEFAULT_LIBRARY = 1 << 9
+
+    @staticmethod
+    def as_str_list() -> list[str]:
+        """Return the string representation of the token."""
+        return [i.name.lower() for i in JacSemTokenModifier if i.name]
+
+
+class Constants(StrEnum):
     """Token constants for Jac."""
 
     JAC_LANG_IMP = "jac"
@@ -28,6 +112,7 @@ class Constants(str, Enum):
 
     PYNLINE = "::py::"
     JAC_GEN_DIR = "__jac_gen__"
+    JAC_MYPY_CACHE = ".jac_mypy_cache"
 
     def __str__(self) -> str:
         """Return the string representation of the token."""
@@ -48,6 +133,18 @@ class Values(int, Enum):
     JAC_ERROR_LINE_RANGE = 3
 
 
+class SymbolAccess(Enum):
+    """Symbol types."""
+
+    PRIVATE = "private"
+    PUBLIC = "public"
+    PROTECTED = "protected"
+
+    def __str__(self) -> str:
+        """Stringify."""
+        return self.value
+
+
 # Done like this for type checker
 # validated synced with test
 class Tokens(str, Enum):
@@ -99,6 +196,7 @@ class Tokens(str, Enum):
     KW_AWAIT = "KW_AWAIT"
     KW_TEST = "KW_TEST"
     KW_ASSERT = "KW_ASSERT"
+    KW_CHECK = "KW_CHECK"
     COLON = "COLON"
     PIPE_FWD = "PIPE_FWD"
     PIPE_BKWD = "PIPE_BKWD"

jaclang/compiler/jac.lark

@@ -30,7 +30,10 @@ sub_name: COLON NAME
 include_stmt: KW_INCLUDE sub_name import_path SEMI
 
 // Architypes
-architype: decorators? (enum | architype_def | architype_decl)
+architype: decorators? architype_decl
+         | architype_def
+         | enum
+
 architype_decl: arch_type access_tag? STRING? NAME inherited_archs? (member_block | SEMI)
 architype_def: abil_to_arch_chain member_block
 decorators: (DECOR_OP atomic_chain)+
@@ -54,8 +57,8 @@ typed_has_clause: named_ref (COLON STRING)? type_tag (EQ expression | KW_BY KW_P
 type_tag: COLON expression
 
 // Enumerations
-enum: enum_def
-     | enum_decl
+enum: decorators? enum_decl
+     | enum_def
 
 enum_decl: KW_ENUM access_tag? STRING? NAME inherited_archs? (enum_block | SEMI)
 enum_def: arch_to_enum_chain enum_block
@@ -64,16 +67,19 @@ enum_block: LBRACE ((enum_stmt COMMA)* enum_stmt)? RBRACE
 enum_stmt: NAME (COLON STRING)? EQ expression
          | NAME (COLON STRING)?
          | py_code_block
+         | free_code
+         | abstract_ability
+         | ability
 
 // Abilities
-ability: decorators? ability_def
-        | decorators? KW_ASYNC? ability_decl
+ability: decorators? KW_ASYNC? ability_decl
         | decorators? genai_ability
+        | ability_def
 
-ability_decl: KW_OVERRIDE? KW_STATIC? KW_CAN access_tag? STRING? any_ref (func_decl | event_clause) (code_block | SEMI)
+ability_decl: KW_OVERRIDE? KW_STATIC? KW_CAN access_tag? STRING? named_ref (func_decl | event_clause) (code_block | SEMI)
 ability_def: arch_to_abil_chain (func_decl | event_clause) code_block
-abstract_ability: KW_OVERRIDE? KW_STATIC? KW_CAN access_tag? STRING? any_ref (func_decl | event_clause) KW_ABSTRACT SEMI
-genai_ability: KW_OVERRIDE? KW_STATIC? KW_CAN access_tag? STRING? any_ref (func_decl) KW_BY atomic_call SEMI
+abstract_ability: KW_OVERRIDE? KW_STATIC? KW_CAN access_tag? STRING? named_ref (func_decl | event_clause) KW_ABSTRACT SEMI
+genai_ability: KW_OVERRIDE? KW_STATIC? KW_CAN access_tag? STRING? named_ref (func_decl) KW_BY atomic_call SEMI
 event_clause: KW_WITH expression? (KW_EXIT | KW_ENTRY) (STRING? RETURN_HINT expression)?
 func_decl: (LPAREN func_decl_params? RPAREN)? (RETURN_HINT (STRING COLON)? expression)?
 func_decl_params: (param_var COMMA)* param_var
@@ -128,9 +134,10 @@ statement: import_stmt
        | nonlocal_ref SEMI
        | typed_ctx_block
        | return_stmt SEMI
-       | yield_expr SEMI
+       | (yield_expr | KW_YIELD) SEMI
        | raise_stmt SEMI
        | assert_stmt SEMI
+       | check_stmt SEMI
        | assignment SEMI
        | delete_stmt SEMI
        | report_stmt SEMI
@@ -219,7 +226,6 @@ return_stmt: KW_RETURN expression?
 
 // Yield statements
 yield_expr: KW_YIELD KW_FROM? expression
-          | KW_YIELD
 
 // Raise statements
 raise_stmt: KW_RAISE (expression (KW_FROM expression)?)?
@@ -227,6 +233,9 @@ raise_stmt: KW_RAISE (expression (KW_FROM expression)?)?
 // Assert statements
 assert_stmt: KW_ASSERT expression (COMMA expression)?
 
+// Check statements
+check_stmt: KW_CHECK expression
+
 // Delete statements
 delete_stmt: KW_DELETE expression
 
@@ -279,7 +288,7 @@ expression: walrus_assign (KW_IF expression KW_ELSE expression)?
           | lambda_expr
 
 // Walrus assignments
-walrus_assign: (any_ref WALRUS_EQ)? pipe
+walrus_assign: (named_ref WALRUS_EQ)? pipe
 
 // Lambda expressions
 lambda_expr: KW_WITH func_decl_params? (RETURN_HINT expression)? KW_CAN expression
@@ -345,7 +354,7 @@ pipe_call: (PIPE_FWD | A_PIPE_FWD | KW_SPAWN | KW_AWAIT)? atomic_chain
 
 // Subscripted and dotted expressions
 atomic_chain: atomic_chain NULL_OK? (filter_compr | assign_compr | index_slice)
-            | atomic_chain NULL_OK? (DOT_BKWD | DOT_FWD | DOT) any_ref
+            | atomic_chain NULL_OK? (DOT_BKWD | DOT_FWD | DOT) named_ref
             | (atomic_call | atom | edge_ref_chain)
 
 index_slice: LSQUARE expression? COLON expression? (COLON expression?)? RSQUARE
@@ -359,10 +368,11 @@ param_list: expr_list COMMA kw_expr_list
           | expr_list
 
 // Atom
-atom: any_ref
+atom: named_ref
     | LPAREN (expression | yield_expr) RPAREN
     | atom_collection
     | atom_literal
+    | type_ref
 
 atom_literal: builtin_type
             | NULL
@@ -375,6 +385,8 @@ atom_literal: builtin_type
             | HEX
             | INT
 
+type_ref: TYPE_OP (named_ref | builtin_type)
+
 multistring: (fstring | STRING | DOC_STRING)+
 
 fstring: FSTR_START fstr_parts FSTR_END
@@ -415,7 +427,7 @@ tuple_list: expression COMMA expr_list COMMA kw_expr_list
           | kw_expr_list
 
 kw_expr_list: (kw_expr_list COMMA)? kw_expr
-kw_expr: any_ref EQ expression | STAR_POW expression
+kw_expr: named_ref EQ expression | STAR_POW expression
 
 // Data Spatial References
 edge_ref_chain: (EDGE_OP|NODE_OP)? LSQUARE expression? (edge_op_ref (filter_compr | expression)?)+ RSQUARE
@@ -438,15 +450,10 @@ typed_filter_compare_list: expression (COLON filter_compare_list)?
 filter_compare_item: named_ref cmp_op expression
 
 // Names and references
-any_ref: named_ref
-       | type_ref
-
 named_ref: special_ref
          | KWESC_NAME
          | NAME
 
-type_ref: TYPE_OP (named_ref | builtin_type)
-
 special_ref: KW_INIT
             | KW_POST_INIT
             | KW_ROOT
@@ -502,6 +509,7 @@ KW_ASYNC: "async"
 KW_AWAIT: "await"
 KW_TEST: "test"
 KW_ASSERT: "assert"
+KW_CHECK: "check"
 KW_IF: "if"
 KW_ELIF: "elif"
 KW_ELSE: "else"