coze-studio/backend/domain/plugin/repository/option.go

86 lines
2.0 KiB
Go

/*
* 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 repository
import (
"github.com/coze-dev/coze-studio/backend/domain/plugin/internal/dal"
)
type PluginSelectedOptions func(*dal.PluginSelectedOption)
func WithPluginID() PluginSelectedOptions {
return func(opts *dal.PluginSelectedOption) {
opts.PluginID = true
}
}
func WithPluginOpenapiDoc() PluginSelectedOptions {
return func(opts *dal.PluginSelectedOption) {
opts.OpenapiDoc = true
}
}
func WithPluginManifest() PluginSelectedOptions {
return func(opts *dal.PluginSelectedOption) {
opts.Manifest = true
}
}
func WithPluginIconURI() PluginSelectedOptions {
return func(opts *dal.PluginSelectedOption) {
opts.IconURI = true
}
}
func WithPluginVersion() PluginSelectedOptions {
return func(opts *dal.PluginSelectedOption) {
opts.Version = true
}
}
type ToolSelectedOptions func(option *dal.ToolSelectedOption)
func WithToolID() ToolSelectedOptions {
return func(opts *dal.ToolSelectedOption) {
opts.ToolID = true
}
}
func WithToolMethod() ToolSelectedOptions {
return func(opts *dal.ToolSelectedOption) {
opts.ToolMethod = true
}
}
func WithToolSubURL() ToolSelectedOptions {
return func(opts *dal.ToolSelectedOption) {
opts.ToolSubURL = true
}
}
func WithToolActivatedStatus() ToolSelectedOptions {
return func(opts *dal.ToolSelectedOption) {
opts.ActivatedStatus = true
}
}
func WithToolDebugStatus() ToolSelectedOptions {
return func(opts *dal.ToolSelectedOption) {
opts.DebugStatus = true
}
}