.
,-. ,-. . . ,-. ,-. ,-.    ,-. ,-. ,-,-. ,-,-. . . ,-. . |- . .
`-. | | | | |   |   |-'    |   | | | | | | | | | | | | | |  | |
`-' `-' `-^ '   `-' `-' :: `-' `-' ' ' ' ' ' ' `-^ ' ' ' `' `-|
                                                             /|
                                                            `-'

git clone https://source.community/Fuwn/space.git

────╮
   1│ // Copyright (C) 2021-2021 Fuwn
   2│ // SPDX-License-Identifier: GPL-3.0-only
   3│ 
   4│ package main
   5│ 
   6│ import (
   7│ 	"io"
   8│ 	"text/template"
   9│ 
  10│ 	"github.com/fuwn/space/pkg/database"
  11│ 	"github.com/pitr/gig"
  12│ 	"github.com/spf13/viper"
  13│ )
  14│ 
  15│ type Template struct {
  16│ 	Templates *template.Template
  17│ }
  18│ 
  19│ type IndexTemplate struct {
  20│ 	Content string
  21│ 	Quote   string
  22│ 	Hits    int
  23│ 	// SystemInfo string
  24│ 	Copyright string
  25│ }
  26│ type ErrorTemplate struct {
  27│ 	Error     string
  28│ 	Content   string
  29│ 	Quote     string
  30│ 	Hits      int
  31│ 	Copyright string
  32│ }
  33│ 
  34│ // Lazy...
  35│ func isHitsEnabled() bool {
  36│ 	return viper.GetBool("space.hits")
  37│ }
  38│ func (_ IndexTemplate) HitsEnabled() bool {
  39│ 	return isHitsEnabled()
  40│ }
  41│ func (_ ErrorTemplate) HitsEnabled() bool {
  42│ 	return isHitsEnabled()
  43│ }
  44│ 
  45│ func (t *Template) Render(w io.Writer, name string, data interface{}, c gig.Context) error {
  46│ 	// Check if the route is present in the hits tracker, if it isn't present, add
  47│ 	// it, but either way: increment it.
  48│ 	hits := database.Get(c.Path())
  49│ 	if hits == 0 {
  50│ 		database.Create(c.Path())
  51│ 	}
  52│ 	database.Increment(c.Path())
  53│ 
  54│ 	return t.Templates.ExecuteTemplate(w, name, data)
  55│ }
────╯