mirror of https://github.com/mikefarah/yq.git
43 lines
1.2 KiB
Go
43 lines
1.2 KiB
Go
package yqlib
|
|
|
|
import (
|
|
"container/list"
|
|
"fmt"
|
|
)
|
|
|
|
func getVariableOperator(d *dataTreeNavigator, context Context, expressionNode *ExpressionNode) (Context, error) {
|
|
variableName := expressionNode.Operation.StringValue
|
|
log.Debug("getVariableOperator %v", variableName)
|
|
result := context.GetVariable(variableName)
|
|
if result == nil {
|
|
result = list.New()
|
|
}
|
|
return context.ChildContext(result), nil
|
|
}
|
|
|
|
type assignVarPreferences struct {
|
|
IsReference bool
|
|
}
|
|
|
|
func assignVariableOperator(d *dataTreeNavigator, context Context, expressionNode *ExpressionNode) (Context, error) {
|
|
lhs, err := d.GetMatchingNodes(context.ReadOnlyClone(), expressionNode.Lhs)
|
|
if err != nil {
|
|
return Context{}, nil
|
|
}
|
|
if expressionNode.Rhs.Operation.OperationType.Type != "GET_VARIABLE" {
|
|
return Context{}, fmt.Errorf("RHS of 'as' operator must be a variable name e.g. $foo")
|
|
}
|
|
variableName := expressionNode.Rhs.Operation.StringValue
|
|
|
|
prefs := expressionNode.Operation.Preferences.(assignVarPreferences)
|
|
|
|
var variableValue *list.List
|
|
if prefs.IsReference {
|
|
variableValue = lhs.MatchingNodes
|
|
} else {
|
|
variableValue = lhs.DeepClone().MatchingNodes
|
|
}
|
|
context.SetVariable(variableName, variableValue)
|
|
return context, nil
|
|
}
|