feat(server): add vpn route to replace /openvpn

This commit is contained in:
Quentin McGaw
2022-08-21 23:28:38 +00:00
parent 48896176e5
commit d685d78e74
4 changed files with 100 additions and 5 deletions

View File

@@ -18,13 +18,14 @@ func newHandler(ctx context.Context, logger infoWarner, logging bool,
) http.Handler {
handler := &handler{}
vpn := newVPNHandler(ctx, vpnLooper, logger)
openvpn := newOpenvpnHandler(ctx, vpnLooper, pfGetter, logger)
dns := newDNSHandler(ctx, unboundLooper, logger)
updater := newUpdaterHandler(ctx, updaterLooper, logger)
publicip := newPublicIPHandler(publicIPLooper, logger)
handler.v0 = newHandlerV0(ctx, logger, vpnLooper, unboundLooper, updaterLooper)
handler.v1 = newHandlerV1(logger, buildInfo, openvpn, dns, updater, publicip)
handler.v1 = newHandlerV1(logger, buildInfo, vpn, openvpn, dns, updater, publicip)
handlerWithLog := withLogMiddleware(handler, logger, logging)
handler.setLogEnabled = handlerWithLog.setEnabled

View File

@@ -10,10 +10,11 @@ import (
)
func newHandlerV1(w warner, buildInfo models.BuildInformation,
openvpn, dns, updater, publicip http.Handler) http.Handler {
vpn, openvpn, dns, updater, publicip http.Handler) http.Handler {
return &handlerV1{
warner: w,
buildInfo: buildInfo,
vpn: vpn,
openvpn: openvpn,
dns: dns,
updater: updater,
@@ -24,6 +25,7 @@ func newHandlerV1(w warner, buildInfo models.BuildInformation,
type handlerV1 struct {
warner warner
buildInfo models.BuildInformation
vpn http.Handler
openvpn http.Handler
dns http.Handler
updater http.Handler
@@ -34,6 +36,8 @@ func (h *handlerV1) ServeHTTP(w http.ResponseWriter, r *http.Request) {
switch {
case r.RequestURI == "/version" && r.Method == http.MethodGet:
h.getVersion(w)
case strings.HasPrefix(r.RequestURI, "/vpn"):
h.vpn.ServeHTTP(w, r)
case strings.HasPrefix(r.RequestURI, "/openvpn"):
h.openvpn.ServeHTTP(w, r)
case strings.HasPrefix(r.RequestURI, "/dns"):

View File

@@ -1,3 +0,0 @@
package server
func stringPtr(s string) *string { return &s }

93
internal/server/vpn.go Normal file
View File

@@ -0,0 +1,93 @@
package server
import (
"context"
"encoding/json"
"net/http"
"strings"
)
func newVPNHandler(ctx context.Context, looper VPNLooper,
w warner) http.Handler {
return &vpnHandler{
ctx: ctx,
looper: looper,
warner: w,
}
}
type vpnHandler struct {
ctx context.Context //nolint:containedctx
looper VPNLooper
warner warner
}
func (h *vpnHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
r.RequestURI = strings.TrimPrefix(r.RequestURI, "/vpn")
switch r.RequestURI {
case "/status":
switch r.Method {
case http.MethodGet:
h.getStatus(w)
case http.MethodPut:
h.setStatus(w, r)
default:
http.Error(w, "method "+r.Method+" not supported", http.StatusBadRequest)
}
case "/settings":
switch r.Method {
case http.MethodGet:
h.getSettings(w)
default:
http.Error(w, "method "+r.Method+" not supported", http.StatusBadRequest)
}
default:
http.Error(w, "route "+r.RequestURI+" not supported", http.StatusBadRequest)
}
}
func (h *vpnHandler) getStatus(w http.ResponseWriter) {
status := h.looper.GetStatus()
encoder := json.NewEncoder(w)
data := statusWrapper{Status: string(status)}
if err := encoder.Encode(data); err != nil {
h.warner.Warn(err.Error())
w.WriteHeader(http.StatusInternalServerError)
return
}
}
func (h *vpnHandler) setStatus(w http.ResponseWriter, r *http.Request) {
decoder := json.NewDecoder(r.Body)
var data statusWrapper
if err := decoder.Decode(&data); err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
status, err := data.getStatus()
if err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
outcome, err := h.looper.ApplyStatus(h.ctx, status)
if err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
encoder := json.NewEncoder(w)
if err := encoder.Encode(outcomeWrapper{Outcome: outcome}); err != nil {
h.warner.Warn(err.Error())
http.Error(w, http.StatusText(http.StatusInternalServerError), http.StatusInternalServerError)
return
}
}
func (h *vpnHandler) getSettings(w http.ResponseWriter) {
settings := h.looper.GetSettings()
encoder := json.NewEncoder(w)
if err := encoder.Encode(settings); err != nil {
h.warner.Warn(err.Error())
w.WriteHeader(http.StatusInternalServerError)
return
}
}