authentication/main.go at main · knockbox/authentication · 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
package main

import (
	"flag"
	"github.com/gorilla/mux"
	"github.com/hashicorp/go-hclog"
	"github.com/joho/godotenv"
	"github.com/knockbox/authentication/internal/handlers"
	"github.com/knockbox/authentication/pkg/keyring"
	"github.com/knockbox/authentication/pkg/middleware"
	"github.com/knockbox/authentication/pkg/utils"
	"os"
)

var bindAddress string
var useDotEnv bool

func init() {
	const (
		usageBindAddress = "the address to bind to, e.g. :9090"
		usageUseDotEnv   = "read variables from a .env file in running directory"
	)

	flag.StringVar(&bindAddress, "bindAddress", ":9090", usageBindAddress)
	flag.StringVar(&bindAddress, "addr", ":9090", usageBindAddress)

	flag.BoolVar(&useDotEnv, "dotenv", false, usageUseDotEnv)
	flag.BoolVar(&useDotEnv, "denv", false, usageUseDotEnv)
}

func main() {
	flag.Parse()

	l := hclog.Default()
	l.SetLevel(hclog.Trace)

	keyset, err := keyring.NewSet(129600, 86400, l)
	if err != nil {
		panic(err)
	}

	keyset.SetCurveTypes(keyring.P521)
	if err := keyset.Generate(3); err != nil {
		panic(err)
	}

	if useDotEnv {
		l.Info("using .env")

		if err := godotenv.Load(); err != nil {
			l.Error(".env", "error", err)
			os.Exit(1)
		}
	}

	sm := mux.NewRouter()
	sm.Use(middleware.UseLogging(l).Middleware)

	// /api grouping
	apiRouter := sm.PathPrefix("/api").Subrouter()
	//apiRouter.Use(middleware.UseCaching(l).Middleware)

	// Routes
	handlers.NewHealthcheck().Route(apiRouter)
	handlers.NewUser(l, keyset).Route(apiRouter)
	handlers.NewToken(l, keyset).Route(apiRouter)

	utils.StartServerWithGracefulShutdown(middleware.CORSMiddleware(sm), bindAddress, l)
}