Create a webserver in node to configure when server is running

Author: agustim

cli/up.go

@@ -16,6 +16,8 @@ import (
 	"syscall"
 	"time"
 
+	"github.com/hyprspace/hyprspace/webserver"
+
 	"github.com/DataDrake/cli-ng/v2/cmd"
 	"github.com/hyprspace/hyprspace/config"
 	"github.com/hyprspace/hyprspace/p2p"
@@ -157,6 +159,11 @@ func UpRun(r *cmd.Root, c *cmd.Sub) {
 	// Configure path for lock
 	lockPath := filepath.Join(filepath.Dir(cfg.Path), cfg.Interface.Name+".lock")
 
+	// Start webserver
+	fmt.Println("[+] Starting Web Server")
+
+	go webserver.CreateServer(cfg)
+
 	// Register the application to listen for SIGINT/SIGTERM
 	go signalExit(host, lockPath)
 

go.mod

@@ -7,6 +7,7 @@ require (
 	github.com/hashicorp/go-version v1.4.0 // indirect
 	github.com/inconshreveable/go-update v0.0.0-20160112193335-8152e7eb6ccf
 	github.com/ipfs/go-datastore v0.5.1
+	github.com/julienschmidt/httprouter v1.3.0
 	github.com/kr/text v0.2.0 // indirect
 	github.com/libp2p/go-libp2p v0.17.0
 	github.com/libp2p/go-libp2p-core v0.13.0

go.sum

@@ -414,6 +414,7 @@ github.com/jstemmer/go-junit-report v0.9.1/go.mod h1:Brl9GWCQeLvo8nXZwPNNblvFj/X
 github.com/jtolds/gls v4.20.0+incompatible h1:xdiiI2gbIgH/gLH7ADydsJ1uDOEzR8yvV7C0MuV77Wo=
 github.com/jtolds/gls v4.20.0+incompatible/go.mod h1:QJZ7F/aHp+rZTRtaJ1ow/lLfFfVYBRgL+9YlvaHOwJU=
 github.com/julienschmidt/httprouter v1.2.0/go.mod h1:SYymIcj16QtmaHHD7aYtjjsJG7VTCxuUUipMqKk8s4w=
+github.com/julienschmidt/httprouter v1.3.0 h1:U0609e9tgbseu3rBINet9P48AI/D3oJs4dN7jwJOQ1U=
 github.com/julienschmidt/httprouter v1.3.0/go.mod h1:JR6WtHb+2LUe8TCKY3cZOxFyyO8IZAc4RVcycCCAKdM=
 github.com/kami-zh/go-capturer v0.0.0-20171211120116-e492ea43421d/go.mod h1:P2viExyCEfeWGU259JnaQ34Inuec4R38JCyBx2edgD0=
 github.com/kisielk/errcheck v1.1.0/go.mod h1:EZBBE59ingxPouuu3KfxchcWSUPOHkagtvWXihfKN4Q=

webserver/webserver.go

@@ -0,0 +1,90 @@
+package webserver
+
+import (
+	"encoding/json"
+	"log"
+
+	"net/http"
+
+	"github.com/hyprspace/hyprspace/config"
+	"github.com/julienschmidt/httprouter"
+)
+
+type Server struct {
+	router *httprouter.Router
+	cfg    *config.Config
+}
+
+// Create a http server to read and edit the config file
+func CreateServer(cfg *config.Config) {
+	// Create a new web server
+	server := &Server{cfg: cfg}
+	server.Init()
+	server.Run()
+}
+
+func (s *Server) Init() {
+	s.router = httprouter.New()
+
+	s.router.GET("/", s.Index)
+	s.router.GET("/routes", s.Routes)
+	s.router.GET("/add-route/:net/:mask/:gateway", s.AddRoute)
+	s.router.GET("/remove-route/:net/:mask/:gateway", s.RemoveRoute)
+}
+
+func (s *Server) Run() {
+	log.Fatal(http.ListenAndServe(":8080", s.router))
+}
+
+func (s *Server) Index(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
+	var PrivateKey string
+	w.WriteHeader(http.StatusOK)
+	w.Header().Set("Content-Type", "application/json")
+	PrivateKey = s.cfg.Interface.PrivateKey
+	s.cfg.Interface.PrivateKey = "<hidden>"
+	jsonResp, err := json.Marshal(s.cfg)
+	s.cfg.Interface.PrivateKey = PrivateKey
+	if err != nil {
+		log.Fatalf("Error happened in JSON marshal. Err: %s", err)
+	}
+	w.Write(jsonResp)
+}
+
+func (s *Server) Routes(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
+	w.WriteHeader(http.StatusOK)
+	w.Header().Set("Content-Type", "application/json")
+	jsonResp, err := json.Marshal(s.cfg.Routes)
+	if err != nil {
+		log.Fatalf("Error happened in JSON marshal. Err: %s", err)
+	}
+	w.Write(jsonResp)
+}
+
+func (s *Server) AddRoute(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
+	network := ps.ByName("net")
+	mask := ps.ByName("mask")
+	cidr := network + "/" + mask
+	gateway := ps.ByName("gateway")
+	s.cfg.Routes[cidr] = config.Route{IP: gateway}
+	w.WriteHeader(http.StatusOK)
+	w.Header().Set("Content-Type", "application/json")
+	jsonResp, err := json.Marshal(s.cfg.Routes)
+	if err != nil {
+		log.Fatalf("Error happened in JSON marshal. Err: %s", err)
+	}
+	w.Write(jsonResp)
+}
+
+func (s *Server) RemoveRoute(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
+	network := ps.ByName("net")
+	mask := ps.ByName("mask")
+	cidr := network + "/" + mask
+	delete(s.cfg.Routes, cidr)
+	w.WriteHeader(http.StatusOK)
+	w.Header().Set("Content-Type", "application/json")
+	jsonResp, err := json.Marshal(s.cfg.Routes)
+	if err != nil {
+		log.Fatalf("Error happened in JSON marshal. Err: %s", err)
+	}
+	w.Write(jsonResp)
+}