From ab9699829ca13f05b5e9c63616dfb85ff63ce783 Mon Sep 17 00:00:00 2001
From: hannah98 <hannah98@users.noreply.github.com>
Date: Sat, 6 Apr 2024 18:21:32 -0500
Subject: [PATCH] Handle tgz (#28)

---
 lib/util.go      | 7 ++-----
 lib/util_test.go | 9 ++++++++-
 2 files changed, 10 insertions(+), 6 deletions(-)

diff --git a/lib/util.go b/lib/util.go
index 0fa5bf3..7b8c878 100644
--- 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
index c10bfbb..8dddb9b 100644
--- 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) {