Extend set type with every protocol

Author: Alejandro Gómez

src/cljx/cats/types.cljx

@@ -1,6 +1,7 @@
 (ns cats.types
   "Monadic types definition."
-  (:require [cats.protocols :as proto]))
+  (:require [clojure.set :as s]
+            [cats.protocols :as proto]))
 
 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
 ;; Either
@@ -194,6 +195,8 @@
 ;; Clojure(Script) types
 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
 
+; TODO: PersistenList
+; TODO: document
 (extend-type #+clj  clojure.lang.LazySeq
              #+cljs cljs.core.LazySeq
   proto/Functor
@@ -210,6 +213,7 @@
   (bind [self f]
     (apply concat (map f self))))
 
+; TODO: test & document
 (extend-type #+clj clojure.lang.PersistentVector
              #+cljs cljs.core.PersistentVector
   proto/Functor
@@ -233,6 +237,32 @@
   proto/MonadPlus
   (mplus [mv mv'] (into mv mv')))
 
+; TODO: test & document
+(extend-type #+clj clojure.lang.PersistentHashSet
+             #+cljs cljs.core.PersistentHashSet
+  proto/Functor
+  (fmap [self f]
+    (set (map f self)))
+
+  proto/Applicative
+  (pure [_ v] #{v})
+
+  (fapply [self av]
+    (set (for [f self
+               v av]
+           (f v))))
+
+  proto/Monad
+  (bind [self f]
+    (apply s/union (map f self)))
+
+  proto/MonadZero
+  (mzero [_] #{})
+
+  proto/MonadPlus
+  (mplus [mv mv']
+    (s/union mv mv')))
+
 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
 ;; Pair (State monad related)
 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;

tests/tests_cats.cljx

@@ -46,7 +46,10 @@
            [[1 3] [1 4] [2 3] [2 4]])))
   (testing "It works with lazy seqs"
     (is (= (m/m-sequence [(lazy-seq [1 2]) (lazy-seq [3 4])])
-           [[1 3] [1 4] [2 3] [2 4]])))
+           '([1 3] [1 4] [2 3] [2 4]))))
+  (testing "It works with sets"
+    (is (= (m/m-sequence [#{1 2} #{3 4}])
+           #{[1 3] [1 4] [2 3] [2 4]})))
   (testing "It works with Maybe values"
     (is (= (m/m-sequence [(t/just 2) (t/just 3)])
            (t/just [2 3])))