chore: updateUI API return 501 when config incomplete

config/update_ui.go

@@ -2,6 +2,7 @@ package config
 
 import (
 	"archive/zip"
+	"errors"
 	"fmt"
 	"io"
 	"os"
@@ -19,15 +20,18 @@ var (
 	ExternalUIFolder string
 	ExternalUIName   string
 )
-
+var (
+	ErrImcompleteConf = errors.New("ExternalUI configure incomplete")
+)
 var xdMutex sync.Mutex
 
 func UpdateUI() error {
 	xdMutex.Lock()
 	defer xdMutex.Unlock()
 
-	if ExternalUIPath == "" || ExternalUIFolder == "" {
-		return fmt.Errorf("ExternalUI configure incomplete")
+	err := prepare()
+	if err != nil {
+		return err
 	}
 
 	data, err := downloadForBytes(ExternalUIURL)
@@ -60,6 +64,25 @@ func UpdateUI() error {
 	return nil
 }
 
+func prepare() error {
+	if ExternalUIPath == "" || ExternalUIURL == "" {
+		return ErrImcompleteConf
+	}
+
+	if ExternalUIName != "" {
+		ExternalUIFolder = filepath.Clean(path.Join(ExternalUIPath, ExternalUIName))
+		if _, err := os.Stat(ExternalUIPath); os.IsNotExist(err) {
+			if err := os.MkdirAll(ExternalUIPath, os.ModePerm); err != nil {
+				return err
+			}
+		}
+	} else {
+		ExternalUIFolder = ExternalUIPath
+	}
+
+	return nil
+}
+
 func unzip(src, dest string) (string, error) {
 	r, err := zip.OpenReader(src)
 	if err != nil {