package mego

import (
	"io"
	"mime/multipart"
	"os"
)

// UploadFile the uploaded file struct
type UploadFile struct {
	FileName string
	Size     int64
	Error    error
	File     multipart.File
	Header   *multipart.FileHeader
}

// Save save the posted file data as a file.
func (file *UploadFile) Save(path string) error {
	if file.Error != nil {
		return file.Error
	}
	f, err := os.OpenFile(path, os.O_RDWR|os.O_CREATE|os.O_EXCL|os.O_TRUNC, 0666)
	if err != nil {
		return err
	}
	defer f.Close()

	if file.File != nil {
		_, err := io.Copy(f, file.File)
		return err
	}
	return nil
}

// SaveAndClose save the posted file as a file and then Close the posted data stream
func (file *UploadFile) SaveAndClose(path string) error {
	err := file.Save(path)
	if file.File != nil {
		err1 := file.File.Close()
		if err == nil {
			err = err1
		}
	}
	return err
}