Merge pull request #52 from drognisep/go-sync-webui

Adding a method to sync the webui dependency without scripts

Author: AlbertShown

v2/sync-webui/gotools.go

@@ -0,0 +1,51 @@
+package main
+
+import (
+	"fmt"
+	"io"
+	"os"
+	"os/exec"
+	"strings"
+)
+
+type Gotools struct {
+	exe string
+}
+
+func GetGo() *Gotools {
+	exe, err := exec.LookPath("go")
+	if err != nil {
+		panic(fmt.Errorf("failed to find go executable on PATH: %w", err))
+	}
+	return &Gotools{
+		exe: exe,
+	}
+}
+
+func (gt *Gotools) Cmd(args ...string) *exec.Cmd {
+	cmd := exec.Command(gt.exe, args...)
+	cmd.Stdin = os.Stdin
+	cmd.Stdout = os.Stdout
+	cmd.Stderr = os.Stderr
+	return cmd
+}
+
+func (gt *Gotools) CmdSilent(args ...string) *exec.Cmd {
+	cmd := gt.Cmd(args...)
+	cmd.Stdout = io.Discard
+	cmd.Stderr = io.Discard
+	return cmd
+}
+
+func (gt *Gotools) Gopath() (string, error) {
+	gopath, ok := os.LookupEnv("GOPATH")
+	if ok {
+		return gopath, nil
+	}
+	output, err := gt.Cmd("env", "GOPATH").Output()
+	if err != nil {
+		return "", fmt.Errorf("failed to check GOPATH from go tools: %w", err)
+	}
+	gopath = strings.TrimSpace(string(output))
+	return gopath, nil
+}

v2/sync-webui/main.go

@@ -0,0 +1,97 @@
+package main
+
+import (
+	"errors"
+	"fmt"
+	"os"
+	"path/filepath"
+	"runtime"
+)
+
+const (
+	goWebuiRepo = "github.com/webui-dev/go-webui/v2"
+	webuiRepo   = "github.com/webui-dev/webui"
+)
+
+func goWebuiVersion(version string) string {
+	return fmt.Sprintf("%s@%s", goWebuiRepo, version)
+}
+
+func webuiVersion(version string) string {
+	return fmt.Sprintf("%s@%s", webuiRepo, version)
+}
+
+func main() {
+	// Ensure the current directory is part of a go module.
+	if !fileExists("go.mod") {
+		errExit(`error: failed to find go.mod file in current directory.
+       To set up the go-webui module, use this script in a module directory.`)
+	}
+	goTools := GetGo()
+	// Run go commands
+	iferrExit(goTools.Cmd("mod", "tidy").Run())
+	iferrExit(goTools.Cmd("get", goWebuiVersion("main")).Run())
+	iferrExit(goTools.CmdSilent("get", webuiVersion("main")).Run())
+
+	// Retrieve GOPATH (use environment variable if defined, otherwise go env)
+	goPath, err := goTools.Gopath()
+	iferrExit(err)
+
+	// Parse the first matching version lines from go.sum
+	// For go-webui/v2:
+	goWebuiFullVersion := field(headN1(grep(goWebuiRepo, "go.sum")), 2)
+	// For webui:
+	webuiFullVersion := field(headN1(grep(webuiRepo, "go.sum")), 2)
+
+	// Construct paths based on the parsed versions
+	goWebuiPath := filepath.Join(goPath, "pkg", "mod", goWebuiVersion(goWebuiFullVersion))
+	webuiPath := filepath.Join(goPath, "pkg", "mod", webuiVersion(webuiFullVersion))
+
+	// Validate that these paths actually exist
+	iferrExit(validatePaths(goWebuiPath, webuiPath))
+	linkName := filepath.Join(goWebuiPath, "webui")
+	// Exit if link already exists in the directory of the used go-webui version.
+	if dirExists(linkName) {
+		return
+	}
+
+	// Store original permissions.
+	// Not strictly necessary, yet we ensure end without changes to the original permissions.
+	ogPerms, err := getPerms(goWebuiPath)
+	iferrExit(err)
+	iferrExit(os.Chmod(goWebuiPath, 0200+ogPerms))
+	// Linking allows using WebUI C even in cases of multiple go-webui versions without creating bloat.
+	if err := os.Symlink(webuiPath, linkName); err != nil {
+		if runtime.GOOS == "windows" {
+			// (Requires Administrator privileges or Developer Mode)
+			errExit("mklink failed. Run as Administrator if needed.")
+		}
+		errExit(err.Error())
+	}
+	// Restore original permissions.
+	iferrExit(os.Chmod(goWebuiPath, ogPerms))
+	iferrExit(goTools.Cmd("mod", "tidy").Run())
+}
+
+func validatePaths(goWebuiPath, webuiPath string) error {
+	var errs []error
+	if !dirExists(goWebuiPath) {
+		errs = append(errs, fmt.Errorf("failed to find go-webui in '%s'", goWebuiPath))
+	}
+	if !dirExists(webuiPath) {
+		errs = append(errs, fmt.Errorf("failed to find webui in '%s'", webuiPath))
+	}
+	return errors.Join(errs...)
+}
+
+func iferrExit(err error) {
+	if err != nil {
+		fmt.Println(err.Error())
+		os.Exit(1)
+	}
+}
+
+func errExit(msg string) {
+	fmt.Println(msg)
+	os.Exit(1)
+}

v2/sync-webui/shellfuncs.go

@@ -0,0 +1,66 @@
+package main
+
+import (
+	"bufio"
+	"os"
+	"strings"
+)
+
+func fileExists(path string) bool {
+	fi, err := os.Stat(path)
+	if err != nil {
+		return false
+	}
+	return !fi.IsDir()
+}
+
+func dirExists(path string) bool {
+	fi, err := os.Stat(path)
+	if err != nil {
+		return false
+	}
+	return fi.IsDir()
+}
+
+func grep(text, path string) []string {
+	var lines []string
+	f, err := os.Open(path)
+	if err != nil {
+		return nil
+	}
+	defer func() {
+		_ = f.Close()
+	}()
+	scanner := bufio.NewScanner(f)
+	for scanner.Scan() {
+		line := scanner.Text()
+		if strings.Contains(line, text) {
+			lines = append(lines, line)
+		}
+	}
+	return lines
+}
+
+func headN1(lines []string) string {
+	if len(lines) == 0 {
+		return ""
+	}
+	return lines[0]
+}
+
+func field(line string, field int) string {
+	line = strings.TrimSpace(line)
+	fields := strings.Split(line, " ")
+	if len(fields) < field {
+		return ""
+	}
+	return fields[field-1]
+}
+
+func getPerms(path string) (os.FileMode, error) {
+	fi, err := os.Stat(path)
+	if err != nil {
+		return 0, err
+	}
+	return fi.Mode(), nil
+}