60 lines
1.1 KiB
Go
60 lines
1.1 KiB
Go
package sign
|
|
|
|
import (
|
|
"io"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"gopkg.in/yaml.v2"
|
|
)
|
|
|
|
var App *AppConfig
|
|
|
|
func ReadInConfig(root string) error {
|
|
cfg := &AppConfig{
|
|
Sites: []*SiteConfig{},
|
|
}
|
|
|
|
if err := filepath.Walk(root, func(fp string, info os.FileInfo, err error) error {
|
|
var site SiteConfig
|
|
if info.IsDir() {
|
|
return nil
|
|
} else if file, err := os.OpenFile(fp, os.O_RDONLY, 0755); err != nil {
|
|
return err
|
|
} else if data, err := io.ReadAll(file); err != nil {
|
|
return err
|
|
} else if err := yaml.Unmarshal(data, &site); err != nil {
|
|
return err
|
|
} else {
|
|
// Extract file name as site id
|
|
site.ID = strings.SplitN(filepath.Base(fp), ".", 2)[0]
|
|
|
|
cfg.Sites = append(cfg.Sites, &site)
|
|
}
|
|
|
|
return nil
|
|
}); err != nil {
|
|
return err
|
|
}
|
|
|
|
App = cfg
|
|
|
|
return nil
|
|
}
|
|
|
|
func SaveInConfig(root string, cfg *AppConfig) error {
|
|
for _, site := range cfg.Sites {
|
|
data, _ := yaml.Marshal(site)
|
|
|
|
fp := filepath.Join(root, site.ID)
|
|
if file, err := os.OpenFile(fp, os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0755); err != nil {
|
|
return err
|
|
} else if _, err := file.Write(data); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|