implement cljs.spec/speced-vars

Author: swannodette

src/main/cljs/cljs/spec.clj

@@ -272,6 +272,23 @@
   [env sym role]
   (symbol (str (ns-qualify env sym) "$" (name role))))
 
+(def ^:private _speced_vars (atom #{}))
+
+(defmacro speced-vars
+  "Returns the set of vars whose namespace is in ns-syms AND
+whose vars have been speced with fdef. If no ns-syms are
+specified, return speced vars from all namespaces."
+  [& ns-syms]
+  (let [ns-match? (if (seq ns-syms)
+                    (set ns-syms)
+                    (constantly true))]
+    (reduce
+      (fn [ret sym]
+        (if (ns-match? (namespace sym))
+          (conj ret (list 'var sym))
+          ret))
+      #{} @_speced_vars)))
+
 (defmacro fdef
   "Takes a symbol naming a function, and one or more of the following:
 
@@ -305,6 +322,7 @@
                  :sym symbol?)
     :ret symbol?)"
   [fn-sym & {:keys [args ret fn] :as m}]
+  (swap! _speced_vars conj (:name (resolve &env fn-sym)))
   (let [env &env
         qn  (ns-qualify env fn-sym)]
     `(do ~@(reduce

src/main/cljs/cljs/spec.cljs

@@ -341,26 +341,6 @@
              (swap! instrumented-vars dissoc v))
            v))
 
-;(defn speced-vars
-;  "Returns the set of vars whose namespace is in ns-syms AND
-;whose vars have been speced with fdef. If no ns-syms are
-;specified, return speced vars from all namespaces."
-;  [& ns-syms]
-;  (let [ns-match? (if (seq ns-syms)
-;                    (set (map str ns-syms))
-;                    (constantly true))]
-;    (reduce-kv
-;      (fn [s k _]
-;        (if (c/and (symbol? k)
-;                   (re-find  #"\$(args|ret)$" (name k))
-;                   (ns-match? (namespace k)))
-;          (if-let [v (resolve (symbol (str/replace (str k) #"\$(args|ret)$" "")))]
-;            (conj s v)
-;            s)
-;          s))
-;      #{}
-;      (registry))))
-;
 ;(defn instrument-ns
 ;  "Call instrument for all speced-vars in namespaces named
 ;by ns-syms. Idempotent."