mattermost/server/public/model/scheme.go
2025-03-31 10:44:34 +02:00

351 lines
12 KiB
Go

// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
// See LICENSE.txt for license information.
package model
import (
"fmt"
"regexp"
"github.com/mattermost/mattermost/server/public/utils/timeutils"
)
const (
SchemeDisplayNameMaxLength = 128
SchemeNameMaxLength = 64
SchemeDescriptionMaxLength = 1024
SchemeScopeTeam = "team"
SchemeScopeChannel = "channel"
SchemeScopePlaybook = "playbook"
SchemeScopeRun = "run"
)
type Scheme struct {
Id string `json:"id"`
Name string `json:"name"`
DisplayName string `json:"display_name"`
Description string `json:"description"`
CreateAt int64 `json:"create_at"`
UpdateAt int64 `json:"update_at"`
DeleteAt int64 `json:"delete_at"`
Scope string `json:"scope"`
DefaultTeamAdminRole string `json:"default_team_admin_role"`
DefaultTeamUserRole string `json:"default_team_user_role"`
DefaultChannelAdminRole string `json:"default_channel_admin_role"`
DefaultChannelUserRole string `json:"default_channel_user_role"`
DefaultTeamGuestRole string `json:"default_team_guest_role"`
DefaultChannelGuestRole string `json:"default_channel_guest_role"`
DefaultPlaybookAdminRole string `json:"default_playbook_admin_role"`
DefaultPlaybookMemberRole string `json:"default_playbook_member_role"`
DefaultRunAdminRole string `json:"default_run_admin_role"`
DefaultRunMemberRole string `json:"default_run_member_role"`
}
func (scheme *Scheme) Auditable() map[string]any {
return map[string]any{
"id": scheme.Id,
"name": scheme.Name,
"display_name": scheme.DisplayName,
"description": scheme.Description,
"create_at": scheme.CreateAt,
"update_at": scheme.UpdateAt,
"delete_at": scheme.DeleteAt,
"scope": scheme.Scope,
"default_team_admin_role": scheme.DefaultTeamAdminRole,
"default_team_user_role": scheme.DefaultTeamUserRole,
"default_channel_admin_role": scheme.DefaultChannelAdminRole,
"default_channel_user_role": scheme.DefaultChannelUserRole,
"default_team_guest_role": scheme.DefaultTeamGuestRole,
"default_channel_guest_role": scheme.DefaultChannelGuestRole,
"default_playbook_admin_role": scheme.DefaultPlaybookAdminRole,
"default_playbook_member_role": scheme.DefaultPlaybookMemberRole,
"default_run_admin_role": scheme.DefaultRunAdminRole,
"default_run_member_role": scheme.DefaultRunMemberRole,
}
}
func (scheme *Scheme) Sanitize() {
scheme.Name = FakeSetting
scheme.DisplayName = FakeSetting
scheme.Description = FakeSetting
}
func (scheme *Scheme) MarshalYAML() (any, error) {
return struct {
Id string `yaml:"id"`
Name string `yaml:"name"`
DisplayName string `yaml:"display_name"`
Description string `yaml:"description"`
CreateAt string `yaml:"create_at"`
UpdateAt string `yaml:"update_at"`
DeleteAt string `yaml:"delete_at"`
Scope string `yaml:"scope"`
DefaultTeamAdminRole string `yaml:"default_team_admin_role"`
DefaultTeamUserRole string `yaml:"default_team_user_role"`
DefaultChannelAdminRole string `yaml:"default_channel_admin_role"`
DefaultChannelUserRole string `yaml:"default_channel_user_role"`
DefaultTeamGuestRole string `yaml:"default_team_guest_role"`
DefaultChannelGuestRole string `yaml:"default_channel_guest_role"`
DefaultPlaybookAdminRole string `yaml:"default_playbook_admin_role"`
DefaultPlaybookMemberRole string `yaml:"default_playbook_member_role"`
DefaultRunAdminRole string `yaml:"default_run_admin_role"`
DefaultRunMemberRole string `yaml:"default_run_member_role"`
}{
Id: scheme.Id,
Name: scheme.Name,
DisplayName: scheme.DisplayName,
Description: scheme.Description,
CreateAt: timeutils.FormatMillis(scheme.CreateAt),
UpdateAt: timeutils.FormatMillis(scheme.UpdateAt),
DeleteAt: timeutils.FormatMillis(scheme.DeleteAt),
Scope: scheme.Scope,
DefaultTeamAdminRole: scheme.DefaultTeamAdminRole,
DefaultTeamUserRole: scheme.DefaultTeamUserRole,
DefaultChannelAdminRole: scheme.DefaultChannelAdminRole,
DefaultChannelUserRole: scheme.DefaultChannelUserRole,
DefaultTeamGuestRole: scheme.DefaultTeamGuestRole,
DefaultChannelGuestRole: scheme.DefaultChannelGuestRole,
DefaultPlaybookAdminRole: scheme.DefaultPlaybookAdminRole,
DefaultPlaybookMemberRole: scheme.DefaultPlaybookMemberRole,
DefaultRunAdminRole: scheme.DefaultRunAdminRole,
DefaultRunMemberRole: scheme.DefaultRunMemberRole,
}, nil
}
func (scheme *Scheme) UnmarshalYAML(unmarshal func(any) error) error {
out := struct {
Id string `yaml:"id"`
Name string `yaml:"name"`
DisplayName string `yaml:"display_name"`
Description string `yaml:"description"`
CreateAt string `yaml:"create_at"`
UpdateAt string `yaml:"update_at"`
DeleteAt string `yaml:"delete_at"`
Scope string `yaml:"scope"`
DefaultTeamAdminRole string `yaml:"default_team_admin_role"`
DefaultTeamUserRole string `yaml:"default_team_user_role"`
DefaultChannelAdminRole string `yaml:"default_channel_admin_role"`
DefaultChannelUserRole string `yaml:"default_channel_user_role"`
DefaultTeamGuestRole string `yaml:"default_team_guest_role"`
DefaultChannelGuestRole string `yaml:"default_channel_guest_role"`
DefaultPlaybookAdminRole string `yaml:"default_playbook_admin_role"`
DefaultPlaybookMemberRole string `yaml:"default_playbook_member_role"`
DefaultRunAdminRole string `yaml:"default_run_admin_role"`
DefaultRunMemberRole string `yaml:"default_run_member_role"`
}{}
err := unmarshal(&out)
if err != nil {
return err
}
createAt, err := timeutils.ParseFormatedMillis(out.CreateAt)
if err != nil {
return err
}
updateAt, err := timeutils.ParseFormatedMillis(out.UpdateAt)
if err != nil {
return err
}
deleteAt, err := timeutils.ParseFormatedMillis(out.DeleteAt)
if err != nil {
return err
}
*scheme = Scheme{
Id: out.Id,
Name: out.Name,
DisplayName: out.DisplayName,
Description: out.Description,
CreateAt: createAt,
UpdateAt: updateAt,
DeleteAt: deleteAt,
Scope: out.Scope,
DefaultTeamAdminRole: out.DefaultTeamAdminRole,
DefaultTeamUserRole: out.DefaultTeamUserRole,
DefaultChannelAdminRole: out.DefaultChannelAdminRole,
DefaultChannelUserRole: out.DefaultChannelUserRole,
DefaultTeamGuestRole: out.DefaultTeamGuestRole,
DefaultChannelGuestRole: out.DefaultChannelGuestRole,
DefaultPlaybookAdminRole: out.DefaultPlaybookAdminRole,
DefaultPlaybookMemberRole: out.DefaultPlaybookMemberRole,
DefaultRunAdminRole: out.DefaultRunAdminRole,
DefaultRunMemberRole: out.DefaultRunMemberRole,
}
return nil
}
type SchemePatch struct {
Name *string `json:"name"`
DisplayName *string `json:"display_name"`
Description *string `json:"description"`
}
func (scheme *SchemePatch) Auditable() map[string]any {
return map[string]any{
"name": scheme.Name,
"display_name": scheme.DisplayName,
"description": scheme.Description,
}
}
type SchemeIDPatch struct {
SchemeID *string `json:"scheme_id"`
}
func (p *SchemeIDPatch) Auditable() map[string]any {
return map[string]any{
"scheme_id": p.SchemeID,
}
}
// SchemeConveyor is used for importing and exporting a Scheme and its associated Roles.
type SchemeConveyor struct {
Name string `json:"name"`
DisplayName string `json:"display_name"`
Description string `json:"description"`
Scope string `json:"scope"`
TeamAdmin string `json:"default_team_admin_role"`
TeamUser string `json:"default_team_user_role"`
TeamGuest string `json:"default_team_guest_role"`
ChannelAdmin string `json:"default_channel_admin_role"`
ChannelUser string `json:"default_channel_user_role"`
ChannelGuest string `json:"default_channel_guest_role"`
PlaybookAdmin string `json:"default_playbook_admin_role"`
PlaybookMember string `json:"default_playbook_member_role"`
RunAdmin string `json:"default_run_admin_role"`
RunMember string `json:"default_run_member_role"`
Roles []*Role `json:"roles"`
}
func (sc *SchemeConveyor) Scheme() *Scheme {
return &Scheme{
DisplayName: sc.DisplayName,
Name: sc.Name,
Description: sc.Description,
Scope: sc.Scope,
DefaultTeamAdminRole: sc.TeamAdmin,
DefaultTeamUserRole: sc.TeamUser,
DefaultTeamGuestRole: sc.TeamGuest,
DefaultChannelAdminRole: sc.ChannelAdmin,
DefaultChannelUserRole: sc.ChannelUser,
DefaultChannelGuestRole: sc.ChannelGuest,
DefaultPlaybookAdminRole: sc.PlaybookAdmin,
DefaultPlaybookMemberRole: sc.PlaybookMember,
DefaultRunAdminRole: sc.RunAdmin,
DefaultRunMemberRole: sc.RunMember,
}
}
type SchemeRoles struct {
SchemeAdmin bool `json:"scheme_admin"`
SchemeUser bool `json:"scheme_user"`
SchemeGuest bool `json:"scheme_guest"`
}
func (s *SchemeRoles) Auditable() map[string]any {
return map[string]any{}
}
func (scheme *Scheme) IsValid() bool {
if !IsValidId(scheme.Id) {
return false
}
return scheme.IsValidForCreate()
}
func (scheme *Scheme) IsValidForCreate() bool {
if scheme.DisplayName == "" || len(scheme.DisplayName) > SchemeDisplayNameMaxLength {
return false
}
if !IsValidSchemeName(scheme.Name) {
return false
}
if len(scheme.Description) > SchemeDescriptionMaxLength {
return false
}
switch scheme.Scope {
case SchemeScopeTeam, SchemeScopeChannel, SchemeScopePlaybook, SchemeScopeRun:
default:
return false
}
if !IsValidRoleName(scheme.DefaultChannelAdminRole) {
return false
}
if !IsValidRoleName(scheme.DefaultChannelUserRole) {
return false
}
if !IsValidRoleName(scheme.DefaultChannelGuestRole) {
return false
}
if scheme.Scope == SchemeScopeTeam {
if !IsValidRoleName(scheme.DefaultTeamAdminRole) {
return false
}
if !IsValidRoleName(scheme.DefaultTeamUserRole) {
return false
}
if !IsValidRoleName(scheme.DefaultTeamGuestRole) {
return false
}
if !IsValidRoleName(scheme.DefaultPlaybookAdminRole) {
return false
}
if !IsValidRoleName(scheme.DefaultPlaybookMemberRole) {
return false
}
if !IsValidRoleName(scheme.DefaultRunAdminRole) {
return false
}
if !IsValidRoleName(scheme.DefaultRunMemberRole) {
return false
}
}
if scheme.Scope == SchemeScopeChannel {
if scheme.DefaultTeamAdminRole != "" {
return false
}
if scheme.DefaultTeamUserRole != "" {
return false
}
if scheme.DefaultTeamGuestRole != "" {
return false
}
}
return true
}
func (scheme *Scheme) Patch(patch *SchemePatch) {
if patch.DisplayName != nil {
scheme.DisplayName = *patch.DisplayName
}
if patch.Name != nil {
scheme.Name = *patch.Name
}
if patch.Description != nil {
scheme.Description = *patch.Description
}
}
func IsValidSchemeName(name string) bool {
re := regexp.MustCompile(fmt.Sprintf("^[a-z0-9_]{2,%d}$", SchemeNameMaxLength))
return re.MatchString(name)
}