Aggiunta la funzione CreaZip.

diff --git a/zakegg.go b/zakegg.go
index 5a6803edfab95d1d2a90f13e2e9eb6d60ca2c3a2..b15c8c027d126035ad71a250110ed48c1b2eb2c1 100644 (file)
--- a/zakegg.go
+++ b/zakegg.go
@@ -3,6 +3,10 @@ package zakegg
 import (
        "fmt"
        "math"
+       "os"
+       "io"
+       "archive/zip"
+       "path/filepath"
 )
 
 func Paginazione(start int, rows int, rows_x_page int, url string) string {
@@ -111,3 +115,42 @@ func Paginazione(start int, rows int, rows_x_page int, url string) string {
 
        return ret
 }
+
+func CreaZip(filename_output string, filenames_input []string, only_basename bool) bool {
+       archive, err := os.Create(filename_output)
+       if err != nil {
+               fmt.Println(err)
+               return false
+       }
+       defer archive.Close()
+
+       zipWriter := zip.NewWriter(archive)
+
+       for _, fn := range filenames_input {
+               f1, err := os.Open(fn)
+               if err != nil {
+                       fmt.Println(err)
+                       continue
+               }
+
+               var fname string
+               if only_basename {
+                       fname = filepath.Base(fn)
+               } else {
+                       fname = fn
+               }
+               w1, err := zipWriter.Create(fname)
+               if err != nil {
+                       fmt.Println(err)
+                       continue
+               }
+               if _, err := io.Copy(w1, f1); err != nil {
+                       fmt.Println(err)
+               }
+
+               f1.Close()
+       }
+
+       zipWriter.Close()
+       return true
+}

diff --git a/zakegg_test.go b/zakegg_test.go
new file mode 100644 (file)
index 0000000..1d4470b
--- /dev/null
+++ b/zakegg_test.go
@@ -0,0 +1,16 @@
+package zakegg
+
+import (
+       "testing"
+)
+
+func TestCreaZip(t *testing.T) {
+       files := []string{"zakegg.go", "go.mod"}
+
+       ret := CreaZip("test.zip", files)
+
+       if !ret {
+               t.Log("File zip non creato correttamente")
+               t.Fail()
+       }
+}