72 lines
1.7 KiB
Go
72 lines
1.7 KiB
Go
package colorschemes
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/xxxserxxx/lingo/v2"
|
|
"github.com/shibukawa/configdir"
|
|
)
|
|
|
|
var registry map[string]Colorscheme
|
|
|
|
func init() {
|
|
if registry == nil {
|
|
registry = make(map[string]Colorscheme)
|
|
}
|
|
}
|
|
|
|
var tr lingo.Translations
|
|
|
|
// Set the translation library
|
|
func SetTr(tra lingo.Translations) {
|
|
tr = tra
|
|
}
|
|
|
|
// FromName loads a Colorscheme by name; confDir is used to search
|
|
// directories for a scheme matching the name. The search order
|
|
// is the same as for config files.
|
|
func FromName(confDir configdir.ConfigDir, c string) (Colorscheme, error) {
|
|
if c == "" {
|
|
c = "default"
|
|
}
|
|
if cs, ok := registry[c]; ok {
|
|
return cs, nil
|
|
}
|
|
cs, err := getCustomColorscheme(confDir, c)
|
|
return cs, err
|
|
}
|
|
|
|
func register(name string, c Colorscheme) {
|
|
if registry == nil {
|
|
registry = make(map[string]Colorscheme)
|
|
}
|
|
c.Name = name
|
|
registry[name] = c
|
|
}
|
|
|
|
// getCustomColorscheme tries to read a custom json colorscheme from <configDir>/<name>.json
|
|
func getCustomColorscheme(confDir configdir.ConfigDir, name string) (Colorscheme, error) {
|
|
var cs Colorscheme
|
|
fn := name + ".json"
|
|
folder := confDir.QueryFolderContainsFile(fn)
|
|
if folder == nil {
|
|
paths := make([]string, 0)
|
|
for _, d := range confDir.QueryFolders(configdir.Existing) {
|
|
paths = append(paths, d.Path)
|
|
}
|
|
return cs, fmt.Errorf(tr.Value("error.colorschemefile", fn, strings.Join(paths, ", ")))
|
|
}
|
|
dat, err := folder.ReadFile(fn)
|
|
if err != nil {
|
|
return cs, fmt.Errorf(tr.Value("error.colorschemeload", filepath.Join(folder.Path, fn), err.Error()))
|
|
}
|
|
err = json.Unmarshal(dat, &cs)
|
|
if err != nil {
|
|
return cs, fmt.Errorf(tr.Value("error.colorschemeparse", err.Error()))
|
|
}
|
|
return cs, nil
|
|
}
|