feat: 完成impl运算符重载

Author: hunter-hongg

bin/util.ml

@@ -93,7 +93,9 @@ let init_env_var_file f =
 
 let init_env_var () = 
   let home = Sys.getenv "HOME" in
-  init_env_var_file (home ^ "/.miver/config/miva.toml");
+  let global_config = home ^ "/.miver/config/miva.toml" in
+  if Sys.file_exists global_config then
+    init_env_var_file global_config;
   init_env_var_file "miva.toml"
 
 

lib/anoncheck.ml

@@ -87,6 +87,7 @@ let check_anon defs =
     | SImportAs (loc, _, _)
     | SImportHere (loc, _)
     | SExport (loc, _)
+    | DImpl (loc, _, _)
     -> (
       match !pre with 
       | Some DCIntro (_, _) -> (

lib/ast.ml

@@ -74,3 +74,14 @@ and def =
   | SImportHere of loc * string
   | DCMagical of loc * string
   | DCIntro of loc * string
+  | DImpl of loc * string * impl_expr list
+
+and impl_op = 
+  | ImAdd 
+  | ImSub
+  | ImMul
+  | ImEq
+  | ImNeq
+
+and impl_expr = 
+  | EImpl of loc * impl_op * string

lib/codegen.ml

@@ -413,6 +413,35 @@ let cxx_def_of_def indent_level ctx def =
   | DTest(_, _, _) -> ""
   | DCMagical (_, _) -> ""
   | DCIntro (_, _) -> ""
+  | DImpl (_, s, n) -> (
+    let ret = ref "" in
+    List.iter (fun n -> (
+      let op, fn = match n with
+      | EImpl (_, op, fn) -> op, fn
+      in
+      let ret_typ = match op with 
+        | ImAdd | ImSub | ImMul 
+          -> s
+        | ImEq | ImNeq 
+          -> "mvp_builtin_boolean" 
+      in
+      let operator = match op with
+        | ImAdd -> "+"
+        | ImSub -> "-"
+        | ImMul -> "*"
+        | ImEq -> "=="
+        | ImNeq -> "!="
+      in
+      let typs = s in
+      let func_sig = 
+        ret_typ ^ " operator" ^ operator ^ 
+          "(const " ^ typs ^ "& ____a, const " ^ 
+          typs ^ "& ____b) { return " ^ fn ^ "(____a, ____b); }\n"
+      in
+      ret := !ret ^ ind ^ func_sig ^ "\n"
+    )) n;
+    !ret
+  )
 
 
 let cxx_func_declaration func_name params ret_typ_opt =

lib/lexer.mll

@@ -78,6 +78,13 @@ rule token = parse
   | "module"            { MODULE }
   | "export"            { EXPORT }
   | "import"            { IMPORT }
+  | "impl"              { IMPLEMENTS }
+
+  | "op_add"            { OP_ADD }
+  | "op_sub"            { OP_SUB }
+  | "op_mul"            { OP_MUL }
+  | "op_eq"             { OP_EQ }
+  | "op_neq"            { OP_NEQ }
 
   | "ptr"               { PTR }
   | "box"               { BOX }

lib/parser.mly

@@ -22,7 +22,8 @@
 %token TEST UNSAFE TRUSTED C_KEYWORD 
 %token EOF 
 %token PTR BOX ADDR DEREF PTRANY
-%token MODULE EXPORT IMPORT
+%token OP_ADD OP_SUB OP_MUL OP_EQ OP_NEQ
+%token MODULE EXPORT IMPORT IMPLEMENTS
 
 %type <Ast.def> def
 %type <Ast.def list> list(def)
@@ -114,6 +115,26 @@ def:
   | m = MAGICAL { DCMagical ( 
     { line = $startpos.Lexing.pos_lnum; col = $startpos.Lexing.pos_cnum - $startpos.Lexing.pos_bol + 1 }
   , m) }
+  | IMPLEMENTS stru = IDENT LBRACE l = separated_nonempty_list(COMMA, impl_decl) RBRACE {
+    DImpl({
+      line = $startpos.Lexing.pos_lnum;
+      col = $startpos.Lexing.pos_cnum - $startpos.Lexing.pos_bol + 1
+    },
+    stru, l)
+  }
+
+impl_decl:
+  | name = reg_op EQ func = IDENT { EImpl ({
+    line = $startpos.Lexing.pos_lnum;
+    col = $startpos.Lexing.pos_cnum - $startpos.Lexing.pos_bol + 1
+  }, name, func) }
+
+reg_op:
+  | OP_ADD { ImAdd }
+  | OP_SUB { ImSub }
+  | OP_MUL { ImMul }
+  | OP_EQ  { ImEq }
+  | OP_NEQ { ImNeq }
 
 func_params:
   | LPAREN RPAREN { [] }