refactor: update visibilities

This commit is contained in:
Steven
2024-08-31 22:04:24 +08:00
parent 8511c09c63
commit b74e4f90c1
43 changed files with 693 additions and 775 deletions

View File

@@ -4,20 +4,10 @@ import (
"context"
v1pb "github.com/yourselfhosted/slash/proto/gen/api/v1"
storepb "github.com/yourselfhosted/slash/proto/gen/store"
"github.com/yourselfhosted/slash/store"
)
func convertRowStatusFromStore(rowStatus store.RowStatus) v1pb.RowStatus {
switch rowStatus {
case store.Normal:
return v1pb.RowStatus_NORMAL
case store.Archived:
return v1pb.RowStatus_ARCHIVED
default:
return v1pb.RowStatus_ROW_STATUS_UNSPECIFIED
}
}
func getCurrentUser(ctx context.Context, s *store.Store) (*store.User, error) {
userID, ok := ctx.Value(userIDContextKey).(int32)
if !ok {
@@ -31,3 +21,48 @@ func getCurrentUser(ctx context.Context, s *store.Store) (*store.User, error) {
}
return user, nil
}
func convertStateFromRowStatus(rowStatus storepb.RowStatus) v1pb.State {
switch rowStatus {
case storepb.RowStatus_NORMAL:
return v1pb.State_ACTIVE
case storepb.RowStatus_ARCHIVED:
return v1pb.State_INACTIVE
default:
return v1pb.State_STATE_UNSPECIFIED
}
}
// ConvertStateToRowStatus converts a v1pb.State to a storepb.RowStatus.
func ConvertStateToRowStatus(state v1pb.State) storepb.RowStatus {
switch state {
case v1pb.State_ACTIVE:
return storepb.RowStatus_NORMAL
case v1pb.State_INACTIVE:
return storepb.RowStatus_ARCHIVED
default:
return storepb.RowStatus_ROW_STATUS_UNSPECIFIED
}
}
func convertVisibilityFromStorepb(visibility storepb.Visibility) v1pb.Visibility {
switch visibility {
case storepb.Visibility_WORKSPACE:
return v1pb.Visibility_WORKSPACE
case storepb.Visibility_PUBLIC:
return v1pb.Visibility_PUBLIC
default:
return v1pb.Visibility_VISIBILITY_UNSPECIFIED
}
}
func convertVisibilityToStorepb(visibility v1pb.Visibility) storepb.Visibility {
switch visibility {
case v1pb.Visibility_PRIVATE, v1pb.Visibility_WORKSPACE:
return storepb.Visibility_WORKSPACE
case v1pb.Visibility_PUBLIC:
return storepb.Visibility_PUBLIC
default:
return storepb.Visibility_VISIBILITY_UNSPECIFIED
}
}