测试2

Author: jdnjk

core/config.go

@@ -0,0 +1,82 @@
+package core
+
+import (
+	"archive/zip"
+	"encoding/json"
+	"fmt"
+	"log"
+)
+
+// InstConfig 结构体表示 inst.json 的内容
+type InstConfig struct {
+	Version        string `json:"version"`
+	Loader         string `json:"loader"`
+	LoaderVersion  string `json:"loaderVersion"`
+	Download       string `json:"download"`
+	MaxConnections int    `json:"maxconnections"`
+}
+
+// Config 定义配置文件的结构
+type Config struct {
+	MaxConnections int `json:"maxconnections"`
+}
+
+// Library 定义库的结构
+type Library struct {
+	Name      string `json:"name"`
+	Downloads struct {
+		Artifact struct {
+			URL  string `json:"url"`
+			Path string `json:"path"`
+			SHA1 string `json:"sha1"`
+		} `json:"artifact"`
+	} `json:"downloads"`
+}
+
+// VersionInfo 定义 version.json 文件的结构
+type VersionInfo struct {
+	Libraries []Library `json:"libraries"`
+}
+
+// 从 JAR 文件中提取 version.json
+func ExtractVersionJson(jarFilePath string) (VersionInfo, error) {
+	var versionInfo VersionInfo
+	r, err := zip.OpenReader(jarFilePath)
+	if err != nil {
+		return versionInfo, fmt.Errorf("无法打开 JAR 文件: %v", err)
+	}
+	defer r.Close()
+
+	for _, f := range r.File {
+		if f.Name == "version.json" {
+			rc, err := f.Open()
+			if err != nil {
+				log.Printf("警告: 无法打开 version.json 文件: %v", err)
+				continue
+			}
+			defer rc.Close()
+			if err := json.NewDecoder(rc).Decode(&versionInfo); err != nil {
+				log.Printf("警告: 无法解析 version.json: %v", err)
+				continue
+			}
+			return versionInfo, nil
+		}
+	}
+	for _, f := range r.File {
+		if f.Name == "install_profile.json" {
+			rc, err := f.Open()
+			if err != nil {
+				return versionInfo, fmt.Errorf("无法打开 install_profile.json 文件: %v", err)
+			}
+			defer rc.Close()
+
+			if err := json.NewDecoder(rc).Decode(&versionInfo); err != nil {
+				return versionInfo, fmt.Errorf("无法解析 install_profile.json: %v", err)
+			}
+
+			return versionInfo, nil
+		}
+	}
+
+	return versionInfo, fmt.Errorf("没有找到文件")
+}

core/download.go

@@ -0,0 +1,83 @@
+package core
+
+import (
+	"fmt"
+	"io"
+	"net/http"
+	"os"
+	"time"
+)
+
+// ProgressReader 用于跟踪 io.Reader 的进度
+type ProgressReader struct {
+	Reader          io.ReadCloser
+	Total           int64
+	Current         int64
+	FilePath        string
+	UpdateInterval  int64 // 更新间隔，单位秒
+	lastUpdatedTime int64
+}
+
+// Read 实现了 io.Reader 接口
+func (pr *ProgressReader) Read(p []byte) (n int, err error) {
+	n, err = pr.Reader.Read(p)
+	pr.Current += int64(n)
+
+	now := time.Now().Unix()
+	if now-pr.lastUpdatedTime >= pr.UpdateInterval || err == io.EOF {
+		pr.lastUpdatedTime = now
+		percent := float64(pr.Current) / float64(pr.Total) * 100
+		fmt.Printf("下载进度: %.2f%% (%s)\n", percent, pr.FilePath)
+	}
+	return
+}
+
+func DownloadFile(url, filePath string) error {
+	const maxRetries = 3
+
+	for i := 0; i < maxRetries; i++ {
+		resp, err := http.Get(url)
+		if err != nil {
+			fmt.Printf("尝试 %d/%d 下载失败: %v\n", i+1, maxRetries, err)
+			continue
+		}
+		defer resp.Body.Close()
+
+		if resp.StatusCode == http.StatusNotFound {
+			fmt.Printf("文件未找到，跳过: %s\n", url)
+			return nil
+		}
+		if resp.StatusCode != http.StatusOK {
+			fmt.Printf("尝试 %d/%d，HTTP 状态码: %d\n", i+1, maxRetries, resp.StatusCode)
+			continue
+		}
+
+		// 直接用 ContentLength，省去第二次请求
+		total := resp.ContentLength
+		if total <= 0 {
+			fmt.Println("无法获取文件大小，将不显示进度")
+		}
+
+		outFile, err := os.Create(filePath)
+		if err != nil {
+			return fmt.Errorf("创建文件失败: %w", err)
+		}
+		defer outFile.Close()
+
+		reader := &ProgressReader{
+			Reader:         resp.Body,
+			Total:          total,
+			FilePath:       filePath,
+			UpdateInterval: 3,
+		}
+
+		if _, err = io.Copy(outFile, reader); err != nil {
+			fmt.Printf("写入失败 %d/%d: %v\n", i+1, maxRetries, err)
+			continue
+		}
+		fmt.Println("下载完成！")
+		return nil
+	}
+
+	return fmt.Errorf("多次尝试下载失败 (共 %d 次)", maxRetries)
+}

