github.com/Azareal/Gosora@v0.0.0-20210729070923-553e66b59003/routes/misc.go

package routes

import (
	"bytes"
	"database/sql"
	"io"
	"net/http"
	"strconv"
	"strings"
	"time"

	//"fmt"

	c "github.com/Azareal/Gosora/common"
	"github.com/Azareal/Gosora/common/phrases"
)

var cacheControlMaxAge = "max-age=" + strconv.Itoa(int(c.Day))      // TODO: Make this a c.Config value
var cacheControlMaxAgeWeek = "max-age=" + strconv.Itoa(int(c.Week)) // TODO: Make this a c.Config value

// GET functions
func StaticFile(w http.ResponseWriter, r *http.Request) {
	file, ok := c.StaticFiles.Get(r.URL.Path)
	if !ok {
		//c.DebugLogf("Failed to find '%s'", r.URL.Path) // TODO: Use MicroNotFound? Might be better than the unneccessary overhead of sprintf
		w.WriteHeader(http.StatusNotFound)
		return
	}
	h := w.Header()

	if file.Length > 300 {
		if h.Get("Range") != "" {
			h.Set("Vary", "Accept-Encoding")
			if len(file.Sha256) != 0 {
				h.Set("Cache-Control", cacheControlMaxAgeWeek)
			} else {
				h.Set("Cache-Control", cacheControlMaxAge) //Cache-Control: max-age=31536000
			}
			ae := r.Header.Get("Accept-Encoding")

			if file.BrLength > 300 && strings.Contains(ae, "br") {
				h.Set("Content-Encoding", "br")
				h.Set("Content-Length", file.StrBrLength)
				http.ServeContent(w, r, r.URL.Path, file.Info.ModTime(), bytes.NewReader(file.BrData))
				return
			}

			if file.GzipLength > 300 && strings.Contains(ae, "gzip") {
				h.Set("Content-Encoding", "gzip")
				h.Set("Content-Length", file.StrGzipLength)
				http.ServeContent(w, r, r.URL.Path, file.Info.ModTime(), bytes.NewReader(file.GzipData))
				return
			}

			h.Set("Content-Length", file.StrLength)
			http.ServeContent(w, r, r.URL.Path, file.Info.ModTime(), bytes.NewReader(file.Data))
			return
		}
	}

	// Surely, there's a more efficient way of doing this?
	t, err := time.Parse(http.TimeFormat, r.Header.Get("If-Modified-Since"))
	if err == nil && file.Info.ModTime().Before(t.Add(1*time.Second)) {
		w.WriteHeader(http.StatusNotModified)
		return
	}
	h.Set("Last-Modified", file.FormattedModTime)
	h.Set("Content-Type", file.Mimetype)
	if len(file.Sha256) != 0 {
		h.Set("Cache-Control", cacheControlMaxAgeWeek)
	} else {
		h.Set("Cache-Control", cacheControlMaxAge) //Cache-Control: max-age=31536000
	}
	h.Set("Vary", "Accept-Encoding")

	if file.BrLength > 0 && strings.Contains(r.Header.Get("Accept-Encoding"), "br") {
		h.Set("Content-Encoding", "br")
		h.Set("Content-Length", file.StrBrLength)
		io.Copy(w, bytes.NewReader(file.BrData)) // Use w.Write instead?
	} else if file.GzipLength > 0 && strings.Contains(r.Header.Get("Accept-Encoding"), "gzip") {
		h.Set("Content-Encoding", "gzip")
		h.Set("Content-Length", file.StrGzipLength)
		io.Copy(w, bytes.NewReader(file.GzipData)) // Use w.Write instead?
	} else {
		h.Set("Content-Length", file.StrLength)
		io.Copy(w, bytes.NewReader(file.Data))
	}
	// Other options instead of io.Copy: io.CopyN(), w.Write(), http.ServeContent()
}

func Overview(w http.ResponseWriter, r *http.Request, u *c.User, h *c.Header) c.RouteError {
	h.Title = phrases.GetTitlePhrase("overview")
	h.Zone = "overview"
	return renderTemplate("overview", w, r, h, c.Page{h, tList, nil})
}

func CustomPage(w http.ResponseWriter, r *http.Request, u *c.User, h *c.Header, name string) c.RouteError {
	h.Zone = "custom_page"
	name = c.SanitiseSingleLine(name)
	page, err := c.Pages.GetByName(name)
	if err == nil {
		h.Title = page.Title
		return renderTemplate("custom_page", w, r, h, c.CustomPagePage{h, page})
	} else if err != sql.ErrNoRows {
		return c.InternalError(err, w, r)
	}
	h.Title = phrases.GetTitlePhrase("page")

	// TODO: Pass the page name to the pre-render hook?
	err = renderTemplate3("page_"+name, "tmpl_page", w, r, h, c.Page{h, tList, nil})
	if err == c.ErrBadDefaultTemplate {
		return c.NotFound(w, r, h)
	} else if err != nil {
		return c.InternalError(err, w, r)
	}
	return nil
}

// TODO: Set the cookie domain
func ChangeTheme(w http.ResponseWriter, r *http.Request, u *c.User) c.RouteError {
	//headerLite, _ := SimpleUserCheck(w, r, u)
	// TODO: Rename js to something else, just in case we rewrite the JS side in WebAssembly?
	js := r.PostFormValue("js") == "1"
	newTheme := c.SanitiseSingleLine(r.PostFormValue("theme"))
	//fmt.Printf("newTheme: %+v\n", newTheme)

	theme, ok := c.Themes[newTheme]
	if !ok || theme.HideFromThemes {
		return c.LocalErrorJSQ("That theme doesn't exist", w, r, u, js)
	}

	cookie := http.Cookie{Name: "current_theme", Value: newTheme, Path: "/", MaxAge: int(c.Year)}
	http.SetCookie(w, &cookie)
	return actionSuccess(w, r, "/", js)
}