forgejo/modules/base/conf.go

307 lines
8.8 KiB
Go
Raw Normal View History

2014-02-12 19:54:09 +00:00
// 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.
2014-03-07 22:22:15 +00:00
package base
2014-02-12 19:54:09 +00:00
import (
"fmt"
"os"
2014-02-19 09:50:53 +00:00
"os/exec"
"path"
"path/filepath"
2014-03-19 08:48:45 +00:00
"strings"
2014-02-12 19:54:09 +00:00
2014-03-11 03:03:17 +00:00
"github.com/Unknwon/com"
2014-02-12 19:54:09 +00:00
"github.com/Unknwon/goconfig"
2014-04-06 17:00:20 +00:00
qlog "github.com/qiniu/log"
2014-03-18 05:33:53 +00:00
2014-03-21 13:06:47 +00:00
"github.com/gogits/cache"
2014-03-22 12:49:53 +00:00
"github.com/gogits/session"
2014-03-21 13:06:47 +00:00
2014-03-18 05:33:53 +00:00
"github.com/gogits/gogs/modules/log"
2014-02-12 19:54:09 +00:00
)
2014-03-18 05:33:53 +00:00
// Mailer represents a mail service.
type Mailer struct {
Name string
Host string
User, Passwd string
}
var (
2014-03-20 20:04:56 +00:00
AppVer string
AppName string
AppLogo string
AppUrl string
Domain string
SecretKey string
2014-03-21 07:27:59 +00:00
RunUser string
2014-03-20 20:04:56 +00:00
RepoRootPath string
2014-03-28 11:26:22 +00:00
InstallLock bool
2014-03-22 20:40:09 +00:00
LogInRememberDays int
CookieUserName string
CookieRememberName string
2014-03-18 05:33:53 +00:00
Cfg *goconfig.ConfigFile
MailService *Mailer
2014-03-21 13:06:47 +00:00
2014-03-22 13:21:57 +00:00
LogMode string
LogConfig string
2014-03-21 13:06:47 +00:00
Cache cache.Cache
CacheAdapter string
CacheConfig string
2014-03-21 16:13:13 +00:00
2014-03-22 12:49:53 +00:00
SessionProvider string
SessionConfig *session.Config
SessionManager *session.Manager
2014-03-22 13:21:57 +00:00
2014-03-23 14:58:50 +00:00
PictureService string
)
2014-02-12 19:54:09 +00:00
2014-03-19 11:21:23 +00:00
var Service struct {
2014-03-21 05:09:22 +00:00
RegisterEmailConfirm bool
DisenableRegisteration bool
RequireSignInView bool
2014-03-22 10:42:19 +00:00
EnableCacheAvatar bool
2014-03-23 23:09:11 +00:00
NotifyMail bool
2014-03-21 05:09:22 +00:00
ActiveCodeLives int
ResetPwdCodeLives int
2014-03-19 11:21:23 +00:00
}
func ExecDir() (string, error) {
2014-02-19 09:50:53 +00:00
file, err := exec.LookPath(os.Args[0])
if err != nil {
return "", err
}
p, err := filepath.Abs(file)
if err != nil {
return "", err
}
2014-03-26 13:22:08 +00:00
return path.Dir(strings.Replace(p, "\\", "/", -1)), nil
2014-02-19 09:50:53 +00:00
}
2014-03-19 08:08:25 +00:00
var logLevels = map[string]string{
"Trace": "0",
"Debug": "1",
"Info": "2",
"Warn": "3",
"Error": "4",
"Critical": "5",
}
2014-03-19 11:21:23 +00:00
func newService() {
Service.ActiveCodeLives = Cfg.MustInt("service", "ACTIVE_CODE_LIVE_MINUTES", 180)
Service.ResetPwdCodeLives = Cfg.MustInt("service", "RESET_PASSWD_CODE_LIVE_MINUTES", 180)
2014-03-21 05:09:22 +00:00
Service.DisenableRegisteration = Cfg.MustBool("service", "DISENABLE_REGISTERATION", false)
Service.RequireSignInView = Cfg.MustBool("service", "REQUIRE_SIGNIN_VIEW", false)
2014-03-22 10:42:19 +00:00
Service.EnableCacheAvatar = Cfg.MustBool("service", "ENABLE_CACHE_AVATAR", false)
2014-03-19 11:21:23 +00:00
}
2014-03-24 15:58:46 +00:00
func newLogService() {
2014-03-19 08:08:25 +00:00
// Get and check log mode.
2014-03-21 16:13:13 +00:00
LogMode = Cfg.MustValue("log", "MODE", "console")
modeSec := "log." + LogMode
2014-03-19 08:08:25 +00:00
if _, err := Cfg.GetSection(modeSec); err != nil {
2014-04-06 17:00:20 +00:00
qlog.Fatalf("Unknown log mode: %s\n", LogMode)
2014-03-19 08:08:25 +00:00
}
// Log level.
2014-03-21 16:13:13 +00:00
levelName := Cfg.MustValue("log."+LogMode, "LEVEL", "Trace")
2014-03-19 08:48:45 +00:00
level, ok := logLevels[levelName]
2014-03-19 08:08:25 +00:00
if !ok {
2014-04-06 17:00:20 +00:00
qlog.Fatalf("Unknown log level: %s\n", levelName)
2014-03-19 08:08:25 +00:00
}
// Generate log configuration.
2014-03-21 16:13:13 +00:00
switch LogMode {
2014-03-19 08:08:25 +00:00
case "console":
2014-03-21 16:13:13 +00:00
LogConfig = fmt.Sprintf(`{"level":%s}`, level)
2014-03-19 08:08:25 +00:00
case "file":
2014-03-20 01:05:48 +00:00
logPath := Cfg.MustValue(modeSec, "FILE_NAME", "log/gogs.log")
os.MkdirAll(path.Dir(logPath), os.ModePerm)
2014-03-21 16:13:13 +00:00
LogConfig = fmt.Sprintf(
`{"level":%s,"filename":"%s","rotate":%v,"maxlines":%d,"maxsize":%d,"daily":%v,"maxdays":%d}`, level,
2014-03-20 01:05:48 +00:00
logPath,
2014-03-19 08:08:25 +00:00
Cfg.MustBool(modeSec, "LOG_ROTATE", true),
Cfg.MustInt(modeSec, "MAX_LINES", 1000000),
1<<uint(Cfg.MustInt(modeSec, "MAX_SIZE_SHIFT", 28)),
Cfg.MustBool(modeSec, "DAILY_ROTATE", true),
Cfg.MustInt(modeSec, "MAX_DAYS", 7))
case "conn":
LogConfig = fmt.Sprintf(`{"level":"%s","reconnectOnMsg":%v,"reconnect":%v,"net":"%s","addr":"%s"}`, level,
2014-03-19 08:08:25 +00:00
Cfg.MustBool(modeSec, "RECONNECT_ON_MSG", false),
Cfg.MustBool(modeSec, "RECONNECT", false),
Cfg.MustValue(modeSec, "PROTOCOL", "tcp"),
Cfg.MustValue(modeSec, "ADDR", ":7020"))
case "smtp":
LogConfig = fmt.Sprintf(`{"level":"%s","username":"%s","password":"%s","host":"%s","sendTos":"%s","subject":"%s"}`, level,
2014-03-19 08:08:25 +00:00
Cfg.MustValue(modeSec, "USER", "example@example.com"),
Cfg.MustValue(modeSec, "PASSWD", "******"),
Cfg.MustValue(modeSec, "HOST", "127.0.0.1:25"),
Cfg.MustValue(modeSec, "RECEIVERS", "[]"),
Cfg.MustValue(modeSec, "SUBJECT", "Diagnostic message from serve"))
2014-03-22 21:59:22 +00:00
case "database":
LogConfig = fmt.Sprintf(`{"level":"%s","driver":"%s","conn":"%s"}`, level,
2014-03-22 21:59:22 +00:00
Cfg.MustValue(modeSec, "Driver"),
Cfg.MustValue(modeSec, "CONN"))
2014-03-19 08:08:25 +00:00
}
2014-04-06 17:54:48 +00:00
log.Info("%s %s", AppName, AppVer)
2014-03-21 16:13:13 +00:00
log.NewLogger(Cfg.MustInt64("log", "BUFFER_LEN", 10000), LogMode, LogConfig)
log.Info("Log Mode: %s(%s)", strings.Title(LogMode), levelName)
2014-03-19 07:24:17 +00:00
}
2014-03-21 14:09:57 +00:00
func newCacheService() {
CacheAdapter = Cfg.MustValue("cache", "ADAPTER", "memory")
switch CacheAdapter {
case "memory":
CacheConfig = fmt.Sprintf(`{"interval":%d}`, Cfg.MustInt("cache", "INTERVAL", 60))
case "redis", "memcache":
CacheConfig = fmt.Sprintf(`{"conn":"%s"}`, Cfg.MustValue("cache", "HOST"))
default:
2014-04-06 17:00:20 +00:00
qlog.Fatalf("Unknown cache adapter: %s\n", CacheAdapter)
2014-03-21 14:09:57 +00:00
}
var err error
Cache, err = cache.NewCache(CacheAdapter, CacheConfig)
if err != nil {
2014-04-06 17:00:20 +00:00
qlog.Fatalf("Init cache system failed, adapter: %s, config: %s, %v\n",
2014-03-21 14:09:57 +00:00
CacheAdapter, CacheConfig, err)
}
log.Info("Cache Service Enabled")
}
2014-03-22 12:49:53 +00:00
func newSessionService() {
SessionProvider = Cfg.MustValue("session", "PROVIDER", "memory")
SessionConfig = new(session.Config)
SessionConfig.ProviderConfig = Cfg.MustValue("session", "PROVIDER_CONFIG")
SessionConfig.CookieName = Cfg.MustValue("session", "COOKIE_NAME", "i_like_gogits")
SessionConfig.CookieSecure = Cfg.MustBool("session", "COOKIE_SECURE")
SessionConfig.EnableSetCookie = Cfg.MustBool("session", "ENABLE_SET_COOKIE", true)
SessionConfig.GcIntervalTime = Cfg.MustInt64("session", "GC_INTERVAL_TIME", 86400)
SessionConfig.SessionLifeTime = Cfg.MustInt64("session", "SESSION_LIFE_TIME", 86400)
SessionConfig.SessionIDHashFunc = Cfg.MustValue("session", "SESSION_ID_HASHFUNC", "sha1")
SessionConfig.SessionIDHashKey = Cfg.MustValue("session", "SESSION_ID_HASHKEY")
2014-03-22 13:21:57 +00:00
if SessionProvider == "file" {
os.MkdirAll(path.Dir(SessionConfig.ProviderConfig), os.ModePerm)
}
2014-03-22 12:49:53 +00:00
var err error
SessionManager, err = session.NewManager(SessionProvider, *SessionConfig)
if err != nil {
2014-04-06 17:00:20 +00:00
qlog.Fatalf("Init session system failed, provider: %s, %v\n",
2014-03-22 12:49:53 +00:00
SessionProvider, err)
}
log.Info("Session Service Enabled")
}
2014-03-19 07:24:17 +00:00
func newMailService() {
// Check mailer setting.
if Cfg.MustBool("mailer", "ENABLED") {
MailService = &Mailer{
Name: Cfg.MustValue("mailer", "NAME", AppName),
2014-03-30 20:01:50 +00:00
Host: Cfg.MustValue("mailer", "HOST"),
User: Cfg.MustValue("mailer", "USER"),
Passwd: Cfg.MustValue("mailer", "PASSWD"),
2014-03-19 07:24:17 +00:00
}
log.Info("Mail Service Enabled")
}
}
2014-03-19 16:50:44 +00:00
func newRegisterMailService() {
2014-03-19 11:21:23 +00:00
if !Cfg.MustBool("service", "REGISTER_EMAIL_CONFIRM") {
return
} else if MailService == nil {
2014-03-19 16:50:44 +00:00
log.Warn("Register Mail Service: Mail Service is not enabled")
2014-03-19 11:21:23 +00:00
return
}
2014-03-19 12:27:27 +00:00
Service.RegisterEmailConfirm = true
2014-03-19 16:50:44 +00:00
log.Info("Register Mail Service Enabled")
2014-03-19 11:21:23 +00:00
}
2014-03-23 23:09:11 +00:00
func newNotifyMailService() {
if !Cfg.MustBool("service", "ENABLE_NOTIFY_MAIL") {
return
} else if MailService == nil {
log.Warn("Notify Mail Service: Mail Service is not enabled")
return
}
Service.NotifyMail = true
log.Info("Notify Mail Service Enabled")
}
2014-03-21 05:48:10 +00:00
func NewConfigContext() {
2014-03-25 08:51:42 +00:00
//var err error
workDir, err := ExecDir()
2014-02-19 09:50:53 +00:00
if err != nil {
2014-04-06 17:00:20 +00:00
qlog.Fatalf("Fail to get work directory: %s\n", err)
2014-02-19 09:50:53 +00:00
}
2014-03-13 01:40:18 +00:00
cfgPath := filepath.Join(workDir, "conf/app.ini")
2014-02-19 09:50:53 +00:00
Cfg, err = goconfig.LoadConfigFile(cfgPath)
2014-02-12 19:54:09 +00:00
if err != nil {
2014-04-06 17:00:20 +00:00
qlog.Fatalf("Cannot load config file(%s): %v\n", cfgPath, err)
2014-02-12 19:54:09 +00:00
}
Cfg.BlockMode = false
2014-03-11 03:03:17 +00:00
2014-03-13 01:40:18 +00:00
cfgPath = filepath.Join(workDir, "custom/conf/app.ini")
2014-03-25 10:52:56 +00:00
if com.IsFile(cfgPath) {
if err = Cfg.AppendFiles(cfgPath); err != nil {
2014-04-06 17:00:20 +00:00
qlog.Fatalf("Cannot load config file(%s): %v\n", cfgPath, err)
2014-03-25 10:52:56 +00:00
}
2014-03-11 03:03:17 +00:00
}
2014-03-18 05:33:53 +00:00
AppName = Cfg.MustValue("", "APP_NAME", "Gogs: Go Git Service")
2014-03-19 11:21:23 +00:00
AppLogo = Cfg.MustValue("", "APP_LOGO", "img/favicon.png")
AppUrl = Cfg.MustValue("server", "ROOT_URL")
2014-03-17 10:13:07 +00:00
Domain = Cfg.MustValue("server", "DOMAIN")
2014-03-19 11:21:23 +00:00
SecretKey = Cfg.MustValue("security", "SECRET_KEY")
2014-03-25 10:52:56 +00:00
InstallLock = Cfg.MustBool("security", "INSTALL_LOCK", false)
2014-03-21 07:27:59 +00:00
RunUser = Cfg.MustValue("", "RUN_USER")
2014-03-25 10:52:56 +00:00
curUser := os.Getenv("USERNAME")
2014-03-25 11:06:56 +00:00
if len(curUser) == 0 {
2014-03-25 11:10:23 +00:00
curUser = os.Getenv("USER")
2014-03-25 11:06:56 +00:00
}
// Does not check run user when the install lock is off.
if InstallLock && RunUser != curUser {
2014-04-06 17:00:20 +00:00
qlog.Fatalf("Expect user(%s) but current user is: %s\n", RunUser, curUser)
2014-03-25 10:52:56 +00:00
}
2014-03-20 20:04:56 +00:00
2014-03-22 20:40:09 +00:00
LogInRememberDays = Cfg.MustInt("security", "LOGIN_REMEMBER_DAYS")
CookieUserName = Cfg.MustValue("security", "COOKIE_USERNAME")
CookieRememberName = Cfg.MustValue("security", "COOKIE_REMEMBER_NAME")
2014-03-22 10:42:19 +00:00
PictureService = Cfg.MustValue("picture", "SERVICE")
2014-03-20 20:04:56 +00:00
// Determine and create root git reposiroty path.
2014-03-28 21:34:07 +00:00
homeDir, err := com.HomeDir()
if err != nil {
2014-04-06 17:00:20 +00:00
qlog.Fatalf("Fail to get home directory): %v\n", err)
2014-03-28 21:34:07 +00:00
}
2014-04-06 17:41:58 +00:00
RepoRootPath = Cfg.MustValue("repository", "ROOT", filepath.Join(homeDir, "gogs-repositories"))
2014-03-20 20:04:56 +00:00
if err = os.MkdirAll(RepoRootPath, os.ModePerm); err != nil {
2014-04-06 17:00:20 +00:00
qlog.Fatalf("Fail to create RepoRootPath(%s): %v\n", RepoRootPath, err)
2014-03-20 20:04:56 +00:00
}
2014-03-19 17:24:46 +00:00
}
2014-03-18 05:33:53 +00:00
2014-03-19 17:24:46 +00:00
func NewServices() {
2014-03-19 14:46:48 +00:00
newService()
2014-03-24 15:58:46 +00:00
newLogService()
2014-03-21 14:09:57 +00:00
newCacheService()
2014-03-22 12:49:53 +00:00
newSessionService()
2014-03-19 07:24:17 +00:00
newMailService()
2014-03-19 16:50:44 +00:00
newRegisterMailService()
2014-03-23 23:09:11 +00:00
newNotifyMailService()
2014-02-12 19:54:09 +00:00
}