-
Notifications
You must be signed in to change notification settings - Fork 0
/
status_endpoint.go
65 lines (54 loc) · 1.51 KB
/
status_endpoint.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
package status_endpoint
//go:generate go run github.com/maxbrunsfeld/counterfeiter/v6 -generate
import (
"code.cloudfoundry.org/lager"
"encoding/json"
"fmt"
"github.com/timotto/ardumower-relay/internal/model"
"net/http"
)
type statusEndpoint struct {
logger lager.Logger
auth Auth
broker Broker
}
//counterfeiter:generate -o fake . Auth
type Auth interface {
LookupUser(req *http.Request) (model.User, error)
}
//counterfeiter:generate -o fake . Broker
type Broker interface {
Find(user model.User) (model.Tunnel, bool)
}
func NewStatusEndpoint(logger lager.Logger, auth Auth, broker Broker) *statusEndpoint {
return &statusEndpoint{
logger: logger.Session("status-endpoint"),
auth: auth,
broker: broker,
}
}
func (s *statusEndpoint) ServeHTTP(w http.ResponseWriter, req *http.Request) {
user, err := s.auth.LookupUser(req)
if err != nil {
s.logger.Info("lookup-user", lager.Data{"error": err.Error()})
w.Header().Set("www-authenticate", `Basic realm="ArduMower Relay"`)
onError(w, http.StatusUnauthorized, fmt.Errorf("unauthorized"))
return
}
tun, found := s.broker.Find(user)
if !found {
onError(w, http.StatusNotFound, fmt.Errorf("not connected"))
return
}
stats := tun.Stats()
bytes, err := json.Marshal(&stats)
if err != nil {
onError(w, http.StatusInternalServerError, fmt.Errorf("encode stats failed: %w", err))
return
}
_, _ = w.Write(bytes)
}
func onError(w http.ResponseWriter, statusCode int, err error) {
w.WriteHeader(statusCode)
_, _ = w.Write([]byte(err.Error()))
}