main.go

package main

import (
	"context"
	"embed"
	"flag"
	"net/http"
	"os"
	"time"

	"github.com/jmoiron/sqlx"
	log "github.com/sirupsen/logrus"
	bolt "go.etcd.io/bbolt"
	"modernc.org/ql"
	_ "modernc.org/ql/driver"

	"git.c3pb.de/gbe/invinoveritas/auth"
)

//go:embed templates/*.tpl
var templateFS embed.FS

//go:embed static/*
var staticFS embed.FS

func httpError(w http.ResponseWriter, msg string, err error, status int) {
	if err != nil {
		msg += ": " + err.Error()
	}

	log.WithField("status", status).
		WithError(err).
		Error(msg)

	http.Error(w, msg, status)
}

type Handler struct {
	sqlDB *sqlx.DB
	ap    authProvider
}

func logRequest(next http.Handler) http.Handler {
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		start := time.Now()
		defer func() {
			d := time.Since(start)

			log.WithFields(log.Fields{
				"method":   r.Method,
				"url":      r.URL,
				"remote":   r.RemoteAddr,
				"duration": d,
				"auth":     auth.Get(r),
			}).Info("handling request")
		}()

		next.ServeHTTP(w, r)
	})
}

func openQLdb(path string) (*sqlx.DB, error) {
	var retried bool

	path += ".ql"

retry:
	db, err := sqlx.Open("ql2", path)
	if err != nil {
		return nil, err
	}

	log.Info("running test-select on db")
	_, err = db.Exec(`SELECT count(*) FROM __Table`)
	if err != nil && !retried {
		log.Println("got err:", err)
		retried = true

		// WAL may be corrupted. Manually remove it and re-try open. See
		// https://gitlab.com/cznic/ql/-/issues/227 for more info.
		err := os.Remove(ql.WalName(path))
		if err != nil {
			return nil, err
		}

		goto retry
	}

	if err != nil {
		return nil, err
	}

	return db, nil
}

func main() {
	formatter := log.TextFormatter{
		DisableColors: true,
	}

	log.StandardLogger().Formatter = &formatter

	dbPath := flag.String("db", "vino", "Path to database file")
	listenAddr := flag.String("listen", "127.0.0.1:7878", "Listening address")
	flag.Parse()

	sqlDB, err := openQLdb(*dbPath)
	if err != nil {
		log.WithError(err).Fatalln("can't open DB:")
	}
	defer sqlDB.Close()

	db, err := bolt.Open(*dbPath+".db", 0644, nil)
	if err != nil {
		log.WithError(err).Fatalln("can't open DB")
	}
	defer db.Close()

	err = initSQLdb(context.TODO(), sqlDB)
	if err != nil {
		log.WithField("wal_name", ql.WalName(*dbPath+".ql")).
			WithError(err).
			Fatalln("can't initalize DB")
	}

	err = initBoltDB(db)
	if err != nil {
		log.WithError(err).Fatalln("can't initialize DB")
	}

	err = migrateBoltToQL(context.TODO(), db, sqlDB)
	if err != nil {
		log.WithError(err).Fatalln("can't migrate bolt to sql db")
	}

	http.HandleFunc("/favicon.ico", http.NotFound)

	http.Handle("/static/", logRequest(http.FileServer(http.FS(staticFS))))

	ap := authProvider{
		db: sqlDB,
	}

	handler := Handler{
		sqlDB: sqlDB,
		ap:    ap,
	}

	http.Handle("/details/img", auth.Require(logRequest(handler.img()), ap))
	http.Handle("/details/", auth.Require(logRequest(handler.details()), ap))
	http.Handle("/user/", auth.Require(logRequest(handler.user()), ap))
	http.Handle("/", auth.Require(logRequest(handler.index()), ap))

	log.Printf("here we go, listening on http://%s", *listenAddr)

	err = http.ListenAndServe(*listenAddr, nil)
	if err != nil {
		log.WithError(err).Fatalln("http handler failed")
	}
}