bench-forgejo/routers/install.go

259 lines
7.2 KiB
Go
Raw Normal View History

2014-03-25 14:21:42 +05:30
// Copyright 2014 The Gogs Authors. All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.
package routers
2014-03-28 16:56:22 +05:30
import (
"errors"
"os"
2014-04-09 00:57:35 +05:30
"os/exec"
2014-05-26 05:41:25 +05:30
"path"
"strings"
2014-07-26 09:54:27 +05:30
"github.com/Unknwon/com"
"github.com/Unknwon/goconfig"
2014-07-26 09:54:27 +05:30
"github.com/Unknwon/macaron"
2014-04-18 19:05:09 +05:30
"github.com/go-xorm/xorm"
2014-03-28 16:56:22 +05:30
"github.com/gogits/gogs/models"
2014-03-29 04:10:31 +05:30
"github.com/gogits/gogs/modules/auth"
2014-03-28 16:56:22 +05:30
"github.com/gogits/gogs/modules/base"
2014-04-13 07:00:09 +05:30
"github.com/gogits/gogs/modules/cron"
"github.com/gogits/gogs/modules/log"
"github.com/gogits/gogs/modules/mailer"
2014-03-28 16:56:22 +05:30
"github.com/gogits/gogs/modules/middleware"
2014-05-26 05:41:25 +05:30
"github.com/gogits/gogs/modules/setting"
2014-04-14 03:42:07 +05:30
"github.com/gogits/gogs/modules/social"
2014-03-28 16:56:22 +05:30
)
const (
INSTALL base.TplName = "install"
)
func checkRunMode() {
2014-05-26 05:41:25 +05:30
switch setting.Cfg.MustValue("", "RUN_MODE") {
case "prod":
2014-07-26 09:54:27 +05:30
macaron.Env = macaron.PROD
2014-05-26 05:41:25 +05:30
setting.ProdMode = true
case "test":
2014-07-26 09:54:27 +05:30
macaron.Env = macaron.TEST
}
2014-07-26 09:54:27 +05:30
log.Info("Run Mode: %s", strings.Title(macaron.Env))
}
2014-04-14 03:42:07 +05:30
func NewServices() {
2014-05-26 05:41:25 +05:30
setting.NewServices()
2014-04-14 03:42:07 +05:30
social.NewOauthService()
}
// GlobalInit is for global configuration reload-able.
func GlobalInit() {
2014-05-26 05:41:25 +05:30
setting.NewConfigContext()
2014-05-26 06:27:01 +05:30
log.Trace("Custom path: %s", setting.CustomPath)
2014-05-30 16:04:24 +05:30
log.Trace("Log path: %s", setting.LogRootPath)
mailer.NewMailerContext()
models.LoadModelsConfig()
2014-04-20 07:43:22 +05:30
NewServices()
2014-03-30 20:17:08 +05:30
2014-05-26 05:41:25 +05:30
if setting.InstallLock {
models.LoadRepoConfig()
models.NewRepoContext()
2014-03-30 20:17:08 +05:30
if err := models.NewEngine(); err != nil {
2014-07-26 09:54:27 +05:30
log.Fatal(4, "Fail to initialize ORM engine: %v", err)
2014-03-30 20:17:08 +05:30
}
models.HasEngine = true
2014-04-13 07:00:09 +05:30
cron.NewCronContext()
2014-06-20 10:44:54 +05:30
log.NewGitLogger(path.Join(setting.LogRootPath, "http.log"))
}
2014-05-31 04:49:30 +05:30
if models.EnableSQLite3 {
log.Info("SQLite3 Enabled")
}
checkRunMode()
}
2014-04-27 10:04:48 +05:30
func renderDbOption(ctx *middleware.Context) {
ctx.Data["DbOptions"] = []string{"MySQL", "PostgreSQL", "SQLite3"}
}
// @router /install [get]
2014-04-11 02:06:50 +05:30
func Install(ctx *middleware.Context, form auth.InstallForm) {
2014-05-26 05:41:25 +05:30
if setting.InstallLock {
ctx.Handle(404, "Install", errors.New("Installation is prohibited"))
2014-03-28 16:56:22 +05:30
return
}
2014-03-25 14:21:42 +05:30
ctx.Data["Title"] = ctx.Tr("install.install")
2014-03-28 16:56:22 +05:30
ctx.Data["PageIsInstall"] = true
2014-05-05 22:38:01 +05:30
// Get and assign values to install form.
if len(form.DbHost) == 0 {
form.DbHost = models.DbCfg.Host
2014-04-11 00:07:43 +05:30
}
if len(form.DbUser) == 0 {
form.DbUser = models.DbCfg.User
2014-04-11 00:07:43 +05:30
}
if len(form.DbPasswd) == 0 {
form.DbPasswd = models.DbCfg.Pwd
2014-04-11 00:07:43 +05:30
}
if len(form.DatabaseName) == 0 {
form.DatabaseName = models.DbCfg.Name
}
if len(form.DatabasePath) == 0 {
form.DatabasePath = models.DbCfg.Path
}
2014-04-11 00:07:43 +05:30
if len(form.RepoRootPath) == 0 {
2014-05-26 05:41:25 +05:30
form.RepoRootPath = setting.RepoRootPath
2014-04-11 00:07:43 +05:30
}
if len(form.RunUser) == 0 {
2014-05-26 05:41:25 +05:30
form.RunUser = setting.RunUser
2014-04-11 00:07:43 +05:30
}
if len(form.Domain) == 0 {
2014-05-26 05:41:25 +05:30
form.Domain = setting.Domain
2014-04-11 00:07:43 +05:30
}
if len(form.AppUrl) == 0 {
2014-05-26 05:41:25 +05:30
form.AppUrl = setting.AppUrl
2014-04-11 00:07:43 +05:30
}
2014-04-27 10:04:48 +05:30
renderDbOption(ctx)
2014-05-05 22:38:01 +05:30
curDbOp := ""
2014-04-27 10:04:48 +05:30
if models.EnableSQLite3 {
2014-05-05 22:38:01 +05:30
curDbOp = "SQLite3" // Default when enabled.
2014-04-27 10:04:48 +05:30
}
2014-05-05 22:38:01 +05:30
ctx.Data["CurDbOption"] = curDbOp
2014-04-27 10:04:48 +05:30
2014-04-11 00:07:43 +05:30
auth.AssignForm(form, ctx.Data)
ctx.HTML(200, INSTALL)
2014-04-11 00:07:43 +05:30
}
2014-04-11 02:06:50 +05:30
func InstallPost(ctx *middleware.Context, form auth.InstallForm) {
2014-05-26 05:41:25 +05:30
if setting.InstallLock {
ctx.Handle(404, "InstallPost", errors.New("Installation is prohibited"))
2014-03-28 16:56:22 +05:30
return
}
ctx.Data["Title"] = ctx.Tr("install.install")
2014-04-11 00:07:43 +05:30
ctx.Data["PageIsInstall"] = true
2014-04-27 10:04:48 +05:30
renderDbOption(ctx)
2014-05-05 22:38:01 +05:30
ctx.Data["CurDbOption"] = form.Database
2014-04-27 10:04:48 +05:30
if ctx.HasError() {
ctx.HTML(200, INSTALL)
return
}
2014-04-09 00:57:35 +05:30
if _, err := exec.LookPath("git"); err != nil {
ctx.RenderWithErr(ctx.Tr("install.test_git_failed", err), INSTALL, &form)
2014-04-09 00:57:35 +05:30
return
}
// Pass basic check, now test configuration.
// Test database setting.
2014-04-27 10:04:48 +05:30
dbTypes := map[string]string{"MySQL": "mysql", "PostgreSQL": "postgres", "SQLite3": "sqlite3"}
models.DbCfg.Type = dbTypes[form.Database]
models.DbCfg.Host = form.DbHost
models.DbCfg.User = form.DbUser
models.DbCfg.Pwd = form.DbPasswd
models.DbCfg.Name = form.DatabaseName
models.DbCfg.SslMode = form.SslMode
models.DbCfg.Path = form.DatabasePath
2014-03-30 20:39:59 +05:30
// Set test engine.
2014-03-30 20:17:08 +05:30
var x *xorm.Engine
if err := models.NewTestEngine(x); err != nil {
// NOTE: should use core.QueryDriver (github.com/go-xorm/core)
2014-03-30 20:39:59 +05:30
if strings.Contains(err.Error(), `Unknown database type: sqlite3`) {
ctx.RenderWithErr(ctx.Tr("install.sqlite3_not_available"), INSTALL, &form)
2014-03-30 19:09:44 +05:30
} else {
ctx.RenderWithErr(ctx.Tr("install.invalid_db_setting", err), INSTALL, &form)
2014-03-30 19:09:44 +05:30
}
return
}
// Test repository root path.
if err := os.MkdirAll(form.RepoRootPath, os.ModePerm); err != nil {
2014-09-15 04:52:52 +05:30
ctx.Data["Err_RepoRootPath"] = true
ctx.RenderWithErr(ctx.Tr("install.invalid_repo_path", err), INSTALL, &form)
return
}
2014-03-30 21:28:21 +05:30
// Check run user.
2014-04-12 09:22:08 +05:30
curUser := os.Getenv("USER")
2014-03-30 21:28:21 +05:30
if len(curUser) == 0 {
2014-04-12 09:22:08 +05:30
curUser = os.Getenv("USERNAME")
2014-03-30 21:28:21 +05:30
}
// Does not check run user when the install lock is off.
if form.RunUser != curUser {
2014-09-15 04:52:52 +05:30
ctx.Data["Err_RunUser"] = true
ctx.RenderWithErr(ctx.Tr("install.run_user_not_match", form.RunUser, curUser), INSTALL, &form)
return
}
// Check admin password.
if form.AdminPasswd != form.ConfirmPasswd {
2014-09-15 04:52:52 +05:30
ctx.Data["Err_AdminPasswd"] = true
ctx.RenderWithErr(ctx.Tr("form.password_not_match"), INSTALL, form)
2014-03-30 21:28:21 +05:30
return
}
// Save settings.
2014-05-26 05:41:25 +05:30
setting.Cfg.SetValue("database", "DB_TYPE", models.DbCfg.Type)
setting.Cfg.SetValue("database", "HOST", models.DbCfg.Host)
setting.Cfg.SetValue("database", "NAME", models.DbCfg.Name)
setting.Cfg.SetValue("database", "USER", models.DbCfg.User)
setting.Cfg.SetValue("database", "PASSWD", models.DbCfg.Pwd)
setting.Cfg.SetValue("database", "SSL_MODE", models.DbCfg.SslMode)
setting.Cfg.SetValue("database", "PATH", models.DbCfg.Path)
setting.Cfg.SetValue("repository", "ROOT", form.RepoRootPath)
setting.Cfg.SetValue("", "RUN_USER", form.RunUser)
setting.Cfg.SetValue("server", "DOMAIN", form.Domain)
setting.Cfg.SetValue("server", "ROOT_URL", form.AppUrl)
2014-03-31 02:04:23 +05:30
if len(strings.TrimSpace(form.SmtpHost)) > 0 {
2014-05-26 05:41:25 +05:30
setting.Cfg.SetValue("mailer", "ENABLED", "true")
setting.Cfg.SetValue("mailer", "HOST", form.SmtpHost)
setting.Cfg.SetValue("mailer", "USER", form.SmtpEmail)
setting.Cfg.SetValue("mailer", "PASSWD", form.SmtpPasswd)
2014-07-26 09:54:27 +05:30
setting.Cfg.SetValue("service", "REGISTER_EMAIL_CONFIRM", com.ToStr(form.RegisterConfirm == "on"))
setting.Cfg.SetValue("service", "ENABLE_NOTIFY_MAIL", com.ToStr(form.MailNotify == "on"))
}
2014-05-26 05:41:25 +05:30
setting.Cfg.SetValue("", "RUN_MODE", "prod")
2014-03-30 21:28:21 +05:30
2014-08-27 14:09:36 +05:30
setting.Cfg.SetValue("log", "MODE", "file")
2014-05-26 05:41:25 +05:30
setting.Cfg.SetValue("security", "INSTALL_LOCK", "true")
2014-10-19 08:56:55 +05:30
setting.Cfg.SetValue("security", "SECRET_KEY", base.GetRandomString(15))
2014-03-30 21:28:21 +05:30
os.MkdirAll("custom/conf", os.ModePerm)
2014-05-26 05:41:25 +05:30
if err := goconfig.SaveConfigFile(setting.Cfg, path.Join(setting.CustomPath, "conf/app.ini")); err != nil {
ctx.RenderWithErr(ctx.Tr("install.save_config_failed", err), INSTALL, &form)
return
}
GlobalInit()
2014-03-30 20:39:59 +05:30
// Create admin account.
2014-07-26 09:54:27 +05:30
if err := models.CreateUser(&models.User{Name: form.AdminName, Email: form.AdminEmail, Passwd: form.AdminPasswd,
2014-03-30 20:39:59 +05:30
IsAdmin: true, IsActive: true}); err != nil {
if err != models.ErrUserAlreadyExist {
2014-05-26 05:41:25 +05:30
setting.InstallLock = false
2014-09-15 04:52:52 +05:30
ctx.Data["Err_AdminName"] = true
ctx.Data["Err_AdminEmail"] = true
ctx.RenderWithErr(ctx.Tr("install.invalid_admin_setting", err), INSTALL, &form)
2014-03-30 20:39:59 +05:30
return
}
2014-03-31 01:31:50 +05:30
log.Info("Admin account already exist")
2014-03-30 20:39:59 +05:30
}
log.Info("First-time run install finished!")
ctx.Flash.Success(ctx.Tr("install.install_success"))
2014-09-20 05:41:34 +05:30
ctx.Redirect(setting.AppSubUrl + "/user/login")
2014-03-25 14:21:42 +05:30
}