gocron/routers/install/install.go

109 lines
2.9 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"
)
// 系统安装
type InstallForm struct {
DbType string `binding:"IN(mysql)"`
DbHost string `binding:"Required"`
DbPort int `binding:"Required"`
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 {
return json.Failure(utils.ResponseFailure, "系统已安装成功")
}
// 写入数据库配置
err := writeConfig(form)
if err != nil {
utils.RecordLog(err)
return json.Failure(utils.ResponseFailure, "数据库配置写入文件失败")
}
// 初始化Db
app.InitDb()
// 创建数据库表
migration := new(models.Migration)
err = migration.Exec(form.DbName)
if err != nil {
utils.RecordLog(err)
return json.Failure(utils.ResponseFailure, "创建数据库表失败")
}
// 创建管理员账号
err = createAdminUser(form)
if err != nil {
utils.RecordLog(err)
return json.Failure(utils.ResponseFailure, "创建管理员账号失败")
}
// 创建安装锁
err = app.CreateInstallLock()
if err != nil {
utils.RecordLog(err)
return json.Failure(utils.ResponseFailure, "创建文件安装锁失败")
}
// 初始化定时任务等
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
}