@ -6,13 +6,46 @@ package routers
import (
import (
"errors"
"errors"
"os"
"strings"
"github.com/Unknwon/goconfig"
"github.com/codegangsta/martini"
"github.com/gogits/gogs/models"
"github.com/gogits/gogs/models"
"github.com/gogits/gogs/modules/auth"
"github.com/gogits/gogs/modules/auth"
"github.com/gogits/gogs/modules/base"
"github.com/gogits/gogs/modules/base"
"github.com/gogits/gogs/modules/log"
"github.com/gogits/gogs/modules/mailer"
"github.com/gogits/gogs/modules/middleware"
"github.com/gogits/gogs/modules/middleware"
)
)
// Check run mode(Default of martini is Dev).
func checkRunMode ( ) {
switch base . Cfg . MustValue ( "" , "RUN_MODE" ) {
case "prod" :
martini . Env = martini . Prod
case "test" :
martini . Env = martini . Test
}
log . Info ( "Run Mode: %s" , strings . Title ( martini . Env ) )
}
// GlobalInit is for global configuration reload-able.
func GlobalInit ( ) {
base . NewConfigContext ( )
mailer . NewMailerContext ( )
models . LoadModelsConfig ( )
models . LoadRepoConfig ( )
models . NewRepoContext ( )
if err := models . NewEngine ( ) ; err != nil && base . InstallLock {
log . Error ( "%v" , err )
os . Exit ( 2 )
}
base . NewServices ( )
checkRunMode ( )
}
func Install ( ctx * middleware . Context , form auth . InstallForm ) {
func Install ( ctx * middleware . Context , form auth . InstallForm ) {
if base . InstallLock {
if base . InstallLock {
ctx . Handle ( 404 , "install.Install" , errors . New ( "Installation is prohibited" ) )
ctx . Handle ( 404 , "install.Install" , errors . New ( "Installation is prohibited" ) )
@ -20,14 +53,114 @@ func Install(ctx *middleware.Context, form auth.InstallForm) {
}
}
ctx . Data [ "Title" ] = "Install"
ctx . Data [ "Title" ] = "Install"
ctx . Data [ "DbCfg" ] = models . DbCfg
ctx . Data [ "RepoRootPath" ] = base . RepoRootPath
ctx . Data [ "RunUser" ] = base . RunUser
ctx . Data [ "AppUrl" ] = base . AppUrl
ctx . Data [ "PageIsInstall" ] = true
ctx . Data [ "PageIsInstall" ] = true
// Get and assign value to install form.
if len ( form . Host ) == 0 {
form . Host = models . DbCfg . Host
}
if len ( form . User ) == 0 {
form . User = models . DbCfg . User
}
if len ( form . Passwd ) == 0 {
form . Passwd = models . DbCfg . Pwd
}
if len ( form . DatabaseName ) == 0 {
form . DatabaseName = models . DbCfg . Name
}
if len ( form . DatabasePath ) == 0 {
form . DatabasePath = models . DbCfg . Path
}
if len ( form . RepoRootPath ) == 0 {
form . RepoRootPath = base . RepoRootPath
}
if len ( form . RunUser ) == 0 {
form . RunUser = base . RunUser
}
if len ( form . Domain ) == 0 {
form . Domain = base . Domain
}
if len ( form . AppUrl ) == 0 {
form . AppUrl = base . AppUrl
}
auth . AssignForm ( form , ctx . Data )
if ctx . Req . Method == "GET" {
if ctx . Req . Method == "GET" {
ctx . HTML ( 200 , "install" )
ctx . HTML ( 200 , "install" )
return
return
}
}
if ctx . HasError ( ) {
ctx . HTML ( 200 , "install" )
return
}
// Pass basic check, now test configuration.
// Test database setting.
dbTypes := map [ string ] string { "mysql" : "mysql" , "pgsql" : "postgres" , "sqlite" : "sqlite3" }
models . DbCfg . Type = dbTypes [ form . Database ]
models . DbCfg . Host = form . Host
models . DbCfg . User = form . User
models . DbCfg . Pwd = form . Passwd
models . DbCfg . Name = form . DatabaseName
models . DbCfg . SslMode = form . SslMode
models . DbCfg . Path = form . DatabasePath
if err := models . NewEngine ( ) ; err != nil {
ctx . RenderWithErr ( "Database setting is not correct: " + err . Error ( ) , "install" , & form )
return
}
// Test repository root path.
if err := os . MkdirAll ( form . RepoRootPath , os . ModePerm ) ; err != nil {
ctx . RenderWithErr ( "Repository root path is invalid: " + err . Error ( ) , "install" , & form )
return
}
// Create admin account.
if _ , err := models . RegisterUser ( & models . User { Name : form . AdminName , Email : form . AdminEmail , Passwd : form . AdminPasswd ,
IsAdmin : true , IsActive : true } ) ; err != nil {
if err != models . ErrUserAlreadyExist {
ctx . RenderWithErr ( "Admin account setting is invalid: " + err . Error ( ) , "install" , & form )
return
}
}
// Save settings.
base . Cfg . SetValue ( "database" , "DB_TYPE" , models . DbCfg . Type )
base . Cfg . SetValue ( "database" , "HOST" , models . DbCfg . Host )
base . Cfg . SetValue ( "database" , "NAME" , models . DbCfg . Name )
base . Cfg . SetValue ( "database" , "USER" , models . DbCfg . User )
base . Cfg . SetValue ( "database" , "PASSWD" , models . DbCfg . Pwd )
base . Cfg . SetValue ( "database" , "SSL_MODE" , models . DbCfg . SslMode )
base . Cfg . SetValue ( "database" , "PATH" , models . DbCfg . Path )
base . Cfg . SetValue ( "repository" , "ROOT" , form . RepoRootPath )
base . Cfg . SetValue ( "" , "RUN_USER" , form . RunUser )
base . Cfg . SetValue ( "server" , "DOMAIN" , form . Domain )
base . Cfg . SetValue ( "server" , "ROOT_URL" , form . AppUrl )
if len ( form . Host ) > 0 {
base . Cfg . SetValue ( "mailer" , "ENABLED" , "true" )
base . Cfg . SetValue ( "mailer" , "HOST" , form . SmtpHost )
base . Cfg . SetValue ( "mailer" , "USER" , form . SmtpEmail )
base . Cfg . SetValue ( "mailer" , "PASSWD" , form . SmtpPasswd )
base . Cfg . SetValue ( "service" , "REGISTER_EMAIL_CONFIRM" , base . ToStr ( form . RegisterConfirm == "on" ) )
base . Cfg . SetValue ( "service" , "ENABLE_NOTIFY_MAIL" , base . ToStr ( form . MailNotify == "on" ) )
}
base . Cfg . SetValue ( "security" , "INSTALL_LOCK" , "true" )
if err := goconfig . SaveConfigFile ( base . Cfg , "custom/conf/app.ini" ) ; err != nil {
ctx . RenderWithErr ( "Fail to save configuration: " + err . Error ( ) , "install" , & form )
return
}
GlobalInit ( )
log . Info ( "First-time run install finished!" )
ctx . Redirect ( "/user/login" )
}
}