Reorganization.

internal/compression/types.go

@@ -0,0 +1,13 @@
+package compression
+
+import "io"
+
+//Compressor is a squashfs decompressor interface. Allows for easy compression.
+type Compressor interface {
+	Compress(io.Reader) ([]byte, error)
+}
+
+//Decompressor is a squashfs decompressor interface. Allows for easy decompression no matter the type of compression.
+type Decompressor interface {
+	Decompress(io.Reader) ([]byte, error)
+}

internal/compression/zlib.go

@@ -0,0 +1,24 @@
+package compression
+
+import (
+	"bytes"
+	"compress/zlib"
+	"io"
+)
+
+//Zlib is a decompressor for gzip type compression
+type Zlib struct{}
+
+//Decompress reads the entirety of the given reader and returns it uncompressed as a byte slice.
+func (z *Zlib) Decompress(r io.Reader) ([]byte, error) {
+	rdr, err := zlib.NewReader(r)
+	if err != nil {
+		return nil, err
+	}
+	var data bytes.Buffer
+	_, err = io.Copy(&data, rdr)
+	if err != nil {
+		return nil, err
+	}
+	return data.Bytes(), nil
+}