mirror of
				https://gitee.com/gitea/gitea
				synced 2025-11-04 08:30:25 +08:00 
			
		
		
		
	
		
			
				
	
	
		
			372 lines
		
	
	
		
			10 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			372 lines
		
	
	
		
			10 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
// 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 base
 | 
						|
 | 
						|
import (
 | 
						|
	"fmt"
 | 
						|
	"os"
 | 
						|
	"os/exec"
 | 
						|
	"path"
 | 
						|
	"path/filepath"
 | 
						|
	"regexp"
 | 
						|
	"strings"
 | 
						|
 | 
						|
	"github.com/Unknwon/com"
 | 
						|
	"github.com/Unknwon/goconfig"
 | 
						|
	qlog "github.com/qiniu/log"
 | 
						|
 | 
						|
	"github.com/gogits/cache"
 | 
						|
	"github.com/gogits/session"
 | 
						|
 | 
						|
	"github.com/gogits/gogs/modules/auth/ldap"
 | 
						|
	"github.com/gogits/gogs/modules/log"
 | 
						|
)
 | 
						|
 | 
						|
// Mailer represents mail service.
 | 
						|
type Mailer struct {
 | 
						|
	Name         string
 | 
						|
	Host         string
 | 
						|
	User, Passwd string
 | 
						|
}
 | 
						|
 | 
						|
type OauthInfo struct {
 | 
						|
	ClientId, ClientSecret string
 | 
						|
	Scopes                 string
 | 
						|
	AuthUrl, TokenUrl      string
 | 
						|
}
 | 
						|
 | 
						|
// Oauther represents oauth service.
 | 
						|
type Oauther struct {
 | 
						|
	GitHub, Google, Tencent,
 | 
						|
	Twitter, Weibo bool
 | 
						|
	OauthInfos map[string]*OauthInfo
 | 
						|
}
 | 
						|
 | 
						|
var (
 | 
						|
	AppVer      string
 | 
						|
	AppName     string
 | 
						|
	AppLogo     string
 | 
						|
	AppUrl      string
 | 
						|
	OfflineMode bool
 | 
						|
	RouterLog   bool
 | 
						|
	ProdMode    bool
 | 
						|
	Domain      string
 | 
						|
	SecretKey   string
 | 
						|
	RunUser     string
 | 
						|
 | 
						|
	RepoRootPath string
 | 
						|
	ScriptType   string
 | 
						|
 | 
						|
	InstallLock bool
 | 
						|
 | 
						|
	LogInRememberDays  int
 | 
						|
	CookieUserName     string
 | 
						|
	CookieRememberName string
 | 
						|
 | 
						|
	Cfg          *goconfig.ConfigFile
 | 
						|
	MailService  *Mailer
 | 
						|
	OauthService *Oauther
 | 
						|
 | 
						|
	LogMode   string
 | 
						|
	LogConfig string
 | 
						|
 | 
						|
	Cache        cache.Cache
 | 
						|
	CacheAdapter string
 | 
						|
	CacheConfig  string
 | 
						|
 | 
						|
	SessionProvider string
 | 
						|
	SessionConfig   *session.Config
 | 
						|
	SessionManager  *session.Manager
 | 
						|
 | 
						|
	PictureService string
 | 
						|
 | 
						|
	EnableRedis    bool
 | 
						|
	EnableMemcache bool
 | 
						|
)
 | 
						|
 | 
						|
var Service struct {
 | 
						|
	RegisterEmailConfirm bool
 | 
						|
	DisableRegistration  bool
 | 
						|
	RequireSignInView    bool
 | 
						|
	EnableCacheAvatar    bool
 | 
						|
	NotifyMail           bool
 | 
						|
	ActiveCodeLives      int
 | 
						|
	ResetPwdCodeLives    int
 | 
						|
	LdapAuth             bool
 | 
						|
}
 | 
						|
 | 
						|
func ExecDir() (string, error) {
 | 
						|
	file, err := exec.LookPath(os.Args[0])
 | 
						|
	if err != nil {
 | 
						|
		return "", err
 | 
						|
	}
 | 
						|
	p, err := filepath.Abs(file)
 | 
						|
	if err != nil {
 | 
						|
		return "", err
 | 
						|
	}
 | 
						|
	return path.Dir(strings.Replace(p, "\\", "/", -1)), nil
 | 
						|
}
 | 
						|
 | 
						|
