Yggdrasil/main.go

package main

import (
	"archive/zip"
	"bytes"
	"embed"
	"fmt"
	"io"
	"log"
	"net/http"
	"net/url"
	"os"
	"path/filepath"
	"regexp"
	"strings"

	"github.com/wailsapp/wails/v2"
	"github.com/wailsapp/wails/v2/pkg/options"
	"github.com/wailsapp/wails/v2/pkg/options/assetserver"
)

var Error *log.Logger
var Warning *log.Logger

func init() {
	log.SetFlags(log.Lmicroseconds | log.Lshortfile)

	Error = log.New(io.MultiWriter(os.Stderr, os.Stdout),
		fmt.Sprintf("%sERROR:%s ", "\033[0;101m", "\033[0m"),
		log.Lmicroseconds|log.Lshortfile)
	Warning = log.New(io.MultiWriter(os.Stdout),
		fmt.Sprintf("%sWarning:%s ", "\033[0;93m", "\033[0m"),
		log.Lmicroseconds|log.Lshortfile)
}

//go:embed all:frontend/dist
var assets embed.FS
var gamePath string
var versionRegex = regexp.MustCompile(`(\d+\.\d+\.\d+)`)

type Addon struct {
	Name string
	URL  *url.URL
}

func NewAddon(name, aurl string) *Addon {
	a := &Addon{Name: name}
	var err error
	a.URL, err = url.Parse(aurl)
	if err != nil {
		Error.Printf("invalid url: %s", aurl)
		return nil
	}
	return a
}

var addons = []*Addon{
	NewAddon("Channeler", "https://git.site.quack-lab.dev/dave/wow_channeler"),
}

