RoadSign/pkg/sign/configurator.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
}