feat(plugin): support saas plugin (#2349)

Co-authored-by: yuwenbinjie <yuwenbinjie@bytedance.com>
Co-authored-by: ski <csu.zengxiaohui@gmail.com>
Co-authored-by: Ryo <fanlv@bytedance.com>
This commit is contained in:
junwen-lee
2025-10-21 10:43:48 +08:00
committed by GitHub
parent d5e913d76d
commit 6e02c861c7
254 changed files with 48874 additions and 2399 deletions

View File

@ -1,3 +1,19 @@
/*
* 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.
*/
// Code generated by MockGen. DO NOT EDIT.
// Source: service.go
//
@ -13,8 +29,8 @@ import (
context "context"
reflect "reflect"
"github.com/coze-dev/coze-studio/backend/crossdomain/contract/plugin/model"
dto0 "github.com/coze-dev/coze-studio/backend/domain/plugin/dto"
model "github.com/coze-dev/coze-studio/backend/crossdomain/contract/plugin/model"
dto "github.com/coze-dev/coze-studio/backend/domain/plugin/dto"
entity "github.com/coze-dev/coze-studio/backend/domain/plugin/entity"
gomock "go.uber.org/mock/gomock"
)
@ -43,18 +59,34 @@ func (m *MockPluginService) EXPECT() *MockPluginServiceMockRecorder {
return m.recorder
}
// BindAgentTools mocks base method.
func (m *MockPluginService) BindAgentTools(ctx context.Context, agentID int64, toolIDs []int64) error {
// BatchGetSaasPluginToolsInfo mocks base method.
func (m *MockPluginService) BatchGetSaasPluginToolsInfo(ctx context.Context, pluginIDs []int64) (map[int64][]*entity.ToolInfo, map[int64]*entity.PluginInfo, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "BindAgentTools", ctx, agentID, toolIDs)
ret := m.ctrl.Call(m, "BatchGetSaasPluginToolsInfo", ctx, pluginIDs)
ret0, _ := ret[0].(map[int64][]*entity.ToolInfo)
ret1, _ := ret[1].(map[int64]*entity.PluginInfo)
ret2, _ := ret[2].(error)
return ret0, ret1, ret2
}
// BatchGetSaasPluginToolsInfo indicates an expected call of BatchGetSaasPluginToolsInfo.
func (mr *MockPluginServiceMockRecorder) BatchGetSaasPluginToolsInfo(ctx, pluginIDs any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGetSaasPluginToolsInfo", reflect.TypeOf((*MockPluginService)(nil).BatchGetSaasPluginToolsInfo), ctx, pluginIDs)
}
// BindAgentTools mocks base method.
func (m *MockPluginService) BindAgentTools(ctx context.Context, agentID int64, bindTools []*model.BindToolInfo) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "BindAgentTools", ctx, agentID, bindTools)
ret0, _ := ret[0].(error)
return ret0
}
// BindAgentTools indicates an expected call of BindAgentTools.
func (mr *MockPluginServiceMockRecorder) BindAgentTools(ctx, agentID, toolIDs any) *gomock.Call {
func (mr *MockPluginServiceMockRecorder) BindAgentTools(ctx, agentID, bindTools any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BindAgentTools", reflect.TypeOf((*MockPluginService)(nil).BindAgentTools), ctx, agentID, toolIDs)
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BindAgentTools", reflect.TypeOf((*MockPluginService)(nil).BindAgentTools), ctx, agentID, bindTools)
}
// CheckPluginToolsDebugStatus mocks base method.
@ -72,10 +104,10 @@ func (mr *MockPluginServiceMockRecorder) CheckPluginToolsDebugStatus(ctx, plugin
}
// ConvertToOpenapi3Doc mocks base method.
func (m *MockPluginService) ConvertToOpenapi3Doc(ctx context.Context, req *dto0.ConvertToOpenapi3DocRequest) *dto0.ConvertToOpenapi3DocResponse {
func (m *MockPluginService) ConvertToOpenapi3Doc(ctx context.Context, req *dto.ConvertToOpenapi3DocRequest) *dto.ConvertToOpenapi3DocResponse {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "ConvertToOpenapi3Doc", ctx, req)
ret0, _ := ret[0].(*dto0.ConvertToOpenapi3DocResponse)
ret0, _ := ret[0].(*dto.ConvertToOpenapi3DocResponse)
return ret0
}
@ -86,10 +118,10 @@ func (mr *MockPluginServiceMockRecorder) ConvertToOpenapi3Doc(ctx, req any) *gom
}
// CopyPlugin mocks base method.
func (m *MockPluginService) CopyPlugin(ctx context.Context, req *dto0.CopyPluginRequest) (*dto0.CopyPluginResponse, error) {
func (m *MockPluginService) CopyPlugin(ctx context.Context, req *dto.CopyPluginRequest) (*dto.CopyPluginResponse, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "CopyPlugin", ctx, req)
ret0, _ := ret[0].(*dto0.CopyPluginResponse)
ret0, _ := ret[0].(*dto.CopyPluginResponse)
ret1, _ := ret[1].(error)
return ret0, ret1
}
@ -101,7 +133,7 @@ func (mr *MockPluginServiceMockRecorder) CopyPlugin(ctx, req any) *gomock.Call {
}
// CreateDraftPlugin mocks base method.
func (m *MockPluginService) CreateDraftPlugin(ctx context.Context, req *dto0.CreateDraftPluginRequest) (int64, error) {
func (m *MockPluginService) CreateDraftPlugin(ctx context.Context, req *dto.CreateDraftPluginRequest) (int64, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "CreateDraftPlugin", ctx, req)
ret0, _ := ret[0].(int64)
@ -116,10 +148,10 @@ func (mr *MockPluginServiceMockRecorder) CreateDraftPlugin(ctx, req any) *gomock
}
// CreateDraftPluginWithCode mocks base method.
func (m *MockPluginService) CreateDraftPluginWithCode(ctx context.Context, req *dto0.CreateDraftPluginWithCodeRequest) (*dto0.CreateDraftPluginWithCodeResponse, error) {
func (m *MockPluginService) CreateDraftPluginWithCode(ctx context.Context, req *dto.CreateDraftPluginWithCodeRequest) (*dto.CreateDraftPluginWithCodeResponse, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "CreateDraftPluginWithCode", ctx, req)
ret0, _ := ret[0].(*dto0.CreateDraftPluginWithCodeResponse)
ret0, _ := ret[0].(*dto.CreateDraftPluginWithCodeResponse)
ret1, _ := ret[1].(error)
return ret0, ret1
}
@ -131,10 +163,10 @@ func (mr *MockPluginServiceMockRecorder) CreateDraftPluginWithCode(ctx, req any)
}
// CreateDraftToolsWithCode mocks base method.
func (m *MockPluginService) CreateDraftToolsWithCode(ctx context.Context, req *dto0.CreateDraftToolsWithCodeRequest) (*dto0.CreateDraftToolsWithCodeResponse, error) {
func (m *MockPluginService) CreateDraftToolsWithCode(ctx context.Context, req *dto.CreateDraftToolsWithCodeRequest) (*dto.CreateDraftToolsWithCodeResponse, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "CreateDraftToolsWithCode", ctx, req)
ret0, _ := ret[0].(*dto0.CreateDraftToolsWithCodeResponse)
ret0, _ := ret[0].(*dto.CreateDraftToolsWithCodeResponse)
ret1, _ := ret[1].(error)
return ret0, ret1
}
@ -224,7 +256,7 @@ func (mr *MockPluginServiceMockRecorder) GetAPPAllPlugins(ctx, appID any) *gomoc
}
// GetAccessToken mocks base method.
func (m *MockPluginService) GetAccessToken(ctx context.Context, oa *dto0.OAuthInfo) (string, error) {
func (m *MockPluginService) GetAccessToken(ctx context.Context, oa *dto.OAuthInfo) (string, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetAccessToken", ctx, oa)
ret0, _ := ret[0].(string)
@ -239,10 +271,10 @@ func (mr *MockPluginServiceMockRecorder) GetAccessToken(ctx, oa any) *gomock.Cal
}
// GetAgentPluginsOAuthStatus mocks base method.
func (m *MockPluginService) GetAgentPluginsOAuthStatus(ctx context.Context, userID, agentID int64) ([]*dto0.AgentPluginOAuthStatus, error) {
func (m *MockPluginService) GetAgentPluginsOAuthStatus(ctx context.Context, userID, agentID int64) ([]*dto.AgentPluginOAuthStatus, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetAgentPluginsOAuthStatus", ctx, userID, agentID)
ret0, _ := ret[0].([]*dto0.AgentPluginOAuthStatus)
ret0, _ := ret[0].([]*dto.AgentPluginOAuthStatus)
ret1, _ := ret[1].(error)
return ret0, ret1
}
@ -254,18 +286,18 @@ func (mr *MockPluginServiceMockRecorder) GetAgentPluginsOAuthStatus(ctx, userID,
}
// GetDraftAgentToolByName mocks base method.
func (m *MockPluginService) GetDraftAgentToolByName(ctx context.Context, agentID int64, toolName string) (*entity.ToolInfo, error) {
func (m *MockPluginService) GetDraftAgentToolByName(ctx context.Context, agentID, pluginID int64, toolName string) (*entity.ToolInfo, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetDraftAgentToolByName", ctx, agentID, toolName)
ret := m.ctrl.Call(m, "GetDraftAgentToolByName", ctx, agentID, pluginID, toolName)
ret0, _ := ret[0].(*entity.ToolInfo)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// GetDraftAgentToolByName indicates an expected call of GetDraftAgentToolByName.
func (mr *MockPluginServiceMockRecorder) GetDraftAgentToolByName(ctx, agentID, toolName any) *gomock.Call {
func (mr *MockPluginServiceMockRecorder) GetDraftAgentToolByName(ctx, agentID, pluginID, toolName any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDraftAgentToolByName", reflect.TypeOf((*MockPluginService)(nil).GetDraftAgentToolByName), ctx, agentID, toolName)
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDraftAgentToolByName", reflect.TypeOf((*MockPluginService)(nil).GetDraftAgentToolByName), ctx, agentID, pluginID, toolName)
}
// GetDraftPlugin mocks base method.
@ -284,10 +316,10 @@ func (mr *MockPluginServiceMockRecorder) GetDraftPlugin(ctx, pluginID any) *gomo
}
// GetOAuthStatus mocks base method.
func (m *MockPluginService) GetOAuthStatus(ctx context.Context, userID, pluginID int64) (*dto0.GetOAuthStatusResponse, error) {
func (m *MockPluginService) GetOAuthStatus(ctx context.Context, userID, pluginID int64) (*dto.GetOAuthStatusResponse, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetOAuthStatus", ctx, userID, pluginID)
ret0, _ := ret[0].(*dto0.GetOAuthStatusResponse)
ret0, _ := ret[0].(*dto.GetOAuthStatusResponse)
ret1, _ := ret[1].(error)
return ret0, ret1
}
@ -358,8 +390,23 @@ func (mr *MockPluginServiceMockRecorder) GetPluginProductAllTools(ctx, pluginID
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPluginProductAllTools", reflect.TypeOf((*MockPluginService)(nil).GetPluginProductAllTools), ctx, pluginID)
}
// GetSaasPluginInfo mocks base method.
func (m *MockPluginService) GetSaasPluginInfo(ctx context.Context, pluginIDs []int64) ([]*entity.PluginInfo, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetSaasPluginInfo", ctx, pluginIDs)
ret0, _ := ret[0].([]*entity.PluginInfo)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// GetSaasPluginInfo indicates an expected call of GetSaasPluginInfo.
func (mr *MockPluginServiceMockRecorder) GetSaasPluginInfo(ctx, pluginIDs any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSaasPluginInfo", reflect.TypeOf((*MockPluginService)(nil).GetSaasPluginInfo), ctx, pluginIDs)
}
// ListCustomOnlinePlugins mocks base method.
func (m *MockPluginService) ListCustomOnlinePlugins(ctx context.Context, spaceID int64, pageInfo dto0.PageInfo) ([]*entity.PluginInfo, int64, error) {
func (m *MockPluginService) ListCustomOnlinePlugins(ctx context.Context, spaceID int64, pageInfo dto.PageInfo) ([]*entity.PluginInfo, int64, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "ListCustomOnlinePlugins", ctx, spaceID, pageInfo)
ret0, _ := ret[0].([]*entity.PluginInfo)
@ -375,10 +422,10 @@ func (mr *MockPluginServiceMockRecorder) ListCustomOnlinePlugins(ctx, spaceID, p
}
// ListDraftPlugins mocks base method.
func (m *MockPluginService) ListDraftPlugins(ctx context.Context, req *dto0.ListDraftPluginsRequest) (*dto0.ListDraftPluginsResponse, error) {
func (m *MockPluginService) ListDraftPlugins(ctx context.Context, req *dto.ListDraftPluginsRequest) (*dto.ListDraftPluginsResponse, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "ListDraftPlugins", ctx, req)
ret0, _ := ret[0].(*dto0.ListDraftPluginsResponse)
ret0, _ := ret[0].(*dto.ListDraftPluginsResponse)
ret1, _ := ret[1].(error)
return ret0, ret1
}
@ -390,10 +437,10 @@ func (mr *MockPluginServiceMockRecorder) ListDraftPlugins(ctx, req any) *gomock.
}
// ListPluginProducts mocks base method.
func (m *MockPluginService) ListPluginProducts(ctx context.Context, req *dto0.ListPluginProductsRequest) (*dto0.ListPluginProductsResponse, error) {
func (m *MockPluginService) ListPluginProducts(ctx context.Context, req *dto.ListPluginProductsRequest) (*dto.ListPluginProductsResponse, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "ListPluginProducts", ctx, req)
ret0, _ := ret[0].(*dto0.ListPluginProductsResponse)
ret0, _ := ret[0].(*dto.ListPluginProductsResponse)
ret1, _ := ret[1].(error)
return ret0, ret1
}
@ -404,6 +451,36 @@ func (mr *MockPluginServiceMockRecorder) ListPluginProducts(ctx, req any) *gomoc
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListPluginProducts", reflect.TypeOf((*MockPluginService)(nil).ListPluginProducts), ctx, req)
}
// ListSaasPluginCategories mocks base method.
func (m *MockPluginService) ListSaasPluginCategories(ctx context.Context, req *dto.ListPluginCategoriesRequest) (*dto.ListPluginCategoriesResponse, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "ListSaasPluginCategories", ctx, req)
ret0, _ := ret[0].(*dto.ListPluginCategoriesResponse)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// ListSaasPluginCategories indicates an expected call of ListSaasPluginCategories.
func (mr *MockPluginServiceMockRecorder) ListSaasPluginCategories(ctx, req any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListSaasPluginCategories", reflect.TypeOf((*MockPluginService)(nil).ListSaasPluginCategories), ctx, req)
}
// ListSaasPluginProducts mocks base method.
func (m *MockPluginService) ListSaasPluginProducts(ctx context.Context, req *dto.ListSaasPluginProductsRequest) (*dto.ListPluginProductsResponse, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "ListSaasPluginProducts", ctx, req)
ret0, _ := ret[0].(*dto.ListPluginProductsResponse)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// ListSaasPluginProducts indicates an expected call of ListSaasPluginProducts.
func (mr *MockPluginServiceMockRecorder) ListSaasPluginProducts(ctx, req any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListSaasPluginProducts", reflect.TypeOf((*MockPluginService)(nil).ListSaasPluginProducts), ctx, req)
}
// MGetAgentTools mocks base method.
func (m *MockPluginService) MGetAgentTools(ctx context.Context, req *model.MGetAgentToolsRequest) ([]*entity.ToolInfo, error) {
m.ctrl.T.Helper()
@ -540,7 +617,7 @@ func (mr *MockPluginServiceMockRecorder) MoveAPPPluginToLibrary(ctx, pluginID an
}
// OAuthCode mocks base method.
func (m *MockPluginService) OAuthCode(ctx context.Context, code string, state *dto0.OAuthState) error {
func (m *MockPluginService) OAuthCode(ctx context.Context, code string, state *dto.OAuthState) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "OAuthCode", ctx, code, state)
ret0, _ := ret[0].(error)
@ -597,7 +674,7 @@ func (mr *MockPluginServiceMockRecorder) PublishPlugin(ctx, req any) *gomock.Cal
}
// RevokeAccessToken mocks base method.
func (m *MockPluginService) RevokeAccessToken(ctx context.Context, meta *dto0.AuthorizationCodeMeta) error {
func (m *MockPluginService) RevokeAccessToken(ctx context.Context, meta *dto.AuthorizationCodeMeta) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "RevokeAccessToken", ctx, meta)
ret0, _ := ret[0].(error)
@ -611,7 +688,7 @@ func (mr *MockPluginServiceMockRecorder) RevokeAccessToken(ctx, meta any) *gomoc
}
// UpdateBotDefaultParams mocks base method.
func (m *MockPluginService) UpdateBotDefaultParams(ctx context.Context, req *dto0.UpdateBotDefaultParamsRequest) error {
func (m *MockPluginService) UpdateBotDefaultParams(ctx context.Context, req *dto.UpdateBotDefaultParamsRequest) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "UpdateBotDefaultParams", ctx, req)
ret0, _ := ret[0].(error)
@ -625,7 +702,7 @@ func (mr *MockPluginServiceMockRecorder) UpdateBotDefaultParams(ctx, req any) *g
}
// UpdateDraftPlugin mocks base method.
func (m *MockPluginService) UpdateDraftPlugin(ctx context.Context, plugin *dto0.UpdateDraftPluginRequest) error {
func (m *MockPluginService) UpdateDraftPlugin(ctx context.Context, plugin *dto.UpdateDraftPluginRequest) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "UpdateDraftPlugin", ctx, plugin)
ret0, _ := ret[0].(error)
@ -639,7 +716,7 @@ func (mr *MockPluginServiceMockRecorder) UpdateDraftPlugin(ctx, plugin any) *gom
}
// UpdateDraftPluginWithCode mocks base method.
func (m *MockPluginService) UpdateDraftPluginWithCode(ctx context.Context, req *dto0.UpdateDraftPluginWithCodeRequest) error {
func (m *MockPluginService) UpdateDraftPluginWithCode(ctx context.Context, req *dto.UpdateDraftPluginWithCodeRequest) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "UpdateDraftPluginWithCode", ctx, req)
ret0, _ := ret[0].(error)
@ -653,7 +730,7 @@ func (mr *MockPluginServiceMockRecorder) UpdateDraftPluginWithCode(ctx, req any)
}
// UpdateDraftTool mocks base method.
func (m *MockPluginService) UpdateDraftTool(ctx context.Context, req *dto0.UpdateDraftToolRequest) error {
func (m *MockPluginService) UpdateDraftTool(ctx context.Context, req *dto.UpdateDraftToolRequest) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "UpdateDraftTool", ctx, req)
ret0, _ := ret[0].(error)