Handle tgz (#28)

marwanhawari · Apr 6, 2024 · ab96998 · ab96998
1 parent 2b6a2b8
commit ab96998
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 6 deletions.
diff --git a/lib/util.go b/lib/util.go
@@ -15,11 +15,8 @@ import (
 )
 
 func isArchiveFile(filePath string) bool {
-	fileExtension := filepath.Ext(filePath)
-	if fileExtension == ".br" || fileExtension == ".bz2" || fileExtension == ".zip" || fileExtension == ".gz" || fileExtension == ".lz4" || fileExtension == ".sz" || fileExtension == ".xz" || fileExtension == ".zst" || fileExtension == ".tar" || fileExtension == ".rar" {
-		return true
-	}
-	return false
+	_, err := archiver.ByExtension(filePath)
+	return err == nil
 }
 
 func isExecutableFile(filePath string) (bool, error) {

diff --git a/lib/util_test.go b/lib/util_test.go
@@ -25,12 +25,19 @@ func Test_isArchiveFile(t *testing.T) {
 			want: false,
 		},
 		{
-			name: "test1",
+			name: "test2",
 			args: args{
 				filePath: "Archive.tar.gz",
 			},
 			want: true,
 		},
+		{
+			name: "test3",
+			args: args{
+				filePath: "Archive.tgz",
+			},
+			want: true,
+		},
 	}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {