package sys import ( "archive/zip" "bufio" "bytes" "compress/bzip2" "compress/gzip" "fmt" "io" "log" "os" "github.com/ulikunitz/xz" ) type Xtract struct { dir string reader *bufio.Reader bar *io.Writer outFile string withBar bool } func NewXtract(dir string, outFile string) *Xtract { return &Xtract{dir: dir, outFile: outFile, withBar: false} } func (x *Xtract) addBar(bar *io.Writer) { x.bar = bar x.withBar = true } func (x *Xtract) setCompressedFile(dumpName string) { path := fmt.Sprintf("%s/%s", x.dir, dumpName) file, err := os.Open(path) if err != nil { panic(err) } x.reader = bufio.NewReader(file) } func (x *Xtract) writeUncompressedData(out *bytes.Buffer) bool { err := os.WriteFile(x.outFile, out.Bytes(), 0644) if err != nil { panic(err) } return true } func (x *Xtract) UnGzip(dumpName string) bool { x.setCompressedFile(dumpName) var out bytes.Buffer r, err := gzip.NewReader(x.reader) if err != nil { panic(err) } if x.withBar { _, err = io.Copy(io.MultiWriter(&out, *x.bar), r) } else { _, err = io.Copy(&out, r) } if err != nil { panic(err) } // if x.withBar { // x.bar.Clear() // } return x.writeUncompressedData(&out) } func (x *Xtract) UnBz2(dumpName string) bool { x.setCompressedFile(dumpName) var out bytes.Buffer r := bzip2.NewReader(x.reader) var err error if x.withBar { _, err = io.Copy(io.MultiWriter(&out, *x.bar), r) } else { _, err = io.Copy(&out, r) } if err != nil { panic(err) } return x.writeUncompressedData(&out) } func (x *Xtract) UnXz(dumpName string) bool { x.setCompressedFile(dumpName) var out bytes.Buffer r, err := xz.NewReader(x.reader) if err != nil { panic(err) } if x.withBar { _, err = io.Copy(io.MultiWriter(&out, *x.bar), r) } else { _, err = io.Copy(&out, r) } if err != nil { panic(err) } return x.writeUncompressedData(&out) } func (x *Xtract) UnZip(dumpName string) bool { var done bool = false path := fmt.Sprintf("%s/%s", x.dir, dumpName) r, err := zip.OpenReader(path) if err != nil { panic(err) } defer r.Close() if len(r.File) > 0 { rc, err := r.File[0].Open() if err != nil { log.Fatal(err) } var out bytes.Buffer if x.withBar { _, err = io.Copy(io.MultiWriter(&out, *x.bar), rc) } else { _, err = io.Copy(&out, rc) } if err != nil { log.Fatal(err) } rc.Close() done = x.writeUncompressedData(&out) } return done }