From 596089aece8383bd374621babff1a14f99413dba Mon Sep 17 00:00:00 2001
From: Andrey Orst <andreyorst@gmail.com>
Date: Sat, 31 Oct 2020 14:27:46 +0300
Subject: feature(macros): implement simple multimethods

---
 macros/core.fnl | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)

(limited to 'macros')

diff --git a/macros/core.fnl b/macros/core.fnl
index be0c1bb..d107ad6 100644
--- a/macros/core.fnl
+++ b/macros/core.fnl
@@ -138,4 +138,33 @@
              :else (error "expected table as first argument"))
            to#))))
 
+(fn first [tbl]
+  (. tbl 1))
+
+(fn rest [tbl]
+  [(unpack tbl 2)])
+
+(fn string? [x]
+  (= (type x) :string))
+
+(fn* core.defmulti
+  [name & opts]
+  (let [docstring (if (string? (first opts)) (first opts))
+        opts (if docstring (rest opts) opts)
+        dispatch-fn (first opts)]
+    `(local ,name
+            (let [multimethods# {}]
+              (setmetatable {} {:__call
+                                (fn [_# ...]
+                                  ,docstring
+                                  ((or (. multimethods# (,dispatch-fn ...))
+                                       (. multimethods# :default)) ...))
+                                :multimethods multimethods#})))))
+
+(fn* core.defmethod
+  [multifn dispatch-val & fn-tail]
+  `(tset (. (getmetatable ,multifn) :multimethods)
+         ,dispatch-val
+         (fn ,(unpack fn-tail))))
+
 core
-- 
cgit v1.2.3