You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
57 lines
1.1 KiB
57 lines
1.1 KiB
2 years ago
|
package utils
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"github.com/mholt/archiver/v3"
|
||
|
|
)
|
||
|
|
|
||
|
|
func IsArchive(file string) bool {
|
||
|
|
uaIface, err := archiver.ByExtension(file)
|
||
|
|
if err != nil {
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
|
||
|
|
_, ok := uaIface.(archiver.Unarchiver)
|
||
|
|
return ok
|
||
|
|
}
|
||
|
|
|
||
|
|
func ExtractArchive(archive, dst string) error {
|
||
|
|
uaIface, err := archiver.ByExtension(archive)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
un, ok := uaIface.(archiver.Unarchiver)
|
||
|
|
if !ok {
|
||
|
|
return fmt.Errorf("format specified by source filename is not an archive format: %s (%T)", archive, uaIface)
|
||
|
|
}
|
||
|
|
|
||
|
|
mytar := &archiver.Tar{
|
||
|
|
OverwriteExisting: true,
|
||
|
|
MkdirAll: true,
|
||
|
|
ImplicitTopLevelFolder: false,
|
||
|
|
ContinueOnError: true,
|
||
|
|
}
|
||
|
|
|
||
|
|
switch v := uaIface.(type) {
|
||
|
|
case *archiver.Tar:
|
||
|
|
uaIface = mytar
|
||
|
|
case *archiver.TarBrotli:
|
||
|
|
v.Tar = mytar
|
||
|
|
case *archiver.TarBz2:
|
||
|
|
v.Tar = mytar
|
||
|
|
case *archiver.TarGz:
|
||
|
|
v.Tar = mytar
|
||
|
|
case *archiver.TarLz4:
|
||
|
|
v.Tar = mytar
|
||
|
|
case *archiver.TarSz:
|
||
|
|
v.Tar = mytar
|
||
|
|
case *archiver.TarXz:
|
||
|
|
v.Tar = mytar
|
||
|
|
case *archiver.TarZstd:
|
||
|
|
v.Tar = mytar
|
||
|
|
}
|
||
|
|
return un.Unarchive(archive, dst)
|
||
|
|
}
|