hongze_yb/services/share_poster.go

package services

import (
	"encoding/json"
	"errors"
	"fmt"
	"hongze/hongze_yb/global"
	"hongze/hongze_yb/utils"
	"io/ioutil"
	"net/http"
	"os"
	"strings"
)

// SharePosterReq 分享海报请求体
type SharePosterReq struct {
	Source string `json:"source" description:"来源"`
	Pars   string `json:"pars" description:"海报动态信息"`
	Scene  string `json:"scene" description:"太阳码scene"`
	Path   string `json:"path" description:"太阳码path"`
}

// CreatePosterFromSource 生成分享海报
func CreatePosterFromSource(source string, pars string) (imgUrl string, err error) {
	var errMsg string
	defer func() {
		if err != nil {
			if global.CONFIG.Serve.RunMode == "release" {
				go SendEmail(utils.APPNAME+"【"+global.CONFIG.Serve.RunMode+"】"+"失败提醒", "生成分享海报失败, "+errMsg+", CreatePosterFromSource: "+err.Error(), utils.EmailSendToUsers)
			}
		}
	}()

	// 根据请求路径及版本号获取本地图片, 无对应图片则生成图片

	// 根据来源生成相应的html文件, 确定生成图片的长宽
	resourceUrl := ""
	filePath := ""
	switch source {
	case "chart_detail":
		filePath, err = CreateChartDetailHtml(pars)
		if err != nil {
			return
		}
	}
	if filePath == "" {
		errMsg = "html资源生成失败"
		err = errors.New(errMsg)
		return
	}
	resourceUrl = "http://127.0.0.1:8612/" + filePath
	fmt.Println(resourceUrl)

	// 请求python服务htm2img
	htm2ImgReq := make(map[string]interface{})
	htm2ImgReq["html_url"] = resourceUrl
	res, err := postHtml2Img(htm2ImgReq)
	if err != nil {
		return
	}
	if res == nil {
		err = errors.New("html转图片请求失败, 未知错误")
		return
	}
	if res.Code != 200 {
		err = errors.New("html转图片失败: " + res.Msg)
		return
	}
	imgUrl = res.Data
	fmt.Println(imgUrl)
	// 清除本地html文件

	// 记录请求路径对应的图片信息

	return
}

type ChartDetailPosterReq struct {
	ChartInfoId int    `json:"chart_info_id"`
	ChartName   string `json:"chart_name"`
	ChartImage  string `json:"chart_image"`
}

// 生成html-图表详情页
func CreateChartDetailHtml(pars string) (filePath string, err error) {
	// 解析html填充的参数
	params := new(ChartDetailPosterReq)
	if err = json.Unmarshal([]byte(pars), &params); err != nil {
		return
	}

	// 获取模板内容
	demoPath := "static/htm2img/chart_detail.html"
	contentByte, err := ioutil.ReadFile(demoPath)
	if err != nil {
		return
	}

	// 获取动态太阳码

	// 替换模板内容
	contentStr := string(contentByte)
	contentStr = strings.ReplaceAll(contentStr, "{{CHART_NAME}}", params.ChartName)
	contentStr = strings.ReplaceAll(contentStr, "{{CHART_IMAGE}}", params.ChartImage)
	fmt.Println("ChartName:", params.ChartName)
	fmt.Println("ChartImage:", params.ChartImage)

	// 生成静态html文件
	filePath = "static/htm2img/" + utils.GetRandStringNoSpecialChar(28) + ".html"
	f, err := os.Create(filePath)
	defer f.Close()
	if err != nil {
		return
	}
	if _, err = f.Write([]byte(contentStr)); err != nil {
		return
	}

	return
}

// 生成静态html文件
func PutHtmlIntoFile(content []byte) (filePath string, err error) {
	filePath = "static/htm2img/" + utils.GetRandStringNoSpecialChar(28) + ".html"
	f, err := os.Create(filePath)
	defer f.Close()
	if err != nil {
		return
	}
	_, err = f.Write(content)
	return
}

type Html2ImgResp struct {
	Code int    `json:"code"`
	Msg  string `json:"msg"`
	Data string `json:"data"`
}

// postHtml2Img 请求htm2img接口
func postHtml2Img(param map[string]interface{}) (resp *Html2ImgResp, err error) {
	pythonUrl := ""
	if global.CONFIG.Serve.RunMode == "release" {
		//edbLibUrl = "http://172.19.173.232:8300/"
	} else {
		pythonUrl = "http://127.0.0.1:8888/"
	}
	postUrl := pythonUrl + "htm2img"
	postData, err := json.Marshal(param)
	if err != nil {
		return
	}
	result, err := Html2ImgHttpPost(postUrl, string(postData), "application/json")
	if err != nil {
		return
	}
	if err = json.Unmarshal(result, &resp); err != nil {
		return
	}
	return resp, nil
}

// Html2ImgHttpPost post请求edb库
func Html2ImgHttpPost(url, postData string, params ...string) ([]byte, error) {
	body := ioutil.NopCloser(strings.NewReader(postData))
	client := &http.Client{}
	req, err := http.NewRequest("POST", url, body)
	if err != nil {
		return nil, err
	}
	contentType := "application/x-www-form-urlencoded;charset=utf-8"
	if len(params) > 0 && params[0] != "" {
		contentType = params[0]
	}
	req.Header.Set("Content-Type", contentType)
	resp, err := client.Do(req)
	if err != nil {
		return nil, err
	}
	defer resp.Body.Close()
	b, err := ioutil.ReadAll(resp.Body)
	fmt.Println("HttpPost:" + string(b))
	return b, err
}