mirror of https://github.com/mindoc-org/mindoc.git
147 lines
3.5 KiB
Go
147 lines
3.5 KiB
Go
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{})
|
|
}
|