core/installer.go

@@ -0,0 +1,90 @@
+package core
+
+import (
+	"fmt"
+	"io"
+	"os"
+	"os/exec"
+	"path/filepath"
+	"runtime"
+	"strings"
+)
+
+func RunInstaller(installerPath string, loader string, version string, loaderVersion string, Download string) error {
+	var cmd *exec.Cmd
+	if Download == "bmclapi" {
+		if loader == "forge" {
+			cmd = exec.Command("java", "-jar", installerPath, "--installServer", "--mirror", "https://bmclapi2.bangbang93.com/maven/")
+		} else if loader == "fabric" {
+			cmd = exec.Command(
+				"java", "-jar", installerPath, "server",
+				"-mavenurl", "https://bmclapi2.bangbang93.com/maven/",
+				"-metaurl", "https://bmclapi2.bangbang93.com/fabric-meta/",
+				"-mcversion", version,
+				"-loader", loaderVersion,
+			)
+		} else {
+			cmd = exec.Command("java", "-jar", installerPath)
+		}
+	} else {
+		if loader == "forge" {
+			cmd = exec.Command("java", "-jar", installerPath, "--installServer")
+		} else if loader == "fabric" {
+			cmd = exec.Command(
+				"java", "-jar", installerPath, "server",
+				"-mcversion", version,
+				"-loader", loaderVersion,
+			)
+		} else {
+			cmd = exec.Command("java", "-jar", installerPath)
+		}
+	}
+
+	stdout, err := cmd.StdoutPipe()
+	if err != nil {
+		return fmt.Errorf("无法获取标准输出: %v", err)
+	}
+	stderr, err := cmd.StderrPipe()
+	if err != nil {
+		return fmt.Errorf("无法获取标准错误: %v", err)
+	}
+	if err := cmd.Start(); err != nil {
+		return fmt.Errorf("启动命令失败: %v", err)
+	}
+	go func() {
+		io.Copy(os.Stdout, stdout)
+	}()
+	go func() {
+		io.Copy(os.Stderr, stderr)
+	}()
+
+	if err := cmd.Wait(); err != nil {
+		return fmt.Errorf("命令执行失败: %v", err)
+	}
+	return nil
+}
+
+func FindJava() (string, bool) {
+	if runtime.GOOS == "linux" {
+		simpfun := "/usr/bin/jdk/jdk1.8.0_361/bin/java"
+		if _, err := os.Stat(simpfun); err == nil {
+			return simpfun, true
+		}
+	}
+
+	javaHome := os.Getenv("JAVA_HOME")
+	if javaHome != "" {
+		javaPath := filepath.Join(javaHome, "bin", "java")
+		if _, err := os.Stat(javaPath); err == nil {
+			return javaPath, false
+		}
+	}
+
+	cmd := exec.Command("java", "-version")
+	output, err := cmd.CombinedOutput()
+	if err == nil && strings.Contains(string(output), "version") {
+		return "java", false
+	}
+
+	return "", false
+}