haskell - 添加语法糖以使 Prismatic Sc​​hema 看起来像 Core.Typed/Haskell

Question

1. 考虑core.typed注释函数的方式：

   (t/ann typed-function [t/Str :-> t/Str])
   

2. 现在考虑 Prismatic Sc​​hema 注释函数的方式：

   (s/defn schema-function :- s/Str
          [arg :- s/Str]
          arg)
   

就个人而言，我发现core.typed注释函数的方式在精神上更清晰，更接近于 Haskell 等强类型语言。

问题：有没有办法使用 Prismatic Sc​​hema 在 clojure 中制作某种宏或函数，具有 (2) 的效果但视觉外观 (1)？也就是说，类似于以下内容：'

(custom-annotation-macro schema-function [s/Str :-> s/Str])
(defn schema-function [arg] arg)

这样的效果仅仅是

(s/defn schema-function :- s/Str
      [arg :- s/Str]
       arg)

Answer 1

为了说明如何使用两个宏来解决这个问题：

(def type-annots (atom (hash-map)))

(defn add-type-annot [fn-name ty]
  (swap! type-annots #(conj % [fn-name ty])))

(defmacro custom-annotation-macro [fn-name ty]
  (add-type-annot fn-name ty)
  nil)

(defn split-fun-type [ty]
  ;; You will need to write this one; 
  ;; it should split [a :-> b :-> c] to [[a b] c]
  ['[s/Str s/Int] 's/Str])

(defmacro defn-typed [fn-name args & body]
  (let [ty (get @type-annots fn-name)]
    (if ty
      (let [[arg-types ret-ty] (split-fun-type ty)
            args-typed (apply vector (apply concat (map vector args arg-types)))]
        `(s/defn ~fn-name :- ~ret-ty ~args-typed ~@body))
      `(defn ~fn-name ~args ~@body))))

我没有费心去实现split-fun-type，因为我真的不了解 Clojure；以上是基于我对它是 Lisp 的理解。