multipartzip.go

multipartzip.go

// Copyright 2014 gandalf authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package multipartzip

import (
	"archive/zip"
	"bytes"
	"io"
	"log"
	"mime/multipart"
	"path/filepath"
)

func CreateZipBuffer(files []struct{ Name, Body string }) (*bytes.Buffer, error) {
	buf := new(bytes.Buffer)
	w := zip.NewWriter(buf)
	for _, file := range files {
		f, err := w.Create(file.Name)
		if err != nil {
			return nil, err
		}
		_, err = f.Write([]byte(file.Body))
		if err != nil {
			return nil, err
		}
	}
	err := w.Close()
	if err != nil {
		return nil, err
	}
	return buf, nil
}

func StreamWriteBuffer(params map[string]string, paramName, path, boundary string, writer *io.PipeWriter, buf *bytes.Buffer) {
	defer writer.Close()
	mpwriter := multipart.NewWriter(writer)
	mpwriter.SetBoundary(boundary)
	part, err := mpwriter.CreateFormFile(paramName, filepath.Base(path))
	if err != nil {
		log.Fatal(err)
		return
	}
	_, err = io.Copy(part, buf)
	if err != nil {
		log.Fatal(err)
		return
	}
	for key, val := range params {
		_ = mpwriter.WriteField(key, val)
	}
	err = mpwriter.Close()
	if err != nil {
		log.Fatal(err)
		return
	}
}