var logLevels = map[string]string{
 | 
						|
	"Trace":    "0",
 | 
						|
	"Debug":    "1",
 | 
						|
	"Info":     "2",
 | 
						|
	"Warn":     "3",
 | 
						|
	"Error":    "4",
 | 
						|
	"Critical": "5",
 | 
						|
}
 | 
						|
 | 
						|
func newService() {
 | 
						|
	Service.ActiveCodeLives = Cfg.MustInt("service", "ACTIVE_CODE_LIVE_MINUTES", 180)
 | 
						|
	Service.ResetPwdCodeLives = Cfg.MustInt("service", "RESET_PASSWD_CODE_LIVE_MINUTES", 180)
 | 
						|
	Service.DisableRegistration = Cfg.MustBool("service", "DISABLE_REGISTRATION", false)
 | 
						|
	Service.RequireSignInView = Cfg.MustBool("service", "REQUIRE_SIGNIN_VIEW", false)
 | 
						|
	Service.EnableCacheAvatar = Cfg.MustBool("service", "ENABLE_CACHE_AVATAR", false)
 | 
						|
}
 | 
						|
 | 
						|
func newLogService() {
 | 
						|
	// Get and check log mode.
 | 
						|
	LogMode = Cfg.MustValue("log", "MODE", "console")
 | 
						|
	modeSec := "log." + LogMode
 | 
						|
	if _, err := Cfg.GetSection(modeSec); err != nil {
 | 
						|
		qlog.Fatalf("Unknown log mode: %s\n", LogMode)
 | 
						|
	}
 | 
						|
 | 
						|
	// Log level.
 | 
						|
	levelName := Cfg.MustValue("log."+LogMode, "LEVEL", "Trace")
 | 
						|
	level, ok := logLevels[levelName]
 | 
						|
	if !ok {
 | 
						|
		qlog.Fatalf("Unknown log level: %s\n", levelName)
 | 
						|
	}
 | 
						|
 | 
						|
	// Generate log configuration.
 | 
						|
	switch LogMode {
 | 
						|
	case "console":
 | 
						|
		LogConfig = fmt.Sprintf(`{"level":%s}`, level)
 | 
						|
	case "file":
 | 
						|
		logPath := Cfg.MustValue(modeSec, "FILE_NAME", "log/gogs.log")
 | 
						|
		os.MkdirAll(path.Dir(logPath), os.ModePerm)
 | 
						|
		LogConfig = fmt.Sprintf(
 | 
						|
			`{"level":%s,"filename":"%s","rotate":%v,"maxlines":%d,"maxsize":%d,"daily":%v,"maxdays":%d}`, level,
 | 
						|
			logPath,
 | 
						|
			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,
 | 
						|
			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,
 | 
						|
			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"))
 | 
						|
	case "database":
 | 
						|
		LogConfig = fmt.Sprintf(`{"level":"%s","driver":"%s","conn":"%s"}`, level,
 | 
						|
			Cfg.MustValue(modeSec, "Driver"),
 | 
						|
			Cfg.MustValue(modeSec, "CONN"))
 | 
						|
	}
 | 
						|
 | 
						|
	log.Info("%s %s", AppName, AppVer)
 | 
						|
	log.NewLogger(Cfg.MustInt64("log", "BUFFER_LEN", 10000), LogMode, LogConfig)
 | 
						|
	log.Info("Log Mode: %s(%s)", strings.Title(LogMode), levelName)
 | 
						|
}
 | 
						|
 | 
						|
func newLdapService() {
 | 
						|
	Service.LdapAuth = Cfg.MustBool("security", "LDAP_AUTH", false)
 | 
						|
	if !Service.LdapAuth {
 | 
						|
		return
 | 
						|
	}
 | 
						|
 | 
						|
	nbsrc := 0
 | 
						|
	for _, v := range Cfg.GetSectionList() {
 | 
						|
		if matched, _ := regexp.MatchString("(?i)^LDAPSOURCE.*", v); matched {
 | 
						|
			ldapname := Cfg.MustValue(v, "name", v)
 | 
						|
			ldaphost := Cfg.MustValue(v, "host")
 | 
						|
			ldapport := Cfg.MustInt(v, "port", 389)
 | 
						|
			ldapbasedn := Cfg.MustValue(v, "basedn", "dc=*,dc=*")
 | 
						|
			ldapattribute := Cfg.MustValue(v, "attribute", "mail")
 | 
						|
			ldapfilter := Cfg.MustValue(v, "filter", "(*)")
 | 
						|
			ldapmsadsaformat := Cfg.MustValue(v, "MSADSAFORMAT", "%s")
 | 
						|
			ldap.AddSource(ldapname, ldaphost, ldapport, ldapbasedn, ldapattribute, ldapfilter, ldapmsadsaformat)
 | 
						|
			nbsrc++
 | 
						|
			log.Debug("%s added as LDAP source", ldapname)
 | 
						|
		}
 | 
						|
	}
 | 
						|
	if nbsrc == 0 {
 | 
						|
		log.Warn("No valide LDAP found, LDAP Authentication NOT enabled")
 | 
						|
		Service.LdapAuth = false
 | 
						|
		return
 | 
						|
	}
 | 
						|
 | 
						|
	log.Info("LDAP Authentication Enabled")
 | 
						|
}
 | 
						|
 | 
						|
func newCacheService() {
 | 
						|
	CacheAdapter = Cfg.MustValue("cache", "ADAPTER", "memory")
 | 
						|
	if EnableRedis {
 | 
						|
		log.Info("Redis Enabled")
 | 
						|
	}
 | 
						|
	if EnableMemcache {
 | 
						|
		log.Info("Memcache Enabled")
 | 
						|
	}
 | 
						|
 | 
						|
	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:
 | 
						|
		qlog.Fatalf("Unknown cache adapter: %s\n", CacheAdapter)
 | 
						|
	}
 | 
						|
 | 
						|
	var err error
 | 
						|
	Cache, err = cache.NewCache(CacheAdapter, CacheConfig)
 | 
						|
	if err != nil {
 | 
						|
		qlog.Fatalf("Init cache system failed, adapter: %s, config: %s, %v\n",
 | 
						|
			CacheAdapter, CacheConfig, err)
 | 
						|
	}
 | 
						|
 | 
						|
	log.Info("Cache Service Enabled")
 | 
						|
}
 | 
						|
 | 
						|
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")
 | 
						|
 | 
						|
	if SessionProvider == "file" {
 | 
						|
		os.MkdirAll(path.Dir(SessionConfig.ProviderConfig), os.ModePerm)
 | 
						|
	}
 | 
						|
 | 
						|
	var err error
 | 
						|
	SessionManager, err = session.NewManager(SessionProvider, *SessionConfig)
 | 
						|
	if err != nil {
 | 
						|
		qlog.Fatalf("Init session system failed, provider: %s, %v\n",
 | 
						|
			SessionProvider, err)
 | 
						|
	}
 | 
						|
 | 
						|
	log.Info("Session Service Enabled")
 | 
						|
}
 | 
						|
 | 
						|
