feat: 打扫垃圾

Author: jdnjk

core/argument.go

@@ -5,14 +5,19 @@ import (
 	"os"
 )
 
-func Argument(gitversion string) {
+func Argument(gitversion string) bool {
+	cleaninst := false
 	if len(os.Args) > 1 && os.Args[1] == "--help" {
 		fmt.Println("--help 获取帮助")
 		fmt.Println("--version 获取版本")
-		return
+		os.Exit(0)
 	}
 	if len(os.Args) > 1 && os.Args[1] == "--version" {
 		fmt.Println("AutoInstall-" + gitversion)
-		return
+		os.Exit(0)
 	}
+	if len(os.Args) > 1 && os.Args[1] == "--clean" {
+		cleaninst = true
+	}
+	return cleaninst
 }

main.go

@@ -66,6 +66,7 @@ func main() {
 	if gitversion == "" {
 		gitversion = "NaN"
 	}
+	cleaninst := core.Argument(gitversion)
 	core.Argument(gitversion)
 	os.MkdirAll(".autoinst/cache", os.ModePerm)
 	instFile := "inst.json"
@@ -89,10 +90,12 @@ func main() {
 		if config.Loader != "vanilla" {
 			fmt.Printf("加载器: %s\n", config.Loader)
 			fmt.Printf("加载器版本: %s\n", config.LoaderVersion)
-			fmt.Println("\033[31m[警告] 加载器版本过新可能会无法正常下载\033[0m")
+			if config.Download == "bmclapi" {
+				fmt.Println("\033[31m[警告] 加载器版本过新可能会无法正常下载\033[0m")
+			}
 		}
 		fmt.Printf("下载源: %s\n", config.Download)
-		pkg.Common(config)
+		pkg.Common(config, cleaninst)
 	} else if os.IsNotExist(err) {
 		log.Println("inst.json 文件不存在")
 	} else {

pkg/common.go

@@ -3,11 +3,12 @@ package pkg
 import (
 	"fmt"
 	"log"
+	"os"
 
 	"github.com/autoinst/AutoInstall/core"
 )
 
-func Common(config core.InstConfig) {
+func Common(config core.InstConfig, cleaninst bool) {
 	javaPath, simpfun, mise := core.FindJava()
 	if simpfun {
 		fmt.Println("已启用 simpfun 环境")
@@ -40,4 +41,23 @@ func Common(config core.InstConfig) {
 		fmt.Println("服务端下载完成")
 		core.RunScript(config.Version, config.Loader, config.LoaderVersion, simpfun, mise)
 	}
+	if cleaninst {
+		fmt.Println("正在清理残留...")
+		if err := os.Remove(".autoinst"); err != nil && !os.IsNotExist(err) {
+			log.Println("删除 .autoinst 文件失败:", err)
+		}
+		files, err := os.ReadDir(".")
+		if err == nil {
+			for _, file := range files {
+				if !file.IsDir() && len(file.Name()) > 4 && file.Name()[len(file.Name())-4:] == ".log" {
+					if err := os.Remove(file.Name()); err != nil {
+						fmt.Println("删除日志文件失败:", file.Name(), err)
+					}
+				}
+			}
+		} else {
+			log.Println("读取目录失败:", err)
+		}
+		fmt.Println("清理完成")
+	}
 }