mirror of https://github.com/ouqiang/gocron
104 lines
2.5 KiB
Go
104 lines
2.5 KiB
Go
|
package install
|
||
|
|
||
|
import (
|
||
|
"gopkg.in/macaron.v1"
|
||
|
"github.com/ouqiang/cron-scheduler/modules/app"
|
||
|
"github.com/ouqiang/cron-scheduler/modules/utils"
|
||
|
"github.com/ouqiang/cron-scheduler/models"
|
||
|
"github.com/ouqiang/cron-scheduler/modules/setting"
|
||
|
"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, "数据库配置写入文件失败")
|
||
|
}
|
||
|
// 创建安装锁
|
||
|
err = app.CreateInstallLock()
|
||
|
if err != nil {
|
||
|
utils.RecordLog(err)
|
||
|
return json.Failure(utils.ResponseFailure, "创建文件安装锁失败")
|
||
|
}
|
||
|
|
||
|
app.InitDb()
|
||
|
// 初始化配置, DB, 定时任务, 创建数据库表
|
||
|
migration := new(models.Migration)
|
||
|
err = migration.Exec()
|
||
|
if err != nil {
|
||
|
utils.RecordLog(err)
|
||
|
return json.Failure(utils.ResponseFailure, "创建数据库表失败")
|
||
|
}
|
||
|
app.InitResource()
|
||
|
|
||
|
// 创建管理员账号
|
||
|
err = createAdminUser(form)
|
||
|
if err != nil {
|
||
|
utils.RecordLog(err)
|
||
|
return json.Failure(utils.ResponseFailure, "创建管理员账号失败")
|
||
|
}
|
||
|
|
||
|
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
|
||
|
}
|