mirror of
https://github.com/mudler/LocalAI.git
synced 2024-12-19 12:47:54 +00:00
57 lines
1.1 KiB
Go
57 lines
1.1 KiB
Go
|
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)
|
||
|
|
}
|