Method CopyExpr

common/ast/factory.go:236–297

(e Expr)

}

func (fac *baseExprFactory) CopyExpr(e Expr) Expr {
	// unwrap navigable expressions to avoid unnecessary allocations during copying.
	if nav, ok := e.(*navigableExprImpl); ok {
		e = nav.Expr
	}
	switch e.Kind() {
	case CallKind:
		c := e.AsCall()
		argsCopy := make([]Expr, len(c.Args()))
		for i, arg := range c.Args() {
			argsCopy[i] = fac.CopyExpr(arg)
		}
		if !c.IsMemberFunction() {
			return fac.NewCall(e.ID(), c.FunctionName(), argsCopy...)
		}
		return fac.NewMemberCall(e.ID(), c.FunctionName(), fac.CopyExpr(c.Target()), argsCopy...)
	case ComprehensionKind:
		compre := e.AsComprehension()
		return fac.NewComprehensionTwoVar(e.ID(),
			fac.CopyExpr(compre.IterRange()),
			compre.IterVar(),
			compre.IterVar2(),
			compre.AccuVar(),
			fac.CopyExpr(compre.AccuInit()),
			fac.CopyExpr(compre.LoopCondition()),
			fac.CopyExpr(compre.LoopStep()),
			fac.CopyExpr(compre.Result()))
	case IdentKind:
		return fac.NewIdent(e.ID(), e.AsIdent())
	case ListKind:
		l := e.AsList()
		elemsCopy := make([]Expr, l.Size())
		for i, elem := range l.Elements() {
			elemsCopy[i] = fac.CopyExpr(elem)
		}
		return fac.NewList(e.ID(), elemsCopy, l.OptionalIndices())
	case LiteralKind:
		return fac.NewLiteral(e.ID(), e.AsLiteral())
	case MapKind:
		m := e.AsMap()
		entriesCopy := make([]EntryExpr, m.Size())
		for i, entry := range m.Entries() {
			entriesCopy[i] = fac.CopyEntryExpr(entry)
		}
		return fac.NewMap(e.ID(), entriesCopy)
	case SelectKind:
		s := e.AsSelect()
		if s.IsTestOnly() {
			return fac.NewPresenceTest(e.ID(), fac.CopyExpr(s.Operand()), s.FieldName())
		}
		return fac.NewSelect(e.ID(), fac.CopyExpr(s.Operand()), s.FieldName())
	case StructKind:
		s := e.AsStruct()
		fieldsCopy := make([]EntryExpr, len(s.Fields()))
		for i, field := range s.Fields() {
			fieldsCopy[i] = fac.CopyEntryExpr(field)
		}
		return fac.NewStruct(e.ID(), s.TypeName(), fieldsCopy)