gotenberg/app/converter/converter.go

// Package converter implements a solution for converting one or more files to PDF.
package converter

import (
	"fmt"
	"net/http"
	"os"

	gfile "github.com/gulien/gotenberg/app/converter/file"
	"github.com/gulien/gotenberg/app/converter/process"
	ghttp "github.com/gulien/gotenberg/app/http"

	"github.com/satori/go.uuid"
)

// Converter handles conversion into PDF of files coming from a request.
type Converter struct {
	files      []*gfile.File
	workingDir string
}

// NoFileToConvertError is raised when the converter has no file
// to convert.
type NoFileToConvertError struct{}

func (e *NoFileToConvertError) Error() string {
	return "There is no file to convert"
}

// FilesKeyNotFoundError is raised when "files" key does not exist
// in the form data
type FilesKeyNotFoundError struct{}

func (e *FilesKeyNotFoundError) Error() string {
	return "\"files\" key was not found in the form data"
}

// NewConverter instantiates a converter by parsing a request.
func NewConverter(r *http.Request, contentType ghttp.ContentType) (*Converter, error) {
	c := &Converter{
		workingDir: fmt.Sprintf("./%s/", uuid.NewV4().String()),
	}

	if err := os.Mkdir(c.workingDir, 0666); err != nil {
		return nil, err
	}

	switch contentType {
	case ghttp.MultipartFormDataContentType:
		err := r.ParseMultipartForm(32 << 20)
		if err != nil {
			return nil, err
		}

		formData := r.MultipartForm
		files, ok := formData.File["files"]
		if !ok {
			return nil, &FilesKeyNotFoundError{}
		}

		for i := range files {
			file, err := files[i].Open()
			if err != nil {
				return nil, err
			}

			defer file.Close()

			f, err := gfile.NewFile(c.workingDir, file)
			if err != nil {
				return nil, err
			}

			c.files = append(c.files, f)
		}
		break
	default:
		f, err := gfile.NewFile(c.workingDir, r.Body)
		if err != nil {
			return nil, err
		}

		c.files = append(c.files, f)
	}

	if len(c.files) == 0 {
		return nil, &NoFileToConvertError{}
	}

	return c, nil
}

// Convert converts its associated files to PDF. If more than one file,
// it will merge all of them into one unique PDF file.
// Returns the new file path or an error if something bad happened.
func (c *Converter) Convert() (string, error) {
	var filesPaths []string
	for _, f := range c.files {
		if f.Type != gfile.PDFType {
			path, err := process.Unconv(c.workingDir, f)
			if err != nil {
				return "", err
			}

			filesPaths = append(filesPaths, path)
		} else {
			filesPaths = append(filesPaths, f.Path)
		}
	}

	if len(filesPaths) == 1 {
		return filesPaths[0], nil
	}

	path, err := process.Merge(c.workingDir, filesPaths)
	if err != nil {
		return "", err
	}

	return path, nil
}

// Clear removes all file inside its working directory.
func (c *Converter) Clear() error {
	return os.RemoveAll(c.workingDir)
}