exhibition-main/internal/model/ocr.go

package model

import (
	"bytes"
	"encoding/json"
	"errors"
	"fmt"
	"io"
	"io/ioutil"
	"net/http"
	"net/url"
	"strconv"
	"time"
	"unicode/utf8"
)

const (
	grant_type = "client_credentials"
)

var accessToken string
var expires_in uint64

type OcrQuery struct {
	IdCardUrl string `json:"idCardUrl"`
	Side      int    `json:"side"` // 1 正面   2 反面
}

type OcrRes struct {
	RealName       string `json:"realName"`
	IDNum          string `json:"iDNum"`
	Path           string `json:"path"`
	Age            int    `json:"age"`
	Birthday       string `json:"birthday"`
	Sex            string `json:"sex"`
	IssueDate      string `json:"issueDate"`
	ExpirationDate string `json:"expirationDate"`
}

const (
	ocr_client_id     = "cLg2dUH1pqopsj22tShw8FQU"
	ocr_client_secret = "diriz5PmhLOB8hwE4KnEEiaBMV6WfBR1"
)

type OrcRes struct {
	Path           string `json:"path"`
	Name           string `json:"name"`
	IdCard         string `json:"idCard"`
	Age            int    `json:"age"`
	Birthday       string `json:"birthday"`
	Sex            string `json:"sex"`
	IssueDate      string `json:"issueDate"`
	ExpirationDate string `json:"expirationDate"`
}

type OcrGetIdCardRes struct {
	WordsResult      WordsResult `json:"words_result"`
	IdcardNumberType int         `json:"idcard_number_type"`
	WordsResultNum   int         `json:"words_result_num"`
	ImageStatus      string      `json:"image_status"`
	LogId            uint64      `json:"log_id"`
}
type WordsResult struct {
	Name           WordsResultDetail `json:"姓名"`
	Address        WordsResultDetail `json:"住址"`
	IdCard         WordsResultDetail `json:"公民身份号码"`
	Birthday       WordsResultDetail `json:"出生"`
	Sex            WordsResultDetail `json:"性别"`
	IssueDate      WordsResultDetail `json:"签发日期"`
	ExpirationDate WordsResultDetail `json:"失效日期"`
}
type WordsResultDetail struct {
	Words string
}

type AccessToken struct {
	Refresh_token     string `json:"refresh_token"`
	Expires_in        uint64 `json:"expires_in"`
	Scope             string `json:"scope"`
	Session_key       string `json:"session_key"`
	Access_token      string `json:"access_token"`
	Session_secret    string `json:"session_secret"`
	Error             string `json:"error"`
	Error_description string `json:"error_description"`
}

func OcrGetIdCard(image string, side string) (*OrcRes, error) {
	var (
		accesstoken string
		response    string
		err         error
	)
	if accesstoken, err = GetOcrAccessToken(); err != nil {
		fmt.Println(err.Error())
		return nil, err
	}
	if response, err = PostForm("https://aip.baidubce.com/rest/2.0/ocr/v1/idcard?access_token=["+accesstoken+"]", url.Values{"image": {image}, "id_card_side": {side}}); err != nil {
		fmt.Println(err.Error())
		return nil, err
	}
	var res OcrGetIdCardRes
	if err = json.Unmarshal([]byte(response), &res); err != nil {
		return nil, err
	}
	fmt.Printf("%+v\n", res)

	switch res.ImageStatus {
	case "normal":

	case "reversed_side":
		return nil, errors.New("身份证正反面颠倒")
	case "non_idcard":
		return nil, errors.New("上传的图片中不包含身份证")
	case "blurred":
		return nil, errors.New("身份证模糊")
	case "other_type_card":
		return nil, errors.New("其他类型证照")
	case "over_exposure":
		return nil, errors.New("身份证关键字段反光或过曝")
	case "over_dark":
		return nil, errors.New("身份证欠曝（亮度过低）")
	case "unknown":
		return nil, errors.New("未知状态")
	default:
		return nil, errors.New("未知状态")
	}
	if side == "front" {
		switch res.IdcardNumberType {
		case -1:
			return nil, errors.New("身份证正面所有字段全为空")
		case 0:
			return nil, errors.New(" 身份证证号不合法，此情况下不返回身份证证号")
		case 1:
		case 2:
			return nil, errors.New("身份证证号和性别、出生信息都不一致")
		case 3:
			return nil, errors.New("身份证证号和出生信息不一致")
		case 4:
			return nil, errors.New("身份证证号和性别信息不一致")
		default:
			return nil, errors.New("未知状态")
		}
	}
	var result OrcRes
	fmt.Println(res.WordsResult.Name.Words)
	if side == "front" {
		result.Name = res.WordsResult.Name.Words
		result.Birthday = res.WordsResult.Birthday.Words
		result.IdCard = res.WordsResult.IdCard.Words
		result.Sex = res.WordsResult.Sex.Words
		result.Path = res.WordsResult.Address.Words
		var age int

		birYear, _ := strconv.Atoi(result.Birthday[0:4])
		birMonth, _ := strconv.Atoi(result.Birthday[4:6])
		fmt.Println(birYear)
		fmt.Println(time.Now().Year())
		age = time.Now().Year() - birYear

		if int(time.Now().Month()) < birMonth {
			age--
		}

		result.Age = age
	} else {
		issueDate := res.WordsResult.IssueDate.Words[:4] + "-" + res.WordsResult.IssueDate.Words[4:6] + "-" + res.WordsResult.IssueDate.Words[6:8]
		result.IssueDate = issueDate
		expirationDate := res.WordsResult.ExpirationDate.Words[:4] + "-" + res.WordsResult.ExpirationDate.Words[4:6] + "-" + res.WordsResult.ExpirationDate.Words[6:8]
		result.ExpirationDate = expirationDate
	}

	return &result, nil
}

