| package utils | |
| import ( | |
| "fmt" | |
| "os" | |
| "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 | |
| } | |
| err = archiver.Walk(archive, func(f archiver.File) error { | |
| if f.FileInfo.Mode()&os.ModeSymlink != 0 { | |
| return fmt.Errorf("archive contains a symlink") | |
| } | |
| return nil | |
| }) | |
| if err != nil { | |
| return err | |
| } | |
| return un.Unarchive(archive, dst) | |
| } | |