feat: manually mirror opencoze's code from bytedance

Change-Id: I09a73aadda978ad9511264a756b2ce51f5761adf

backend/application/shortcutcmd/init.go

@@ -0,0 +1,40 @@
+/*
+ * 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 shortcutcmd
+
+import (
+	"gorm.io/gorm"
+
+	"github.com/coze-dev/coze-studio/backend/domain/shortcutcmd/repository"
+	"github.com/coze-dev/coze-studio/backend/domain/shortcutcmd/service"
+	"github.com/coze-dev/coze-studio/backend/infra/contract/idgen"
+)
+
+var ShortcutCmdSVC *ShortcutCmdApplicationService
+
+func InitService(db *gorm.DB, idGenSVC idgen.IDGenerator) *ShortcutCmdApplicationService {
+
+	components := &service.Components{
+		ShortCutCmdRepo: repository.NewShortCutCmdRepo(db, idGenSVC),
+	}
+	shortcutCmdDomainSVC := service.NewShortcutCommandService(components)
+
+	ShortcutCmdSVC = &ShortcutCmdApplicationService{
+		ShortCutDomainSVC: shortcutCmdDomainSVC,
+	}
+	return ShortcutCmdSVC
+}

backend/application/shortcutcmd/shortcut_cmd.go

@@ -0,0 +1,119 @@
+/*
+ * 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 shortcutcmd
+
+import (
+	"context"
+	"strconv"
+
+	"github.com/coze-dev/coze-studio/backend/api/model/ocean/cloud/playground"
+	"github.com/coze-dev/coze-studio/backend/application/base/ctxutil"
+	"github.com/coze-dev/coze-studio/backend/domain/shortcutcmd/entity"
+	"github.com/coze-dev/coze-studio/backend/domain/shortcutcmd/service"
+	"github.com/coze-dev/coze-studio/backend/pkg/lang/conv"
+)
+
+type ShortcutCmdApplicationService struct {
+	ShortCutDomainSVC service.ShortcutCmd
+}
+
+func (s *ShortcutCmdApplicationService) Handler(ctx context.Context, req *playground.CreateUpdateShortcutCommandRequest) (*playground.ShortcutCommand, error) {
+
+	cr, buildErr := s.buildReq(ctx, req)
+	if buildErr != nil {
+		return nil, buildErr
+	}
+	var err error
+	var cmdDO *entity.ShortcutCmd
+	if cr.CommandID > 0 {
+		cmdDO, err = s.ShortCutDomainSVC.UpdateCMD(ctx, cr)
+	} else {
+		cmdDO, err = s.ShortCutDomainSVC.CreateCMD(ctx, cr)
+	}
+
+	if err != nil {
+		return nil, err
+	}
+
+	if cmdDO == nil {
+		return nil, nil
+	}
+	return s.buildDo2Vo(ctx, cmdDO), nil
+}
+func (s *ShortcutCmdApplicationService) buildReq(ctx context.Context, req *playground.CreateUpdateShortcutCommandRequest) (*entity.ShortcutCmd, error) {
+
+	uid := ctxutil.MustGetUIDFromCtx(ctx)
+
+	var workflowID int64
+	var pluginID int64
+	var err error
+	if req.GetShortcuts().GetWorkFlowID() != "" {
+		workflowID, err = strconv.ParseInt(req.GetShortcuts().GetWorkFlowID(), 10, 64)
+		if err != nil {
+			return nil, err
+		}
+	}
+
+	if req.GetShortcuts().GetPluginID() != "" {
+		pluginID, err = strconv.ParseInt(req.GetShortcuts().GetPluginID(), 10, 64)
+		if err != nil {
+			return nil, err
+		}
+	}
+
+	return &entity.ShortcutCmd{
+		ObjectID:        req.GetObjectID(),
+		CommandID:       req.GetShortcuts().CommandID,
+		CommandName:     req.GetShortcuts().CommandName,
+		ShortcutCommand: req.GetShortcuts().ShortcutCommand,
+		Description:     req.GetShortcuts().Description,
+		SendType:        int32(req.GetShortcuts().SendType),
+		ToolType:        int32(req.GetShortcuts().ToolType),
+		WorkFlowID:      workflowID,
+		PluginID:        pluginID,
+		Components:      req.GetShortcuts().ComponentsList,
+		CardSchema:      req.GetShortcuts().CardSchema,
+		ToolInfo:        req.GetShortcuts().ToolInfo,
+		CreatorID:       uid,
+		PluginToolID:    req.GetShortcuts().PluginAPIID,
+		PluginToolName:  req.GetShortcuts().PluginAPIName,
+		TemplateQuery:   req.GetShortcuts().TemplateQuery,
+		ShortcutIcon:    req.GetShortcuts().ShortcutIcon,
+	}, nil
+}
+
+func (s *ShortcutCmdApplicationService) buildDo2Vo(ctx context.Context, do *entity.ShortcutCmd) *playground.ShortcutCommand {
+
+	return &playground.ShortcutCommand{
+		ObjectID:        do.ObjectID,
+		CommandID:       do.CommandID,
+		CommandName:     do.CommandName,
+		ShortcutCommand: do.ShortcutCommand,
+		Description:     do.Description,
+		SendType:        playground.SendType(do.SendType),
+		ToolType:        playground.ToolType(do.ToolType),
+		WorkFlowID:      conv.Int64ToStr(do.WorkFlowID),
+		PluginID:        conv.Int64ToStr(do.PluginID),
+		ComponentsList:  do.Components,
+		CardSchema:      do.CardSchema,
+		ToolInfo:        do.ToolInfo,
+		PluginAPIID:     do.PluginToolID,
+		PluginAPIName:   do.PluginToolName,
+		TemplateQuery:   do.TemplateQuery,
+		ShortcutIcon:    do.ShortcutIcon,
+	}
+}