mirror of
https://github.com/andwati/zitadel.git
synced 2026-11-09 07:35:13 +00:00
Livio AmstutzandGitHub
32f8545082
* fix: check correct permission for label policy in console policy grid * fix: add next steps for label policy detail * fix: add label policy link in console home * fix: i18n of EN features * fix: set font color on preview * fix: handle label policy remove in write model * fix: add more content-types for font upload * fix: add label policy link in console home * fix: dont return error when bucket does not exists * fix: iam access pref * fix: EN i18n for apply button
383 lines
11 KiB
Go
383 lines
11 KiB
Go
package assets
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
|
|
"github.com/caos/zitadel/internal/api/authz"
|
|
"github.com/caos/zitadel/internal/command"
|
|
"github.com/caos/zitadel/internal/domain"
|
|
"github.com/caos/zitadel/internal/iam/model"
|
|
"github.com/caos/zitadel/internal/id"
|
|
"github.com/caos/zitadel/internal/management/repository"
|
|
)
|
|
|
|
func (h *Handler) UploadDefaultLabelPolicyLogo() Uploader {
|
|
return &labelPolicyLogoUploader{h.idGenerator, false, true, []string{"image/"}, 1 << 19}
|
|
}
|
|
|
|
func (h *Handler) UploadDefaultLabelPolicyLogoDark() Uploader {
|
|
return &labelPolicyLogoUploader{h.idGenerator, true, true, []string{"image/"}, 1 << 19}
|
|
}
|
|
|
|
func (h *Handler) UploadOrgLabelPolicyLogo() Uploader {
|
|
return &labelPolicyLogoUploader{h.idGenerator, false, false, []string{"image/"}, 1 << 19}
|
|
}
|
|
|
|
func (h *Handler) UploadOrgLabelPolicyLogoDark() Uploader {
|
|
return &labelPolicyLogoUploader{h.idGenerator, true, false, []string{"image/"}, 1 << 19}
|
|
}
|
|
|
|
type labelPolicyLogoUploader struct {
|
|
idGenerator id.Generator
|
|
darkMode bool
|
|
defaultPolicy bool
|
|
contentTypes []string
|
|
maxSize int64
|
|
}
|
|
|
|
func (l *labelPolicyLogoUploader) ContentTypeAllowed(contentType string) bool {
|
|
for _, ct := range l.contentTypes {
|
|
if strings.HasPrefix(contentType, ct) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (l *labelPolicyLogoUploader) MaxFileSize() int64 {
|
|
return l.maxSize
|
|
}
|
|
|
|
func (l *labelPolicyLogoUploader) ObjectName(_ authz.CtxData) (string, error) {
|
|
suffixID, err := l.idGenerator.Next()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
prefix := domain.LabelPolicyLogoPath
|
|
if l.darkMode {
|
|
return prefix + "-" + domain.Dark + "-" + suffixID, nil
|
|
}
|
|
return prefix + "-" + suffixID, nil
|
|
}
|
|
|
|
func (l *labelPolicyLogoUploader) BucketName(ctxData authz.CtxData) string {
|
|
if l.defaultPolicy {
|
|
return domain.IAMID
|
|
}
|
|
return ctxData.OrgID
|
|
}
|
|
|
|
func (l *labelPolicyLogoUploader) Callback(ctx context.Context, info *domain.AssetInfo, orgID string, commands *command.Commands) error {
|
|
if l.defaultPolicy {
|
|
if l.darkMode {
|
|
_, err := commands.AddLogoDarkDefaultLabelPolicy(ctx, info.Key)
|
|
return err
|
|
}
|
|
_, err := commands.AddLogoDefaultLabelPolicy(ctx, info.Key)
|
|
return err
|
|
}
|
|
if l.darkMode {
|
|
_, err := commands.AddLogoDarkLabelPolicy(ctx, orgID, info.Key)
|
|
return err
|
|
}
|
|
_, err := commands.AddLogoLabelPolicy(ctx, orgID, info.Key)
|
|
return err
|
|
}
|
|
|
|
func (h *Handler) GetDefaultLabelPolicyLogo() Downloader {
|
|
return &labelPolicyLogoDownloader{org: h.orgRepo, darkMode: false, defaultPolicy: true, preview: false}
|
|
}
|
|
|
|
func (h *Handler) GetDefaultLabelPolicyLogoDark() Downloader {
|
|
return &labelPolicyLogoDownloader{org: h.orgRepo, darkMode: true, defaultPolicy: true, preview: false}
|
|
}
|
|
|
|
func (h *Handler) GetPreviewDefaultLabelPolicyLogo() Downloader {
|
|
return &labelPolicyLogoDownloader{org: h.orgRepo, darkMode: false, defaultPolicy: true, preview: true}
|
|
}
|
|
|
|
func (h *Handler) GetPreviewDefaultLabelPolicyLogoDark() Downloader {
|
|
return &labelPolicyLogoDownloader{org: h.orgRepo, darkMode: true, defaultPolicy: true, preview: true}
|
|
}
|
|
|
|
func (h *Handler) GetOrgLabelPolicyLogo() Downloader {
|
|
return &labelPolicyLogoDownloader{org: h.orgRepo, darkMode: false, defaultPolicy: false, preview: false}
|
|
}
|
|
|
|
func (h *Handler) GetOrgLabelPolicyLogoDark() Downloader {
|
|
return &labelPolicyLogoDownloader{org: h.orgRepo, darkMode: true, defaultPolicy: false, preview: false}
|
|
}
|
|
|
|
func (h *Handler) GetPreviewOrgLabelPolicyLogo() Downloader {
|
|
return &labelPolicyLogoDownloader{org: h.orgRepo, darkMode: false, defaultPolicy: false, preview: true}
|
|
}
|
|
|
|
func (h *Handler) GetPreviewOrgLabelPolicyLogoDark() Downloader {
|
|
return &labelPolicyLogoDownloader{org: h.orgRepo, darkMode: true, defaultPolicy: false, preview: true}
|
|
}
|
|
|
|
type labelPolicyLogoDownloader struct {
|
|
org repository.OrgRepository
|
|
darkMode bool
|
|
defaultPolicy bool
|
|
preview bool
|
|
}
|
|
|
|
func (l *labelPolicyLogoDownloader) ObjectName(ctx context.Context, path string) (string, error) {
|
|
policy, err := getLabelPolicy(ctx, l.defaultPolicy, l.preview, l.org)
|
|
if err != nil {
|
|
return "", nil
|
|
}
|
|
if l.darkMode {
|
|
return policy.LogoDarkURL, nil
|
|
}
|
|
return policy.LogoURL, nil
|
|
}
|
|
|
|
func (l *labelPolicyLogoDownloader) BucketName(ctx context.Context, id string) string {
|
|
return getLabelPolicyBucketName(ctx, l.defaultPolicy, l.preview, l.org)
|
|
}
|
|
|
|
func (h *Handler) UploadDefaultLabelPolicyIcon() Uploader {
|
|
return &labelPolicyIconUploader{h.idGenerator, false, true, []string{"image/"}, 1 << 19}
|
|
}
|
|
|
|
func (h *Handler) UploadDefaultLabelPolicyIconDark() Uploader {
|
|
return &labelPolicyIconUploader{h.idGenerator, true, true, []string{"image/"}, 1 << 19}
|
|
}
|
|
|
|
func (h *Handler) UploadOrgLabelPolicyIcon() Uploader {
|
|
return &labelPolicyIconUploader{h.idGenerator, false, false, []string{"image/"}, 1 << 19}
|
|
}
|
|
|
|
func (h *Handler) UploadOrgLabelPolicyIconDark() Uploader {
|
|
return &labelPolicyIconUploader{h.idGenerator, true, false, []string{"image/"}, 1 << 19}
|
|
}
|
|
|
|
type labelPolicyIconUploader struct {
|
|
idGenerator id.Generator
|
|
darkMode bool
|
|
defaultPolicy bool
|
|
contentTypes []string
|
|
maxSize int64
|
|
}
|
|
|
|
func (l *labelPolicyIconUploader) ContentTypeAllowed(contentType string) bool {
|
|
for _, ct := range l.contentTypes {
|
|
if strings.HasPrefix(contentType, ct) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (l *labelPolicyIconUploader) MaxFileSize() int64 {
|
|
return l.maxSize
|
|
}
|
|
|
|
func (l *labelPolicyIconUploader) ObjectName(_ authz.CtxData) (string, error) {
|
|
suffixID, err := l.idGenerator.Next()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
prefix := domain.LabelPolicyIconPath
|
|
if l.darkMode {
|
|
return prefix + "-" + domain.Dark + "-" + suffixID, nil
|
|
}
|
|
return prefix + "-" + suffixID, nil
|
|
}
|
|
|
|
func (l *labelPolicyIconUploader) BucketName(ctxData authz.CtxData) string {
|
|
if l.defaultPolicy {
|
|
return domain.IAMID
|
|
}
|
|
return ctxData.OrgID
|
|
}
|
|
|
|
func (l *labelPolicyIconUploader) Callback(ctx context.Context, info *domain.AssetInfo, orgID string, commands *command.Commands) error {
|
|
if l.defaultPolicy {
|
|
if l.darkMode {
|
|
_, err := commands.AddIconDarkDefaultLabelPolicy(ctx, info.Key)
|
|
return err
|
|
}
|
|
_, err := commands.AddIconDefaultLabelPolicy(ctx, info.Key)
|
|
return err
|
|
}
|
|
|
|
if l.darkMode {
|
|
_, err := commands.AddIconDarkLabelPolicy(ctx, orgID, info.Key)
|
|
return err
|
|
}
|
|
_, err := commands.AddIconLabelPolicy(ctx, orgID, info.Key)
|
|
return err
|
|
}
|
|
|
|
func (h *Handler) GetDefaultLabelPolicyIcon() Downloader {
|
|
return &labelPolicyIconDownloader{org: h.orgRepo, darkMode: false, defaultPolicy: true, preview: false}
|
|
}
|
|
|
|
func (h *Handler) GetDefaultLabelPolicyIconDark() Downloader {
|
|
return &labelPolicyIconDownloader{org: h.orgRepo, darkMode: true, defaultPolicy: true, preview: false}
|
|
}
|
|
|
|
func (h *Handler) GetPreviewDefaultLabelPolicyIcon() Downloader {
|
|
return &labelPolicyIconDownloader{org: h.orgRepo, darkMode: false, defaultPolicy: true, preview: true}
|
|
}
|
|
|
|
func (h *Handler) GetPreviewDefaultLabelPolicyIconDark() Downloader {
|
|
return &labelPolicyIconDownloader{org: h.orgRepo, darkMode: true, defaultPolicy: true, preview: true}
|
|
}
|
|
|
|
func (h *Handler) GetOrgLabelPolicyIcon() Downloader {
|
|
return &labelPolicyIconDownloader{org: h.orgRepo, darkMode: false, defaultPolicy: false, preview: false}
|
|
}
|
|
|
|
func (h *Handler) GetOrgLabelPolicyIconDark() Downloader {
|
|
return &labelPolicyIconDownloader{org: h.orgRepo, darkMode: true, defaultPolicy: false, preview: false}
|
|
}
|
|
|
|
func (h *Handler) GetPreviewOrgLabelPolicyIcon() Downloader {
|
|
return &labelPolicyIconDownloader{org: h.orgRepo, darkMode: false, defaultPolicy: false, preview: true}
|
|
}
|
|
|
|
func (h *Handler) GetPreviewOrgLabelPolicyIconDark() Downloader {
|
|
return &labelPolicyIconDownloader{org: h.orgRepo, darkMode: true, defaultPolicy: false, preview: true}
|
|
}
|
|
|
|
type labelPolicyIconDownloader struct {
|
|
org repository.OrgRepository
|
|
darkMode bool
|
|
defaultPolicy bool
|
|
preview bool
|
|
}
|
|
|
|
func (l *labelPolicyIconDownloader) ObjectName(ctx context.Context, path string) (string, error) {
|
|
policy, err := getLabelPolicy(ctx, l.defaultPolicy, l.preview, l.org)
|
|
if err != nil {
|
|
return "", nil
|
|
}
|
|
if l.darkMode {
|
|
return policy.IconDarkURL, nil
|
|
}
|
|
return policy.IconURL, nil
|
|
}
|
|
|
|
func (l *labelPolicyIconDownloader) BucketName(ctx context.Context, id string) string {
|
|
return getLabelPolicyBucketName(ctx, l.defaultPolicy, l.preview, l.org)
|
|
}
|
|
|
|
func (h *Handler) UploadDefaultLabelPolicyFont() Uploader {
|
|
return &labelPolicyFontUploader{h.idGenerator, true, []string{"font/", "application/octet-stream"}, 1 << 19}
|
|
}
|
|
|
|
func (h *Handler) UploadOrgLabelPolicyFont() Uploader {
|
|
return &labelPolicyFontUploader{h.idGenerator, false, []string{"font/", "application/octet-stream"}, 1 << 19}
|
|
}
|
|
|
|
type labelPolicyFontUploader struct {
|
|
idGenerator id.Generator
|
|
defaultPolicy bool
|
|
contentTypes []string
|
|
maxSize int64
|
|
}
|
|
|
|
func (l *labelPolicyFontUploader) ContentTypeAllowed(contentType string) bool {
|
|
for _, ct := range l.contentTypes {
|
|
if strings.HasPrefix(contentType, ct) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (l *labelPolicyFontUploader) MaxFileSize() int64 {
|
|
return l.maxSize
|
|
}
|
|
|
|
func (l *labelPolicyFontUploader) ObjectName(_ authz.CtxData) (string, error) {
|
|
suffixID, err := l.idGenerator.Next()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
prefix := domain.LabelPolicyFontPath
|
|
return prefix + "-" + suffixID, nil
|
|
}
|
|
|
|
func (l *labelPolicyFontUploader) BucketName(ctxData authz.CtxData) string {
|
|
if l.defaultPolicy {
|
|
return domain.IAMID
|
|
}
|
|
return ctxData.OrgID
|
|
}
|
|
|
|
func (l *labelPolicyFontUploader) Callback(ctx context.Context, info *domain.AssetInfo, orgID string, commands *command.Commands) error {
|
|
if l.defaultPolicy {
|
|
_, err := commands.AddFontDefaultLabelPolicy(ctx, info.Key)
|
|
return err
|
|
}
|
|
_, err := commands.AddFontLabelPolicy(ctx, orgID, info.Key)
|
|
return err
|
|
}
|
|
|
|
func (h *Handler) GetDefaultLabelPolicyFont() Downloader {
|
|
return &labelPolicyFontDownloader{org: h.orgRepo, defaultPolicy: true, preview: false}
|
|
}
|
|
|
|
func (h *Handler) GetPreviewDefaultLabelPolicyFont() Downloader {
|
|
return &labelPolicyFontDownloader{org: h.orgRepo, defaultPolicy: true, preview: true}
|
|
}
|
|
|
|
func (h *Handler) GetOrgLabelPolicyFont() Downloader {
|
|
return &labelPolicyFontDownloader{org: h.orgRepo, defaultPolicy: false, preview: false}
|
|
}
|
|
|
|
func (h *Handler) GetPreviewOrgLabelPolicyFont() Downloader {
|
|
return &labelPolicyFontDownloader{org: h.orgRepo, defaultPolicy: true, preview: true}
|
|
}
|
|
|
|
type labelPolicyFontDownloader struct {
|
|
org repository.OrgRepository
|
|
defaultPolicy bool
|
|
preview bool
|
|
}
|
|
|
|
func (l *labelPolicyFontDownloader) ObjectName(ctx context.Context, path string) (string, error) {
|
|
policy, err := getLabelPolicy(ctx, l.defaultPolicy, l.preview, l.org)
|
|
if err != nil {
|
|
return "", nil
|
|
}
|
|
return policy.FontURL, nil
|
|
}
|
|
|
|
func (l *labelPolicyFontDownloader) BucketName(ctx context.Context, id string) string {
|
|
return getLabelPolicyBucketName(ctx, l.defaultPolicy, l.preview, l.org)
|
|
}
|
|
|
|
func getLabelPolicy(ctx context.Context, defaultPolicy, preview bool, orgRepo repository.OrgRepository) (*model.LabelPolicyView, error) {
|
|
if defaultPolicy {
|
|
if preview {
|
|
return orgRepo.GetPreviewDefaultLabelPolicy(ctx)
|
|
}
|
|
return orgRepo.GetDefaultLabelPolicy(ctx)
|
|
}
|
|
if preview {
|
|
return orgRepo.GetPreviewLabelPolicy(ctx)
|
|
}
|
|
return orgRepo.GetLabelPolicy(ctx)
|
|
}
|
|
|
|
func getLabelPolicyBucketName(ctx context.Context, defaultPolicy, preview bool, org repository.OrgRepository) string {
|
|
if defaultPolicy {
|
|
return domain.IAMID
|
|
}
|
|
policy, err := getLabelPolicy(ctx, defaultPolicy, preview, org)
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
if policy.Default {
|
|
return domain.IAMID
|
|
}
|
|
return authz.GetCtxData(ctx).OrgID
|
|
}
|