@@ -155,7 +155,7 @@ func (r *Engine) Verify(ctx context.Context, policy *engine.Policy, input []byte
155155 }
156156
157157 // Get raw results first
158- if err := executeQuery (fmt . Sprintf ( "%s \n " , parsedModule .Package .Path ), r .operatingMode == EnvironmentModeRestrictive ); err != nil {
158+ if err := executeQuery (getRuleName ( parsedModule .Package .Path , "" ), r .operatingMode == EnvironmentModeRestrictive ); err != nil {
159159 return nil , err
160160 }
161161
@@ -165,15 +165,15 @@ func (r *Engine) Verify(ctx context.Context, policy *engine.Policy, input []byte
165165 }
166166
167167 // Try the main rule first
168- if err := executeQuery (fmt . Sprintf ( "%v.%s \n " , parsedModule .Package .Path , mainRule ), r .operatingMode == EnvironmentModeRestrictive ); err != nil {
168+ if err := executeQuery (getRuleName ( parsedModule .Package .Path , mainRule ), r .operatingMode == EnvironmentModeRestrictive ); err != nil {
169169 return nil , err
170170 }
171171
172172 // If res is nil, it means that the rule hasn't been found
173173 // TODO: Remove when this deprecated rule is not used anymore
174174 if res == nil {
175175 // Try with the deprecated main rule
176- if err := executeQuery (fmt . Sprintf ( "%v.%s \n " , parsedModule .Package .Path , deprecatedRule ), r .operatingMode == EnvironmentModeRestrictive ); err != nil {
176+ if err := executeQuery (getRuleName ( parsedModule .Package .Path , deprecatedRule ), r .operatingMode == EnvironmentModeRestrictive ); err != nil {
177177 return nil , err
178178 }
179179
@@ -325,3 +325,10 @@ func regoResultSetToRawResults(res rego.ResultSet) map[string]interface{} {
325325 }
326326 return raw
327327}
328+
329+ func getRuleName (packagePath ast.Ref , rule string ) string {
330+ if rule == "" {
331+ return fmt .Sprintf ("%s\n " , packagePath )
332+ }
333+ return fmt .Sprintf ("%v.%s\n " , packagePath , rule )
334+ }
0 commit comments