feat: add extension to getting page of static

Author: LinuxSuRen

cmd/server.go

@@ -366,6 +366,7 @@ func (o *serverOption) runE(cmd *cobra.Command, args []string) (err error) {
 		mux.HandlePath(http.MethodGet, "/healthz", frontEndHandlerWithLocation(o.consolePath))
 		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, "/get", o.getAtestBinary)
 		mux.HandlePath(http.MethodPost, "/runner/{suite}/{case}", service.WebRunnerHandler)
 		mux.HandlePath(http.MethodGet, "/api/v1/sbom", service.SBomHandler)
@@ -531,6 +532,22 @@ func debugHandler(mux *runtime.ServeMux, remoteServer server.RunnerServer) {
 	})
 }
 
+func (o *serverOption) dataFromExtension(extServer server.UIExtensionServer) func(w http.ResponseWriter,
+	r *http.Request, pathParams map[string]string) {
+	return func(w http.ResponseWriter, r *http.Request, pathParams map[string]string) {
+		ctx := r.Context()
+		for k, v := range r.Header {
+			if !strings.HasPrefix(k, "X-Extension-") {
+				continue
+			}
+			ctx = context.WithValue(ctx, k, v)
+		}
+		extServer.GetPageOfStatic(ctx, &server.SimpleName{
+			Name: pathParams["data"],
+		})
+	}
+}
+
 func (o *serverOption) getAtestBinary(w http.ResponseWriter, r *http.Request, pathParams map[string]string) {
 	name := util.EmptyThenDefault(r.URL.Query().Get("name"), "atest")
 

pkg/server/remote_server.go

@@ -1531,6 +1531,22 @@ func (s *server) GetPageOfCSS(ctx context.Context, in *SimpleName) (result *Comm
 	return
 }
 
+func (s *server) GetPageOfStatic(ctx context.Context, in *SimpleName) (result *CommonResult, err error) {
+	result = &CommonResult{}
+	extName := ctx.Value("X-Extension-Name")
+	if loader, ok := uiExtensionLoaders[extName.(string)]; ok {
+		if js, err := loader.GetPageOfStatic(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/server/server.pb.go

@@ -328,6 +328,11 @@ service UIExtension {
             get: "/api/v1/extension/pages/{name}/css"
         };
     }
+    rpc GetPageOfStatic(SimpleName) returns (CommonResult) {
+        option (google.api.http) = {
+            get: "/api/v1/extension/pages/{name}/static"
+        };
+    }
 }
 
 message Menu {

pkg/server/server.pb.gw.go

@@ -36,6 +36,7 @@ type Loader interface {
 	GetMenus() ([]*Menu, error)
 	GetPageOfJS(string) (string, error)
 	GetPageOfCSS(string) (string, error)
+	GetPageOfStatic(string) (string, error)
 }
 
 type Menu struct {

pkg/server/server.proto

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

pkg/server/server_grpc.pb.go

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

pkg/testing/loader.go

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