func newMailService() {
 | 
						|
	// Check mailer setting.
 | 
						|
	if !Cfg.MustBool("mailer", "ENABLED") {
 | 
						|
		return
 | 
						|
	}
 | 
						|
 | 
						|
	MailService = &Mailer{
 | 
						|
		Name:   Cfg.MustValue("mailer", "NAME", AppName),
 | 
						|
		Host:   Cfg.MustValue("mailer", "HOST"),
 | 
						|
		User:   Cfg.MustValue("mailer", "USER"),
 | 
						|
		Passwd: Cfg.MustValue("mailer", "PASSWD"),
 | 
						|
	}
 | 
						|
	log.Info("Mail Service Enabled")
 | 
						|
}
 | 
						|
 | 
						|
func newRegisterMailService() {
 | 
						|
	if !Cfg.MustBool("service", "REGISTER_EMAIL_CONFIRM") {
 | 
						|
		return
 | 
						|
	} else if MailService == nil {
 | 
						|
		log.Warn("Register Mail Service: Mail Service is not enabled")
 | 
						|
		return
 | 
						|
	}
 | 
						|
	Service.RegisterEmailConfirm = true
 | 
						|
	log.Info("Register Mail Service Enabled")
 | 
						|
}
 | 
						|
 | 
						|
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")
 | 
						|
}
 | 
						|
 | 
						|
