core/sys/xtract.go

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
}