package domainService

import (
	"fmt"
	pgTransaction "github.com/linmadan/egglib-go/transaction/pg"
	"gitlab.fjmaimaimai.com/allied-creation/character-library-metadata-bastion/pkg/domain"
	"gitlab.fjmaimaimai.com/allied-creation/character-library-metadata-bastion/pkg/domain/bytecore"
	"gitlab.fjmaimaimai.com/allied-creation/character-library-metadata-bastion/pkg/infrastructure/api/bytelib"
	"gitlab.fjmaimaimai.com/allied-creation/character-library-metadata-bastion/pkg/infrastructure/repository"
)

type PreviewDataTableService struct {
	FileId             int
	transactionContext *pgTransaction.TransactionContext
}

// Preview 预览  【data-table】
func (ptr *PreviewDataTableService) Preview(ctx *domain.Context, fileId int, fields []*domain.Field, where domain.Where) (interface{}, error) {
	fileRepository, _ := repository.NewFileRepository(ptr.transactionContext)
	file, err := fileRepository.FindOne(map[string]interface{}{"fileId": fileId})
	if err != nil {
		return nil, fmt.Errorf("文件不存在")
	}
	isSourceFile := false
	fileUrl := ""
	// Copy to TemporaryFile
	if file.FileType != domain.TemporaryFile.ToString() {
		file = file.CopyTo(domain.TemporaryFile, ctx)
		if file, err = fileRepository.Save(file); err != nil {
			return nil, err
		}
		isSourceFile = true
		fileUrl = file.FileInfo.Url
	}
	//TEST
	ptr.FileId = file.FileId

	// Load Data From Excel(python api)
	byteCore, _ := CreateByteCoreService()
	response, err := byteCore.LoadDataTable(bytecore.ReqLoadDataTable{
		FileId:   file.FileId,
		FileName: file.FileInfo.Name,
		Url:      file.FileInfo.Url,
		Ext:      file.FileInfo.Ext,
		Where:    where,

		OriginalTableId:     fmt.Sprintf("%v", file.FileId),
		IsFromOriginalTable: isSourceFile,
		TableFileUrl:        fileUrl,
		ColumnSchemas:       bytelib.DomainFieldsToColumnSchemas(fields),
		QueryParameters:     make(map[string]interface{}),
		SortParameters:      make(map[string]interface{}),
	})
	if err != nil {
		return nil, err
	}
	response.ObjectType = domain.ObjectFile
	response.FileId = file.FileId
	response.TableType = domain.ExcelTable.ToString()
	return response, nil
}

func (ptr *PreviewDataTableService) GetFileId() int {
	return ptr.FileId
}

func NewLoadDataTableService(transactionContext *pgTransaction.TransactionContext) (*PreviewDataTableService, error) {
	if transactionContext == nil {
		return nil, fmt.Errorf("transactionContext参数不能为nil")
	} else {
		return &PreviewDataTableService{
			transactionContext: transactionContext,
		}, nil
	}
}