init.go 1.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687
  1. package config
  2. import (
  3. "fmt"
  4. "log"
  5. "os"
  6. "path"
  7. "path/filepath"
  8. "runtime"
  9. "strings"
  10. "github.com/IceWhaleTech/CasaOS/model"
  11. "github.com/go-ini/ini"
  12. )
  13. //系统配置
  14. var SysInfo = &model.SysInfoModel{}
  15. //用户相关
  16. var UserInfo = &model.UserModel{}
  17. //用户相关
  18. var AppInfo = &model.APPModel{}
  19. //var RedisInfo = &model.RedisModel{}
  20. //server相关
  21. var ServerInfo = &model.ServerModel{}
  22. var SystemConfigInfo = &model.SystemConfig{}
  23. var CasaOSGlobalVariables = &model.CasaOSGlobalVariables{}
  24. var FileSettingInfo = &model.FileSetting{}
  25. var Cfg *ini.File
  26. //初始化设置,获取系统的部分信息。
  27. func InitSetup(config string) {
  28. var configDir = USERCONFIGURL
  29. if len(config) > 0 {
  30. configDir = config
  31. }
  32. var err error
  33. //读取文件
  34. Cfg, err = ini.Load(configDir)
  35. if err != nil {
  36. fmt.Printf("Fail to read file: %v", err)
  37. os.Exit(1)
  38. }
  39. mapTo("user", UserInfo)
  40. mapTo("app", AppInfo)
  41. //mapTo("redis", RedisInfo)
  42. mapTo("server", ServerInfo)
  43. mapTo("system", SystemConfigInfo)
  44. mapTo("file", FileSettingInfo)
  45. SystemConfigInfo.ConfigPath = configDir
  46. // AppInfo.ProjectPath = getCurrentDirectory() //os.Getwd()
  47. }
  48. //映射
  49. func mapTo(section string, v interface{}) {
  50. err := Cfg.Section(section).MapTo(v)
  51. if err != nil {
  52. log.Fatalf("Cfg.MapTo %s err: %v", section, err)
  53. }
  54. }
  55. // 获取当前执行文件绝对路径(go run)
  56. func getCurrentAbPathByCaller() string {
  57. var abPath string
  58. _, filename, _, ok := runtime.Caller(0)
  59. if ok {
  60. abPath = path.Dir(filename)
  61. }
  62. return abPath
  63. }
  64. func getCurrentDirectory() string {
  65. dir, err := filepath.Abs(filepath.Dir(os.Args[0]))
  66. if err != nil {
  67. log.Fatal(err)
  68. }
  69. return strings.Replace(dir, "\\", "/", -1)
  70. }