The function findInStores() didn't consider localStore

This commit is contained in:
Celestino Amoroso 2025-06-18 14:28:03 +02:00
parent c480e81c0a
commit f0e600dd8b
2 changed files with 39 additions and 27 deletions

View File

@ -44,30 +44,43 @@ func NewDictSetContextV(flags DictSetFlag, varStores []map[string]string) *DictS
return ctx
}
func (self *DictSetContext) GetVar(name string) (value string, exists bool) {
if value, exists = self.localStore[name]; !exists {
value, exists = findInStores(name, self.varStores)
func (ctx *DictSetContext) GetVar(name string) (value string, exists bool) {
if value, exists = ctx.localStore[name]; !exists {
value, exists = ctx.findInStores(name)
}
return
}
func (self *DictSetContext) SetVar(name string, value string) (repeatedValue string) {
self.localStore[name] = value
func (ctx *DictSetContext) SetVar(name string, value string) (repeatedValue string) {
ctx.localStore[name] = value
return value
}
func findInStores(key string, stores []map[string]string) (value string, found bool) {
for _, vars := range stores {
func (ctx *DictSetContext) findInStores(key string) (value string, found bool) {
if value, found = ctx.localStore[key]; !found {
for _, vars := range ctx.varStores {
if vars != nil {
if value, found = vars[key]; found {
break
}
}
}
}
return
}
func (self *DictSetContext) Handle(spec string, scanFlags ScannerFlag) (value string, err error) {
// func findInStores(key string, stores []map[string]string) (value string, found bool) {
// for _, vars := range stores {
// if vars != nil {
// if value, found = vars[key]; found {
// break
// }
// }
// }
// return
// }
func (ctx *DictSetContext) Handle(spec string, scanFlags ScannerFlag) (value string, err error) {
var found bool
var rawValue string
@ -81,30 +94,30 @@ func (self *DictSetContext) Handle(spec string, scanFlags ScannerFlag) (value st
rawValue = name
found = true
} else {
rawValue, found = findInStores(name, self.varStores)
rawValue, found = ctx.findInStores(name)
}
}
self.SetVar(pipeNameKey, name) // NOTE Maybe this line can be deleted
ctx.SetVar(pipeNameKey, name) // NOTE Maybe this line can be deleted
origVarFound := found
for _, part := range parts {
var expandedPart string
self.varStores[0][pipeValueKey] = rawValue
self.SetVar(pipeValueKey, rawValue)
if expandedPart, err = Scan(self, part); err != nil {
ctx.varStores[0][pipeValueKey] = rawValue
ctx.SetVar(pipeValueKey, rawValue)
if expandedPart, err = Scan(ctx, part); err != nil {
break
}
self.SetVar(pipeNameKey, name)
if rawValue, err = self.evalContent(origVarFound, rawValue, expandedPart); err != nil {
ctx.SetVar(pipeNameKey, name)
if rawValue, err = ctx.evalContent(origVarFound, rawValue, expandedPart); err != nil {
break
}
origVarFound = true
}
if err == nil {
if len(rawValue) == 0 && (self.flags&KeepVar) != 0 {
if len(rawValue) == 0 && (ctx.flags&KeepVar) != 0 {
value = Var(name)
} else {
value, err = Scan(self, rawValue)
value, err = Scan(ctx, rawValue)
//value = rawValue
}
}
@ -116,7 +129,7 @@ func (self *DictSetContext) Handle(spec string, scanFlags ScannerFlag) (value st
return
}
func (self *DictSetContext) evalContent(varFound bool, rawValue string, alternateValue string) (value string, err error) {
func (ctx *DictSetContext) evalContent(varFound bool, rawValue string, alternateValue string) (value string, err error) {
var args []string
if args, err = sq.Split(alternateValue); err != nil {
return
@ -127,8 +140,8 @@ func (self *DictSetContext) evalContent(varFound bool, rawValue string, alternat
op := args[0]
if strings.HasPrefix(op, "__") {
var f ExpanderFunc
if f, err = self.GetFunc(op); err == nil {
value, err = f(self, rawValue, args[1:])
if f, err = ctx.GetFunc(op); err == nil {
value, err = f(ctx, rawValue, args[1:])
}
} else if varFound {
value = rawValue

View File

@ -86,7 +86,6 @@ func TestNewDictSetContext(t *testing.T) {
if (gotErr != nil && input.wantErr == nil) || (gotErr == nil && input.wantErr != nil) || (gotErr.Error() != input.wantErr.Error()) {
t.Errorf("%d: %s(%q)/err = <%v>, want <%v>", i, funcName, input.source, gotErr, input.wantErr)
}
}
}