/*

Copyright 2017 Mikael Berthe

Licensed under the MIT license.  Please see the LICENSE file is this directory.

*/

package madon

import (

	"bytes"

	"encoding/json"

	"fmt"

	"io"

	"mime/multipart"

	"os"

	"path/filepath"

	"github.com/sendgrid/rest"

)

// UploadMedia uploads the given file and returns an attachment

func (mc *Client) UploadMedia(filePath string) (*Attachment, error) {

	var b bytes.Buffer

	if filePath == "" {

		return nil, ErrInvalidParameter

	}

	f, err := os.Open(filePath)

	if err != nil {

		return nil, fmt.Errorf("cannot read file: %s", err.Error())

	}

	defer f.Close()

	w := multipart.NewWriter(&b)

	formWriter, err := w.CreateFormFile("file", filepath.Base(filePath))

	if err != nil {

		return nil, fmt.Errorf("media upload: cannot create form: %s", err.Error())

	}

	if _, err = io.Copy(formWriter, f); err != nil {

		return nil, fmt.Errorf("media upload: cannot create form: %s", err.Error())

	}

	w.Close()

	req, err := mc.prepareRequest("media", rest.Post, nil)

	if err != nil {

		return nil, fmt.Errorf("media prepareRequest failed: %s", err.Error())

	}

	req.Headers["Content-Type"] = w.FormDataContentType()

	req.Body = b.Bytes()

	// Make API call

	r, err := restAPI(req)

	if err != nil {

		return nil, fmt.Errorf("media upload failed: %s", err.Error())

	}

	// Check for error reply

	var errorResult Error

	if err := json.Unmarshal([]byte(r.Body), &errorResult); err == nil {

		// The empty object is not an error

		if errorResult.Text != "" {

			return nil, fmt.Errorf("%s", errorResult.Text)

		}

	}

	// Not an error reply; let's unmarshal the data

	var attachment Attachment

	err = json.Unmarshal([]byte(r.Body), &attachment)

	if err != nil {

		return nil, fmt.Errorf("cannot decode API response (media): %s", err.Error())

	}

	return &attachment, nil

}