[script] Add decompress script

1 files changed, 114 insertions, 0 deletions

diff --git a/.local/bin/decompress b/.local/bin/decompress
new file mode 100755
index 0000000..a2023c3
--- /dev/null
+++ b/.local/bin/decompress
@@ -0,0 +1,114 @@
+#!/usr/bin/env fish
+
+set options 'h/help'
+set archiver_formats tar
+set bzip2 bz2 tar.bz2 tbz
+set gzip gz tar.gz tgz
+set lz4 lz4 tar.lz4
+set xz xz tar.xz txz
+set zstd zst tar.zst tzst
+set compressor_formats $bzip2 $gzip $lz4 $xz $zstd
+set arch_and_comp_formats 7z rar zip
+set formats $archiver_formats $compressor_formats $arch_and_comp_formats
+
+function print_help
+    echo -es "Unarchive and uncompress wrapper\n" \
+        "Supported formats: $formats\n" \
+        "Also able to use MIME types and detect ZIP magic sequence"
+    echo "Usage: uncompress" \
+        "[-h | --help]" \
+        "file1 [file2, file3, ...]" >&2
+    echo -es \
+        "-h, --help\t\tPrint this help"
+end
+
+function get_extension -a file
+    set bname (basename $file)
+    set extension (string join '.' (string split '.' $bname)[2..])
+    echo $extension
+end
+
+function extract_by_extension -a archive extension
+    switch $extension
+        case tar
+            tar --force-local -xvf $archive
+        case tar.bz2 tbz
+            bzcat $archive | tar xf -
+        case bz2
+            bzip2 -d $archive
+        case tar.gz tgz
+            zcat $archive | tar xf -
+        case gz
+            gzip -d $archive
+        case tar.lz4
+            lz4cat $archive | tar xf -
+        case lz4
+            lz4 $archive
+        case tar.xz txz
+            xzcat $archive | tar xf -
+        case xz
+            xz -d $archive
+        case tar.zst tzst
+            zstdcat $archive | tar xf -
+        case zst
+            zstd -d $archive
+        case 7z
+            7z x $archive
+        case rar
+            if type -q rar
+                rar x $archive
+            else
+                unrar x $archive
+            end
+        case zip
+            unzip $archive
+    end
+    or return 1
+end
+
+function extract_by_mime -a archive
+    set possible_extensions (string split ' ' (file --extension $archive))[2]
+    set possible_extensions (string split '/' $possible_extensions)
+
+    for extension in $possible_extensions
+        extract_by_extension $archive $extension
+        and return
+    end
+
+    return 1
+end
+
+# Script main code
+
+argparse $options -- $argv
+or return 1
+
+if set -q _flag_help; or test (count $argv) -eq 0
+    print_help
+    return
+end
+
+set final_status 0
+
+for archive in $argv
+    set extension (get_extension $archive)
+
+    if contains $extension $formats
+        extract_by_extension $archive $extension
+    else
+        echo "Unable to match $archive extension, trying MIME-type detection" >&2
+        extract_by_mime $archive
+        and continue
+
+        echo "Checking if $archive is secretly a ZIP archive" >&2
+        if test (head -c 4 $archive | base64) = "UEsDBA=="
+            unzip $archive
+            continue
+        end
+
+        echo "Extraction of $archive failed" >&2
+        set final_status 1
+    end
+end
+
+return $final_status