yamldb.go 557 B

123456789101112131415161718192021222324252627282930313233343536373839
  1. package main
  2. import (
  3. "io/ioutil"
  4. "github.com/go-yaml/yaml"
  5. )
  6. func loadLock() ConfigFile {
  7. f, err := ioutil.ReadFile("lock.yaml")
  8. if err != nil {
  9. ioutil.WriteFile("lock.yaml", []byte(""), 0777)
  10. f = []byte{}
  11. }
  12. var cf ConfigFile
  13. err = yaml.Unmarshal(f, &cf)
  14. if err != nil {
  15. die(err)
  16. }
  17. if cf.Sites == nil {
  18. cf.Sites = make(map[string]*coreMatch)
  19. }
  20. return cf
  21. }
  22. func writeLock(c ConfigFile) {
  23. b, err := yaml.Marshal(c)
  24. if err != nil {
  25. panic(err)
  26. }
  27. err = ioutil.WriteFile("lock.yaml", b, 0777)
  28. if err != nil {
  29. panic(err)
  30. }
  31. }