117 lines
		
	
	
		
			3.4 KiB
		
	
	
	
		
			Go
		
	
	
	
			
		
		
	
	
			117 lines
		
	
	
		
			3.4 KiB
		
	
	
	
		
			Go
		
	
	
	
| // Copyright 2020 The Gitea 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 install
 | |
| 
 | |
| import (
 | |
| 	"fmt"
 | |
| 	"net/http"
 | |
| 	"path"
 | |
| 
 | |
| 	"code.gitea.io/gitea/modules/log"
 | |
| 	"code.gitea.io/gitea/modules/public"
 | |
| 	"code.gitea.io/gitea/modules/setting"
 | |
| 	"code.gitea.io/gitea/modules/templates"
 | |
| 	"code.gitea.io/gitea/modules/web"
 | |
| 	"code.gitea.io/gitea/modules/web/middleware"
 | |
| 	"code.gitea.io/gitea/routers/common"
 | |
| 	"code.gitea.io/gitea/services/forms"
 | |
| 
 | |
| 	"gitea.com/go-chi/session"
 | |
| )
 | |
| 
 | |
| type dataStore map[string]interface{}
 | |
| 
 | |
| func (d *dataStore) GetData() map[string]interface{} {
 | |
| 	return *d
 | |
| }
 | |
| 
 | |
| func installRecovery() func(next http.Handler) http.Handler {
 | |
| 	rnd := templates.HTMLRenderer()
 | |
| 	return func(next http.Handler) http.Handler {
 | |
| 		return http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
 | |
| 			defer func() {
 | |
| 				// Why we need this? The first recover will try to render a beautiful
 | |
| 				// error page for user, but the process can still panic again, then
 | |
| 				// we have to just recover twice and send a simple error page that
 | |
| 				// should not panic any more.
 | |
| 				defer func() {
 | |
| 					if err := recover(); err != nil {
 | |
| 						combinedErr := fmt.Sprintf("PANIC: %v\n%s", err, log.Stack(2))
 | |
| 						log.Error("%s", combinedErr)
 | |
| 						if setting.IsProd {
 | |
| 							http.Error(w, http.StatusText(http.StatusInternalServerError), http.StatusInternalServerError)
 | |
| 						} else {
 | |
| 							http.Error(w, combinedErr, http.StatusInternalServerError)
 | |
| 						}
 | |
| 					}
 | |
| 				}()
 | |
| 
 | |
| 				if err := recover(); err != nil {
 | |
| 					combinedErr := fmt.Sprintf("PANIC: %v\n%s", err, log.Stack(2))
 | |
| 					log.Error("%s", combinedErr)
 | |
| 
 | |
| 					lc := middleware.Locale(w, req)
 | |
| 					store := dataStore{
 | |
| 						"Language":       lc.Language(),
 | |
| 						"CurrentURL":     setting.AppSubURL + req.URL.RequestURI(),
 | |
| 						"i18n":           lc,
 | |
| 						"SignedUserID":   int64(0),
 | |
| 						"SignedUserName": "",
 | |
| 					}
 | |
| 
 | |
| 					w.Header().Set(`X-Frame-Options`, setting.CORSConfig.XFrameOptions)
 | |
| 
 | |
| 					if !setting.IsProd {
 | |
| 						store["ErrorMsg"] = combinedErr
 | |
| 					}
 | |
| 					err = rnd.HTML(w, http.StatusInternalServerError, "status/500", templates.BaseVars().Merge(store))
 | |
| 					if err != nil {
 | |
| 						log.Error("%v", err)
 | |
| 					}
 | |
| 				}
 | |
| 			}()
 | |
| 
 | |
| 			next.ServeHTTP(w, req)
 | |
| 		})
 | |
| 	}
 | |
| }
 | |
| 
 | |
| // Routes registers the install routes
 | |
| func Routes() *web.Route {
 | |
| 	r := web.NewRoute()
 | |
| 	for _, middle := range common.Middlewares() {
 | |
| 		r.Use(middle)
 | |
| 	}
 | |
| 
 | |
| 	r.Use(web.WrapWithPrefix(public.AssetsURLPathPrefix, public.AssetsHandlerFunc(&public.Options{
 | |
| 		Directory: path.Join(setting.StaticRootPath, "public"),
 | |
| 		Prefix:    public.AssetsURLPathPrefix,
 | |
| 	}), "InstallAssetsHandler"))
 | |
| 
 | |
| 	r.Use(session.Sessioner(session.Options{
 | |
| 		Provider:       setting.SessionConfig.Provider,
 | |
| 		ProviderConfig: setting.SessionConfig.ProviderConfig,
 | |
| 		CookieName:     setting.SessionConfig.CookieName,
 | |
| 		CookiePath:     setting.SessionConfig.CookiePath,
 | |
| 		Gclifetime:     setting.SessionConfig.Gclifetime,
 | |
| 		Maxlifetime:    setting.SessionConfig.Maxlifetime,
 | |
| 		Secure:         setting.SessionConfig.Secure,
 | |
| 		SameSite:       setting.SessionConfig.SameSite,
 | |
| 		Domain:         setting.SessionConfig.Domain,
 | |
| 	}))
 | |
| 
 | |
| 	r.Use(installRecovery())
 | |
| 	r.Use(Init)
 | |
| 	r.Get("/", Install)
 | |
| 	r.Post("/", web.Bind(forms.InstallForm{}), SubmitInstall)
 | |
| 
 | |
| 	r.NotFound(web.Wrap(installNotFound))
 | |
| 	return r
 | |
| }
 | |
| 
 | |
| func installNotFound(w http.ResponseWriter, req *http.Request) {
 | |
| 	http.Redirect(w, req, setting.AppURL, http.StatusFound)
 | |
| }
 |