granthpal/server.go at main · PICT-LibraryAutomation/granthpal · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
package main

import (
	"net/http"
	"os"

	"github.com/99designs/gqlgen/graphql/handler"
	"github.com/99designs/gqlgen/graphql/playground"
	"github.com/PICT-LibraryAutomation/granthpal/auth"
	"github.com/PICT-LibraryAutomation/granthpal/graph"
	"github.com/PICT-LibraryAutomation/granthpal/graph/resolvers"
	"github.com/PICT-LibraryAutomation/granthpal/remote"
	"github.com/PICT-LibraryAutomation/granthpal/search"
	"github.com/PICT-LibraryAutomation/granthpal/sessions"
	"github.com/go-chi/chi/v5"
	"github.com/go-chi/cors"
	"github.com/joho/godotenv"
	"github.com/rs/zerolog"
	"github.com/rs/zerolog/log"
)

func main() {
	zerolog.TimeFieldFormat = zerolog.TimeFormatUnix
	log.Logger = log.Output(zerolog.ConsoleWriter{Out: os.Stderr})

	err := godotenv.Load()
	if err != nil {
		log.Fatal().Err(err).Msg("Couldn't load .env file")
	}

	addr := os.Getenv("GRANTHPAL_ADDR")
	if addr == "" {
		log.Fatal().Msg("GRANTHPAL_ADDR not provided")
	}

	remoteDB := remote.ConnectToRemote()
	sm := sessions.NewSessionManager()
	s := search.ConnectToMeilisearch()

	router := chi.NewRouter()

	router.Use(cors.Handler(cors.Options{
		AllowedOrigins:   []string{"https://*", "http://*"},
		AllowedMethods:   []string{"GET", "POST", "PUT", "DELETE", "OPTIONS"},
		AllowedHeaders:   []string{"Accept", "Authorization", "Content-Type", "X-CSRF-Token"},
		ExposedHeaders:   []string{"Link"},
		AllowCredentials: true,
	}))

	router.Use(remote.RemoteMiddleware(remoteDB))
	router.Use(sessions.SessionManagerMiddleware(sm))
	router.Use(auth.AuthMiddleware())
	router.Use(search.SearchMiddleware(s))

	srv := handler.NewDefaultServer(graph.NewExecutableSchema(graph.Config{
		Resolvers: &resolvers.Resolver{
			Remote:         remoteDB,
			SessionManager: sm,
			Search:         s,
		},
		Directives: graph.DirectiveRoot{
			IsAuthenticated: resolvers.IsAuthenticated,
			IsKind:          resolvers.IsKind,
		},
	}))

	router.Handle("/", playground.ApolloSandboxHandler("GraphQL Sandbox", "/gql"))
	router.Handle("/gql", srv)

	router.Post("/auth/login", auth.Login)
	router.Post("/auth/logout", auth.Logout)

	log.Info().
		Str("Address", addr).
		Msg("Listening...")
	err = http.ListenAndServe(addr, router)
	log.Fatal().AnErr("Error", err).Msg("Server terminated")
}