package dao

import (
	"errors"

	"github.com/fonchain/fonchain-artistinfo/cmd/model"
	statement "github.com/fonchain/fonchain-artistinfo/pb/artistinfoStatement"
	db "github.com/fonchain/fonchain-artistinfo/pkg/db"
	"github.com/fonchain/fonchain-artistinfo/pkg/m"
	"go.uber.org/zap"
)

func StatementList(artistUid string, state int32) (contractData []*statement.Contracts, err error) {

	var contracts []model.Contract
	if err = db.DB.Where("artist_uid = ? AND  state = ?", artistUid, state).Where("type == 4 OR type == 7").Find(&contracts).Error; err != nil {
		zap.L().Error("get contracts info err", zap.Error(err))
		err = errors.New(m.ERROR_SELECT)
		return
	}

	for _, v := range contracts {
		Contract := &statement.Contracts{
			ContractUid:   v.Uid,
			ArtistUid:     v.ArtistUid,
			ArtworkUid:    v.ArtworkUid,
			ContractId:    v.ContractId,
			TransactionId: v.TransactionId,
			Type:          v.Type,
			ViewUrl:       v.ViewUrl,
			DownloadUrl:   v.DownloadUrl,
			State:         v.State,
			Status:        int32(v.Status),
			LockTime:      v.LockTime,
			SignTime:      v.SignTime,
			BatchTime:     v.BatchTime,
			BatchUid:      v.BatchUid,
		}

		contractData = append(contractData, Contract)
	}

	return
}

func GetTxArtistNameByBatchUid(batchUid string) (artworkTx model.ArtworkTx, err error) {
	if err = db.DB.Where("uid = ?", batchUid).Find(&artworkTx).Error; err != nil {
		zap.L().Error("get artworkTx info err", zap.Error(err))
		err = errors.New(m.ERROR_SELECT)
		return
	}

	return
}

func GetArtworkTxDetailByBatchUid(batchUid string) (artworkTxDetail []model.ArtworkTxDetail, err error) {
	if err = db.DB.Where("batch_uid = ?", batchUid).Find(&artworkTxDetail).Error; err != nil {
		zap.L().Error("get artworkTxDetail infos err", zap.Error(err))
		err = errors.New(m.ERROR_SELECT)
		return
	}

	return
}

func GetArtworkSoldTxDetailByBatchUid(batchUid string) (artworkSoldTxDetail []model.ArtworkSoldTxDetail, err error) {
	if err = db.DB.Where("batch_uid = ?", batchUid).Find(&artworkSoldTxDetail).Error; err != nil {
		zap.L().Error("get artworkSoldTxDetail infos err", zap.Error(err))
		err = errors.New(m.ERROR_SELECT)
		return
	}

	return
}

func GetCopyArtistNameByBatchUid(batchUid string) (artworkCopy model.ArtworkCopy, err error) {
	if err = db.DB.Where("uid = ?", batchUid).Find(&artworkCopy).Error; err != nil {
		zap.L().Error("get artworkCopy info err", zap.Error(err))
		err = errors.New(m.ERROR_SELECT)
		return
	}

	return
}

func GetArtworkCopyDetailByBatchUid(batchUid string) (artworkCopyDetail []model.ArtworkCopyDetail, err error) {
	if err = db.DB.Where("batch_uid = ?", batchUid).Find(&artworkCopyDetail).Error; err != nil {
		zap.L().Error("get artworkCopyDetail infos err", zap.Error(err))
		err = errors.New(m.ERROR_SELECT)
		return
	}

	return
}

func GetArtworkSoldCopyDetailByBatchUid(batchUid string) (artworkSoldCopyDetail []model.ArtworkSoldCopyDetail, err error) {
	if err = db.DB.Where("batch_uid = ?", batchUid).Find(&artworkSoldCopyDetail).Error; err != nil {
		zap.L().Error("get artworkSoldCopyDetail infos err", zap.Error(err))
		err = errors.New(m.ERROR_SELECT)
		return
	}

	return
}