settings.go 1.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061
  1. package settings
  2. import (
  3. "gopkg.in/ini.v1"
  4. "log"
  5. "os"
  6. )
  7. var Conf *ini.File
  8. type Server struct {
  9. HttpPort string
  10. RunMode string
  11. WebSocketToken string
  12. JwtSecret string
  13. HTTPChallengePort string
  14. Email string
  15. Database string
  16. }
  17. var ServerSettings = &Server{
  18. HttpPort: "9000",
  19. RunMode: "debug",
  20. HTTPChallengePort: "9180",
  21. Database: "database",
  22. }
  23. var ConfPath string
  24. func Init(confPath string) {
  25. ConfPath = confPath
  26. if _, err := os.Stat(ConfPath); os.IsExist(err) {
  27. Setup()
  28. }
  29. }
  30. func Setup() {
  31. var err error
  32. Conf, err = ini.Load(ConfPath)
  33. if err != nil {
  34. log.Fatalf("setting.Setup, fail to parse '%s': %v", ConfPath, err)
  35. }
  36. mapTo("server", ServerSettings)
  37. }
  38. func mapTo(section string, v interface{}) {
  39. err := Conf.Section(section).MapTo(v)
  40. if err != nil {
  41. log.Fatalf("Cfg.MapTo %s err: %v", section, err)
  42. }
  43. }
  44. func Save() (err error) {
  45. err = Conf.SaveTo(ConfPath)
  46. if err != nil {
  47. return
  48. }
  49. Setup()
  50. return
  51. }