marvin/marvinblum
1
0
Fork 0
mirror of https://github.com/Kugelschieber/marvinblum.git synced 2026-01-18 06:40:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added experimental version of pirsch for server side tracking.

Browse Source
This commit is contained in:
Marvin Blum
2020-06-21 16:01:42 +02:00
committed by Marvin Blum
parent a23faafc25
commit 186f1213a6
10 changed files with 106 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
main.go
View File

@@ -1,29 +1,36 @@
package main
import (
"database/sql"
"fmt"
"github.com/Kugelschieber/marvinblum.de/blog"
"github.com/Kugelschieber/marvinblum.de/tpl"
"github.com/NYTimes/gziphandler"
"github.com/caddyserver/certmagic"
emvi "github.com/emvi/api-go"
"github.com/emvi/logbuch"
"github.com/emvi/pirsch"
"github.com/gorilla/mux"
_ "github.com/lib/pq"
"github.com/rs/cors"
"html/template"
"net/http"
"os"
"strconv"
"strings"
"time"
)
const (
staticDir = "static"
staticDirPrefix = "/static/"
logTimeFormat = "2006-01-02_15:04:05"
envPrefix = "MB_"
staticDir = "static"
staticDirPrefix = "/static/"
logTimeFormat = "2006-01-02_15:04:05"
envPrefix = "MB_"
connectionString = `host=%s port=%s user=%s password=%s dbname=%s sslmode=%s sslcert=%s sslkey=%s sslrootcert=%s connectTimeout=%s timezone=%s`
)
var (
tracker *pirsch.Tracker
blogInstance *blog.Blog
)
@@ -50,8 +57,37 @@ func logEnvConfig() {
}
}
func setupTracker() {
logbuch.Info("Connecting to database...")
host := os.Getenv("MB_DB_HOST")
port := os.Getenv("MB_DB_PORT")
user := os.Getenv("MB_DB_USER")
password := os.Getenv("MB_DB_PASSWORD")
schema := os.Getenv("MB_DB_SCHEMA")
sslMode := os.Getenv("MB_DB_SSLMODE")
sslCert := os.Getenv("MB_DB_SSLCERT")
sslKey := os.Getenv("MB_DB_SSLKEY")
sslRootCert := os.Getenv("MB_DB_SSLROOTCERT")
zone, offset := time.Now().Zone()
timezone := zone + strconv.Itoa(-offset/3600)
logbuch.Info("Setting time zone", logbuch.Fields{"timezone": timezone})
connectionStr := fmt.Sprintf(connectionString, host, port, user, password, schema, sslMode, sslCert, sslKey, sslRootCert, "30", timezone)
db, err := sql.Open("postgres", connectionStr)
if err != nil {
logbuch.Fatal("Error connecting to database", logbuch.Fields{"err": err})
}
if err := db.Ping(); err != nil {
logbuch.Fatal("Error pinging database", logbuch.Fields{"err": err})
}
tracker = pirsch.NewTracker(pirsch.NewPostgresStore(db))
}
func serveAbout() http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
tracker.Hit(r)
data := struct {
Articles []emvi.Article
}{
@@ -67,6 +103,7 @@ func serveAbout() http.HandlerFunc {
func serveBlogPage() http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
tracker.Hit(r)
data := struct {
Articles map[int][]emvi.Article
}{
@@ -82,6 +119,7 @@ func serveBlogPage() http.HandlerFunc {
func serveBlogArticle() http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
tracker.Hit(r)
vars := mux.Vars(r)
slug := strings.Split(vars["slug"], "-")
@@ -119,9 +157,9 @@ func setupRouter() *mux.Router {
router.PathPrefix(staticDirPrefix).Handler(http.StripPrefix(staticDirPrefix, gziphandler.GzipHandler(http.FileServer(http.Dir(staticDir)))))
router.Handle("/blog/{slug}", serveBlogArticle())
router.Handle("/blog", serveBlogPage())
router.Handle("/legal", tpl.ServeTemplate("legal.html"))
router.Handle("/legal", tpl.ServeTemplate("legal.html", tracker))
router.Handle("/", serveAbout())
router.NotFoundHandler = tpl.ServeTemplate("notfound.html")
router.NotFoundHandler = tpl.ServeTemplate("notfound.html", tracker)
return router
}
@@ -162,6 +200,7 @@ func main() {
configureLog()
logEnvConfig()
tpl.LoadTemplate()
setupTracker()
blogInstance = blog.NewBlog()
router := setupRouter()
corsConfig := configureCors(router)