func (m *OcrRes) CheckIdAndName() {
	if utf8.RuneCountInString(m.RealName) == 18 && utf8.RuneCountInString(m.IDNum) != 18 {
		m.IDNum, m.RealName = m.RealName, m.IDNum
	}
}

func GetOcrAccessToken() (string, error) {
	var (
		resObj AccessToken
		err    error
		// daoAccessToken entity.AccessToken
	)
	if expires_in == 0 || expires_in < uint64(time.Now().Unix()) {
		fmt.Println(1)
		// if daoAccessToken, err = dao.GetAccessToken("baidu", "accesstoken"); err != nil { //查询是否有记录
		// 	return "", err
		// }
		// fmt.Println(2)
		if resObj, err = getOcrAccessToken(); err != nil { //从链上获取
			return "", err
		}
		// var tmp = entity.AccessToken{
		// 	Platform:   "baidu",
		// 	Types:      "accesstoken",
		// 	Txt:        resObj.Access_token,
		// 	Expires_in: uint64(time.Now().Unix()) + resObj.Expires_in,
		// }
		// if daoAccessToken.Uid == 0 {
		// 	if err = dao.AddAccessToken(tmp); err != nil {
		// 		return "", err
		// 	}
		// } else {
		// 	if _, err = dao.UpdateAccessToken(tmp); err != nil {
		// 		return "", err
		// 	}
		// }
		accessToken = resObj.Access_token
		expires_in = resObj.Expires_in
	}
	return accessToken, nil

}

func getOcrAccessToken() (AccessToken, error) {
	var (
		resObj AccessToken
		err    error
	)
	url := "https://aip.baidubce.com/oauth/2.0/token"
	urlReq := "?grant_type=" + grant_type + "&client_id=" + ocr_client_id + "&client_secret=" + ocr_client_secret
	fmt.Println(urlReq)
	res := Get(url + urlReq)
	if err = json.Unmarshal([]byte(res), &resObj); err != nil {
		return resObj, err
	}
	if resObj.Error != "" {
		return resObj, errors.New(resObj.Error_description)
	}
	return resObj, err
}

func PostForm(urlStr string, data url.Values) (string, error) {
	resp, err := http.PostForm(urlStr, data)

	if err != nil {
		// handle error
	}
	defer resp.Body.Close()
	body, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		// handle error
	}
	return string(body), err
}

func Get(url string) string {

	// 超时时间：5秒
	client := &http.Client{Timeout: 5 * time.Second}
	resp, err := client.Get(url)
	if err != nil {
		panic(err)
	}
	defer resp.Body.Close()
	var buffer [512]byte
	result := bytes.NewBuffer(nil)
	for {
		n, err := resp.Body.Read(buffer[0:])
		result.Write(buffer[0:n])
		if err != nil && err == io.EOF {
			break
		} else if err != nil {
			panic(err)
		}
	}

	return result.String()
}