gocron/routers/install/install.go

136 lines
3.8 KiB
Go

package install
import (
"github.com/ouqiang/cron-scheduler/models"
"github.com/ouqiang/cron-scheduler/modules/app"
"github.com/ouqiang/cron-scheduler/modules/setting"
"github.com/ouqiang/cron-scheduler/modules/utils"
"gopkg.in/macaron.v1"
"strconv"
"github.com/ouqiang/cron-scheduler/modules/logger"
)
// 系统安装
type InstallForm struct {
DbType string `binding:"IN(mysql)"`
DbHost string `binding:"Required"`
DbPort int `binding:"Required;Range(1-65535)"` // todo 限制端口范围1-65535, 为什么规则未生效?
DbUsername string `binding:"Required"`
DbPassword string `binding:"Required"`
DbName string `binding:"Required"`
DbTablePrefix string
AdminUsername string `binding:"Required;MinSize(3)"`
AdminPassword string `binding:"Required;MinSize(6)"`
AdminEmail string `binding:"Email"`
}
// 显示安装页面
func Show(ctx *macaron.Context) {
if app.Installed {
ctx.Redirect("/")
}
ctx.Data["Title"] = "安装"
ctx.Data["DisableNav"] = true
ctx.HTML(200, "install/show")
}
// 安装,
func Install(ctx *macaron.Context, form InstallForm) string {
json := utils.Json{}
if app.Installed {
logger.Warn("系统重复安装")
return json.Failure(utils.ResponseFailure, "系统已安装!")
}
err := testDbConnection(form)
if err != nil {
logger.Error(err)
return json.Failure(utils.ResponseFailure, "数据库连接失败")
}
// 写入数据库配置
err = writeConfig(form)
if err != nil {
logger.Error(err)
return json.Failure(utils.ResponseFailure, "数据库配置写入文件失败")
}
app.InitDb()
// 创建数据库表
migration := new(models.Migration)
err = migration.Exec(form.DbName)
if err != nil {
logger.Error(err)
return json.Failure(utils.ResponseFailure, "创建数据库表失败")
}
// 创建管理员账号
err = createAdminUser(form)
if err != nil {
logger.Error(err)
return json.Failure(utils.ResponseFailure, "创建管理员账号失败")
}
// 创建安装锁
err = app.CreateInstallLock()
if err != nil {
logger.Error(err)
return json.Failure(utils.ResponseFailure, "创建文件安装锁失败")
}
app.Installed = true
// 初始化定时任务等
app.InitResource()
return json.Success("安装成功", nil)
}
// 数据库配置写入文件
func writeConfig(form InstallForm) error {
dbConfig := map[string]map[string]string{
"db": map[string]string{
"engine": form.DbType,
"host": form.DbHost,
"port": strconv.Itoa(form.DbPort),
"user": form.DbUsername,
"password": form.DbPassword,
"database": form.DbName,
"prefix": form.DbTablePrefix,
"charset": "utf8",
},
}
return setting.Write(dbConfig, app.AppConfig)
}
// 创建管理员账号
func createAdminUser(form InstallForm) error {
user := new(models.User)
user.Name = form.AdminUsername
user.Password = form.AdminPassword
user.Email = form.AdminEmail
user.IsAdmin = 1
_, err := user.Create()
return err
}
// 测试数据库连接
func testDbConnection(form InstallForm) error {
var dbConfig map[string]string = make(map[string]string)
dbConfig["engine"] = form.DbType
dbConfig["host"] = form.DbHost
dbConfig["port"] = strconv.Itoa(form.DbPort)
dbConfig["user"] = form.DbUsername
dbConfig["password"] = form.DbPassword
dbConfig["charset"] = "utf8"
db, err := models.CreateTmpDb(dbConfig)
if err != nil {
return err
}
defer db.Close()
err = db.Ping()
return err
}