UPDATE: Add ldiff

internal/git-service/launcher.go

@@ -0,0 +1,93 @@
+package gitService
+
+import (
+	"encoding/json"
+	"firefly-launcher/pkg/constant"
+	"firefly-launcher/pkg/models"
+	"io"
+	"net/http"
+	"github.com/minio/selfupdate"
+
+)
+
+type GitService struct{}
+
+
+func (g *GitService) GetLatestLauncherVersion() (bool, string, string) {
+	resp, err := http.Get(constant.LauncherGitUrl)
+	if err != nil {
+		return false, "", err.Error()
+	}
+	defer resp.Body.Close()
+
+	body, _ := io.ReadAll(resp.Body)
+
+	var releases []models.ReleaseType
+	err = json.Unmarshal(body, &releases)
+	if err != nil {
+		return false, "", err.Error()
+	}
+
+	if len(releases) == 0 {
+		return false, "", "no releases found"
+	}
+
+	return true, releases[0].TagName, ""
+}
+
+
+
+func (g *GitService) UpdateLauncherProgress(version string) (bool, string) {
+	resp, err := http.Get(constant.LauncherGitUrl)
+	if err != nil {
+		panic(err)
+	}
+	defer resp.Body.Close()
+
+	body, _ := io.ReadAll(resp.Body)
+
+	var releases []*models.ReleaseType
+	err = json.Unmarshal(body, &releases)
+	if err != nil {
+		return false, err.Error()
+	}
+
+	if len(releases) == 0 {
+		return false, "no releases found"
+	}
+
+	var releaseData *models.ReleaseType
+	for _, release := range releases {
+		if release.TagName == version {
+			releaseData = release
+			break
+		}
+	}
+
+	if releaseData == nil || releaseData.TagName == "" {
+		return false, "no release found"
+	}
+
+	var assetWin models.AssetType
+	for _, asset := range releaseData.Assets {
+		if asset.Name == constant.LauncherFile {
+			assetWin = asset
+			break
+		}
+	}
+
+	if assetWin.Name == "" {
+		return false, "no assets found"
+	}
+
+	resp, err = http.Get(assetWin.BrowserDownloadURL)
+	if err != nil {
+		return false, err.Error()
+	}
+	defer resp.Body.Close()
+	err = selfupdate.Apply(resp.Body, selfupdate.Options{})
+	if err != nil {
+		return false, err.Error()
+	}
+	return true, ""
+}

internal/git-service/proxy.go

@@ -0,0 +1,106 @@
+package gitService
+
+import (
+	"firefly-launcher/pkg/constant"
+	"firefly-launcher/pkg/models"
+	"fmt"
+	"io"
+	"net/http"
+	"os"
+	"path/filepath"
+	"encoding/json"
+	"github.com/wailsapp/wails/v3/pkg/application"
+)
+
+func (g *GitService) GetLatestProxyVersion() (bool, string, string) {
+	resp, err := http.Get(constant.ProxyGitUrl)
+	if err != nil {
+		return false, "", err.Error()
+	}
+	defer resp.Body.Close()
+
+	body, _ := io.ReadAll(resp.Body)
+
+	var releases []models.ReleaseType
+	err = json.Unmarshal(body, &releases)
+	if err != nil {
+		return false, "", err.Error()
+	}
+
+	if len(releases) == 0 {
+		return false, "", "no releases found"
+	}
+
+	return true, releases[0].TagName, ""
+}
+
+func (g *GitService) DownloadProxyProgress(version string) (bool, string) {
+	resp, err := http.Get(constant.ProxyGitUrl)
+	if err != nil {
+		panic(err)
+	}
+	defer resp.Body.Close()
+
+	body, _ := io.ReadAll(resp.Body)
+
+	var releases []*models.ReleaseType
+	err = json.Unmarshal(body, &releases)
+	if err != nil {
+		return false, err.Error()
+	}
+
+	if len(releases) == 0 {
+		return false, "no releases found"
+	}
+
+	var releaseData *models.ReleaseType
+	for _, release := range releases {
+		if release.TagName == version {
+			releaseData = release
+			break
+		}
+	}
+
+	if releaseData == nil || releaseData.TagName == "" {
+		return false, "no release found"
+	}
+
+	var assetWin models.AssetType
+	for _, asset := range releaseData.Assets {
+		if asset.Name == constant.ProxyZipFile {
+			assetWin = asset
+			break
+		}
+	}
+
+	if assetWin.Name == "" {
+		return false, "no assets found"
+	}
+
+	if err := os.Mkdir(constant.ProxyStorageUrl, 0755); err != nil {
+		if !os.IsExist(err) {
+			return false, err.Error()
+		}
+	}
+	saveFile := filepath.Join(constant.ProxyStorageUrl, assetWin.Name)
+
+	resp, err = http.Get(assetWin.BrowserDownloadURL)
+	if err != nil {
+		return false, err.Error()
+	}
+	defer resp.Body.Close()
+
+	DownloadFile(saveFile, assetWin.BrowserDownloadURL, func(percent float64, speed float64) {
+		application.Get().Event.Emit("download:proxy", map[string]interface{}{
+			"percent": fmt.Sprintf("%.2f", percent),
+			"speed":   fmt.Sprintf("%.2f", speed),
+		})
+
+	})
+	return true, ""
+}
+
+func (g *GitService) UnzipProxy() {
+	unzipParallel(filepath.Join(constant.ProxyStorageUrl, constant.ProxyZipFile), constant.ProxyStorageUrl)
+	os.Remove(filepath.Join(constant.ProxyStorageUrl, constant.ProxyZipFile))
+}

