bench-forgejo/routers/install.go

385 lines
12 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"
"path/filepath"
"strings"
2014-07-26 09:54:27 +05:30
"github.com/Unknwon/com"
2014-04-18 19:05:09 +05:30
"github.com/go-xorm/xorm"
"gopkg.in/ini.v1"
2015-10-16 06:58:12 +05:30
"gopkg.in/macaron.v1"
2014-03-28 16:56:22 +05:30
2015-12-16 03:55:45 +05:30
"github.com/gogits/git-module"
2015-11-27 04:03:45 +05:30
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"
"github.com/gogits/gogs/modules/cron"
"github.com/gogits/gogs/modules/log"
"github.com/gogits/gogs/modules/mailer"
2016-02-21 03:40:05 +05:30
"github.com/gogits/gogs/modules/markdown"
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"
"github.com/gogits/gogs/modules/ssh"
"github.com/gogits/gogs/modules/template/highlight"
"github.com/gogits/gogs/modules/user"
2014-03-28 16:56:22 +05:30
)
const (
INSTALL base.TplName = "install"
)
func checkRunMode() {
2014-12-31 16:07:29 +05:30
switch setting.Cfg.Section("").Key("RUN_MODE").String() {
case "prod":
2014-07-26 09:54:27 +05:30
macaron.Env = macaron.PROD
2015-07-16 00:17:51 +05:30
macaron.ColorLog = false
2014-05-26 05:41:25 +05:30
setting.ProdMode = true
2015-12-09 06:36:12 +05:30
default:
git.Debug = true
}
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()
mailer.NewContext()
2014-04-14 03:42:07 +05:30
}
// GlobalInit is for global configuration reload-able.
func GlobalInit() {
2015-09-17 08:38:46 +05:30
setting.NewContext()
highlight.NewContext()
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)
2015-09-17 08:38:46 +05:30
models.LoadConfigs()
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
2015-09-17 08:38:46 +05:30
cron.NewContext()
2015-07-25 19:02:04 +05:30
models.InitDeliverHooks()
models.InitTestPullRequests()
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 {
2015-08-02 10:06:35 +05:30
log.Info("SQLite3 Supported")
2014-05-31 04:49:30 +05:30
}
if models.EnableTidb {
log.Info("TiDB Supported")
}
2015-12-18 16:19:28 +05:30
if setting.SupportMiniWinService {
log.Info("Builtin Windows Service Supported")
}
checkRunMode()
if setting.StartSSHServer {
ssh.Listen(setting.SSHPort)
log.Info("SSH server started on :%v", setting.SSHPort)
}
2016-01-28 02:18:57 +05:30
// Build Sanitizer
2016-02-21 03:40:05 +05:30
markdown.BuildSanitizer()
}
func InstallInit(ctx *middleware.Context) {
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
dbOpts := []string{"MySQL", "PostgreSQL"}
if models.EnableSQLite3 {
dbOpts = append(dbOpts, "SQLite3")
}
if models.EnableTidb {
dbOpts = append(dbOpts, "TiDB")
}
ctx.Data["DbOptions"] = dbOpts
}
func Install(ctx *middleware.Context) {
form := auth.InstallForm{}
2015-07-09 10:47:48 +05:30
// Database settings
form.DbHost = models.DbCfg.Host
form.DbUser = models.DbCfg.User
form.DbName = models.DbCfg.Name
form.DbPath = models.DbCfg.Path
2015-09-12 06:30:12 +05:30
ctx.Data["CurDbOption"] = "MySQL"
switch models.DbCfg.Type {
case "postgres":
ctx.Data["CurDbOption"] = "PostgreSQL"
case "sqlite3":
if models.EnableSQLite3 {
2015-09-13 01:01:36 +05:30
ctx.Data["CurDbOption"] = "SQLite3"
}
case "tidb":
if models.EnableTidb {
ctx.Data["CurDbOption"] = "TiDB"
2015-09-12 06:30:12 +05:30
}
2015-07-09 10:47:48 +05:30
}
// Application general settings
form.AppName = setting.AppName
form.RepoRootPath = setting.RepoRootPath
// Note(unknwon): it's hard for Windows users change a running user,
// so just use current one if config says default.
if setting.IsWindows && setting.RunUser == "git" {
form.RunUser = user.CurrentUsername()
} else {
form.RunUser = setting.RunUser
2014-04-11 00:07:43 +05:30
}
form.Domain = setting.Domain
form.SSHPort = setting.SSHPort
form.HTTPPort = setting.HttpPort
form.AppUrl = setting.AppUrl
2016-02-12 19:49:45 +05:30
form.LogRootPath = setting.LogRootPath
2015-07-09 10:47:48 +05:30
// E-mail service settings
if setting.MailService != nil {
form.SMTPHost = setting.MailService.Host
2015-07-09 13:40:31 +05:30
form.SMTPFrom = setting.MailService.From
2015-07-09 10:47:48 +05:30
form.SMTPEmail = setting.MailService.User
2014-04-27 10:04:48 +05:30
}
2015-07-09 10:47:48 +05:30
form.RegisterConfirm = setting.Service.RegisterEmailConfirm
form.MailNotify = setting.Service.EnableNotifyMail
// Server and other services settings
form.OfflineMode = setting.OfflineMode
form.DisableGravatar = setting.DisableGravatar
2015-07-09 10:47:48 +05:30
form.DisableRegistration = setting.Service.DisableRegistration
2015-09-13 21:44:32 +05:30
form.EnableCaptcha = setting.Service.EnableCaptcha
2015-07-09 10:47:48 +05:30
form.RequireSignInView = setting.Service.RequireSignInView
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) {
ctx.Data["CurDbOption"] = form.DbType
2014-04-27 10:04:48 +05:30
if ctx.HasError() {
2015-07-08 17:17:56 +05:30
if ctx.HasValue("Err_SMTPEmail") {
ctx.Data["Err_SMTP"] = true
}
if ctx.HasValue("Err_AdminName") ||
ctx.HasValue("Err_AdminPasswd") ||
ctx.HasValue("Err_AdminEmail") {
ctx.Data["Err_Admin"] = true
}
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.
dbTypes := map[string]string{"MySQL": "mysql", "PostgreSQL": "postgres", "SQLite3": "sqlite3", "TiDB": "tidb"}
models.DbCfg.Type = dbTypes[form.DbType]
models.DbCfg.Host = form.DbHost
models.DbCfg.User = form.DbUser
models.DbCfg.Passwd = form.DbPasswd
models.DbCfg.Name = form.DbName
models.DbCfg.SSLMode = form.SSLMode
models.DbCfg.Path = form.DbPath
2015-09-13 01:01:36 +05:30
if (models.DbCfg.Type == "sqlite3" || models.DbCfg.Type == "tidb") &&
len(models.DbCfg.Path) == 0 {
ctx.Data["Err_DbPath"] = true
ctx.RenderWithErr(ctx.Tr("install.err_empty_db_path"), INSTALL, &form)
return
} else if models.DbCfg.Type == "tidb" &&
strings.ContainsAny(path.Base(models.DbCfg.Path), ".-") {
2015-07-08 17:17:56 +05:30
ctx.Data["Err_DbPath"] = true
2015-09-13 01:01:36 +05:30
ctx.RenderWithErr(ctx.Tr("install.err_invalid_tidb_name"), INSTALL, &form)
2015-07-08 17:17:56 +05:30
return
}
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 {
2014-03-30 20:39:59 +05:30
if strings.Contains(err.Error(), `Unknown database type: sqlite3`) {
2015-07-08 17:17:56 +05:30
ctx.Data["Err_DbType"] = true
2014-11-10 16:00:07 +05:30
ctx.RenderWithErr(ctx.Tr("install.sqlite3_not_available", "http://gogs.io/docs/installation/install_from_binary.html"), INSTALL, &form)
2014-03-30 19:09:44 +05:30
} else {
2015-07-08 17:17:56 +05:30
ctx.Data["Err_DbSetting"] = true
ctx.RenderWithErr(ctx.Tr("install.invalid_db_setting", err), INSTALL, &form)
2014-03-30 19:09:44 +05:30
}
return
}
// Test repository root path.
form.RepoRootPath = strings.Replace(form.RepoRootPath, "\\", "/", -1)
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
}
2016-02-12 19:49:45 +05:30
// Test log root path.
form.LogRootPath = strings.Replace(form.LogRootPath, "\\", "/", -1)
if err := os.MkdirAll(form.LogRootPath, os.ModePerm); err != nil {
ctx.Data["Err_LogRootPath"] = true
ctx.RenderWithErr(ctx.Tr("install.invalid_log_root_path", err), INSTALL, &form)
return
}
2014-03-30 21:28:21 +05:30
// Check run user.
curUser := user.CurrentUsername()
2014-03-30 21:28:21 +05:30
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
}
2015-09-13 01:01:36 +05:30
// Check logic loophole between disable self-registration and no admin account.
if form.DisableRegistration && len(form.AdminName) == 0 {
ctx.Data["Err_Services"] = true
ctx.Data["Err_Admin"] = true
ctx.RenderWithErr(ctx.Tr("install.no_admin_and_disable_registration"), INSTALL, form)
return
}
// Check admin password.
2015-09-13 01:01:36 +05:30
if len(form.AdminName) > 0 && len(form.AdminPasswd) == 0 {
ctx.Data["Err_Admin"] = true
ctx.Data["Err_AdminPasswd"] = true
ctx.RenderWithErr(ctx.Tr("install.err_empty_admin_password"), INSTALL, form)
return
}
if form.AdminPasswd != form.AdminConfirmPasswd {
2015-08-02 10:06:35 +05:30
ctx.Data["Err_Admin"] = true
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
}
if form.AppUrl[len(form.AppUrl)-1] != '/' {
form.AppUrl += "/"
}
// Save settings.
cfg := ini.Empty()
if com.IsFile(setting.CustomConf) {
// Keeps custom settings if there is already something.
if err := cfg.Append(setting.CustomConf); err != nil {
log.Error(4, "Fail to load custom conf '%s': %v", setting.CustomConf, err)
}
}
cfg.Section("database").Key("DB_TYPE").SetValue(models.DbCfg.Type)
cfg.Section("database").Key("HOST").SetValue(models.DbCfg.Host)
cfg.Section("database").Key("NAME").SetValue(models.DbCfg.Name)
cfg.Section("database").Key("USER").SetValue(models.DbCfg.User)
cfg.Section("database").Key("PASSWD").SetValue(models.DbCfg.Passwd)
cfg.Section("database").Key("SSL_MODE").SetValue(models.DbCfg.SSLMode)
cfg.Section("database").Key("PATH").SetValue(models.DbCfg.Path)
2015-07-09 10:47:48 +05:30
cfg.Section("").Key("APP_NAME").SetValue(form.AppName)
cfg.Section("repository").Key("ROOT").SetValue(form.RepoRootPath)
cfg.Section("").Key("RUN_USER").SetValue(form.RunUser)
cfg.Section("server").Key("DOMAIN").SetValue(form.Domain)
cfg.Section("server").Key("HTTP_PORT").SetValue(form.HTTPPort)
cfg.Section("server").Key("ROOT_URL").SetValue(form.AppUrl)
if form.SSHPort == 0 {
cfg.Section("server").Key("DISABLE_SSH").SetValue("true")
} else {
cfg.Section("server").Key("DISABLE_SSH").SetValue("false")
cfg.Section("server").Key("SSH_PORT").SetValue(com.ToStr(form.SSHPort))
}
if len(strings.TrimSpace(form.SMTPHost)) > 0 {
cfg.Section("mailer").Key("ENABLED").SetValue("true")
cfg.Section("mailer").Key("HOST").SetValue(form.SMTPHost)
2015-07-09 13:40:31 +05:30
cfg.Section("mailer").Key("FROM").SetValue(form.SMTPFrom)
cfg.Section("mailer").Key("USER").SetValue(form.SMTPEmail)
cfg.Section("mailer").Key("PASSWD").SetValue(form.SMTPPasswd)
2015-07-09 10:47:48 +05:30
} else {
cfg.Section("mailer").Key("ENABLED").SetValue("false")
}
2015-07-09 10:47:48 +05:30
cfg.Section("service").Key("REGISTER_EMAIL_CONFIRM").SetValue(com.ToStr(form.RegisterConfirm))
cfg.Section("service").Key("ENABLE_NOTIFY_MAIL").SetValue(com.ToStr(form.MailNotify))
cfg.Section("server").Key("OFFLINE_MODE").SetValue(com.ToStr(form.OfflineMode))
cfg.Section("picture").Key("DISABLE_GRAVATAR").SetValue(com.ToStr(form.DisableGravatar))
2015-07-09 10:47:48 +05:30
cfg.Section("service").Key("DISABLE_REGISTRATION").SetValue(com.ToStr(form.DisableRegistration))
2015-09-13 21:44:32 +05:30
cfg.Section("service").Key("ENABLE_CAPTCHA").SetValue(com.ToStr(form.EnableCaptcha))
2015-07-09 10:47:48 +05:30
cfg.Section("service").Key("REQUIRE_SIGNIN_VIEW").SetValue(com.ToStr(form.RequireSignInView))
cfg.Section("").Key("RUN_MODE").SetValue("prod")
2014-03-30 21:28:21 +05:30
cfg.Section("session").Key("PROVIDER").SetValue("file")
2014-12-21 09:21:16 +05:30
cfg.Section("log").Key("MODE").SetValue("file")
cfg.Section("log").Key("LEVEL").SetValue("Info")
2016-02-12 19:49:45 +05:30
cfg.Section("log").Key("ROOT_PATH").SetValue(form.LogRootPath)
2014-08-27 14:09:36 +05:30
cfg.Section("security").Key("INSTALL_LOCK").SetValue("true")
cfg.Section("security").Key("SECRET_KEY").SetValue(base.GetRandomString(15))
os.MkdirAll(filepath.Dir(setting.CustomConf), os.ModePerm)
if err := cfg.SaveTo(setting.CustomConf); err != nil {
ctx.RenderWithErr(ctx.Tr("install.save_config_failed", err), INSTALL, &form)
return
}
GlobalInit()
// Create admin account
2015-07-08 17:17:56 +05:30
if len(form.AdminName) > 0 {
u := &models.User{
2015-07-08 17:17:56 +05:30
Name: form.AdminName,
Email: form.AdminEmail,
Passwd: form.AdminPasswd,
IsAdmin: true,
IsActive: true,
}
if err := models.CreateUser(u); err != nil {
2015-07-08 17:17:56 +05:30
if !models.IsErrUserAlreadyExist(err) {
setting.InstallLock = false
ctx.Data["Err_AdminName"] = true
ctx.Data["Err_AdminEmail"] = true
ctx.RenderWithErr(ctx.Tr("install.invalid_admin_setting", err), INSTALL, &form)
return
}
log.Info("Admin account already exist")
u, _ = models.GetUserByName(u.Name)
2014-03-30 20:39:59 +05:30
}
// Auto-login for admin
ctx.Session.Set("uid", u.Id)
ctx.Session.Set("uname", u.Name)
2014-03-30 20:39:59 +05:30
}
log.Info("First-time run install finished!")
ctx.Flash.Success(ctx.Tr("install.install_success"))
ctx.Redirect(form.AppUrl + "user/login")
2014-03-25 14:21:42 +05:30
}