func (a *Addon) GetRemoteTocURL() *url.URL {
	return a.URL.JoinPath("raw", "branch", "master", a.Name+".toc")
}
func (a *Addon) GetRemoteReleaseURL() *url.URL {
	return a.URL.JoinPath("media", "branch", "master", a.Name+".zip")
}
func (a *Addon) GetRemoteRelease() (body []byte, err error) {
	url := a.GetRemoteReleaseURL()
	response, err := http.Get(url.String())
	if err != nil {
		return nil, fmt.Errorf("error getting remote release: %w", err)
	}
	return io.ReadAll(response.Body)
}
func (a *Addon) HasBreakingChanges(lhsToc, rhsToc string) bool {
	lhsLines := strings.Split(lhsToc, "\n")
	rhsLines := strings.Split(rhsToc, "\n")
	for i, lhsLine := range lhsLines {
		rhsLine := rhsLines[i]
		lhsLine = strings.TrimSpace(lhsLine)
		rhsLine = strings.TrimSpace(rhsLine)
		// We don't care about ## lines
		if strings.HasPrefix(lhsLine, "#") && strings.HasPrefix(rhsLine, "#") {
			continue
		}
		// Nor do we care about empty lines
		if lhsLine == "" && rhsLine == "" {
			continue
		}

		log.Printf("%s %s", lhsLine, rhsLine)
		if i >= len(rhsLines) || rhsLine != lhsLine {
			return true
		}
	}
	return false
}
func UpdateFile(localPath string, data []byte) (err error) {
	err = os.MkdirAll(filepath.Dir(localPath), 0755)
	if err != nil {
		return fmt.Errorf("error creating directory: %w", err)
	}
	fileHandle, err := os.OpenFile(localPath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0644)
	if err != nil {
		return fmt.Errorf("error opening file: %w", err)
	}
	defer fileHandle.Close()
	_, err = fileHandle.Write(data)
	if err != nil {
		return fmt.Errorf("error writing file: %w", err)
	}

	return nil
}
func (a *Addon) Update(body []byte) (err error) {
	log.Printf("Updating %s", a.Name)
	zipReader, err := zip.NewReader(bytes.NewReader(body), int64(len(body)))
	if err != nil {
		return fmt.Errorf("error creating zip reader: %w", err)
	}
	log.Printf("Found %d files", len(zipReader.File))
	hasBreakingChanges := false
	for _, file := range zipReader.File {
		if file.FileInfo().IsDir() {
			continue
		}
		log.Printf("Found file %s", file.Name)
		fileHandle, err := file.Open()
		if err != nil {
			return fmt.Errorf("error opening file: %w", err)
		}
		fileData, err := io.ReadAll(fileHandle)
		if err != nil {
			return fmt.Errorf("error reading file: %w", err)
		}

		if strings.HasSuffix(file.Name, ".toc") {
			localToc, err := os.ReadFile(a.GetLocalTocPath())
			if err != nil {
				if os.IsNotExist(err) {
					localToc = []byte{}
				} else {
					return fmt.Errorf("error reading local toc: %w", err)
				}
			}
			if a.HasBreakingChanges(string(localToc), string(fileData)) {
				log.Printf("Has breaking changes")
				hasBreakingChanges = true
			}
		}

		localPath := filepath.Join(gamePath, "Interface", "AddOns", file.Name)
		log.Printf("Updating file %s", localPath)
		err = UpdateFile(localPath, fileData)
		if err != nil {
			return fmt.Errorf("error updating file: %w", err)
		}
	}
	if hasBreakingChanges {
		Warning.Printf("Has breaking changes")
	}
	return nil
}
func (a *Addon) GetLocalTocPath() string {
	return filepath.Join(gamePath, "Interface", "AddOns", a.Name, a.Name+".toc")
}
func (a *Addon) GetRemoteVersion() (version string, err error) {
	url := a.GetRemoteTocURL()
	log.Printf("Fetching remote version from %s", url.String())
	response, err := http.Get(url.String())
	if err != nil {
		return "", fmt.Errorf("error getting remote version: %w", err)
	}
	if response.StatusCode != http.StatusOK {
		return "", fmt.Errorf("error getting remote version with status: %s", response.Status)
	}
	defer response.Body.Close()
	body, err := io.ReadAll(response.Body)
	if err != nil {
		return "", fmt.Errorf("error reading remote version: %w", err)
	}
	return GetVersion(string(body))
}
func (a *Addon) GetLocalVersion() (version string, err error) {
	file, err := os.Open(a.GetLocalTocPath())
	if err != nil {
		return "", fmt.Errorf("error opening local toc: %w", err)
	}
	defer file.Close()
	body, err := io.ReadAll(file)
	if err != nil {
		return "", fmt.Errorf("error reading local toc: %w", err)
	}
	return GetVersion(string(body))
}
func (a *Addon) IsUpToDate() bool {
	remoteVersion, err := a.GetRemoteVersion()
	if err != nil {
		return false
	}
	localVersion, err := a.GetLocalVersion()
	if err != nil {
		return false
	}
	return remoteVersion == localVersion
}

func main() {
	gamePath = filepath.Join("C:\\", "Games", "WoWRuski")
	//	for _, addon := range addons {
	//		log.Printf("%#v", addon.IsUpToDate())
	//		log.Printf("%#v", addon.GetRemoteReleaseURL())
	//	}
	file, err := os.ReadFile("Heimdall.zip")
	if err != nil {
		log.Printf("error reading file: %s", err)
		return
	}
	err = addons[0].Update(file)
	if err != nil {
		log.Printf("error updating addon: %s", err)
		return
	}
	log.Printf("Addon updated")
	return
	app := NewApp()

	err = wails.Run(&options.App{
		Title:  "wails-template",
		Width:  1024,
		Height: 768,
		AssetServer: &assetserver.Options{
			Assets: assets,
		},
		BackgroundColour: &options.RGBA{R: 27, G: 38, B: 54, A: 1},
		OnStartup:        app.startup,
		Bind: []interface{}{
			app,
		},
	})

	if err != nil {
		println("Error:", err.Error())
	}
}