command.go 3.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146
  1. package commands
  2. import (
  3. "encoding/gob"
  4. "fmt"
  5. "net/url"
  6. "os"
  7. "time"
  8. "github.com/astaxie/beego"
  9. "github.com/astaxie/beego/logs"
  10. "github.com/astaxie/beego/orm"
  11. "github.com/lifei6671/gocaptcha"
  12. "github.com/lifei6671/godoc/conf"
  13. "github.com/lifei6671/godoc/models"
  14. "strings"
  15. "github.com/lifei6671/godoc/commands/migrate"
  16. )
  17. // RegisterDataBase 注册数据库
  18. func RegisterDataBase() {
  19. adapter := beego.AppConfig.String("db_adapter")
  20. if adapter == "mysql" {
  21. host := beego.AppConfig.String("db_host")
  22. database := beego.AppConfig.String("db_database")
  23. username := beego.AppConfig.String("db_username")
  24. password := beego.AppConfig.String("db_password")
  25. timezone := beego.AppConfig.String("timezone")
  26. port := beego.AppConfig.String("db_port")
  27. dataSource := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=true&loc=%s", username, password, host, port, database, url.QueryEscape(timezone))
  28. orm.RegisterDataBase("default", "mysql", dataSource)
  29. location, err := time.LoadLocation(timezone)
  30. if err == nil {
  31. orm.DefaultTimeLoc = location
  32. } else {
  33. fmt.Println(err)
  34. }
  35. }else if adapter == "sqlite3" {
  36. database := beego.AppConfig.String("db_database")
  37. orm.RegisterDataBase("default", "sqlite3", database)
  38. }
  39. }
  40. // RegisterModel 注册Model
  41. func RegisterModel() {
  42. orm.RegisterModelWithPrefix(conf.GetDatabasePrefix(),
  43. new(models.Member),
  44. new(models.Book),
  45. new(models.Relationship),
  46. new(models.Option),
  47. new(models.Document),
  48. new(models.Attachment),
  49. new(models.Logger),
  50. new(models.MemberToken),
  51. new(models.DocumentHistory),
  52. new(models.Migration),
  53. )
  54. migrate.RegisterMigration()
  55. }
  56. // RegisterLogger 注册日志
  57. func RegisterLogger() {
  58. logs.SetLogFuncCall(true)
  59. logs.SetLogger("console")
  60. logs.EnableFuncCallDepth(true)
  61. logs.Async()
  62. if _, err := os.Stat("logs/log.log"); os.IsNotExist(err) {
  63. os.MkdirAll("./logs", 0777)
  64. if f, err := os.Create("logs/log.log"); err == nil {
  65. f.Close()
  66. beego.SetLogger("file", `{"filename":"logs/log.log"}`)
  67. }
  68. }
  69. beego.SetLogFuncCall(true)
  70. beego.BeeLogger.Async()
  71. }
  72. // RunCommand 注册orm命令行工具
  73. func RegisterCommand() {
  74. Install()
  75. CheckUpdate()
  76. migrate.RunMigration()
  77. }
  78. func RegisterFunction() {
  79. beego.AddFuncMap("config", models.GetOptionValue)
  80. beego.AddFuncMap("cdn", func(p string) string {
  81. cdn := beego.AppConfig.DefaultString("cdn","")
  82. if strings.HasPrefix(p,"/") && strings.HasSuffix(cdn,"/"){
  83. return cdn + string(p[1:])
  84. }
  85. if !strings.HasPrefix(p,"/") && !strings.HasSuffix(cdn,"/"){
  86. return cdn + "/" + p
  87. }
  88. return cdn + p
  89. });
  90. beego.AddFuncMap("cdnjs", func(p string) string {
  91. cdn := beego.AppConfig.DefaultString("cdnjs","")
  92. if strings.HasPrefix(p,"/") && strings.HasSuffix(cdn,"/"){
  93. return cdn + string(p[1:])
  94. }
  95. if !strings.HasPrefix(p,"/") && !strings.HasSuffix(cdn,"/"){
  96. return cdn + "/" + p
  97. }
  98. return cdn + p
  99. });
  100. beego.AddFuncMap("cdncss", func(p string) string {
  101. cdn := beego.AppConfig.DefaultString("cdncss","")
  102. if strings.HasPrefix(p,"/") && strings.HasSuffix(cdn,"/"){
  103. return cdn + string(p[1:])
  104. }
  105. if !strings.HasPrefix(p,"/") && !strings.HasSuffix(cdn,"/"){
  106. return cdn + "/" + p
  107. }
  108. return cdn + p
  109. });
  110. beego.AddFuncMap("cdnimg", func(p string) string {
  111. cdn := beego.AppConfig.DefaultString("cdnimg","")
  112. if strings.HasPrefix(p,"/") && strings.HasSuffix(cdn,"/"){
  113. return cdn + string(p[1:])
  114. }
  115. if !strings.HasPrefix(p,"/") && !strings.HasSuffix(cdn,"/"){
  116. return cdn + "/" + p
  117. }
  118. return cdn + p
  119. });
  120. }
  121. func init() {
  122. gocaptcha.ReadFonts("./static/fonts", ".ttf")
  123. gob.Register(models.Member{})
  124. }