You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
100 lines
2.5 KiB
100 lines
2.5 KiB
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
"net/http"
|
|
"runtime/debug"
|
|
"strings"
|
|
|
|
"git.aiterp.net/rpdata/api/models"
|
|
"git.aiterp.net/rpdata/api/models/changes"
|
|
|
|
"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")
|
|
}()
|
|
|
|
go logListedChanges()
|
|
|
|
log.Fatal(http.ListenAndServe(":8081", nil))
|
|
}
|
|
|
|
func logListedChanges() {
|
|
sub := changes.Subscribe(context.Background(), nil, true)
|
|
|
|
for change := range sub {
|
|
log.Printf("Change: Author=%#+v Model=%#+v Op=%#+v", change.Author, change.Model, change.Op)
|
|
for _, object := range change.Objects() {
|
|
switch object := object.(type) {
|
|
case models.Log:
|
|
log.Printf("\tLog: %s (%s)", object.ID, object.ShortID)
|
|
case models.Character:
|
|
log.Printf("\tCharacter: %s", object.ID)
|
|
case models.Channel:
|
|
log.Printf("\tChannel: %s", object.Name)
|
|
case models.Post:
|
|
log.Printf("\tPost: %s", object.ID)
|
|
case models.Story:
|
|
log.Printf("\tStory: %s", object.ID)
|
|
case models.Tag:
|
|
log.Printf("\tTag: (%s,%s)", object.Kind, object.Name)
|
|
case models.Chapter:
|
|
log.Printf("\tChapter: %s", object.ID)
|
|
}
|
|
}
|
|
}
|
|
|
|
log.Println("Change subscription closed.")
|
|
}
|
|
|
|
func queryHandler() http.HandlerFunc {
|
|
handler := handler.GraphQL(
|
|
graph2.New(),
|
|
handler.RecoverFunc(func(ctx context.Context, err interface{}) error {
|
|
log.Println(err)
|
|
log.Println(string(debug.Stack()))
|
|
|
|
return fmt.Errorf("The server failed to serve your request due to an internal error, it has been logged")
|
|
}),
|
|
)
|
|
|
|
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)
|
|
}
|
|
}
|