OliveTin/internal/httpservers/webuiServer.go

package httpservers

import (
	"encoding/json"
	//	cors "github.com/OliveTin/OliveTin/internal/cors"
	log "github.com/sirupsen/logrus"
	"net/http"
	"os"
	"path"
	"path/filepath"

	config "github.com/OliveTin/OliveTin/internal/config"
	installationinfo "github.com/OliveTin/OliveTin/internal/installationinfo"
	sv "github.com/OliveTin/OliveTin/internal/stringvariables"
)

var (
	customThemeCss     []byte
	customThemeCssRead = false
)

type webUISettings struct {
	Rest                   string
	ShowFooter             bool
	ShowNavigation         bool
	ShowNewVersions        bool
	AvailableVersion       string
	CurrentVersion         string
	PageTitle              string
	SectionNavigationStyle string
	DefaultIconForBack     string
	SshFoundKey            string
	SshFoundConfig         string
	EnableCustomJs         bool
}

func findWebuiDir() string {
	directoriesToSearch := []string{
		cfg.WebUIDir,
		"../webui/",
		"/usr/share/OliveTin/webui/",
		"/var/www/OliveTin/",
		"/var/www/olivetin/",
		"/etc/OliveTin/webui/",
	}

	// Use a classic i := 0 style for loop here instead of range, as the
	// search order must be deterministic - the order that the slice was defined in.
	for i := 0; i < len(directoriesToSearch); i++ {
		dir := directoriesToSearch[i]
		absdir, _ := filepath.Abs(dir)

		if _, err := os.Stat(absdir); !os.IsNotExist(err) {
			log.WithFields(log.Fields{
				"dir": absdir,
			}).Infof("Found the webui directory")

			sv.Set("internal.webuidir", absdir+" ("+dir+")")

			return dir
		}
	}

	log.Warnf("Did not find the webui directory, you will probably get 404 errors.")

	return "./webui" // Should not exist
}

func findCustomWebuiDir() string {
	dir := path.Join(cfg.GetDir(), "custom-webui")

	return dir
}

func setupCustomWebuiDir() {
	dir := findCustomWebuiDir()

	err := os.MkdirAll(path.Join(dir, "themes/"), 0775)

	if err != nil {
		log.Warnf("Could not create themes directory: %v", err)
		sv.Set("internal.themesdir", err.Error())
	} else {
		sv.Set("internal.themesdir", dir)
	}
}

func generateThemeCss(w http.ResponseWriter, r *http.Request) {
	themeCssFilename := path.Join(findCustomWebuiDir(), "themes", cfg.ThemeName, "theme.css")

	if !customThemeCssRead || cfg.ThemeCacheDisabled {
		customThemeCssRead = true

		if _, err := os.Stat(themeCssFilename); err == nil {
			customThemeCss, err = os.ReadFile(themeCssFilename)
		} else {
			log.Debugf("Theme CSS not read: %v", err)
			customThemeCss = []byte("/* not found */")
		}
	}

	w.Header().Add("Content-Type", "text/css")
	w.Write(customThemeCss)
}

func generateWebUISettings(w http.ResponseWriter, r *http.Request) {
	jsonRet, _ := json.Marshal(webUISettings{
		Rest:                   cfg.ExternalRestAddress + "/api/",
		ShowFooter:             cfg.ShowFooter,
		ShowNavigation:         cfg.ShowNavigation,
		ShowNewVersions:        cfg.ShowNewVersions,
		AvailableVersion:       installationinfo.Runtime.AvailableVersion,
		CurrentVersion:         installationinfo.Build.Version,
		PageTitle:              cfg.PageTitle,
		SectionNavigationStyle: cfg.SectionNavigationStyle,
		DefaultIconForBack:     cfg.DefaultIconForBack,
		SshFoundKey:            installationinfo.Runtime.SshFoundKey,
		SshFoundConfig:         installationinfo.Runtime.SshFoundConfig,
		EnableCustomJs:         cfg.EnableCustomJs,
	})

	w.Header().Add("Content-Type", "application/json")
	_, err := w.Write([]byte(jsonRet))

	if err != nil {
		log.Warnf("Could not write webui settings: %v", err)
	}
}

func startWebUIServer(cfg *config.Config) {
	log.WithFields(log.Fields{
		"address": cfg.ListenAddressWebUI,
	}).Info("Starting WebUI server")

	setupCustomWebuiDir()

	mux := http.NewServeMux()
	mux.Handle("/custom-webui/", http.StripPrefix("/custom-webui/", http.FileServer(http.Dir(findCustomWebuiDir()))))
	mux.HandleFunc("/theme.css", generateThemeCss)
	mux.HandleFunc("/webUiSettings.json", generateWebUISettings)

	webuiDir := findWebuiDir()
	mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
		dirName := path.Dir(r.URL.Path)

		// Mangle requests for any path like /logs or /config to load the webui index.html
		if path.Ext(r.URL.Path) == "" && r.URL.Path != "/" {
			log.Debugf("Mangling request for %s to /index.html", r.URL.Path)

			http.ServeFile(w, r, path.Join(webuiDir, "index.html"))
		} else {
			http.StripPrefix(dirName, http.FileServer(http.Dir(webuiDir))).ServeHTTP(w, r)
		}
	})

	srv := &http.Server{
		Addr:    cfg.ListenAddressWebUI,
		Handler: mux,
	}

	log.Fatal(srv.ListenAndServe())
}