mirror of
https://github.com/usememos/memos.git
synced 2025-10-31 16:59:30 +08:00
263 lines
10 KiB
Go
263 lines
10 KiB
Go
package v1
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
|
|
v1pb "github.com/usememos/memos/proto/gen/api/v1"
|
|
storepb "github.com/usememos/memos/proto/gen/store"
|
|
"github.com/usememos/memos/store"
|
|
)
|
|
|
|
func (s *APIV1Service) GetWorkspaceSetting(ctx context.Context, request *v1pb.GetWorkspaceSettingRequest) (*v1pb.WorkspaceSetting, error) {
|
|
workspaceSettingKeyString, err := ExtractWorkspaceSettingKeyFromName(request.Name)
|
|
if err != nil {
|
|
return nil, status.Errorf(codes.InvalidArgument, "invalid workspace setting name: %v", err)
|
|
}
|
|
|
|
workspaceSettingKey := storepb.WorkspaceSettingKey(storepb.WorkspaceSettingKey_value[workspaceSettingKeyString])
|
|
// Get workspace setting from store with default value.
|
|
switch workspaceSettingKey {
|
|
case storepb.WorkspaceSettingKey_BASIC:
|
|
_, err = s.Store.GetWorkspaceBasicSetting(ctx)
|
|
case storepb.WorkspaceSettingKey_GENERAL:
|
|
_, err = s.Store.GetWorkspaceGeneralSetting(ctx)
|
|
case storepb.WorkspaceSettingKey_MEMO_RELATED:
|
|
_, err = s.Store.GetWorkspaceMemoRelatedSetting(ctx)
|
|
case storepb.WorkspaceSettingKey_STORAGE:
|
|
_, err = s.Store.GetWorkspaceStorageSetting(ctx)
|
|
default:
|
|
return nil, status.Errorf(codes.InvalidArgument, "unsupported workspace setting key: %v", workspaceSettingKey)
|
|
}
|
|
if err != nil {
|
|
return nil, status.Errorf(codes.Internal, "failed to get workspace setting: %v", err)
|
|
}
|
|
|
|
workspaceSetting, err := s.Store.GetWorkspaceSetting(ctx, &store.FindWorkspaceSetting{
|
|
Name: workspaceSettingKey.String(),
|
|
})
|
|
if err != nil {
|
|
return nil, status.Errorf(codes.Internal, "failed to get workspace setting: %v", err)
|
|
}
|
|
if workspaceSetting == nil {
|
|
return nil, status.Errorf(codes.NotFound, "workspace setting not found")
|
|
}
|
|
|
|
// For storage setting, only host can get it.
|
|
if workspaceSetting.Key == storepb.WorkspaceSettingKey_STORAGE {
|
|
user, err := s.GetCurrentUser(ctx)
|
|
if err != nil {
|
|
return nil, status.Errorf(codes.Internal, "failed to get current user: %v", err)
|
|
}
|
|
if user == nil || user.Role != store.RoleHost {
|
|
return nil, status.Errorf(codes.PermissionDenied, "permission denied")
|
|
}
|
|
}
|
|
|
|
return convertWorkspaceSettingFromStore(workspaceSetting), nil
|
|
}
|
|
|
|
func (s *APIV1Service) SetWorkspaceSetting(ctx context.Context, request *v1pb.SetWorkspaceSettingRequest) (*v1pb.WorkspaceSetting, error) {
|
|
user, err := s.GetCurrentUser(ctx)
|
|
if err != nil {
|
|
return nil, status.Errorf(codes.Internal, "failed to get current user: %v", err)
|
|
}
|
|
if user.Role != store.RoleHost {
|
|
return nil, status.Errorf(codes.PermissionDenied, "permission denied")
|
|
}
|
|
|
|
updateSetting := convertWorkspaceSettingToStore(request.Setting)
|
|
// Don't allow to update workspace general setting in demo mode.
|
|
// Such as disallow user registration, disallow password auth, etc.
|
|
if s.Profile.Mode == "demo" && updateSetting.Key == storepb.WorkspaceSettingKey_GENERAL {
|
|
return nil, status.Errorf(codes.InvalidArgument, "setting workspace setting is not allowed in demo mode")
|
|
}
|
|
|
|
workspaceSetting, err := s.Store.UpsertWorkspaceSetting(ctx, updateSetting)
|
|
if err != nil {
|
|
return nil, status.Errorf(codes.Internal, "failed to upsert workspace setting: %v", err)
|
|
}
|
|
|
|
return convertWorkspaceSettingFromStore(workspaceSetting), nil
|
|
}
|
|
|
|
func convertWorkspaceSettingFromStore(setting *storepb.WorkspaceSetting) *v1pb.WorkspaceSetting {
|
|
workspaceSetting := &v1pb.WorkspaceSetting{
|
|
Name: fmt.Sprintf("%s%s", WorkspaceSettingNamePrefix, setting.Key.String()),
|
|
}
|
|
switch setting.Value.(type) {
|
|
case *storepb.WorkspaceSetting_GeneralSetting:
|
|
workspaceSetting.Value = &v1pb.WorkspaceSetting_GeneralSetting{
|
|
GeneralSetting: convertWorkspaceGeneralSettingFromStore(setting.GetGeneralSetting()),
|
|
}
|
|
case *storepb.WorkspaceSetting_StorageSetting:
|
|
workspaceSetting.Value = &v1pb.WorkspaceSetting_StorageSetting{
|
|
StorageSetting: convertWorkspaceStorageSettingFromStore(setting.GetStorageSetting()),
|
|
}
|
|
case *storepb.WorkspaceSetting_MemoRelatedSetting:
|
|
workspaceSetting.Value = &v1pb.WorkspaceSetting_MemoRelatedSetting{
|
|
MemoRelatedSetting: convertWorkspaceMemoRelatedSettingFromStore(setting.GetMemoRelatedSetting()),
|
|
}
|
|
}
|
|
return workspaceSetting
|
|
}
|
|
|
|
func convertWorkspaceSettingToStore(setting *v1pb.WorkspaceSetting) *storepb.WorkspaceSetting {
|
|
settingKeyString, _ := ExtractWorkspaceSettingKeyFromName(setting.Name)
|
|
workspaceSetting := &storepb.WorkspaceSetting{
|
|
Key: storepb.WorkspaceSettingKey(storepb.WorkspaceSettingKey_value[settingKeyString]),
|
|
Value: &storepb.WorkspaceSetting_GeneralSetting{
|
|
GeneralSetting: convertWorkspaceGeneralSettingToStore(setting.GetGeneralSetting()),
|
|
},
|
|
}
|
|
switch workspaceSetting.Key {
|
|
case storepb.WorkspaceSettingKey_GENERAL:
|
|
workspaceSetting.Value = &storepb.WorkspaceSetting_GeneralSetting{
|
|
GeneralSetting: convertWorkspaceGeneralSettingToStore(setting.GetGeneralSetting()),
|
|
}
|
|
case storepb.WorkspaceSettingKey_STORAGE:
|
|
workspaceSetting.Value = &storepb.WorkspaceSetting_StorageSetting{
|
|
StorageSetting: convertWorkspaceStorageSettingToStore(setting.GetStorageSetting()),
|
|
}
|
|
case storepb.WorkspaceSettingKey_MEMO_RELATED:
|
|
workspaceSetting.Value = &storepb.WorkspaceSetting_MemoRelatedSetting{
|
|
MemoRelatedSetting: convertWorkspaceMemoRelatedSettingToStore(setting.GetMemoRelatedSetting()),
|
|
}
|
|
}
|
|
return workspaceSetting
|
|
}
|
|
|
|
func convertWorkspaceGeneralSettingFromStore(setting *storepb.WorkspaceGeneralSetting) *v1pb.WorkspaceGeneralSetting {
|
|
if setting == nil {
|
|
return nil
|
|
}
|
|
generalSetting := &v1pb.WorkspaceGeneralSetting{
|
|
DisallowUserRegistration: setting.DisallowUserRegistration,
|
|
DisallowPasswordAuth: setting.DisallowPasswordAuth,
|
|
AdditionalScript: setting.AdditionalScript,
|
|
AdditionalStyle: setting.AdditionalStyle,
|
|
WeekStartDayOffset: setting.WeekStartDayOffset,
|
|
DisallowChangeUsername: setting.DisallowChangeUsername,
|
|
DisallowChangeNickname: setting.DisallowChangeNickname,
|
|
}
|
|
if setting.CustomProfile != nil {
|
|
generalSetting.CustomProfile = &v1pb.WorkspaceCustomProfile{
|
|
Title: setting.CustomProfile.Title,
|
|
Description: setting.CustomProfile.Description,
|
|
LogoUrl: setting.CustomProfile.LogoUrl,
|
|
Locale: setting.CustomProfile.Locale,
|
|
Appearance: setting.CustomProfile.Appearance,
|
|
}
|
|
}
|
|
return generalSetting
|
|
}
|
|
|
|
func convertWorkspaceGeneralSettingToStore(setting *v1pb.WorkspaceGeneralSetting) *storepb.WorkspaceGeneralSetting {
|
|
if setting == nil {
|
|
return nil
|
|
}
|
|
generalSetting := &storepb.WorkspaceGeneralSetting{
|
|
DisallowUserRegistration: setting.DisallowUserRegistration,
|
|
DisallowPasswordAuth: setting.DisallowPasswordAuth,
|
|
AdditionalScript: setting.AdditionalScript,
|
|
AdditionalStyle: setting.AdditionalStyle,
|
|
WeekStartDayOffset: setting.WeekStartDayOffset,
|
|
DisallowChangeUsername: setting.DisallowChangeUsername,
|
|
DisallowChangeNickname: setting.DisallowChangeNickname,
|
|
}
|
|
if setting.CustomProfile != nil {
|
|
generalSetting.CustomProfile = &storepb.WorkspaceCustomProfile{
|
|
Title: setting.CustomProfile.Title,
|
|
Description: setting.CustomProfile.Description,
|
|
LogoUrl: setting.CustomProfile.LogoUrl,
|
|
Locale: setting.CustomProfile.Locale,
|
|
Appearance: setting.CustomProfile.Appearance,
|
|
}
|
|
}
|
|
return generalSetting
|
|
}
|
|
|
|
func convertWorkspaceStorageSettingFromStore(settingpb *storepb.WorkspaceStorageSetting) *v1pb.WorkspaceStorageSetting {
|
|
if settingpb == nil {
|
|
return nil
|
|
}
|
|
setting := &v1pb.WorkspaceStorageSetting{
|
|
StorageType: v1pb.WorkspaceStorageSetting_StorageType(settingpb.StorageType),
|
|
FilepathTemplate: settingpb.FilepathTemplate,
|
|
UploadSizeLimitMb: settingpb.UploadSizeLimitMb,
|
|
}
|
|
if settingpb.S3Config != nil {
|
|
setting.S3Config = &v1pb.WorkspaceStorageSetting_S3Config{
|
|
AccessKeyId: settingpb.S3Config.AccessKeyId,
|
|
AccessKeySecret: settingpb.S3Config.AccessKeySecret,
|
|
Endpoint: settingpb.S3Config.Endpoint,
|
|
Region: settingpb.S3Config.Region,
|
|
Bucket: settingpb.S3Config.Bucket,
|
|
UsePathStyle: settingpb.S3Config.UsePathStyle,
|
|
}
|
|
}
|
|
return setting
|
|
}
|
|
|
|
func convertWorkspaceStorageSettingToStore(setting *v1pb.WorkspaceStorageSetting) *storepb.WorkspaceStorageSetting {
|
|
if setting == nil {
|
|
return nil
|
|
}
|
|
settingpb := &storepb.WorkspaceStorageSetting{
|
|
StorageType: storepb.WorkspaceStorageSetting_StorageType(setting.StorageType),
|
|
FilepathTemplate: setting.FilepathTemplate,
|
|
UploadSizeLimitMb: setting.UploadSizeLimitMb,
|
|
}
|
|
if setting.S3Config != nil {
|
|
settingpb.S3Config = &storepb.StorageS3Config{
|
|
AccessKeyId: setting.S3Config.AccessKeyId,
|
|
AccessKeySecret: setting.S3Config.AccessKeySecret,
|
|
Endpoint: setting.S3Config.Endpoint,
|
|
Region: setting.S3Config.Region,
|
|
Bucket: setting.S3Config.Bucket,
|
|
UsePathStyle: setting.S3Config.UsePathStyle,
|
|
}
|
|
}
|
|
return settingpb
|
|
}
|
|
|
|
func convertWorkspaceMemoRelatedSettingFromStore(setting *storepb.WorkspaceMemoRelatedSetting) *v1pb.WorkspaceMemoRelatedSetting {
|
|
if setting == nil {
|
|
return nil
|
|
}
|
|
return &v1pb.WorkspaceMemoRelatedSetting{
|
|
DisallowPublicVisibility: setting.DisallowPublicVisibility,
|
|
DisplayWithUpdateTime: setting.DisplayWithUpdateTime,
|
|
ContentLengthLimit: setting.ContentLengthLimit,
|
|
EnableAutoCompact: setting.EnableAutoCompact,
|
|
EnableDoubleClickEdit: setting.EnableDoubleClickEdit,
|
|
EnableLinkPreview: setting.EnableLinkPreview,
|
|
EnableComment: setting.EnableComment,
|
|
EnableLocation: setting.EnableLocation,
|
|
DefaultVisibility: setting.DefaultVisibility,
|
|
Reactions: setting.Reactions,
|
|
DisableMarkdownShortcuts: setting.DisableMarkdownShortcuts,
|
|
}
|
|
}
|
|
|
|
func convertWorkspaceMemoRelatedSettingToStore(setting *v1pb.WorkspaceMemoRelatedSetting) *storepb.WorkspaceMemoRelatedSetting {
|
|
if setting == nil {
|
|
return nil
|
|
}
|
|
return &storepb.WorkspaceMemoRelatedSetting{
|
|
DisallowPublicVisibility: setting.DisallowPublicVisibility,
|
|
DisplayWithUpdateTime: setting.DisplayWithUpdateTime,
|
|
ContentLengthLimit: setting.ContentLengthLimit,
|
|
EnableAutoCompact: setting.EnableAutoCompact,
|
|
EnableDoubleClickEdit: setting.EnableDoubleClickEdit,
|
|
EnableLinkPreview: setting.EnableLinkPreview,
|
|
EnableComment: setting.EnableComment,
|
|
EnableLocation: setting.EnableLocation,
|
|
DefaultVisibility: setting.DefaultVisibility,
|
|
Reactions: setting.Reactions,
|
|
DisableMarkdownShortcuts: setting.DisableMarkdownShortcuts,
|
|
}
|
|
}
|