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) }