func NewConfigContext() {
 | 
						|
	workDir, err := ExecDir()
 | 
						|
	if err != nil {
 | 
						|
		qlog.Fatalf("Fail to get work directory: %s\n", err)
 | 
						|
	}
 | 
						|
 | 
						|
	cfgPath := filepath.Join(workDir, "conf/app.ini")
 | 
						|
	Cfg, err = goconfig.LoadConfigFile(cfgPath)
 | 
						|
	if err != nil {
 | 
						|
		qlog.Fatalf("Cannot load config file(%s): %v\n", cfgPath, err)
 | 
						|
	}
 | 
						|
	Cfg.BlockMode = false
 | 
						|
 | 
						|
	cfgPath = filepath.Join(workDir, "custom/conf/app.ini")
 | 
						|
	if com.IsFile(cfgPath) {
 | 
						|
		if err = Cfg.AppendFiles(cfgPath); err != nil {
 | 
						|
			qlog.Fatalf("Cannot load config file(%s): %v\n", cfgPath, err)
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	AppName = Cfg.MustValue("", "APP_NAME", "Gogs: Go Git Service")
 | 
						|
	AppLogo = Cfg.MustValue("", "APP_LOGO", "img/favicon.png")
 | 
						|
	AppUrl = Cfg.MustValue("server", "ROOT_URL")
 | 
						|
	Domain = Cfg.MustValue("server", "DOMAIN")
 | 
						|
	OfflineMode = Cfg.MustBool("server", "OFFLINE_MODE", false)
 | 
						|
	RouterLog = Cfg.MustBool("server", "ROUTER_LOG", true)
 | 
						|
	SecretKey = Cfg.MustValue("security", "SECRET_KEY")
 | 
						|
 | 
						|
	InstallLock = Cfg.MustBool("security", "INSTALL_LOCK", false)
 | 
						|
 | 
						|
	RunUser = Cfg.MustValue("", "RUN_USER")
 | 
						|
	curUser := os.Getenv("USER")
 | 
						|
	if len(curUser) == 0 {
 | 
						|
		curUser = os.Getenv("USERNAME")
 | 
						|
	}
 | 
						|
	// Does not check run user when the install lock is off.
 | 
						|
	if InstallLock && RunUser != curUser {
 | 
						|
		qlog.Fatalf("Expect user(%s) but current user is: %s\n", RunUser, curUser)
 | 
						|
	}
 | 
						|
 | 
						|
	LogInRememberDays = Cfg.MustInt("security", "LOGIN_REMEMBER_DAYS")
 | 
						|
	CookieUserName = Cfg.MustValue("security", "COOKIE_USERNAME")
 | 
						|
	CookieRememberName = Cfg.MustValue("security", "COOKIE_REMEMBER_NAME")
 | 
						|
	PictureService = Cfg.MustValue("picture", "SERVICE")
 | 
						|
 | 
						|
	// Determine and create root git reposiroty path.
 | 
						|
	homeDir, err := com.HomeDir()
 | 
						|
	if err != nil {
 | 
						|
		qlog.Fatalf("Fail to get home directory): %v\n", err)
 | 
						|
	}
 | 
						|
	RepoRootPath = Cfg.MustValue("repository", "ROOT", filepath.Join(homeDir, "gogs-repositories"))
 | 
						|
	if err = os.MkdirAll(RepoRootPath, os.ModePerm); err != nil {
 | 
						|
		qlog.Fatalf("Fail to create RepoRootPath(%s): %v\n", RepoRootPath, err)
 | 
						|
	}
 | 
						|
	ScriptType = Cfg.MustValue("repository", "SCRIPT_TYPE", "bash")
 | 
						|
}
 | 
						|
 | 
						|
func NewBaseServices() {
 | 
						|
	newService()
 | 
						|
	newLogService()
 | 
						|
	newLdapService()
 | 
						|
	newCacheService()
 | 
						|
	newSessionService()
 | 
						|
	newMailService()
 | 
						|
	newRegisterMailService()
 | 
						|
	newNotifyMailService()
 | 
						|
}
 |