Refactor

2025-09-12 13:03:41 +00:00 · 2020-01-30 22:37:59 -07:00
parent 19c6732de0
commit 8b42d0aff8
71 changed files with 4024 additions and 507 deletions
--- a/pkg/schemaserver/writer/html.go
+++ b/pkg/schemaserver/writer/html.go
@@ -0,0 +1,85 @@
+package writer
+
+import (
+	"strings"
+
+	"github.com/rancher/steve/pkg/schemaserver/types"
+)
+
+const (
+	JSURL          = "https://releases.rancher.com/api-ui/%API_UI_VERSION%/ui.min.js"
+	CSSURL         = "https://releases.rancher.com/api-ui/%API_UI_VERSION%/ui.min.css"
+	DefaultVersion = "1.1.8"
+)
+
+var (
+	start = `
+<!DOCTYPE html>
+<!-- If you are reading this, there is a good chance you would prefer sending an
+"Accept: application/json" header and receiving actual JSON responses. -->
+<link rel="stylesheet" type="text/css" href="%CSSURL%" />
+<script src="%JSURL%"></script>
+<script>
+var user = "admin";
+var curlUser='${CATTLE_ACCESS_KEY}:${CATTLE_SECRET_KEY}';
+var schemas="%SCHEMAS%";
+var data =
+`
+	end = []byte(`</script>
+`)
+)
+
+type StringGetter func() string
+
+type HTMLResponseWriter struct {
+	EncodingResponseWriter
+	CSSURL       StringGetter
+	JSURL        StringGetter
+	APIUIVersion StringGetter
+}
+
+func (h *HTMLResponseWriter) start(apiOp *types.APIRequest, code int) {
+	AddCommonResponseHeader(apiOp)
+	apiOp.Response.Header().Set("content-type", "text/html")
+	apiOp.Response.WriteHeader(code)
+}
+
+func (h *HTMLResponseWriter) Write(apiOp *types.APIRequest, code int, obj types.APIObject) {
+	h.write(apiOp, code, obj)
+}
+
+func (h *HTMLResponseWriter) WriteList(apiOp *types.APIRequest, code int, list types.APIObjectList) {
+	h.write(apiOp, code, list)
+}
+
+func (h *HTMLResponseWriter) write(apiOp *types.APIRequest, code int, obj interface{}) {
+	h.start(apiOp, code)
+	schemaSchema := apiOp.Schemas.Schemas["schema"]
+	headerString := start
+	if schemaSchema != nil {
+		headerString = strings.Replace(headerString, "%SCHEMAS%", apiOp.URLBuilder.Collection(schemaSchema), 1)
+	}
+	var jsurl, cssurl string
+	if h.CSSURL != nil && h.JSURL != nil && h.CSSURL() != "" && h.JSURL() != "" {
+		jsurl = h.JSURL()
+		cssurl = h.CSSURL()
+	} else if h.APIUIVersion != nil && h.APIUIVersion() != "" {
+		jsurl = strings.Replace(JSURL, "%API_UI_VERSION%", h.APIUIVersion(), 1)
+		cssurl = strings.Replace(CSSURL, "%API_UI_VERSION%", h.APIUIVersion(), 1)
+	} else {
+		jsurl = strings.Replace(JSURL, "%API_UI_VERSION%", DefaultVersion, 1)
+		cssurl = strings.Replace(CSSURL, "%API_UI_VERSION%", DefaultVersion, 1)
+	}
+	headerString = strings.Replace(headerString, "%JSURL%", jsurl, 1)
+	headerString = strings.Replace(headerString, "%CSSURL%", cssurl, 1)
+
+	apiOp.Response.Write([]byte(headerString))
+	if apiObj, ok := obj.(types.APIObject); ok {
+		h.Body(apiOp, apiOp.Response, apiObj)
+	} else if list, ok := obj.(types.APIObjectList); ok {
+		h.BodyList(apiOp, apiOp.Response, list)
+	}
+	if schemaSchema != nil {
+		apiOp.Response.Write(end)
+	}
+}