49 lines
1.1 KiB
Go
Executable File
49 lines
1.1 KiB
Go
Executable File
package extract
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"os"
|
|
"path/filepath"
|
|
)
|
|
|
|
var (
|
|
ErrUnsupportedFormat = errors.New("unsupported archive format")
|
|
ErrUnsafeEntry = errors.New("archive contains an unsafe path")
|
|
ErrTooLarge = errors.New("archive exceeds the extraction size limit")
|
|
ErrTooManyEntries = errors.New("archive contains too many entries")
|
|
)
|
|
|
|
type Limits struct {
|
|
MaxBytes int64
|
|
MaxEntries int
|
|
}
|
|
|
|
type Progress interface {
|
|
SetTotal(total int64)
|
|
Add(delta int64)
|
|
}
|
|
|
|
func Archive(ctx context.Context, archivePath, destination string, limits Limits, progress Progress) error {
|
|
archiveFormat, _ := detect(filepath.Base(archivePath))
|
|
if archiveFormat == formatNone {
|
|
return ErrUnsupportedFormat
|
|
}
|
|
if err := os.Mkdir(destination, 0o755); err != nil {
|
|
return err
|
|
}
|
|
|
|
writer := &entryWriter{ctx: ctx, root: destination, limits: limits, progress: progress}
|
|
var err error
|
|
if archiveFormat == formatZip {
|
|
err = extractZip(archivePath, writer)
|
|
} else {
|
|
err = extractTar(archivePath, archiveFormat, writer)
|
|
}
|
|
|
|
if err != nil {
|
|
os.RemoveAll(destination)
|
|
}
|
|
return err
|
|
}
|