123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146 |
- package commands
- import (
- "encoding/gob"
- "fmt"
- "net/url"
- "os"
- "time"
- "github.com/astaxie/beego"
- "github.com/astaxie/beego/logs"
- "github.com/astaxie/beego/orm"
- "github.com/lifei6671/gocaptcha"
- "github.com/lifei6671/godoc/conf"
- "github.com/lifei6671/godoc/models"
- "strings"
- "github.com/lifei6671/godoc/commands/migrate"
- )
- // RegisterDataBase 注册数据库
- func RegisterDataBase() {
- adapter := beego.AppConfig.String("db_adapter")
- if adapter == "mysql" {
- host := beego.AppConfig.String("db_host")
- database := beego.AppConfig.String("db_database")
- username := beego.AppConfig.String("db_username")
- password := beego.AppConfig.String("db_password")
- timezone := beego.AppConfig.String("timezone")
- port := beego.AppConfig.String("db_port")
- dataSource := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=true&loc=%s", username, password, host, port, database, url.QueryEscape(timezone))
- orm.RegisterDataBase("default", "mysql", dataSource)
- location, err := time.LoadLocation(timezone)
- if err == nil {
- orm.DefaultTimeLoc = location
- } else {
- fmt.Println(err)
- }
- }else if adapter == "sqlite3" {
- database := beego.AppConfig.String("db_database")
- orm.RegisterDataBase("default", "sqlite3", database)
- }
- }
- // RegisterModel 注册Model
- func RegisterModel() {
- orm.RegisterModelWithPrefix(conf.GetDatabasePrefix(),
- new(models.Member),
- new(models.Book),
- new(models.Relationship),
- new(models.Option),
- new(models.Document),
- new(models.Attachment),
- new(models.Logger),
- new(models.MemberToken),
- new(models.DocumentHistory),
- new(models.Migration),
- )
- migrate.RegisterMigration()
- }
- // RegisterLogger 注册日志
- func RegisterLogger() {
- logs.SetLogFuncCall(true)
- logs.SetLogger("console")
- logs.EnableFuncCallDepth(true)
- logs.Async()
- if _, err := os.Stat("logs/log.log"); os.IsNotExist(err) {
- os.MkdirAll("./logs", 0777)
- if f, err := os.Create("logs/log.log"); err == nil {
- f.Close()
- beego.SetLogger("file", `{"filename":"logs/log.log"}`)
- }
- }
- beego.SetLogFuncCall(true)
- beego.BeeLogger.Async()
- }
- // RunCommand 注册orm命令行工具
- func RegisterCommand() {
- Install()
- CheckUpdate()
- migrate.RunMigration()
- }
- func RegisterFunction() {
- beego.AddFuncMap("config", models.GetOptionValue)
- beego.AddFuncMap("cdn", func(p string) string {
- cdn := beego.AppConfig.DefaultString("cdn","")
- if strings.HasPrefix(p,"/") && strings.HasSuffix(cdn,"/"){
- return cdn + string(p[1:])
- }
- if !strings.HasPrefix(p,"/") && !strings.HasSuffix(cdn,"/"){
- return cdn + "/" + p
- }
- return cdn + p
- });
- beego.AddFuncMap("cdnjs", func(p string) string {
- cdn := beego.AppConfig.DefaultString("cdnjs","")
- if strings.HasPrefix(p,"/") && strings.HasSuffix(cdn,"/"){
- return cdn + string(p[1:])
- }
- if !strings.HasPrefix(p,"/") && !strings.HasSuffix(cdn,"/"){
- return cdn + "/" + p
- }
- return cdn + p
- });
- beego.AddFuncMap("cdncss", func(p string) string {
- cdn := beego.AppConfig.DefaultString("cdncss","")
- if strings.HasPrefix(p,"/") && strings.HasSuffix(cdn,"/"){
- return cdn + string(p[1:])
- }
- if !strings.HasPrefix(p,"/") && !strings.HasSuffix(cdn,"/"){
- return cdn + "/" + p
- }
- return cdn + p
- });
- beego.AddFuncMap("cdnimg", func(p string) string {
- cdn := beego.AppConfig.DefaultString("cdnimg","")
- if strings.HasPrefix(p,"/") && strings.HasSuffix(cdn,"/"){
- return cdn + string(p[1:])
- }
- if !strings.HasPrefix(p,"/") && !strings.HasSuffix(cdn,"/"){
- return cdn + "/" + p
- }
- return cdn + p
- });
- }
- func init() {
- gocaptcha.ReadFonts("./static/fonts", ".ttf")
- gob.Register(models.Member{})
- }
|