1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106

package theme

import (
	"path/filepath"

	"github.com/pkg/errors"
	"github.com/samber/lo"

	"{{{ .Package }}}/app/lib/filesystem"
	"{{{ .Package }}}/app/util"
)

const KeyNew = "new"

type Service struct {
	root  string
	files filesystem.FileLoader
	cache Themes
}

func NewService(files filesystem.FileLoader) *Service {
	return &Service{root: "themes", files: files}
}

func (s *Service) All(logger util.Logger) Themes {
	s.loadIfNeeded(logger)
	return s.cache
}

func (s *Service) Clear() {
	s.cache = nil
}

func (s *Service) Get(theme string, logger util.Logger) *Theme {
	return lo.FindOrElse(s.All(logger), Default, func(t *Theme) bool {
		return t.Key == theme
	})
}

func (s *Service) Save(t *Theme, originalKey string, logger util.Logger) error {
	if t.Key == Default.Key {
		return errors.New("can't overwrite default theme")
	}
	if t.Key == "" || t.Key == KeyNew {
		t.Key = util.RandomString(12)
	}
	if originalKey != t.Key {
		err := s.Remove(originalKey, logger)
		if err != nil {
			return err
		}
	}
	b := util.ToJSONBytes(t, true)
	err := s.files.WriteFile(filepath.Join(s.root, t.Key+".json"), b, filesystem.DefaultMode, true)
	if err != nil {
		logger.Warnf("can't save theme [%s]: %+v", t.Key, err)
	}
	s.cache = s.cache.Replace(t)
	return nil
}

func (s *Service) loadIfNeeded(logger util.Logger) {
	if s.cache == nil {
		s.cache = Themes{Default}
		if s.files.IsDir(s.root) {
			lo.ForEach(s.files.ListJSON(s.root, nil, true, logger), func(key string, _ int) {
				t := &Theme{}
				b, err := s.files.ReadFile(filepath.Join(s.root, key+".json"))
				if err != nil {
					logger.Warnf("can't load theme [%s]: %+v", key, err)
				}
				err = util.FromJSON(b, t)
				if err != nil {
					logger.Warnf("can't load theme [%s]: %+v", key, err)
				}
				t.Key = key
				s.cache = append(s.cache, t)
			})
		}{{{ if .HasModule "themecatalog" }}}
		s.cache = append(s.cache, lo.Filter(CatalogThemes, func(t *Theme, _ int) bool {
			return !s.cache.Contains(t.Key)
		})...){{{ end }}}
		s.cache.Sort()
	}
}

func (s *Service) Remove(key string, logger util.Logger) error {
	s.cache = s.cache.Remove(key)
	if !s.FileExists(key) {
		return nil
	}
	return s.files.Remove(filepath.Join(s.root, key+".json"), logger)
}

func (s *Service) FileExists(key string) bool {
	return s.files.Exists(filepath.Join(s.root, key+".json"))
}

func ApplyMap(frm util.ValueMap) *Theme {
	orig := Default

	l := orig.Light.Clone().ApplyMap(frm, "light-")
	d := orig.Dark.Clone().ApplyMap(frm, "dark-")

	return &Theme{Light: l, Dark: d}
}