/
main.go
96 lines (83 loc) · 2.2 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
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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
package main
import (
"log"
"net/http"
"foxygo.at/foxtrot/pkg/foxtrot"
"github.com/alecthomas/kong"
"github.com/gorilla/websocket"
_ "github.com/mattn/go-sqlite3"
)
var (
// Semver holds the version exposed at api/version and passed via linker flag on CI.
Semver = "undefined"
// CommitSha holds the commit exposed at api/version and passed via linker flag on CI.
CommitSha = "undefined"
)
func main() {
cfg := &foxtrot.Config{
Version: foxtrot.Version{Semver: Semver, CommitSha: CommitSha, App: "foxtrot"},
}
kong.Parse(cfg, kong.Description("Foxtrot Server"))
mux := http.NewServeMux()
mux.Handle("/", http.FileServer(http.Dir("frontend/public")))
if _, err := foxtrot.NewApp(cfg, mux); err != nil {
log.Fatal(err)
}
port := ":8080"
log.Printf("Listening on port %s", port)
if err := http.ListenAndServe(port, handler(logHTTP(mux))); err != nil {
log.Fatal(err)
}
}
func handler(h http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
if r.URL.Path == "/ws" {
socketHandler(w, r)
} else {
h.ServeHTTP(w, r)
}
})
}
func logHTTP(h http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
ww := &responseWriter{
ResponseWriter: w,
statusCode: http.StatusOK,
}
h.ServeHTTP(ww, r)
log.Printf("%d %-4s %s %s\n", ww.statusCode, r.Method, r.URL, r.RemoteAddr)
})
}
type responseWriter struct {
http.ResponseWriter
statusCode int
}
func (w *responseWriter) WriteHeader(code int) {
w.statusCode = code
w.ResponseWriter.WriteHeader(code)
}
// upgrader holds the websocket connection.
var upgrader = websocket.Upgrader{
ReadBufferSize: 1024,
WriteBufferSize: 1024,
}
// socketHandler echos websocket messages back to the client.
func socketHandler(w http.ResponseWriter, r *http.Request) {
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Printf("upgrader.Upgrade: %v", err)
return
}
defer conn.Close() //nolint: errcheck
for {
messageType, p, err := conn.ReadMessage()
if err != nil {
log.Printf("conn.ReadMessage: %v", err)
return
}
if err := conn.WriteMessage(messageType, p); err != nil {
log.Printf("conn.WriteMessage: %v", err)
return
}
}
}