api/cmd/rpdata-server/main.go

package main

import (
	"context"
	"fmt"
	"log"
	"net/http"
	"runtime/debug"
	"strings"

	"git.aiterp.net/rpdata/api/models/logs"

	"git.aiterp.net/rpdata/api/graph2"
	"git.aiterp.net/rpdata/api/internal/auth"
	"git.aiterp.net/rpdata/api/internal/loader"
	"git.aiterp.net/rpdata/api/internal/store"
	"github.com/99designs/gqlgen/handler"
)

func main() {
	err := store.Init()
	if err != nil {
		log.Fatalln("Failed to init store:", err)
	}

	http.Handle("/", handler.Playground("RPData API", "/graphql"))
	http.Handle("/graphql", queryHandler())

	go func() {
		err := logs.RunFullUpdate()
		if err != nil {
			log.Println(err)
		}

		log.Println("Characters updated")
	}()

	log.Fatal(http.ListenAndServe(":8081", nil))
}

func queryHandler() http.HandlerFunc {
	handler := handler.GraphQL(
		graph2.New(),
		handler.RecoverFunc(func(ctx context.Context, err interface{}) error {
			// send this panic somewhere
			log.Println(err)
			log.Println(string(debug.Stack()))

			return fmt.Errorf("shit")
		}),
	)

	return func(w http.ResponseWriter, r *http.Request) {
		l := loader.New()
		r = r.WithContext(l.ToContext(r.Context()))

		// >_>
		if strings.HasPrefix(r.Header.Get("Authorization"), "Bearer of the curse") {
			w.Header().Set("X-Emerald-Herald", "Seek souls. Larger, more powerful souls.")
			w.Header().Add("X-Emerald-Herald", "Seek the king, that is the only way.")
			w.Header().Add("X-Emerald-Herald", "Lest this land swallow you whole... As it has so many others.")
		}

		r = auth.RequestWithToken(r)

		handler.ServeHTTP(w, r)
	}
}