add extension server endpoint

Author: LinuxSuRen

cmd/server.go

@@ -367,8 +367,25 @@ func (o *serverOption) runE(cmd *cobra.Command, args []string) (err error) {
 		mux.HandlePath(http.MethodGet, "/favicon.ico", frontEndHandlerWithLocation(o.consolePath))
 		mux.HandlePath(http.MethodGet, "/swagger.json", frontEndHandlerWithLocation(o.consolePath))
 		mux.HandlePath(http.MethodGet, "/data/{data}", o.dataFromExtension(remoteServer.(server.UIExtensionServer)))
-		mux.HandlePath(http.MethodGet, "/extensionProxy/{*}", func(w http.ResponseWriter, r *http.Request, pathParams map[string]string) {
+		mux.HandlePath(http.MethodGet, "/extensionProxy/{extension}/{path}", func(w http.ResponseWriter, r *http.Request, pathParams map[string]string) {
 			fmt.Println(pathParams)
+			extServer := remoteServer.(server.UIExtensionServer)
+			fmt.Println(extServer)
+
+			ctx := r.Context()
+			for k, v := range r.Header {
+				if !strings.HasPrefix(k, "X-Extension-") {
+					continue
+				}
+				ctx = context.WithValue(ctx, k, v)
+			}
+
+			resp, err := extServer.GetPageOfServer(ctx, &server.SimpleName{Name: pathParams["extension"]})
+			if err != nil {
+				fmt.Println(err)
+				return
+			}
+			fmt.Println(resp)
 		})
 		mux.HandlePath(http.MethodGet, "/get", o.getAtestBinary)
 		mux.HandlePath(http.MethodPost, "/runner/{suite}/{case}", service.WebRunnerHandler)

pkg/server/remote_server.go

@@ -1492,6 +1492,7 @@ func (s *server) GetMenus(ctx context.Context, _ *Empty) (result *MenuList, err
 					Index: menu.Index,
 				})
 				uiExtensionLoaders[menu.Index] = loader
+				fmt.Println("menu", menu.Index)
 			}
 		}
 	}
@@ -1570,6 +1571,21 @@ func (s *server) GetPageOfStatic(ctx context.Context, in *SimpleName) (result *C
 	return
 }
 
+func (s *server) GetPageOfServer(ctx context.Context, in *SimpleName) (result *CommonResult, err error) {
+	result = &CommonResult{}
+	if loader, ok := uiExtensionLoaders[in.Name]; ok {
+		if js, err := loader.GetPageOfServer(in.Name); err == nil {
+			result.Message = js
+			result.Success = true
+		} else {
+			result.Message = err.Error()
+		}
+	} else {
+		result.Message = fmt.Sprintf("not found loader for %s", in.Name)
+	}
+	return
+}
+
 // implement the mock server
 
 // Start starts the mock server

pkg/testing/loader.go

@@ -37,6 +37,7 @@ type Loader interface {
 	GetPageOfJS(string) (string, error)
 	GetPageOfCSS(string) (string, error)
 	GetPageOfStatic(string) (string, error)
+	GetPageOfServer(string) (string, error)
 }
 
 type Menu struct {

pkg/testing/loader_file.go

@@ -611,3 +611,8 @@ func (l *fileLoader) GetPageOfStatic(s string) (string, error) {
 	// no need to implement
 	return "", nil
 }
+
+func (l *fileLoader) GetPageOfServer(s string) (string, error) {
+	// no need to implement
+	return "", nil
+}

pkg/testing/loader_non.go

@@ -200,3 +200,7 @@ func (l *nonLoader) GetPageOfCSS(s string) (string, error) {
 func (l *nonLoader) GetPageOfStatic(s string) (string, error) {
 	return "", fmt.Errorf("not support")
 }
+
+func (l *nonLoader) GetPageOfServer(s string) (string, error) {
+	return "", fmt.Errorf("not support")
+}

pkg/testing/remote/grpc_store.go

@@ -447,6 +447,16 @@ func (g *gRPCLoader) GetPageOfStatic(name string) (result string, err error) {
 	return
 }
 
+func (g *gRPCLoader) GetPageOfServer(name string) (result string, err error) {
+	var data *server.CommonResult
+	if data, err = g.client.GetPageOfServer(g.ctx, &server.SimpleName{
+		Name: name,
+	}); err == nil && data != nil {
+		result = data.Message
+	}
+	return
+}
+
 func (g *gRPCLoader) Parse() (server *mock.Server, err error) {
 	return
 }