snake_case

webcontroller/web_controller.go

@@ -0,0 +1,114 @@
+package webcontroller
+
+import (
+	"errors"
+	"net/http"
+
+	"github.com/google/uuid"
+
+	"fornaxian.com/pixeldrain-web/init/conf"
+	"github.com/Fornaxian/log"
+	"github.com/julienschmidt/httprouter"
+)
+
+// WebController controls how requests are handled and makes sure they have
+// proper context when running
+type WebController struct {
+	conf              *conf.PixelWebConfig
+	templates         *TemplateManager
+	staticResourceDir string
+
+	// page-specific variables
+	captchaSiteKey string
+}
+
+// New initializes a new WebController by registering all the request handlers
+// and parsing all templates in the resource directory
+func New(r *httprouter.Router, prefix string, conf *conf.PixelWebConfig) *WebController {
+	var wc = &WebController{
+		conf:              conf,
+		staticResourceDir: conf.StaticResourceDir,
+	}
+	wc.templates = NewTemplateManager(
+		conf.TemplateDir,
+		conf.APIURLExternal,
+		conf.DebugMode,
+	)
+	wc.templates.ParseTemplates(false)
+
+	var p = prefix
+
+	// Serve static files
+	r.ServeFiles(p+"/res/*filepath", http.Dir(wc.staticResourceDir))
+
+	// General navigation
+	r.GET(p+"/" /*                */, wc.serveTemplate("home", false))
+	r.GET(p+"/favicon.ico" /*     */, wc.serveFile("/favicon.ico"))
+	r.GET(p+"/global.css" /*      */, wc.globalCSSHandler)
+	r.GET(p+"/api" /*             */, wc.serveTemplate("apidoc", false))
+	r.GET(p+"/history" /*         */, wc.serveTemplate("history_cookies", false))
+	r.GET(p+"/u/:id" /*           */, wc.serveFileViewer)
+	r.GET(p+"/u/:id/preview" /*   */, wc.serveFilePreview)
+	r.GET(p+"/l/:id" /*           */, wc.serveListViewer)
+	r.GET(p+"/t" /*               */, wc.serveTemplate("paste", false))
+
+	// User account pages
+	r.GET(p+"/register" /*        */, wc.serveRegister)
+	r.GET(p+"/login" /*           */, wc.serveTemplate("login", false))
+	r.GET(p+"/logout" /*          */, wc.serveTemplate("logout", true))
+	r.POST(p+"/logout" /*         */, wc.serveLogout)
+	r.GET(p+"/user" /*            */, wc.serveTemplate("user_home", true))
+	r.GET(p+"/user/files" /*      */, wc.serveTemplate("user_files", true))
+	r.GET(p+"/user/lists" /*      */, wc.serveTemplate("user_lists", true))
+	r.GET(p+"/user/filemanager" /**/, wc.serveTemplate("file_manager", true))
+
+	r.NotFound = http.HandlerFunc(wc.serveNotFound)
+
+	return wc
+}
+
+func (wc *WebController) serveTemplate(
+	tpl string,
+	requireAuth bool,
+) httprouter.Handle {
+	return func(
+		w http.ResponseWriter,
+		r *http.Request,
+		p httprouter.Params,
+	) {
+		var tpld = wc.newTemplateData(w, r)
+		if requireAuth && !tpld.Authenticated {
+			http.Redirect(w, r, "/login", http.StatusSeeOther)
+			return
+		}
+		err := wc.templates.Get().ExecuteTemplate(w, tpl, tpld)
+		if err != nil {
+			log.Error("Error executing template '%s': %s", tpl, err)
+		}
+	}
+}
+
+func (wc *WebController) serveFile(path string) httprouter.Handle {
+	return func(
+		w http.ResponseWriter,
+		r *http.Request,
+		p httprouter.Params,
+	) {
+		http.ServeFile(w, r, wc.staticResourceDir+"/favicon.ico")
+	}
+}
+
+func (wc *WebController) serveNotFound(w http.ResponseWriter, r *http.Request) {
+	log.Debug("Not Found: %s", r.URL)
+	w.WriteHeader(http.StatusNotFound)
+	wc.templates.Get().ExecuteTemplate(w, "404", wc.newTemplateData(w, r))
+}
+
+func (wc *WebController) getAPIKey(r *http.Request) (key string, err error) {
+	if cookie, err := r.Cookie("pd_auth_key"); err == nil {
+		if _, err := uuid.Parse(cookie.Value); err == nil {
+			return cookie.Value, nil
+		}
+	}
+	return "", errors.New("not a valid pixeldrain authentication cookie")
+}