mindoc/commands/command.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()
	Update()
	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{})
}