feat: support sets in partial evaluation

Project.toml

@@ -1,7 +1,7 @@
 name = "OpenPolicyAgent"
 uuid = "8f257efb-743c-4ebc-8197-d291a1f743b4"
 authors = ["JuliaHub Inc.", "Tanmay Mohapatra <tanmaykm@gmail.com>"]
-version = "0.3.1"
+version = "0.3.2"
 
 [deps]
 Dates = "ade2ca70-3891-5945-98fb-dc099432e06a"

src/utils/sql.jl

@@ -159,7 +159,7 @@ function visit(visitor::SQLVisitor, ref::AST.OPARef)
  return nothing
 end
 
-function visit(visitor::SQLVisitor, arr::AST.OPAArray)
+function visit(visitor::SQLVisitor, arr::Union{AST.OPAArray,AST.OPASet})
  push!(visitor.result_stack, string("(", join([walk(visitor, v) for v in arr.value], ", "), ")"))
  return nothing
 end

test/sql_translate.jl

@@ -207,7 +207,7 @@ function to_sql(ref::OPARef)
  end
 end
 
-function to_sql(arr::OPAArray)
+function to_sql(arr::Union{OPAArray,OPASet})
  return string("(", join(to_sql.(arr.value), ", "), ")")
 end
 

test/test_data.jl

@@ -233,6 +233,25 @@ const PARTIAL_COMPILE_CASES = [
  unknowns = ["data.reports"],
  sql = "( ( public.juliahub_reports.clearance_level - 4 ) >= 0 )",
  ),
+ (
+ policy = """package example
+ import future.keywords.in
+ allow {
+ data.reports[_].category in {"public"}
+ }
+ """,
+ query = "data.example.allow == true",
+ input = Dict{String,Any}(
+ "subject" => Dict{String,Any}(
+ "clearance_level" => 4
+ )
+ ),
+ options = Dict{String,Any}(
+ "disableInlining" => []
+ ),
+ unknowns = ["data.reports"],
+ sql = "( public.juliahub_reports.category in ('public') )",
+ ),
 ]
 
 const EXAMPLE_QUERY = """input.servers[i].ports[_] = "p2"; input.servers[i].name = name"""