package ai_model_cli

import (
	"goskeleton/app/global/my_errors"
	"goskeleton/app/global/variable"
	"goskeleton/app/utils/baidubce"
	"net/http"
	"net/url"
	"strings"

	"github.com/gin-gonic/gin"
)

/**
 * 向百度智能云ocr(基础精准版)发送消息, 获得识别消息
 * @return 识别出的信息的内容
 */
func RequestOCR(context *gin.Context) (r bool, c interface{}) {
	path := "https://aip.baidubce.com/rest/2.0/ocr/v1/accurate_basic?access_token=" + baidubce.GetAccessToken()
	// 获得图片数据，需要 base64 和urlencode处理
	// TODO:linlnf
	content := context.PostForm("pic")

	// 组装 body 数据，参数在该网址：https://cloud.baidu.com/doc/OCR/s/1k3h7y3db
	data := make(url.Values)
	data.Set("detect_direction", "false")
	data.Set("paragraph", "false")
	data.Set("probability", "false")
	data.Set("multidirectional_recognize", "false")
	data.Set("image", content)
	payload := strings.NewReader(data.Encode())
	client := &http.Client{}
	// 请求数据
	req, err := http.NewRequest("POST", path, payload)
	if err != nil {
		variable.ZapLog.Error(my_errors.ErrorBaiduCEUseOCRFail + err.Error())
		return false, nil
	}
	req.Header.Add("Content-Type", "application/x-www-form-urlencoded")
	req.Header.Add("Accept", "application/json")
	res, err := client.Do(req)

	if err != nil {
		variable.ZapLog.Error(my_errors.ErrorBaiduCEUseOCRFail + err.Error())
		return false, nil
	}
	defer res.Body.Close()
	//从res中提取识别出的字符信息
	mbody, err := baidubce.DecodeResBody(res, "ocr")
	if err != nil {
		variable.ZapLog.Error(my_errors.ErrorBaiduCEUseOCRFail + err.Error())
		return false, nil
	}
	return true, gin.H{
		"words": decodeOCRBody2Str(mbody),
	}
}

/*
 * 将返回的内容中的所有识别出的文字信息进行组合
 * @return 识别出的信息的内容字符串
 */
func decodeOCRBody2Str(mbody map[string]interface{}) (str string) {
	words, ok := mbody["words_result"]
	if ok {
		sliceWords, ok := words.([]interface{})
		if ok {
			// 遍历切片
			for _, word := range sliceWords {
				str += word.(map[string]interface{})["words"].(string) + "\n"
			}
		}
		return str
	}
	return ""
}