mirror of https://github.com/ouqiang/gocron
136 lines
3.8 KiB
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
|
|
|
|
} |