summaryrefslogtreecommitdiffstats
path: root/main.go
blob: f79ffc265c98170a3c28ca09caa706aa73f0e766 (plain) (blame)
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
package main

import (
	"flag"
	"fmt"
	"log/slog"
	"net/http"
	"os"
	"strings"
	"time"

	"git.wilk.cx/cr-web/registry"
	"git.wilk.cx/cr-web/web"
)

func main() {
	httpAddr := flag.String("http", ":8080", "HTTP listen address")
	defaultVerbosity := os.Getenv("VERBOSITY")
	verbosity := flag.String("verbosity", defaultVerbosity, "Log verbosity: ERR, WARN, INFO (env VERBOSITY)")
	flag.Parse()

	var level slog.Level
	switch strings.ToUpper(strings.TrimSpace(*verbosity)) {
	case "ERROR":
		level = slog.LevelError
	case "WARN":
		level = slog.LevelWarn
	case "INFO":
		level = slog.LevelInfo
	case "":
		level = slog.LevelInfo
	default:
		fmt.Fprintf(os.Stderr, "invalid verbosity %q: expected ERROR, WARN, or INFO\n", *verbosity)
		os.Exit(2)
	}
	slog.SetDefault(slog.New(slog.NewTextHandler(os.Stderr, &slog.HandlerOptions{Level: level})))

	registryURL := os.Getenv("REGISTRY_URL")
	if registryURL == "" {
		registryURL = "http://localhost:5000"
	}
	extRegistryURL := os.Getenv("EXT_REGISTRY_URL")
	if extRegistryURL == "" {
		extRegistryURL = registryURL
	}
	registryHost := strings.TrimRight(strings.TrimPrefix(strings.TrimPrefix(extRegistryURL, "https://"), "http://"), "/")

	provider := &registry.HTTPProvider{
		BaseURL: registryURL,
		Client: &http.Client{
			Timeout: 10 * time.Second,
		},
	}

	server, err := web.NewServer(provider, registryHost)
	if err != nil {
		slog.Error("initialize server", "error", err)
		os.Exit(1)
	}

	httpServer := &http.Server{
		Addr:              *httpAddr,
		Handler:           server.Handler(),
		ReadHeaderTimeout: 5 * time.Second,
	}

	slog.Info("listening", "address", *httpAddr)
	if err := httpServer.ListenAndServe(); err != nil {
		slog.Error("HTTP server stopped", "error", err)
		os.Exit(1)
	}
}