main.go

package main

import (
  "os"
  "log"

  "github.com/gofiber/fiber/v2"
  "github.com/gofiber/fiber/v2/middleware/logger"
  "github.com/gofiber/fiber/v2/middleware/compress"
  "github.com/gofiber/template/django"
)

var is_prod = os.Getenv("ENV") == "prod"

func main() {
  engine := django.New("./views", ".django")

  app := fiber.New(fiber.Config{
    Prefork: is_prod,
    Views: engine,
    PassLocalsToViews: true,
  })

  if !is_prod {
    app.Use(logger.New())
  }
  app.Use(compress.New())

  app.Use(func(c *fiber.Ctx) error {
    c.Set("X-XSS-Protection", "1; mode=block")
    c.Set("X-Content-Type-Options", "nosniff")
    c.Set("X-Download-Options", "noopen")
    c.Set("Strict-Transport-Security", "max-age=63072000; includeSubdomains")
    c.Set("X-Frame-Options", "SAMEORIGIN")
    c.Set("X-DNS-Prefetch-Control", "off")
    c.Set("referrer-policy", "no-referrer, strict-origin-when-cross-origin")

    return c.Next()
  })

  app.Static("/images", "./public/images", fiber.Static{
    Compress: true,
    ByteRange: true,
    MaxAge: 31536000,
  })

  app.Static("/", "./public", fiber.Static{
    Compress: true,
    ByteRange: true,
    MaxAge: 3600,
  })

  app.Get("/", func(c *fiber.Ctx) error {
    theme := c.Cookies("theme", "day")
    c.Set("Link", "<https://jackketcham.com>; rel=\"canonical\"")

    return c.Render("index", fiber.Map{
      "theme": theme,
      "theme_is_day": theme == "day",
    }, "layouts/main")
  })

  app.Get("/projects/bike-data", func(c *fiber.Ctx) error {
    return c.Render("bike_data", fiber.Map{}, "layouts/main")
  })

  app.Get("/projects", func(c *fiber.Ctx) error {
    return c.Render("projects", fiber.Map{}, "layouts/main")
  })

  app.Post("/theme", func(c *fiber.Ctx) error {
    cookie := new(fiber.Cookie)
    cookie.Name = "theme"

    if c.Cookies("theme", "day") == "day" {
      cookie.Value = "night"
    } else {
      cookie.Value = "day"
    }

    c.Cookie(cookie)

    c.Status(fiber.StatusOK)
    return c.Redirect("/")
  })

  app.Use(func(c *fiber.Ctx) error {
    theme := c.Cookies("theme", "day")

    return c.Status(404).Render("404", fiber.Map{
      "theme": theme,
      "theme_is_day": theme == "day",
    }, "layouts/main")
  })

  log.Fatal(app.Listen(":3000"))
}