feat: manually mirror opencoze's code from bytedance

Change-Id: I09a73aadda978ad9511264a756b2ce51f5761adf

backend/domain/workflow/internal/nodes/variableassigner/variable_assign.go

@@ -0,0 +1,146 @@
+/*
+ * Copyright 2025 coze-dev Authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package variableassigner
+
+import (
+	"context"
+	"fmt"
+	"strings"
+	"sync"
+
+	"github.com/cloudwego/eino/compose"
+
+	"github.com/coze-dev/coze-studio/backend/domain/workflow/crossdomain/variable"
+	"github.com/coze-dev/coze-studio/backend/domain/workflow/entity/vo"
+	"github.com/coze-dev/coze-studio/backend/domain/workflow/internal/execute"
+	"github.com/coze-dev/coze-studio/backend/domain/workflow/internal/nodes"
+	"github.com/coze-dev/coze-studio/backend/pkg/errorx"
+	"github.com/coze-dev/coze-studio/backend/types/errno"
+)
+
+type AppVariables struct {
+	vars map[string]any
+	mu   sync.RWMutex
+}
+
+func NewAppVariables() *AppVariables {
+	return &AppVariables{
+		vars: make(map[string]any),
+	}
+}
+
+func (av *AppVariables) Set(key string, value any) {
+	av.mu.Lock()
+	av.vars[key] = value
+	av.mu.Unlock()
+}
+
+func (av *AppVariables) Get(key string) (any, bool) {
+	av.mu.RLock()
+	defer av.mu.RUnlock()
+
+	if value, ok := av.vars[key]; ok {
+		return value, ok
+	}
+	return nil, false
+}
+
+type AppVariableStore interface {
+	GetAppVariableValue(key string) (any, bool)
+	SetAppVariableValue(key string, value any)
+}
+
+type VariableAssigner struct {
+	config *Config
+}
+
+type Config struct {
+	Pairs   []*Pair
+	Handler *variable.Handler
+}
+
+type Pair struct {
+	Left  vo.Reference
+	Right compose.FieldPath
+}
+
+func NewVariableAssigner(_ context.Context, conf *Config) (*VariableAssigner, error) {
+	for _, pair := range conf.Pairs {
+		if pair.Left.VariableType == nil {
+			return nil, fmt.Errorf("cannot assign to output of nodes in VariableAssigner, ref: %v", pair.Left)
+		}
+
+		if *pair.Left.VariableType == vo.GlobalSystem {
+			return nil, fmt.Errorf("cannot assign to global system variables in VariableAssigner because they are read-only, ref: %v", pair.Left)
+		}
+
+		vType := *pair.Left.VariableType
+		if vType != vo.GlobalAPP && vType != vo.GlobalUser {
+			return nil, fmt.Errorf("cannot assign to variable type %s in VariableAssigner", vType)
+		}
+	}
+
+	return &VariableAssigner{
+		config: conf,
+	}, nil
+}
+
+func (v *VariableAssigner) Assign(ctx context.Context, in map[string]any) (map[string]any, error) {
+	for _, pair := range v.config.Pairs {
+		right, ok := nodes.TakeMapValue(in, pair.Right)
+		if !ok {
+			return nil, vo.NewError(errno.ErrInputFieldMissing, errorx.KV("name", strings.Join(pair.Right, ".")))
+		}
+
+		vType := *pair.Left.VariableType
+		switch vType {
+		case vo.GlobalAPP:
+			err := compose.ProcessState(ctx, func(ctx context.Context, appVarsStore AppVariableStore) error {
+				if len(pair.Left.FromPath) != 1 {
+					return fmt.Errorf("can only assign to top level variable: %v", pair.Left.FromPath)
+				}
+				appVarsStore.SetAppVariableValue(pair.Left.FromPath[0], right)
+				return nil
+			})
+			if err != nil {
+				return nil, err
+			}
+		case vo.GlobalUser:
+			opts := make([]variable.OptionFn, 0, 1)
+			if exeCtx := execute.GetExeCtx(ctx); exeCtx != nil {
+				exeCfg := exeCtx.RootCtx.ExeCfg
+				opts = append(opts, variable.WithStoreInfo(variable.StoreInfo{
+					AgentID:      exeCfg.AgentID,
+					AppID:        exeCfg.AppID,
+					ConnectorID:  exeCfg.ConnectorID,
+					ConnectorUID: exeCfg.ConnectorUID,
+				}))
+			}
+			err := v.config.Handler.Set(ctx, *pair.Left.VariableType, pair.Left.FromPath, right, opts...)
+			if err != nil {
+				return nil, vo.WrapIfNeeded(errno.ErrVariablesAPIFail, err)
+			}
+		default:
+			panic("impossible")
+		}
+	}
+
+	// TODO if not error considered successful
+	return map[string]any{
+		"isSuccess": true,
+	}, nil
+}