feat(server): add vpn route to replace /openvpn
This commit is contained in:
@@ -18,13 +18,14 @@ func newHandler(ctx context.Context, logger infoWarner, logging bool,
|
|||||||
) http.Handler {
|
) http.Handler {
|
||||||
handler := &handler{}
|
handler := &handler{}
|
||||||
|
|
||||||
|
vpn := newVPNHandler(ctx, vpnLooper, logger)
|
||||||
openvpn := newOpenvpnHandler(ctx, vpnLooper, pfGetter, logger)
|
openvpn := newOpenvpnHandler(ctx, vpnLooper, pfGetter, logger)
|
||||||
dns := newDNSHandler(ctx, unboundLooper, logger)
|
dns := newDNSHandler(ctx, unboundLooper, logger)
|
||||||
updater := newUpdaterHandler(ctx, updaterLooper, logger)
|
updater := newUpdaterHandler(ctx, updaterLooper, logger)
|
||||||
publicip := newPublicIPHandler(publicIPLooper, logger)
|
publicip := newPublicIPHandler(publicIPLooper, logger)
|
||||||
|
|
||||||
handler.v0 = newHandlerV0(ctx, logger, vpnLooper, unboundLooper, updaterLooper)
|
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)
|
handlerWithLog := withLogMiddleware(handler, logger, logging)
|
||||||
handler.setLogEnabled = handlerWithLog.setEnabled
|
handler.setLogEnabled = handlerWithLog.setEnabled
|
||||||
|
|||||||
@@ -10,10 +10,11 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
func newHandlerV1(w warner, buildInfo models.BuildInformation,
|
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{
|
return &handlerV1{
|
||||||
warner: w,
|
warner: w,
|
||||||
buildInfo: buildInfo,
|
buildInfo: buildInfo,
|
||||||
|
vpn: vpn,
|
||||||
openvpn: openvpn,
|
openvpn: openvpn,
|
||||||
dns: dns,
|
dns: dns,
|
||||||
updater: updater,
|
updater: updater,
|
||||||
@@ -24,6 +25,7 @@ func newHandlerV1(w warner, buildInfo models.BuildInformation,
|
|||||||
type handlerV1 struct {
|
type handlerV1 struct {
|
||||||
warner warner
|
warner warner
|
||||||
buildInfo models.BuildInformation
|
buildInfo models.BuildInformation
|
||||||
|
vpn http.Handler
|
||||||
openvpn http.Handler
|
openvpn http.Handler
|
||||||
dns http.Handler
|
dns http.Handler
|
||||||
updater http.Handler
|
updater http.Handler
|
||||||
@@ -34,6 +36,8 @@ func (h *handlerV1) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|||||||
switch {
|
switch {
|
||||||
case r.RequestURI == "/version" && r.Method == http.MethodGet:
|
case r.RequestURI == "/version" && r.Method == http.MethodGet:
|
||||||
h.getVersion(w)
|
h.getVersion(w)
|
||||||
|
case strings.HasPrefix(r.RequestURI, "/vpn"):
|
||||||
|
h.vpn.ServeHTTP(w, r)
|
||||||
case strings.HasPrefix(r.RequestURI, "/openvpn"):
|
case strings.HasPrefix(r.RequestURI, "/openvpn"):
|
||||||
h.openvpn.ServeHTTP(w, r)
|
h.openvpn.ServeHTTP(w, r)
|
||||||
case strings.HasPrefix(r.RequestURI, "/dns"):
|
case strings.HasPrefix(r.RequestURI, "/dns"):
|
||||||
|
|||||||
@@ -1,3 +0,0 @@
|
|||||||
package server
|
|
||||||
|
|
||||||
func stringPtr(s string) *string { return &s }
|
|
||||||
93
internal/server/vpn.go
Normal file
93
internal/server/vpn.go
Normal 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
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user