mlogclub
5d7c10aeab
- Updated runtime configuration to use __CS_AI_AGENT_WIDGET_CONFIG__ instead of __CS_AGENT_WIDGET_CONFIG__. - Changed message types in support host bridge from "cs-agent" to "cs-ai-agent". - Minified SDK script updated to reflect new AI agent naming conventions. - Adjusted scrollbar styles in main.scss to use .cs-ai-agent-scrollbar instead of .cs-agent-scrollbar.
51 lines
1.2 KiB
Go
51 lines
1.2 KiB
Go
package storage
|
|
|
|
import (
|
|
"cs-ai-agent/internal/pkg/config"
|
|
"cs-ai-agent/internal/pkg/enums"
|
|
"cs-ai-agent/internal/pkg/errorsx"
|
|
"io"
|
|
)
|
|
|
|
var providers = make(map[enums.AssetProvider]FileStorageProvider)
|
|
|
|
type FileStorageProvider interface {
|
|
ProviderType() enums.AssetProvider
|
|
Upload(reader io.Reader, key string, info UploadInfo) (*StoredFile, error)
|
|
GetURL(key string) string
|
|
GetSignedURL(key string) string
|
|
Delete(key string) error
|
|
Read(key string) (io.ReadCloser, error)
|
|
}
|
|
|
|
func GetDefault() (FileStorageProvider, error) {
|
|
return NewProvider(config.Current().Storage.Default)
|
|
}
|
|
|
|
func GetProvider(providerType enums.AssetProvider) (FileStorageProvider, error) {
|
|
if provider, exists := providers[providerType]; exists {
|
|
return provider, nil
|
|
}
|
|
provider, err := NewProvider(providerType)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
providers[providerType] = provider
|
|
|
|
return provider, nil
|
|
}
|
|
|
|
func NewProvider(provider enums.AssetProvider) (FileStorageProvider, error) {
|
|
cfg := config.Current().Storage
|
|
|
|
switch provider {
|
|
case "", enums.AssetProviderLocal:
|
|
return NewLocalStorage(cfg.Local), nil
|
|
case enums.AssetProviderOSS:
|
|
return NewOSSStorage(cfg.OSS), nil
|
|
default:
|
|
return nil, errorsx.InvalidParam("不支持的文件存储类型")
|
|
}
|
|
}
|