feat: manually mirror opencoze's code from bytedance

Change-Id: I09a73aadda978ad9511264a756b2ce51f5761adf

backend/application/singleagent/duplicate.go

@@ -0,0 +1,177 @@
+/*
+ * 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 singleagent
+
+import (
+	"context"
+
+	intelligence "github.com/coze-dev/coze-studio/backend/api/model/intelligence/common"
+	"github.com/coze-dev/coze-studio/backend/api/model/ocean/cloud/developer_api"
+	"github.com/coze-dev/coze-studio/backend/api/model/project_memory"
+	"github.com/coze-dev/coze-studio/backend/application/base/ctxutil"
+	"github.com/coze-dev/coze-studio/backend/crossdomain/contract/crossplugin"
+	"github.com/coze-dev/coze-studio/backend/domain/agent/singleagent/entity"
+	searchEntity "github.com/coze-dev/coze-studio/backend/domain/search/entity"
+	shortcutCMDEntity "github.com/coze-dev/coze-studio/backend/domain/shortcutcmd/entity"
+	"github.com/coze-dev/coze-studio/backend/pkg/lang/conv"
+	"github.com/coze-dev/coze-studio/backend/pkg/lang/slices"
+)
+
+type duplicateAgentResourceFn func(ctx context.Context, appContext *ServiceComponents, oldAgent, newAgent *entity.SingleAgent) (*entity.SingleAgent, error)
+
+func (s *SingleAgentApplicationService) DuplicateDraftBot(ctx context.Context, req *developer_api.DuplicateDraftBotRequest) (*developer_api.DuplicateDraftBotResponse, error) {
+	draftAgent, err := s.ValidateAgentDraftAccess(ctx, req.BotID)
+	if err != nil {
+		return nil, err
+	}
+
+	newAgentID, err := s.appContext.IDGen.GenID(ctx)
+	if err != nil {
+		return nil, err
+	}
+
+	userID := ctxutil.MustGetUIDFromCtx(ctx)
+	duplicateInfo := &entity.DuplicateInfo{
+		NewAgentID: newAgentID,
+		SpaceID:    req.GetSpaceID(),
+		UserID:     userID,
+		DraftAgent: draftAgent,
+	}
+
+	newAgent, err := s.DomainSVC.DuplicateInMemory(ctx, duplicateInfo)
+	if err != nil {
+		return nil, err
+	}
+
+	duplicateFns := []duplicateAgentResourceFn{
+		duplicateVariables,
+		duplicatePlugin,
+		duplicateShortCommand,
+	}
+
+	for _, fn := range duplicateFns {
+		newAgent, err = fn(ctx, s.appContext, draftAgent, newAgent)
+		if err != nil {
+			return nil, err
+		}
+	}
+
+	_, err = s.DomainSVC.CreateSingleAgentDraftWithID(ctx, userID, newAgentID, newAgent)
+	if err != nil {
+		return nil, err
+	}
+
+	userInfo, err := s.appContext.UserDomainSVC.GetUserInfo(ctx, userID)
+	if err != nil {
+		return nil, err
+	}
+
+	err = s.appContext.EventBus.PublishProject(ctx, &searchEntity.ProjectDomainEvent{
+		OpType: searchEntity.Created,
+		Project: &searchEntity.ProjectDocument{
+			Status:  intelligence.IntelligenceStatus_Using,
+			Type:    intelligence.IntelligenceType_Bot,
+			ID:      newAgent.AgentID,
+			SpaceID: &req.SpaceID,
+			OwnerID: &userID,
+			Name:    &newAgent.Name,
+		},
+	})
+	if err != nil {
+		return nil, err
+	}
+
+	return &developer_api.DuplicateDraftBotResponse{
+		Data: &developer_api.DuplicateDraftBotData{
+			BotID: newAgent.AgentID,
+			Name:  newAgent.Name,
+			UserInfo: &developer_api.Creator{
+				ID:             userID,
+				Name:           userInfo.Name,
+				AvatarURL:      userInfo.IconURL,
+				Self:           userID == draftAgent.CreatorID,
+				UserUniqueName: userInfo.UniqueName,
+				UserLabel:      nil,
+			},
+		},
+		Code: 0,
+	}, nil
+}
+
+func duplicateVariables(ctx context.Context, appContext *ServiceComponents, oldAgent, newAgent *entity.SingleAgent) (*entity.SingleAgent, error) {
+	if oldAgent.VariablesMetaID == nil || *oldAgent.VariablesMetaID <= 0 {
+		return newAgent, nil
+	}
+
+	vars, err := appContext.VariablesDomainSVC.GetVariableMetaByID(ctx, *oldAgent.VariablesMetaID)
+	if err != nil {
+		return nil, err
+	}
+
+	vars.ID = 0
+	vars.BizID = conv.Int64ToStr(newAgent.AgentID)
+	vars.BizType = project_memory.VariableConnector_Bot
+	vars.Version = ""
+	vars.CreatorID = newAgent.CreatorID
+
+	varMetaID, err := appContext.VariablesDomainSVC.UpsertMeta(ctx, vars)
+	if err != nil {
+		return nil, err
+	}
+
+	newAgent.VariablesMetaID = &varMetaID
+
+	return newAgent, nil
+}
+
+func duplicatePlugin(ctx context.Context, _ *ServiceComponents, oldAgent, newAgent *entity.SingleAgent) (*entity.SingleAgent, error) {
+	err := crossplugin.DefaultSVC().DuplicateDraftAgentTools(ctx, oldAgent.AgentID, newAgent.AgentID)
+	if err != nil {
+		return nil, err
+	}
+	return newAgent, nil
+}
+
+func duplicateShortCommand(ctx context.Context, appContext *ServiceComponents, oldAgent, newAgent *entity.SingleAgent) (*entity.SingleAgent, error) {
+	metas, err := appContext.ShortcutCMDDomainSVC.ListCMD(ctx, &shortcutCMDEntity.ListMeta{
+		SpaceID:  oldAgent.SpaceID,
+		ObjectID: oldAgent.AgentID,
+		IsOnline: 0,
+		CommandIDs: slices.Transform(oldAgent.ShortcutCommand, func(a string) int64 {
+			return conv.StrToInt64D(a, 0)
+		}),
+	})
+	if err != nil {
+		return nil, err
+	}
+
+	shortcutCommandIDs := make([]string, 0, len(metas))
+	for _, meta := range metas {
+		meta.ObjectID = newAgent.AgentID
+		meta.CreatorID = newAgent.CreatorID
+		do, err := appContext.ShortcutCMDDomainSVC.CreateCMD(ctx, meta)
+		if err != nil {
+			return nil, err
+		}
+
+		shortcutCommandIDs = append(shortcutCommandIDs, conv.Int64ToStr(do.CommandID))
+	}
+
+	newAgent.ShortcutCommand = shortcutCommandIDs
+
+	return newAgent, nil
+}