Add is_set builtin

fugue · Nov 16, 2019 · 27e363c · 27e363c
1 parent 9359917
commit 27e363c
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/lib/Fregot/Eval/Builtins.hs b/lib/Fregot/Eval/Builtins.hs
@@ -222,6 +222,7 @@ defaultBuiltins = HMS.fromList
     , (NamedFunction (BuiltinName "is_array"),                  builtin_is_array)
     , (NamedFunction (BuiltinName "is_boolean"),                builtin_is_boolean)
     , (NamedFunction (BuiltinName "is_object"),                 builtin_is_object)
+    , (NamedFunction (BuiltinName "is_set"),                    builtin_is_set)
     , (NamedFunction (BuiltinName "is_string"),                 builtin_is_string)
     , (NamedFunction (QualifiedName "json" "unmarshal"),        builtin_json_unmarshal)
     , (NamedFunction (BuiltinName "lower"),                     builtin_lower)
@@ -333,6 +334,12 @@ builtin_is_object = Builtin (In Out) $ pure $
         ObjectV _ -> return True
         _         -> return False
 
+builtin_is_set :: Monad m => Builtin m
+builtin_is_set = Builtin (In Out) $ pure $
+    \(Cons val Nil) -> case val of
+        SetV _ -> return True
+        _      -> return False
+
 builtin_is_string :: Monad m => Builtin m
 builtin_is_string = Builtin (In Out) $ pure $
     \(Cons val Nil) -> case val of

diff --git a/tests/rego/builtins-06.rego b/tests/rego/builtins-06.rego
@@ -7,3 +7,9 @@ test_is_boolean {
   not is_boolean(1)
   not is_boolean([true])
 }
+
+test_is_set {
+  is_set(set())
+  is_set({set()})
+  not is_set([set()])
+}