fix: 测试

Author: Weidows

.github/workflows/test.yml

@@ -26,7 +26,7 @@ jobs:
       - name: Add msbuild to PATH
         uses: microsoft/setup-msbuild@v1.1
       - name: Set up go
-        uses: actions/setup-go@v2
+        uses: actions/setup-go@v6
         with:
           go-version: "1.23"
       - name: Go test

cmd/wutils/zip/crack.go

@@ -18,7 +18,7 @@ const (
 	passwordFile      = "password-dict.txt"
 )
 
-func getPasswordFilePath() (string, error) {
+func getHomePasswordFilePath() (string, error) {
 	home, err := os.UserHomeDir()
 	if err != nil {
 		return "", fmt.Errorf("failed to get home directory: %w", err)
@@ -42,7 +42,7 @@ func getPasswordFilePath() (string, error) {
 }
 
 func CrackPassword(archivePath string) string {
-	passwordFilePath, err := getPasswordFilePath()
+	passwordFilePath, err := getHomePasswordFilePath()
 	if err != nil {
 		fmt.Printf("Error: %v\n", err)
 		return ""

cmd/wutils/zip/crack_test.go

@@ -7,12 +7,13 @@ import (
 )
 
 func TestCrackPassword(t *testing.T) {
-	home, _ := os.UserHomeDir()
-	configDir := filepath.Join(home, ".config", "wutils")
-	passwordDictPath := filepath.Join(configDir, "password-dict.txt")
+	localDictPath := "./password-dict.txt"
+	os.WriteFile(localDictPath, []byte("test\nwrong\n123456"), 0644)
+	defer os.Remove(localDictPath)
 
-	os.MkdirAll(configDir, 0755)
-	os.WriteFile(passwordDictPath, []byte("test\nwrong\n123456"), 0644)
+	home, _ := os.UserHomeDir()
+	homeDictPath := filepath.Join(home, ".config", "wutils", "password-dict.txt")
+	os.WriteFile(homeDictPath, []byte("test\nwrong\n123456"), 0644)
 
 	type args struct {
 		archivePath string
@@ -25,7 +26,7 @@ func TestCrackPassword(t *testing.T) {
 		{
 			name: "test",
 			args: args{
-				archivePath: "test.zip",
+				archivePath: "test/test.zip",
 			},
 			want: "test",
 		},

cmd/wutils/zip/password.txt

@@ -53,8 +53,8 @@ func (a *Archive) Unzip(destDir string) Result {
 	}
 }
 
-// TryUnzip attempts to unzip the archive with the given password
-// Returns true if successful, false otherwise (deprecated: use Unzip instead)
+// TryUnzip verifies if the password is correct without extracting files
+// Returns true if successful, false otherwise
 func (a *Archive) TryUnzip() bool {
 	ext := strings.ToLower(filepath.Ext(a.archivePath))
 	switch ext {
@@ -65,26 +65,32 @@ func (a *Archive) TryUnzip() bool {
 		}
 		defer reader.Close()
 
+		if len(reader.File) == 0 {
+			return false
+		}
+
 		for _, file := range reader.File {
-			if err := a.extractZipFile(file, "."); err != nil {
+			file.SetPassword(a.password)
+			rc, err := file.Open()
+			if err != nil {
+				return false
+			}
+			buf := make([]byte, 1)
+			_, err = rc.Read(buf)
+			rc.Close()
+			if err != nil {
 				return false
 			}
-			return true
 		}
+		return true
 
 	case ".7z":
 		reader, err := sevenzip.OpenReaderWithPassword(a.archivePath, a.password)
 		if err != nil {
 			return false
 		}
 		defer reader.Close()
-
-		for _, file := range reader.File {
-			if err := a.extract7zFile(file, "."); err != nil {
-				return false
-			}
-			return true
-		}
+		return true
 	}
 
 	return false