x

core/node_os.go

@@ -4,6 +4,7 @@ import (
 	"errors"
 	"os"
 	"path/filepath"
+	"runtime"
 
 	"github.com/dop251/goja"
 )
@@ -41,5 +42,7 @@ func getJsOs(vm *goja.Runtime, running func() bool) *goja.Object {
 		}
 		return dir
 	})
+	jsos.Set("name", runtime.GOOS)
+	jsos.Set("arch", runtime.GOARCH)
 	return jsos
 }

core/web.go

@@ -7,6 +7,7 @@ import (
 	"io"
 	"net"
 	"net/http"
+	"os"
 	"regexp"
 	"strings"
 	"time"
@@ -43,6 +44,21 @@ func Cors() gin.HandlerFunc {
 var Server *gin.Engine
 
 func init() {
+	for _, arg := range os.Args { //处理升级
+		if arg == "-r" { //准备程序->原程序
+			rfix := ".ready.exe"
+			ofix := ".exe"
+			if strings.Contains(os.Args[0], rfix) {
+				err := utils.CopyFile(utils.ProcessName, strings.Replace(utils.ProcessName, rfix, ofix, -1))
+				if err == nil {
+					utils.Daemon("reset")
+				}
+			} else {
+				os.Remove(strings.ReplaceAll(os.Args[0], ofix, rfix))
+			}
+			continue
+		}
+	}
 	gin.SetMode(gin.ReleaseMode)
 	Server = gin.New()
 	// Server.Use(gin.Recovery())

main.go

@@ -6,7 +6,6 @@ import (
 	"os"
 	"os/exec"
 	"runtime"
-	"strings"
 	"time"
 
 	_ "github.com/cdle/sillyplus/adapters/qq"
@@ -19,7 +18,6 @@ import (
 var sillyplus = core.MakeBucket("sillyplus")
 
 func main() {
-
 	loc, _ := time.LoadLocation("Asia/Shanghai")
 	time.Local = loc
 	core.Init()
@@ -31,19 +29,6 @@ func main() {
 		if arg == "-d" {
 			d = true
 		}
-		if arg == "-r" { //准备程序->原程序
-			rfix := ".ready.exe"
-			ofix := ".exe"
-			if strings.Contains(os.Args[0], rfix) {
-				err := utils.CopyFile(utils.ProcessName, strings.Replace(utils.ProcessName, rfix, ofix, -1))
-				if err == nil {
-					utils.Daemon("reset")
-				}
-			} else {
-				os.Remove(strings.ReplaceAll(os.Args[0], ofix, rfix))
-			}
-			continue
-		}
 	}
 	go func() { //弹出浏览器
 		if runtime.GOOS != "windows" {