feat: add gateway feature

cmd/gateway.go

@@ -0,0 +1,39 @@
+package cmd
+
+import (
+	"github.com/linuxsuren/atest-ext-collector/pkg/gateway"
+	"github.com/spf13/cobra"
+)
+
+func createGatewayCmd() (cmd *cobra.Command) {
+	opt := &gatewayOptions{}
+	cmd = &cobra.Command{
+		Use:  "gateway",
+		RunE: opt.runE,
+	}
+	opt.setupFlags(cmd)
+	return
+}
+
+type gatewayOptions struct {
+	certFile string
+	keyFile  string
+}
+
+func (o *gatewayOptions) setupFlags(cmd *cobra.Command) {
+	flags := cmd.Flags()
+	flags.StringVarP(&o.certFile, "cert", "c", "", "The cert file")
+	flags.StringVarP(&o.keyFile, "key", "k", "", "The key file")
+}
+
+func (o *gatewayOptions) runE(cmd *cobra.Command, args []string) (err error) {
+	var gw *gateway.Gateway
+	if gw, err = gateway.ParseGateway(args[0]); err != nil {
+		return
+	}
+	server := gateway.GatewayServer{}
+	server.WithGateway(gw)
+	server.WithTLS(o.certFile, o.keyFile)
+	server.Start()
+	return
+}

cmd/root.go

@@ -24,6 +24,6 @@ func CreateRootCmd() (c *cobra.Command) {
 		Short: "A collector for API testing, it will start a HTTP proxy server",
 	}
 	c.AddCommand(createCollectorCmd(), createControllerCmd(),
-		createProxyCmd(), createDNSCmd())
+		createProxyCmd(), createDNSCmd(), createGatewayCmd())
 	return
 }

go.mod

