package wechatminiprogram

import (
	"gopkg.in/dtapps/go-library.v3/utils/gohttp"
	"os"
)

type SaveImgResponse struct {
	Path string
	Name string
}

func (app *App) SaveImg(resp gohttp.Response, dir, saveName string) SaveImgResponse {
	// 返回是二进制图片，或者json错误
	if resp.Header.Get("Content-Type") == "image/jpeg" || resp.Header.Get("Content-Type") == "image/png" {
		// 保存在output目录
		outputFileName := saveName

		if resp.Header.Get("Content-Type") == "image/jpeg" {
			outputFileName = outputFileName + ".jpg"
		} else {
			outputFileName = outputFileName + ".png"
		}
	here:
		f, err := os.OpenFile(dir+outputFileName, os.O_CREATE|os.O_RDWR, 0666)
		if err != nil {
			os.Mkdir(dir, 0666)
			goto here
		}
		f.Write(resp.Body)
		f.Close()
		return SaveImgResponse{
			Path: dir + outputFileName,
			Name: outputFileName,
		}
	}
	return SaveImgResponse{}
}