also trim in context query

realshuting · Nov 25, 2020 · 75bd8e2 · 75bd8e2
1 parent 54f816c
commit 75bd8e2
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/pkg/engine/context/evaluate.go b/pkg/engine/context/evaluate.go
@@ -10,9 +10,14 @@ import (
 
 //Query the JSON context with JMESPATH search path
 func (ctx *Context) Query(query string) (interface{}, error) {
+	query = strings.TrimSpace(query)
+	if query == "" {
+		return nil, fmt.Errorf("invalid query (nil)")
+	}
+
 	var emptyResult interface{}
 	// check for white-listed variables
-	if !ctx.isWhiteListed(query) {
+	if !ctx.isBuiltInVariable(query) {
 		return emptyResult, fmt.Errorf("variable %s cannot be used", query)
 	}
 
@@ -40,7 +45,7 @@ func (ctx *Context) Query(query string) (interface{}, error) {
 	return result, nil
 }
 
-func (ctx *Context) isWhiteListed(variable string) bool {
+func (ctx *Context) isBuiltInVariable(variable string) bool {
 	if len(ctx.builtInVars) == 0 {
 		return true
 	}