fonchain-artistinfo/cmd/internal/dao/contract.go

package dao

import (
	"errors"
	"time"

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

func FinishContract(id string) (err error) {
	var contracts model.Contract
	if err = db.DB.Where("transaction_id = ?", id).First(&contracts).Error; err != nil {
		zap.L().Error("get contract info err", zap.Error(err))
		err = errors.New(m.ERROR_SELECT)
		return
	}

	if err = db.DB.Model(&contracts).Update("state", 3).Error; err != nil {
		zap.L().Error("user contract failed", zap.Error(err))
		err = errors.New(m.UPDATE_FAILED)
		return
	}

	return
}

func ContractList(req *contract.ContractListRequest) (rep *contract.ContractListRespond, err error) {
	rep = &contract.ContractListRespond{}
	var Data []*contract.ContractData

	var user model.User
	if err = db.DB.Where("id = ? ", int32(req.ID)).First(&user).Error; err != nil {
		zap.L().Error("get user info err", zap.Error(err))
		err = errors.New(m.ERROR_SELECT)
		return
	}

	// todo
	//var realName model.RealName
	//if err = db.DB.Where("id = ? ", user.RealNameID).First(&realName).Error; err != nil {
	//	zap.L().Error("get realName info err", zap.Error(err))
	//	err = errors.New(m.ERROR_SELECT)
	//	return
	//}

	var args []interface{}
	var sqlWhere = "user_id = ?  and type !=4 "
	args = append(args, req.ID)

	if req.State != 0 {
		sqlWhere += " and state = ? "
		if req.State == 2 {
			req.State = 3
		}
		args = append(args, req.State)
	}

	var contractModel []model.Contract
	if err = db.DB.Where(sqlWhere, args...).Find(&contractModel).Error; err != nil {
		zap.L().Error("get contractModels info err", zap.Error(err))
		err = errors.New(m.ERROR_SELECT)
		return
	}

	for _, v := range contractModel {
		contract := &contract.ContractData{}
		contract.ID = uint64(v.ID)
		contract.ContractId = v.ContractId
		contract.DownloadUrl = v.DownloadUrl
		contract.MgmtUserId = v.MgmtUserId
		contract.State = int64(v.State)
		contract.TransactionId = v.TransactionId
		contract.Type = int64(v.Type)
		contract.UserId = int64(v.UserId)
		contract.ViewUrl = v.ViewUrl
		contract.ExpirationTime = v.CreatedAt.Add(86400 * time.Second).Format("2006-01-02 15:04:05")
		contract.SignTime = v.UpdatedAt.Format("2006-01-02")
		Data = append(Data, contract)
	}

	rep.Data = Data

	return
}

func ContractTxList(req *contract.ContractTxListRequest) (rep *contract.ContractTxListRespond, err error) {
	rep = &contract.ContractTxListRespond{}
	var Data []*contract.ContractData

	var user model.User
	if err = db.DB.Where("id = ? ", int32(req.ID)).First(&user).Error; err != nil {
		zap.L().Error("get user info err", zap.Error(err))
		err = errors.New(m.ERROR_SELECT)
		return
	}
	// todo
	//var realName model.RealName
	//if err = db.DB.Where("id = ? ", user.RealNameID).First(&realName).Error; err != nil {
	//	zap.L().Error("get realName info err", zap.Error(err))
	//	err = errors.New(m.ERROR_SELECT)
	//	return
	//}
	//
	//var args []interface{}
	//var sqlWhere = "  card_id = ? and (type = 4 or type = 7) "
	//args = append(args, realName.IDNum)
	//if req.State != 0 {
	//	sqlWhere += " and state = ? "
	//	args = append(args, req.State)
	//}
	//
	//var contractModel []model.Contract
	//if err = db.DB.Where(sqlWhere, args...).Find(&contractModel).Error; err != nil {
	//	zap.L().Error("get contractModels info err", zap.Error(err))
	//	err = errors.New(m.ERROR_SELECT)
	//	return
	//}
	//
	//for _, v := range contractModel {
	//	contractTmp := &contract.ContractData{}
	//	contractTmp.ID = uint64(v.ID)
	//	contractTmp.ContractId = v.ContractId
	//	contractTmp.DownloadUrl = v.DownloadUrl
	//	contractTmp.MgmtUserId = v.MgmtUserId
	//	contractTmp.State = int64(v.State)
	//	contractTmp.TransactionId = v.TransactionId
	//	contractTmp.Type = int64(v.Type)
	//	var b string
	//	if v.Type == 4 {
	//		b = "物权"
	//	}
	//	if v.Type == 7 {
	//		b = "版权"
	//	}
	//	contractTmp.BatchName = b + v.BatchName[0:4] + "年" + v.BatchName[4:6] + "月" + v.BatchName[6:11] + "年" + v.BatchName[11:13] + "月"
	//	contractTmp.UserId = int64(v.UserId)
	//	contractTmp.ViewUrl = v.ViewUrl
	//	contractTmp.ExpirationTime = v.CreatedAt.Add(86400 * time.Second).Format("2006-01-02 15:04:05")
	//	contractTmp.SignTime = v.UpdatedAt.Format("2006-01-02")
	//	Data = append(Data, contractTmp)
	//}

	rep.Data = Data

	return
}

func GetContract(id int32) (rep *contract.ContractData, err error) {
	var con model.Contract
	if err = db.DB.Where("id = ? ", id).First(&con).Error; err != nil {
		zap.L().Error("get contract info err", zap.Error(err))
		err = errors.New(m.ERROR_SELECT)
		return
	}
	rep.ID = uint64(con.ID)
	rep.UserId = int64(con.UserId)
	rep.CardId = con.CardId
	rep.MgmtUserId = con.MgmtUserId
	rep.ArtworkId = con.ArtworkId
	rep.ContractId = con.ContractId
	rep.TransactionId = con.TransactionId
	rep.Type = int64(con.Type)
	rep.BatchName = con.BatchName
	rep.ViewUrl = con.ViewUrl
	rep.DownloadUrl = con.DownloadUrl
	rep.State = int64(con.State)

	return
}

// 更新交易id
func UpdateContract(req *contract.UpdateContractRequest) error {
	//数据库操作异常
	var con model.Contract
	if err := db.DB.Model(&con).Updates(&model.Contract{ContractId: req.ContractId, ViewUrl: req.ViewUrl, DownloadUrl: req.DownloadUrl}).Error; err != nil {
		return err
	}
	return nil
}

// 更新交易id
func UpdateContractTx(txId string, contractId int32) (err error) {
	var con model.Contract
	con.ID = contractId
	if err = db.DB.Model(&con).Update("transaction_id", txId).Error; err != nil {
		return
	}
	if err = db.DB.Model(&con).Update("state", 1).Error; err != nil {
		return
	}
	return
}