internal/git-service/server.go

@@ -0,0 +1,107 @@
+package gitService
+
+import (
+	"encoding/json"
+	"firefly-launcher/pkg/constant"
+	"firefly-launcher/pkg/models"
+	"fmt"
+	"io"
+	"net/http"
+	"os"
+	"path/filepath"
+
+	"github.com/wailsapp/wails/v3/pkg/application"
+)
+
+func (g *GitService) GetLatestServerVersion() (bool, string, string) {
+	resp, err := http.Get(constant.ServerGitUrl)
+	if err != nil {
+		return false, "", err.Error()
+	}
+	defer resp.Body.Close()
+
+	body, _ := io.ReadAll(resp.Body)
+
+	var releases []models.ReleaseType
+	err = json.Unmarshal(body, &releases)
+	if err != nil {
+		return false, "", err.Error()
+	}
+
+	if len(releases) == 0 {
+		return false, "", "no releases found"
+	}
+
+	return true, releases[0].TagName, ""
+}
+
+func (g *GitService) DownloadServerProgress(version string) (bool, string) {
+	resp, err := http.Get(constant.ServerGitUrl)
+	if err != nil {
+		panic(err)
+	}
+	defer resp.Body.Close()
+
+	body, _ := io.ReadAll(resp.Body)
+
+	var releases []*models.ReleaseType
+	err = json.Unmarshal(body, &releases)
+	if err != nil {
+		return false, err.Error()
+	}
+
+	if len(releases) == 0 {
+		return false, "no releases found"
+	}
+
+	var releaseData *models.ReleaseType
+	for _, release := range releases {
+		if release.TagName == version {
+			releaseData = release
+			break
+		}
+	}
+
+	if releaseData == nil || releaseData.TagName == "" {
+		return false, "no release found"
+	}
+
+	var assetWin models.AssetType
+	for _, asset := range releaseData.Assets {
+		if asset.Name == constant.ServerZipFile {
+			assetWin = asset
+			break
+		}
+	}
+
+	if assetWin.Name == "" {
+		return false, "no assets found"
+	}
+
+	if err := os.Mkdir(constant.ServerStorageUrl, 0755); err != nil {
+		if !os.IsExist(err) {
+			return false, err.Error()
+		}
+	}
+
+	saveFile := filepath.Join(constant.ServerStorageUrl, assetWin.Name)
+
+	resp, err = http.Get(assetWin.BrowserDownloadURL)
+	if err != nil {
+		return false, err.Error()
+	}
+	defer resp.Body.Close()
+
+	DownloadFile(saveFile, assetWin.BrowserDownloadURL, func(percent float64, speed float64) {
+		application.Get().Event.Emit("download:server", map[string]interface{}{
+			"percent": fmt.Sprintf("%.2f", percent),
+			"speed":   fmt.Sprintf("%.2f", speed),
+		})
+	})
+	return true, ""
+}
+
+func (g *GitService) UnzipServer() {
+	unzipParallel(filepath.Join(constant.ServerStorageUrl, constant.ServerZipFile), constant.ServerStorageUrl)
+	os.Remove(filepath.Join(constant.ServerStorageUrl, constant.ServerZipFile))
+}

internal/git-service/util.go

