refactor: type

src/ty/internal/primop.rs

@@ -0,0 +1,67 @@
+use derive_more::Constructor;
+
+use super::Value;
+
+#[derive(Debug, Clone, Constructor)]
+pub struct PrimOp {
+    pub name: &'static str,
+    arity: u8,
+    func: fn(Vec<Value>) -> Value,
+}
+
+impl PartialEq for PrimOp {
+    fn eq(&self, _: &Self) -> bool {
+        false
+    }
+}
+
+impl PrimOp {
+    pub fn call(self, args: Vec<Value>) -> Value {
+        if (args.len() as u8) < self.arity {
+            Value::PartialPrimOp(PartialPrimOp {
+                name: self.name,
+                arity: self.arity - args.len() as u8,
+                args,
+                func: self.func,
+            })
+        } else if args.len() as u8 == self.arity {
+            (self.func)(args)
+        } else {
+            unimplemented!()
+        }
+    }
+}
+
+#[derive(Debug, Clone)]
+pub struct PartialPrimOp {
+    pub name: &'static str,
+    arity: u8,
+    args: Vec<Value>,
+    func: fn(Vec<Value>) -> Value,
+}
+
+impl PartialEq for PartialPrimOp {
+    fn eq(&self, _: &Self) -> bool {
+        false
+    }
+}
+
+impl PartialPrimOp {
+    pub fn call(mut self, args: Vec<Value>) -> Value {
+        let len = args.len() as u8;
+        self.args.extend(args);
+        if len < self.arity {
+            Value::PartialPrimOp(PartialPrimOp {
+                name: self.name,
+                arity: self.arity - len,
+                args: self.args,
+                func: self.func,
+            })
+        } else if len == self.arity {
+            (self.func)(self.args)
+        } else {
+            unimplemented!()
+        }
+    }
+}
+