imagini/internal/api/routes.go

package api

import (
	"encoding/json"
	"net/http"

	"github.com/99designs/gqlgen/graphql/handler"
	"github.com/99designs/gqlgen/graphql/playground"

	"reichard.io/imagini/graph"
	"reichard.io/imagini/graph/generated"
)

func (api *API) registerRoutes() {

	// Set up Directives
	graphConfig := generated.Config{
		Resolvers: &graph.Resolver{
			DB:   api.DB,
			Auth: api.Auth,
		},
		Directives: generated.DirectiveRoot{
			Meta:       api.metaDirective,
			HasMinRole: api.hasMinRoleDirective,
		},
	}
	srv := handler.NewDefaultServer(generated.NewExecutableSchema(graphConfig))

	// Handle GraphQL
	api.Router.Handle("/playground", playground.Handler("GraphQL playground", "/query"))
	api.Router.Handle("/query", api.injectContextMiddleware(srv))

	// Handle Resource Route
	api.Router.HandleFunc("/media/", multipleMiddleware(
		api.mediaHandler,
		api.authMiddleware,
	))

}

func errorJSON(w http.ResponseWriter, err string, code int) {
	errStruct := &APIResponse{Error: &APIError{Message: err, Code: int64(code)}}
	responseJSON(w, errStruct, code)
}

func responseJSON(w http.ResponseWriter, msg interface{}, code int) {
	w.Header().Set("Content-Type", "application/json; charset=utf-8")
	w.Header().Set("X-Content-Type-Options", "nosniff")
	w.WriteHeader(code)
	json.NewEncoder(w).Encode(msg)
}

func successJSON(w http.ResponseWriter, msg string, code int) {
	w.Header().Set("Content-Type", "application/json; charset=utf-8")
	w.Header().Set("X-Content-Type-Options", "nosniff")
	w.WriteHeader(code)
	json.NewEncoder(w).Encode(map[string]interface{}{"success": msg})
}