wip: bad implementation of unzip

3 files changed, 36 insertions, 0 deletions

diff --git a/cmd/dbasik-api/return.go b/cmd/dbasik-api/return.go
index 0973fc9..f172452 100644
--- a/cmd/dbasik-api/return.go
+++ b/cmd/dbasik-api/return.go
@@ -1,6 +1,7 @@
 package main
 
 import (
+	"archive/zip"
 	"fmt"
 	"github.com/tealeg/xlsx/v3"
 	"path/filepath"
@@ -133,6 +134,29 @@ func (fp *DirectoryFilePackage) Prepare() ([]string, error) {
 	return files, nil
 }
 
+type ZipFilePackage struct {
+	FilePath string
+}
+
+func (fp *ZipFilePackage) Prepare() ([]string, error) {
+	// return a slice of the files from inside the zip file pointed to by fh.FilePath
+	// and an error if any
+	files, err := zip.OpenReader(fp.FilePath)
+	if err != nil {
+		return nil, err
+	}
+	defer files.Close()
+	out := []string{}
+	for _, file := range files.File {
+		out = append(out, file.Name)
+	}
+	return out, nil
+}
+
 func NewDirectoryFilePackage(filePath string) *DirectoryFilePackage {
 	return &DirectoryFilePackage{FilePath: filePath}
 }
+
+func NewZipFilePackage(filePath string) *ZipFilePackage {
+	return &ZipFilePackage{FilePath: filePath}
+}
diff --git a/cmd/dbasik-api/return_test.go b/cmd/dbasik-api/return_test.go
index bd2ffd9..d6c9daa 100644
--- a/cmd/dbasik-api/return_test.go
+++ b/cmd/dbasik-api/return_test.go
@@ -179,3 +179,15 @@ func TestPrepareFiles(t *testing.T) {
 		t.Errorf("Prepare() did not return ../../testdata/valid_excel.xlsx")
 	}
 }
+
+func TestUnzipFiles(t *testing.T) {
+	fp := NewZipFilePackage("../../testdata/test.zip")
+	files, err := fp.Prepare()
+	if err != nil {
+		t.Error(err)
+	}
+	t.Log(files)
+	if !slices.Contains(files, "valid_excel.xlsx") {
+		t.Errorf("Prepare() did not return test.xlsx")
+	}
+}
diff --git a/testdata/test.zip b/testdata/test.zip
new file mode 100644
index 0000000..b712934
--- /dev/null
+++ b/testdata/test.zip