@@ -0,0 +1,188 @@
+package gitService
+
+import (
+	"archive/zip"
+	"fmt"
+	"io"
+	"math"
+	"net/http"
+	"os"
+	"path/filepath"
+	"runtime"
+	"sync"
+	"time"
+)
+
+func humanFormat(bytes int64) string {
+	n := float64(bytes)
+	for _, unit := range []string{"", "Ki", "Mi", "Gi"} {
+		if math.Abs(n) < 1024.0 {
+			return fmt.Sprintf("%3.1f%sB", n, unit)
+		}
+		n /= 1024.0
+	}
+	return fmt.Sprintf("%.1fTiB", n)
+}
+
+type WriteCounter struct {
+	Total             uint64
+	StartTime         time.Time
+	OnEmit            func(percent float64, speedMBps float64)
+	TotalSize         int64
+	lastLoggedPercent int
+}
+
+func NewWriteCounter(total int64, onEmit func(percent float64, speedMBps float64)) *WriteCounter {
+	return &WriteCounter{
+		StartTime:         time.Now(),
+		TotalSize:         total,
+		lastLoggedPercent: -1,
+		OnEmit:            onEmit,
+	}
+}
+
+func (wc *WriteCounter) Write(p []byte) (int, error) {
+	n := len(p)
+	wc.Total += uint64(n)
+	wc.PrintProgress()
+	return n, nil
+}
+
+func (wc *WriteCounter) PrintProgress() {
+	elapsed := time.Since(wc.StartTime).Seconds()
+	if elapsed < 0.001 {
+		elapsed = 0.001
+	}
+
+	speed := float64(wc.Total) / 1024 / 1024 / elapsed // MB/s
+	percent := float64(wc.Total) / float64(wc.TotalSize) * 100
+	if wc.OnEmit != nil {
+		wc.OnEmit(percent, speed)
+	}
+}
+
+func DownloadFile(filepath string, url string, onEmit func(percent float64, speed float64)) error {
+	tmpPath := filepath + ".tmp"
+
+	resp, err := http.Get(url)
+	if err != nil {
+		return fmt.Errorf("failed to get file: %w", err)
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != http.StatusOK {
+		return fmt.Errorf("bad status: %s", resp.Status)
+	}
+
+	out, err := os.Create(tmpPath)
+	if err != nil {
+		return fmt.Errorf("failed to create tmp file: %w", err)
+	}
+
+	counter := NewWriteCounter(resp.ContentLength, onEmit)
+	_, err = io.Copy(out, io.TeeReader(resp.Body, counter))
+	if closeErr := out.Close(); closeErr != nil {
+		return fmt.Errorf("failed to close tmp file: %w", closeErr)
+	}
+	if err != nil {
+		return fmt.Errorf("failed to download file: %w", err)
+	}
+
+	// Delete destination file if it exists
+	if _, err := os.Stat(filepath); err == nil {
+		if err := os.Remove(filepath); err != nil {
+			return fmt.Errorf("failed to remove existing file: %w", err)
+		}
+	}
+
+	for i := 0; i < 3; i++ {
+		err = os.Rename(tmpPath, filepath)
+		if err == nil {
+			break
+		}
+		time.Sleep(300 * time.Millisecond)
+	}
+	if err != nil {
+		return fmt.Errorf("failed to rename after retries: %w", err)
+	}
+
+	return nil
+}
+
+func unzipParallel(src string, dest string) error {
+	numCPU := runtime.NumCPU()
+
+	reserved := 1
+	if numCPU > 4 {
+		reserved = 2
+	}
+	maxWorkers := numCPU - reserved
+	if maxWorkers < 1 {
+		maxWorkers = 1
+	}
+	r, err := zip.OpenReader(src)
+	if err != nil {
+		return err
+	}
+	defer r.Close()
+
+	err = os.MkdirAll(dest, 0755)
+	if err != nil {
+		return err
+	}
+
+	type job struct {
+		f *zip.File
+	}
+	jobs := make(chan job)
+	var wg sync.WaitGroup
+
+	// Worker pool
+	for i := 0; i < maxWorkers; i++ {
+		wg.Go(func() {
+			for j := range jobs {
+				err := extractFile(j.f, dest)
+				if err != nil {
+					fmt.Printf("Error extracting %s: %v\n", j.f.Name, err)
+				}
+			}
+		})
+	}
+
+	// Feed jobs
+	for _, f := range r.File {
+		jobs <- job{f}
+	}
+	close(jobs)
+	wg.Wait()
+
+	return nil
+}
+
+func extractFile(f *zip.File, dest string) error {
+	fp := filepath.Join(dest, f.Name)
+
+	if f.FileInfo().IsDir() {
+		return os.MkdirAll(fp, f.Mode())
+	}
+
+	err := os.MkdirAll(filepath.Dir(fp), 0755)
+	if err != nil {
+		return err
+	}
+
+	rc, err := f.Open()
+	if err != nil {
+		return err
+	}
+	defer rc.Close()
+
+	out, err := os.OpenFile(fp, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, f.Mode())
+	if err != nil {
+		return err
+	}
+	defer out.Close()
+
+	_, err = io.Copy(out, rc)
+	return err
+}