package aliyun

import (
	"encoding/json"
	"fmt"
	"github.com/alibabacloud-go/docmind-api-20220711/client"
	aliyunSdk "github.com/fonchain_enterprise/fonchain-main/pkg/aliyun"
	"github.com/fonchain_enterprise/fonchain-main/pkg/e"
	"gopkg.in/errgo.v2/fmt/errors"
	"path/filepath"
)

func DocStruct(fileUrl, fileName string) (docId string, err error) {
	if fileName == "" {
		fileName = filepath.Base(fileUrl)
	}
	request := client.SubmitDocStructureJobRequest{
		FileUrl:  &fileUrl,
		FileName: &fileName,
	}
	response, err := aliyunSdk.DocClient.SubmitDocStructureJob(&request)
	if err != nil {
		err = errors.New(e.GetMsg(e.ERROR_ALIYUN_DOC_SUBMIT))
		return
	}
	httpOk := response.StatusCode
	if *httpOk != 200 {
		err = errors.New(e.GetMsg(e.ERROR_ALIYUN_DOC_SUBMIT))
		return
	}
	docId = *response.Body.Data.Id
	fmt.Println(response.Body.Data.Id)
	fmt.Println(docId)
	return

}

func DocResult(docId string) (complete bool, texts []string, err error) {
	// 调用查询接口
	request := client.GetDocStructureResultRequest{Id: &docId}
	response, err := aliyunSdk.DocClient.GetDocStructureResult(&request)
	if err != nil {
		err = errors.New(e.GetMsg(e.ERROR_ALIYUN_DOC_PARSE))
		return
	}
	ok := response.Body.Completed
	if *ok != true {
		return
	}
	complete = true
	var layoutsSlice []map[string]interface{}
	layouts := response.Body.Data["layouts"]
	layoutsBytes, _ := json.Marshal(layouts)
	_ = json.Unmarshal(layoutsBytes, &layoutsSlice)
	for _, v := range layoutsSlice {
		texts = append(texts, fmt.Sprint(v["text"]))
	}
	return
}