package go_aliyun_oss

import (
	"fmt"
	uuid "github.com/satori/go.uuid"
	"io/ioutil"
	"mime/multipart"
	"os"
	"path/filepath"
	"reflect"
	"strings"
)

type OssFile struct {
	FileByte []byte
	FileOldName string
	FileNewName string
	FileType string
	File	interface{}
}

type OssFileInterface interface {
	FileTypeTransForm() (*OssFile,error)
	GetFileType() *OssFile
}

//	@method: FileTypeTransForm
//	@description: 文件类型转换
//	@author: mr.x 2021-06-19 00:28:46
//	@return: *OssFile
//	@return: error
func (ossFile *OssFile) FileTypeTransForm() (*OssFile,error) {
	var err error

	switch ossFile.File.(type) {
	case *os.File:

		ossFile.FileByte,err = ioutil.ReadAll(ossFile.File.(*os.File))

		if err != nil {
			panic("read os type file failed:" + err.Error())
		}

		_,ossFile.FileOldName = filepath.Split(ossFile.File.(*os.File).Name())

		break

	case *multipart.FileHeader:

		fileResources,err := ossFile.File.(*multipart.FileHeader).Open()

		if err != nil {
			panic("open multipart file failed:" + err.Error())
		}

		defer fileResources.Close()

		ossFile.FileByte,err = ioutil.ReadAll(fileResources)

		if err != nil {
			panic("read multipart file failed:" + err.Error())
		}

		ossFile.FileOldName = ossFile.File.(*multipart.FileHeader).Filename

		break

	case string:
		newFile,err := os.Open(ossFile.File.(string))

		if err != nil {
			panic("open file path failed:" + err.Error())
		}

		defer newFile.Close()

		ossFile.FileByte,err = ioutil.ReadAll(newFile)

		_,ossFile.FileOldName = filepath.Split(newFile.Name())

		break

	// 支持[]byte数组传递 因为无法解析文件类型 默认直接给出文件类型为.png
	case []byte:
		ossFile.FileByte = ossFile.File.([]byte)

		//判断是否指定了文件的类型 如果没有指定默认为png格式
		if ossFile.FileType == "" || len(ossFile.FileType) <= 0 {
			ossFile.FileOldName = uuid.NewV4().String() + ".png"
		} else {
			ossFile.FileOldName = uuid.NewV4().String() + ossFile.FileType
		}

		break

	default:
		fmt.Println(reflect.TypeOf(ossFile.File))
		panic("file type is not support" )
	}

	ossFile.GetFileType()

	return ossFile,nil
}

//	@method: GetFileType
//	@description: 截取文件类型
//	@author: mr.x 2021-06-19 00:28:52
//	@return: *OssFile
func (ossFile *OssFile) GetFileType() *OssFile {

	// 当没有传递文件类型时去文件名中截取出文件类型
	if ossFile.FileType == "" || len(ossFile.FileType) <= 0 {
		//from oldFileName split file type
		fileTypeSufIndex := strings.Index(ossFile.FileOldName,".")

		fileType := ossFile.FileOldName[fileTypeSufIndex:]

		ossFile.FileType = fileType
	}

	//generate only file name
	ossFile.FileNewName = uuid.NewV5(uuid.NewV4(),ossFile.FileOldName).String() + ossFile.FileType

	return ossFile
}