RWKV-Runner/backend-golang/app.go

package backend_golang

import (
	"context"
	"net/http"
	"os"
	"os/exec"
	"path/filepath"
	"runtime"

	"github.com/minio/selfupdate"
	wruntime "github.com/wailsapp/wails/v2/pkg/runtime"
)

// App struct
type App struct {
	ctx       context.Context
	exDir     string
	cmdPrefix string
}

// NewApp creates a new App application struct
func NewApp() *App {
	return &App{}
}

// startup is called when the app starts. The context is saved
// so we can call the runtime methods
func (a *App) OnStartup(ctx context.Context) {
	a.ctx = ctx
	a.exDir = ""
	a.cmdPrefix = ""

	if runtime.GOOS == "darwin" {
		ex, _ := os.Executable()
		a.exDir = filepath.Dir(ex) + "/../../../"
		a.cmdPrefix = "cd " + a.exDir + " && "
	}

	a.downloadLoop()
}

func (a *App) UpdateApp(url string) (broken bool, err error) {
	resp, err := http.Get(url)
	if err != nil {
		return false, err
	}
	defer resp.Body.Close()
	err = selfupdate.Apply(resp.Body, selfupdate.Options{})
	if err != nil {
		if rerr := selfupdate.RollbackError(err); rerr != nil {
			return true, rerr
		}
		return false, err
	}
	name, err := os.Executable()
	if err != nil {
		return false, err
	}
	exec.Command(name, os.Args[1:]...).Start()
	wruntime.Quit(a.ctx)
	return false, nil
}

func (a *App) GetPlatform() string {
	return runtime.GOOS
}