/
main.go
49 lines (39 loc) · 1.08 KB
/
main.go
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
package main
import (
logger "github.com/chi-middleware/logrus-logger"
"github.com/go-chi/chi"
"github.com/go-chi/chi/middleware"
log "github.com/sirupsen/logrus"
"net/http"
"os"
)
// Default handler, returns 500 Internal Server Error
func invalidHandler(w http.ResponseWriter, _ *http.Request) {
http.Error(w, "Invalid Request", http.StatusInternalServerError)
}
type Config struct {
birdSocket string
listen string
}
func main() {
conf := Config{
"/var/run/bird.ctl",
":8000",
}
if birdSocketEnv := os.Getenv("BIRD_SOCKET"); birdSocketEnv != "" {
conf.birdSocket = birdSocketEnv
}
if listenEnv := os.Getenv("BIRDLG_LISTEN"); listenEnv != "" {
conf.listen = listenEnv
}
r := chi.NewRouter()
r.Use(logger.Logger("router", log.StandardLogger()))
r.Use(middleware.Recoverer)
// Start HTTP server
r.Get("/", invalidHandler)
r.Get("/bird", birdHandler(conf.birdSocket))
r.HandleFunc("/traceroute", tracerouteIPv4Wrapper)
r.HandleFunc("/traceroute6", tracerouteIPv6Wrapper)
log.Infof("Listening on %s", conf.listen)
log.Fatal(http.ListenAndServe(conf.listen, r))
}