support dynamic distro config

internal/resource/distrores/strings.go

@@ -8,4 +8,4 @@ import (
 	"github.com/pingcap/tidb-dashboard/util/distro"
 )
 
-var resource = distro.DistributionResource{}
+var resource = distro.StringsRes()

pkg/uiserver/uiserver.go

@@ -5,17 +5,22 @@ package uiserver
 import (
 	"bytes"
 	"compress/gzip"
+	"encoding/base64"
+	"encoding/json"
 	"html"
 	"io"
 	"io/ioutil"
 	"net/http"
+	"os"
+	"path"
 	"strings"
 
 	"github.com/pingcap/log"
 	"github.com/shurcooL/httpgzip"
 	"go.uber.org/zap"
 
 	"github.com/pingcap/tidb-dashboard/pkg/config"
+	"github.com/pingcap/tidb-dashboard/util/distro"
 )
 
 type UpdateContentFunc func(fs http.FileSystem, oldFile http.File, path, newContent string, zippedBytes []byte)
@@ -39,6 +44,9 @@ func RewriteAssets(fs http.FileSystem, cfg *config.Config, updater UpdateContent
 		tmplText := string(bs)
 		updated := strings.ReplaceAll(tmplText, "__PUBLIC_PATH_PREFIX__", html.EscapeString(cfg.PublicPathPrefix))
 
+		distroStrings, _ := json.Marshal(distro.R()) // this will never fail
+		updated = strings.ReplaceAll(updated, "__DISTRO_STRINGS_RES__", base64.StdEncoding.EncodeToString(distroStrings))
+
 		var b bytes.Buffer
 		w := gzip.NewWriter(&b)
 		if _, err := w.Write([]byte(updated)); err != nil {
@@ -53,6 +61,63 @@ func RewriteAssets(fs http.FileSystem, cfg *config.Config, updater UpdateContent
 
 	rewrite("/index.html")
 	rewrite("/diagnoseReport.html")
+
+	overrideDistroAssetsRes(fs, cfg, updater)
+}
+
+func overrideDistroAssetsRes(fs http.FileSystem, cfg *config.Config, updater UpdateContentFunc) {
+	exePath, err := os.Executable()
+	if err != nil {
+		log.Fatal("Failed to get work dir", zap.Error(err))
+	}
+
+	distroResDir := path.Join(path.Dir(exePath), distro.DistroResFolderName)
+	info, err := os.Stat(distroResDir)
+	if err != nil || !info.IsDir() {
+		// just ignore
+		return
+	}
+
+	override := func(assetName string) {
+		assetPath := path.Join("/", distro.DistroResFolderName, assetName)
+		targetFile, err := fs.Open(assetPath)
+		if err != nil {
+			// has no target asset to be overried, skip
+			return
+		}
+		defer targetFile.Close()
+
+		sourceFile, err := os.Open(path.Join(distroResDir, assetName))
+		if err != nil {
+			log.Fatal("Failed to open source file", zap.String("path", assetName), zap.Error(err))
+		}
+		defer sourceFile.Close()
+
+		data, err := ioutil.ReadAll(sourceFile)
+		if err != nil {
+			log.Fatal("Failed to read asset", zap.String("path", assetName), zap.Error(err))
+		}
+
+		var b bytes.Buffer
+		w := gzip.NewWriter(&b)
+		if _, err := w.Write(data); err != nil {
+			log.Fatal("Failed to zip asset", zap.Error(err))
+		}
+		if err := w.Close(); err != nil {
+			log.Fatal("Failed to zip asset", zap.Error(err))
+		}
+
+		updater(fs, targetFile, assetPath, string(data), b.Bytes())
+	}
+
+	// traverse
+	files, err := ioutil.ReadDir(distroResDir)
+	if err != nil {
+		log.Fatal("Failed to read dir", zap.String("dir", distroResDir), zap.Error(err))
+	}
+	for _, file := range files {
+		override(file.Name())
+	}
 }
 
 func Handler(root http.FileSystem) http.Handler {

scripts/go.mod

@@ -3,12 +3,15 @@ module scripts
 go 1.13
 
 require (
+	github.com/pingcap/errors v0.11.5-0.20200917111840-a15ef68f753d // indirect
 	github.com/pingcap/log v0.0.0-20210906054005-afc726e70354
-	github.com/pingcap/tidb-dashboard v0.0.0-00010101000000-000000000000
 	github.com/shurcooL/httpfs v0.0.0-20190707220628-8d4bc4ba7749 // indirect
 	github.com/shurcooL/vfsgen v0.0.0-20200824052919-0d455de96546
 	github.com/swaggo/swag v1.6.6-0.20200529100950-7c765ddd0476
 	go.uber.org/zap v1.19.0
+	golang.org/x/lint v0.0.0-20201208152925-83fdc39ff7b5 // indirect
+	golang.org/x/net v0.0.0-20210405180319-a5a99cb37ef4 // indirect
+	golang.org/x/tools v0.0.0-20210112230658-8b4aab62c064 // indirect
 )
 
 replace github.com/pingcap/tidb-dashboard => ../