thor/wg-ui
1
0
Fork 0
mirror of https://github.com/Threnklyn/wg-ui.git synced 2026-05-19 05:13:30 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add developer mode to proxy static assets from frontend server during development

Browse Source
This commit is contained in:
Daniel Lundin
2019-06-06 18:27:18 +02:00
parent abd26f948d
commit a7dde97c47
1 changed files with 21 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
server.go
+21 -7
View File
@@ -6,6 +6,8 @@ import (
"fmt"
"net"
"net/http"
"net/http/httputil"
"net/url"
"os"
"path"
"strconv"
@@ -36,6 +38,8 @@ var (
wgListenPort = kingpin.Flag("wg-listen-port", "Wireguard UDP port to listen to").Default("51820").Int()
wgEndpoint = kingpin.Flag("wg-endpoint", "Wireguard endpoint address").Default("127.0.0.1:51820").String()
wgAllowedIPs = kingpin.Flag("wg-allowed-ips", "Wireguard client allowed ips").Default("0.0.0.0/0").Strings()
devUIServer = kingpin.Flag("dev-ui-server", "Developer mode: If specified, proxy all static assets to this endpoint").String()
)
type Server struct {
@@ -274,14 +278,29 @@ func (s *Server) Start() error {
}
router := httprouter.New()
router.GET("/", s.Index)
router.GET("/img/*filepath", s.Index)
router.GET("/api/v1/users/:user/devices/:device", s.withAuth(s.GetDevice))
router.PUT("/api/v1/users/:user/devices/:device", s.withAuth(s.EditDevice))
router.DELETE("/api/v1/users/:user/devices/:device", s.withAuth(s.DeleteDevice))
router.GET("/api/v1/users/:user/devices", s.withAuth(s.GetDevices))
router.POST("/api/v1/users/:user/devices", s.withAuth(s.CreateDevice))
if *devUIServer != "" {
log.Debug("Serving static assets proxying from development server: ", *devUIServer)
devProxy := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
url, _ := url.Parse(*devUIServer)
proxy := httputil.NewSingleHostReverseProxy(url)
r.URL.Host = url.Host
r.URL.Scheme = url.Scheme
r.Header.Set("X-Forwarded-Host", r.Header.Get("Host"))
r.Host = url.Host
proxy.ServeHTTP(w, r)
})
router.NotFound = devProxy
} else {
log.Debug("Serving static assets embedded in binary")
router.NotFound = s.assets
}
log.WithField("listenAddr", *listenAddr).Info("Starting server")
return http.ListenAndServe(*listenAddr, router)
}
@@ -342,11 +361,6 @@ func (s *Server) withAuth(handler httprouter.Handle) httprouter.Handle {
}
}
func (s *Server) Index(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
log.Debug("Index")
s.assets.ServeHTTP(w, r)
}
func (s *Server) GetDevices(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
user := r.Context().Value("user")
log.Debug(user)