@@ -21,6 +21,7 @@ require (
 	github.com/davecgh/go-spew v1.1.1 // indirect
 	github.com/ghodss/yaml v1.0.0 // indirect
 	github.com/google/uuid v1.3.0 // indirect
+	github.com/gorilla/mux v1.8.1 // indirect
 	github.com/huandu/xstrings v1.3.3 // indirect
 	github.com/imdario/mergo v0.3.11 // indirect
 	github.com/inconshreveable/mousetrap v1.1.0 // indirect

go.sum

@@ -21,6 +21,8 @@ github.com/google/gopacket v1.1.19/go.mod h1:iJ8V8n6KS+z2U1A8pUwu8bW5SyEMkXJB8Yo
 github.com/google/uuid v1.1.1/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
 github.com/google/uuid v1.3.0 h1:t6JiXgmwXMjEs8VusXIJk2BXHsn+wx8BZdTaoZ5fu7I=
 github.com/google/uuid v1.3.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
+github.com/gorilla/mux v1.8.1 h1:TuBL49tXwgrFYWhqrNgrUNEY92u81SPhu7sTdzQEiWY=
+github.com/gorilla/mux v1.8.1/go.mod h1:AKf9I4AEqPTmMytcMc0KkNouC66V3BtZ4qD5fmWSiMQ=
 github.com/h2non/gock v1.2.0 h1:K6ol8rfrRkUOefooBC8elXoaNGYkpp7y2qcxGG6BzUE=
 github.com/h2non/parth v0.0.0-20190131123155-b4df798d6542 h1:2VTzZjLZBgl62/EtslCrtky5vbi9dd7HrQPQIx6wqiw=
 github.com/huandu/xstrings v1.3.3 h1:/Gcsuc1x8JVbJ9/rlye4xZnVAbEkGauT8lbebqcQws4=

pkg/gateway/parser.go

@@ -0,0 +1,15 @@
+package gateway
+
+import (
+	"gopkg.in/yaml.v3"
+	"os"
+)
+
+func ParseGateway(config string) (gw *Gateway, err error) {
+	gw = &Gateway{}
+	var data []byte
+	if data, err = os.ReadFile(config); err == nil {
+		err = yaml.Unmarshal(data, gw)
+	}
+	return
+}

pkg/gateway/server.go

@@ -0,0 +1,104 @@
+package gateway
+
+import (
+	"fmt"
+	"github.com/gorilla/mux"
+	"io"
+	"log"
+	"net/http"
+	"net/url"
+	"strings"
+)
+
+type GatewayServer struct {
+	Router   *mux.Router
+	Gateway  *Gateway
+	certFile string
+	keyFile  string
+}
+
+func (g *GatewayServer) WithGateway(gw *Gateway) {
+	g.Gateway = gw
+}
+
+func (g *GatewayServer) WithTLS(certFile, keyFile string) {
+	g.certFile = certFile
+	g.keyFile = keyFile
+}
+
+func (g *GatewayServer) Start() {
+	g.Router = mux.NewRouter()
+
+	g.Router.HandleFunc("/{any:.*}", func(writer http.ResponseWriter, r *http.Request) {
+		fmt.Println(r.RequestURI)
+		if g.Gateway == nil {
+			return
+		}
+
+		for _, server := range g.Gateway.Servers {
+			if server.Domain == r.Host {
+
+				for _, route := range server.Routes {
+					fmt.Println(r.URL.Path, route.Path)
+					if !strings.HasPrefix(r.URL.Path, route.Path) {
+						continue
+					}
+
+					targetURL, err := url.Parse(route.ProxyPass + strings.TrimSuffix(r.URL.Path, route.Path))
+					for k, v := range r.URL.Query() {
+						targetURL.Query().Set(k, strings.Join(v, ","))
+					}
+					newRequest, err := http.NewRequestWithContext(r.Context(), r.Method, targetURL.RequestURI(), r.Body)
+					//if err := r.ParseForm(); err == nil {
+					//	newRequest.Form = url.Values{}
+					//	for k, v := range r.Form {
+					//		newRequest.Form[k] = v
+					//	}
+					//	newRequest.PostForm = url.Values{}
+					//	for k, v := range r.PostForm {
+					//		newRequest.PostForm[k] = v
+					//	}
+					//} else {
+					//	log.Println("failed to parse form", err)
+					//}
+					newRequest.Header = r.Header
+
+					fmt.Println("proxy pass the new request", targetURL.RequestURI())
+					resp, err := http.DefaultClient.Do(newRequest)
+					if err != nil {
+						log.Println(err, resp)
+					}
+					if resp == nil {
+						return
+					}
+
+					if resp.Header != nil {
+						for k, v := range resp.Header {
+							writer.Header().Add(k, strings.Join(v, ","))
+						}
+					}
+					defer resp.Body.Close()
+					io.Copy(writer, resp.Body)
+					return
+				}
+			}
+		}
+	})
+
+	http.Handle("/", g.Router)
+	srv := &http.Server{
+		Handler: g.Router,
+		Addr:    "0.0.0.0:80",
+	}
+	go func() {
+		if (g.certFile != "") && (g.keyFile != "") {
+			srvTLS := &http.Server{
+				Handler: g.Router,
+				Addr:    "0.0.0.0:443",
+			}
+			fmt.Println("Starting HTTPS server on ", srv.Addr)
+			srvTLS.ListenAndServeTLS(g.certFile, g.keyFile)
+		}
+	}()
+	srv.ListenAndServe()
+}

pkg/gateway/type.go

@@ -0,0 +1,16 @@
+package gateway
+
+type Gateway struct {
+	Servers []Server `yaml:"servers"`
+}
+
+type Server struct {
+	Protocol string  `yaml:"protocol"`
+	Domain   string  `yaml:"domain"`
+	Routes   []Route `yaml:"routes"`
+}
+
+type Route struct {
+	Path      string `yaml:"path"`
+	ProxyPass string `yaml:"proxyPass"`
+}