Initial orientator

internal/server/combined.go

@@ -0,0 +1,52 @@
+package server
+
+import (
+	"archive/zip"
+	"bytes"
+	"io"
+
+	"orientator/internal/pipeline"
+)
+
+type combinedZip struct {
+	results []*pipeline.Result
+	errors  map[string]error
+}
+
+func (c *combinedZip) add(r *pipeline.Result) {
+	c.results = append(c.results, r)
+}
+
+func (c *combinedZip) addError(name string, err error) {
+	if c.errors == nil {
+		c.errors = map[string]error{}
+	}
+	c.errors[name] = err
+}
+
+func (c *combinedZip) finalize() ([]byte, []pipeline.PageReport, error) {
+	var buf bytes.Buffer
+	zw := zip.NewWriter(&buf)
+	var allReports []pipeline.PageReport
+	for _, r := range c.results {
+		w, err := zw.Create(r.OutputName)
+		if err != nil {
+			return nil, nil, err
+		}
+		if _, err := io.Copy(w, bytes.NewReader(r.OutputBytes)); err != nil {
+			return nil, nil, err
+		}
+		allReports = append(allReports, r.Reports...)
+	}
+	for name, err := range c.errors {
+		w, e := zw.Create(name + ".error.txt")
+		if e != nil {
+			return nil, nil, e
+		}
+		_, _ = w.Write([]byte(err.Error()))
+	}
+	if err := zw.Close(); err != nil {
+		return nil, nil, err
+	}
+	return buf.Bytes(), allReports, nil
+}