package domain_service

import (
	"fmt"
	"github.com/go-pg/pg/v10"
	pgTransaction "github.com/linmadan/egglib-go/transaction/pg"
	"github.com/linmadan/egglib-go/utils/tool_funs"
	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-cooperation/pkg/domain"
	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-cooperation/pkg/infrastructure/dao"
	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-cooperation/pkg/infrastructure/repository"
	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-cooperation/pkg/infrastructure/utils"
	"strconv"
	"time"
)

const (
	// 合约分红列表查询
	SearchContractDividends = "SearchContractDividends"
	// 获取分红合约详情
	GetContractDividends = "GetContractDividends"
	// 分红统计(分红明细)
	DividendsStatistics = "DividendsStatistics"

	// 企业-商品统计
	CooperationGoodsStatistics = "cooperationGoodsStatistics"
	// 企业-共创模式统计
	CooperationModeStatistics = "CooperationModeStatistics"
	// 企业-分红统计
	CompanyDividendsStatistics = "CompanyDividendsStatistics"
	// 企业、个人 - 分红预算列表
	SearchDividendsEstimates = "SearchDividendsEstimates"
	// 公司 - 共创用户统计
	CompanyCooperationUsersStatistics = "CompanyCooperationUsersStatistics"
	//公司 - 共创用户模式统计
	CooperationUserModeStatistics = "CooperationUserModeStatistics"
	// 公司 - 共创用户分红支付统计
	CompanyPaymentHistoryStatistics = "CompanyPaymentHistoryStatistics"

	// 个人 - 共创企业统计
	CooperationCompanyStatistics = "CooperationCompanyStatistics"
	// 个人 - 用户合约统计
	PersonCooperationContractStatistics = "PersonCooperationContractStatistics"
	// 个人 - 企业支付统计
	PersonCompanyPaymentHistoryStatistics = "PersonCompanyPaymentHistoryStatistics"

	// 账期结算单统计
	CreditAccountStatistics = "CreditAccountStatistics"
	// 公司/个人 - 支付历史统计直方图
	PaymentHistoryHistogramStatistics = "PaymentHistoryHistogramStatistics"
	// 用户相关的项目合约编号列表查询
	RelevantCooperationContractNumbers = "RelevantCooperationContractNumbers"
)

//  CooperationStatisticsService 共创统计服务
type CooperationStatisticsService struct {
	transactionContext *pgTransaction.TransactionContext
}

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

/*****   1.合约分红模块  *****/

/*1.1 分红合约搜索*/

// 分红合约搜索
//
// queryOptions 查询参数
func (ptr *CooperationStatisticsService) SearchContractDividends(queryOptions map[string]interface{}) (interface{}, error) {
	var request = struct {
		//企业
		CompanyId int64 `json:"companyId"`
		OrgId     int64 `json:"orgId"`
		UserId    int64 `json:"userId"`
		//个人
		UserBaseId int64 `json:"userBaseId"`
		Offset     int   `json:"offset"`
		Limit      int   `json:"limit"`
	}{}
	if err := LoadQueryObject(queryOptions, &request); err != nil {
		return nil, err
	}
	queryOptions = tool_funs.SimpleStructToMap(&request)

	// 1.根据个人、企业查询合约列表
	var contracts []*domain.CooperationContract
	var err error
	if request.UserBaseId > 0 {
		contracts, err = ptr.getUserContracts(request.UserBaseId, queryOptions)
	} else if request.OrgId > 0 {
		queryOptions["orgId"] = request.OrgId
		contracts, err = ptr.getCompanyContracts(request.UserId, queryOptions)
	}
	if err != nil {
		return nil, err
	}

	// 2.根据合约查询分红预算
	var numbers []string
	var results = make([]*searchContractDividendsResult, 0)
	for i := range contracts {
		item := contracts[i]
		resultItem := &searchContractDividendsResult{
			CooperationContractId:     item.CooperationContractId,
			CooperationContractName:   item.CooperationContractName,
			CooperationContractNumber: item.CooperationContractNumber,
			Status:                    item.Status,
			CreatedAt:                 item.CreatedAt.Unix() * 1000,
		}
		results = append(results, resultItem)
		numbers = append(numbers, item.CooperationContractNumber)
	}
	mapCreditAccount, err := ptr.getContractsCreditAccount(numbers)
	if err != nil {
		return nil, err
	}
	for i := range results {
		if v, ok := mapCreditAccount[results[i].CooperationContractNumber]; ok {
			results[i].DividendsAmount = v.SettlementAmount
			results[i].DividendsOrderAmount = v.GoodAmountCount
		}
	}
	return results, nil
}

// getUserContracts  获取用户的合约列表
//
// p1 p1_desc
func (ptr *CooperationStatisticsService) getUserContracts(userBaseId int64, queryOptions map[string]interface{}) ([]*domain.CooperationContract, error) {
	var contracts = make([]*domain.CooperationContract, 0)
	if userBaseId == 0 {
		return contracts, nil
	}
	contractNumbers, err := ptr.getRelevantContracts(map[string]interface{}{"userBaseId": userBaseId})
	if len(contractNumbers) == 0 {
		return contracts, nil
	}
	if err != nil {
		return contracts, err
	}
	queryOptions["cooperationContractNumbers"] = contractNumbers
	contractRepository, _ := dao.NewCooperationContractDao(ptr.transactionContext)
	_, contracts, err = contractRepository.Find(queryOptions)
	return contracts, err
}

// getCompanyContracts  获取组织合约列表
//
// p1 p1_desc
func (ptr *CooperationStatisticsService) getCompanyContracts(userId int64, queryOptions map[string]interface{}) ([]*domain.CooperationContract, error) {
	var contracts = make([]*domain.CooperationContract, 0)
	if userId == 0 {
		return contracts, nil
	}
	contractNumbers, err := ptr.getRelevantContracts(map[string]interface{}{"userId": userId})
	if len(contractNumbers) == 0 {
		return contracts, nil
	}
	if err != nil {
		return contracts, err
	}
	queryOptions["cooperationContractNumbers"] = contractNumbers
	contractRepository, _ := dao.NewCooperationContractDao(ptr.transactionContext)
	_, contracts, err = contractRepository.Find(queryOptions)
	return contracts, err
}

//getRelevantContracts  获取相关人的合约
func (ptr *CooperationStatisticsService) getRelevantContracts(queryOptions map[string]interface{}) ([]string, error) {
	undertakerRepository, _ := repository.NewCooperationContractRelevantRepository(ptr.transactionContext)
	_, undertakers, err := undertakerRepository.Find(queryOptions)
	var numbers []string
	for i := range undertakers {
		numbers = append(numbers, undertakers[i].CooperationContractNumber)
	}
	if len(numbers) == 0 {
		return []string{}, nil
	}
	return numbers, err
}

// getContractsDividendsEstimate 合约分红预算
func (ptr *CooperationStatisticsService) getContractsDividendsEstimate(numbers []string) (map[string]*domain.DividendsEstimate, error) {
	var estimates []*domain.DividendsEstimate
	var resMap = make(map[string]*domain.DividendsEstimate)
	if len(numbers) == 0 {
		return resMap, nil
	}
	_, err := ptr.transactionContext.PgDd.Query(&estimates, `select cooperation_contract_number,sum(dividends_amount) dividends_amount from dividends_estimates 
where cooperation_contract_number in (?)
group by cooperation_contract_number
`, pg.In(numbers))
	if err != nil {
		return nil, err
	}

	for i := range estimates {
		resMap[estimates[i].CooperationContractNumber] = estimates[i]
	}
	return resMap, nil
}

// getContractsDividendsEstimate 合约分红预算
func (ptr *CooperationStatisticsService) getContractsCreditAccount(numbers []string) (map[string]*domain.CreditAccount, error) {
	var creditAccounts []*domain.CreditAccount
	var resMap = make(map[string]*domain.CreditAccount)
	if len(numbers) == 0 {
		return resMap, nil
	}
	var creditAccount = new(domain.CreditAccount)

	query := ptr.transactionContext.PgDd.Model(creditAccount)
	query.Column("cooperation_contract_number")
	query.ColumnExpr("sum(good_amount_count) good_amount_count")
	query.ColumnExpr("sum(settlement_amount) settlement_amount")
	query.Where("cooperation_contract_number in (?)", pg.In(numbers))
	query.Where("deleted_at is null")
	query.Group("cooperation_contract_number")
	err := query.Select(&creditAccounts)
	if err != nil {
		return nil, err
	}

	for i := range creditAccounts {
		resMap[creditAccounts[i].CooperationContractNumber] = creditAccounts[i]
	}
	return resMap, nil
}

type searchContractDividendsResult struct {
	// 共创合约ID
	CooperationContractId int64 `json:"cooperationContractId,string"`
	// 共创合约名称
	CooperationContractName string `json:"cooperationContractName"`
	// 共创合约编号
	CooperationContractNumber string `json:"cooperationContractNumber"`
	// 合约状态，1恢复，2暂停
	Status int32 `json:"cooperationContractStatus"`
	// 分红预算
	DividendsAmount float64 `json:"dividendsAmount"`
	// 分红订单金额
	DividendsOrderAmount float64 `json:"dividendsOrderAmount"`
	// 创建时间
	CreatedAt int64 `json:"createdAt"`
}

/*1.2 分红合约详情*/

//GetContractDividends 分红合约详情
//
// queryOptions 查询参数
func (ptr *CooperationStatisticsService) GetContractDividends(queryOptions map[string]interface{}) (interface{}, error) {
	// 1.合约详情
	contractRepository, _ := repository.NewCooperationContractRepository(ptr.transactionContext)
	if _, ok := queryOptions["contractId"]; !ok {
		return nil, fmt.Errorf("合约ID(contractId)不能为空")
	}
	var res = make(map[string]interface{})
	contract, err := contractRepository.FindOne(map[string]interface{}{"cooperationContractId": queryOptions["contractId"]})
	if err != nil {
		return nil, err
	}
	res["cooperationContract"] = map[string]interface{}{
		"cooperationContractDescription": contract.CooperationContractDescription,
		"cooperationContractId":          contract.CooperationContractId,
		"cooperationContractName":        contract.CooperationContractName,
		"cooperationContractNumber":      contract.CooperationContractNumber,
		"createdAt":                      contract.CreatedAt.Unix() * 1000,
		"status":                         contract.Status,
	}

	// 2.合约分红列表
	creditAccountRepository, _ := repository.NewCreditAccountRepository(ptr.transactionContext)
	queryOptions["cooperationContractNumber"] = contract.CooperationContractNumber
	_, creditAccounts, err := creditAccountRepository.Find(queryOptions)
	if err != nil {
		return res, err
	}

	// 3.取订单商品列表
	var orderGoodIds []int64
	for i := range creditAccounts {
		a := creditAccounts[i]
		for j := range a.AccountDetail {
			if a.AccountDetail[j].OrderGoodId > 0 {
				orderGoodIds = append(orderGoodIds, a.AccountDetail[j].OrderGoodId)
			}
		}
	}

	orderGoodRepository, _ := repository.NewOrderGoodRepository(ptr.transactionContext)
	_, orderGoods, err := orderGoodRepository.Find(map[string]interface{}{"orderGoodIds": orderGoodIds})
	if err != nil {
		return res, err
	}
	var mapOrderGoods = make(map[int64]*domain.OrderGood)
	for i := range orderGoods {
		mapOrderGoods[orderGoods[i].OrderGoodId] = orderGoods[i]
	}

	var dividends = make([]interface{}, 0)
	for i := range creditAccounts {
		a := creditAccounts[i]
		for j := range a.AccountDetail {
			participateType, _ := strconv.Atoi(a.ParticipateType)
			detail := a.AccountDetail[j]
			item := map[string]interface{}{
				"creditAccountId": a.CreditAccountId,
				"orderGoodName":   "",
				"dividendsType":   0,
				"dividendsRatio":  0,
				"dividendsAmount": detail.DividendsAmount,
				"dividendsUser": map[string]interface{}{
					"userInfo": a.Participator.UserInfo,
					"userId":   a.Participator.UserId,
				},
				"dividendsParticipateType": participateType,
				"dividendsAccountStatus":   a.PaymentStatus,
				"dividendsEstimateTime":    a.CreatedAt.Unix() * 1000,
				"orderOrReturnedOrderNum":  a.CreditAccountOrderNum,
			}
			if detail.OrderGoodAmount > 0 {
				item["dividendsRatio"] = utils.Round(detail.DividendsAmount/detail.OrderGoodAmount*100, 2)
			}
			item["dividendsType"] = a.AccountDetail[j].DividendsType
			if v, ok := mapOrderGoods[detail.OrderGoodId]; ok {
				item["orderGoodName"] = v.OrderGoodName
			}

			dividends = append(dividends, item)
		}

	}
	res["dividends"] = dividends
	return res, nil
}

// 分红明细
func (ptr *CooperationStatisticsService) DividendsStatistics(queryOptions map[string]interface{}) (interface{}, error) {
	// 参数验证
	var request = struct {
		//企业
		CompanyId int64 `json:"companyId"`
		OrgId     int64 `json:"orgId"`
		UserId    int64 `json:"userId"`
		//个人
		UserBaseId int64 `json:"userBaseId"`
	}{}
	if err := LoadQueryObject(queryOptions, &request); err != nil {
		return nil, err
	}
	queryOptions = tool_funs.SimpleStructToMap(&request)

	type response struct {
		Total      float64 `json:"total"`
		Accounting float64 `json:"accounting"`
		Accounted  float64 `json:"accounted"`
		Paid       float64 `json:"paid"`
	}
	creditAccountDao, _ := dao.NewCreditAccountDao(ptr.transactionContext)
	var contractNumbers []string
	var err error
	if request.OrgId > 0 && request.UserId > 0 {
		contractNumbers, err = ptr.getRelevantContracts(map[string]interface{}{"orgId": request.OrgId, "userId": request.UserId})
		if err != nil {
			return struct{}{}, err
		}
	} else if request.UserBaseId > 0 {
		contractNumbers, err = ptr.getRelevantContracts(map[string]interface{}{"orgId": request.OrgId, "userBaseId": request.UserBaseId})
		if err != nil {
			return struct{}{}, err
		}
	}

	queryOptions["cooperationContractNumbers"] = contractNumbers

	var allDividends = &CreditAccountStatisticsResponse{}
	var annualDividends = &CreditAccountStatisticsResponse{}
	var quarterDividends = &CreditAccountStatisticsResponse{}

	if len(contractNumbers) > 0 {
		if err := creditAccountDao.DividendsStatistics(queryOptions, allDividends); err != nil {
			return nil, err
		}
		allDividends.Accounting = allDividends.Total - allDividends.Accounted

		queryOptions["beginTime"] = time.Date(time.Now().Year(), 1, 1, 0, 0, 0, 0, time.Local)
		queryOptions["endTime"] = time.Date(time.Now().Year(), 1, 1, 0, 0, 0, 0, time.Local).AddDate(1, 0, 0)
		if err := creditAccountDao.DividendsStatistics(queryOptions, annualDividends); err != nil {
			return nil, err
		}
		annualDividends.Accounting = annualDividends.Total - annualDividends.Accounted

		queryOptions["beginTime"], queryOptions["endTime"] = quarterBeginEnd()
		if err := creditAccountDao.DividendsStatistics(queryOptions, quarterDividends); err != nil {
			return nil, err
		}
		quarterDividends.Accounting = quarterDividends.Total - quarterDividends.Accounted
	}

	ret := map[string]interface{}{
		"allDividends":     allDividends,
		"annualDividends":  annualDividends,
		"quarterDividends": quarterDividends,
	}
	return ret, nil
}

func quarterBeginEnd() (time.Time, time.Time) {
	y := time.Now().Year()
	m := time.Now().Month()
	var mBegin, mEnd int
	var begin, end time.Time
	switch m {
	case 4, 5, 6:
		mBegin = 4
		mEnd = 6
	case 7, 8, 9:
		mBegin = 7
		mEnd = 9
	case 10, 11, 12:
		mBegin = 10
		mEnd = 12
	case 1, 2, 3:
		mBegin = 1
		mEnd = 3
	}
	begin = time.Date(y, time.Month(mBegin), 1, 0, 0, 0, 0, time.Local)
	end = time.Date(y, time.Month(mEnd), 1, 0, 0, 0, 0, time.Local)
	return begin, end
}

// 分红预算列表
func (ptr *CooperationStatisticsService) SearchDividendsEstimates(queryOptions map[string]interface{}) (interface{}, error) {
	// 参数验证
	var request = struct {
		CompanyId int64 `json:"companyId"`
		OrgId     int64 `json:"orgId"`
		UserId    int64 `json:"userId"`

		UserBaseId             int64 `json:"userBaseId"`
		DividendsAccountStatus int32 `json:"dividendsAccountStatus"` // 分红结算状态
		Offset                 int64 `json:"offset"`
		Limit                  int64 `json:"limit"`
	}{}
	if err := LoadQueryObject(queryOptions, &request); err != nil {
		return nil, err
	}
	queryOptions = tool_funs.SimpleStructToMap(&request)

	var contractNumbers []string
	var err error
	if request.OrgId > 0 && request.UserId > 0 {
		contractNumbers, err = ptr.getRelevantContracts(map[string]interface{}{"orgId": request.OrgId, "userId": request.UserId})
		if err != nil {
			return struct{}{}, err
		}
	} else if request.UserBaseId > 0 {
		contractNumbers, err = ptr.getRelevantContracts(map[string]interface{}{"orgId": request.OrgId, "userBaseId": request.UserBaseId})
		if err != nil {
			return struct{}{}, err
		}
	}
	type response struct {
		// 承接人分红预算记录ID
		DividendsEstimateId int64 `json:"dividendsEstimateId,string"`
		// 分红结算状态，1待结算，2已结算
		DividendsAccountStatus int32 `json:"dividendsAccountStatus"`
		// 分红金额
		DividendsAmount float64 `json:"dividendsAmount"`
		// 参与分红类型，1承接人，2推荐人，3关联业务员
		DividendsParticipateType int32 `json:"dividendsParticipateType"`
		// 分红类型，1订单分红，2退货冲销，3金额激励
		DividendsType int32 `json:"dividendsType"`
		// 分红用户（共创参与）
		DividendsUser *domain.User `json:"dividendsUser"`
		// 分红订单金额
		DividendsOrderAmount float64 `json:"dividendsOrderAmount"`
		//OrderGoodName string `json:"orderGoodName"`
		CustomerName string `json:"customerName"`
	}
	var results = make([]response, 0)
	if len(contractNumbers) == 0 {
		return results, nil
	}

	queryOptions["cooperationContractNumbers"] = contractNumbers

	dividendsEstimateRepository, _ := repository.NewDividendsEstimateRepository(ptr.transactionContext)
	_, estimates, err := dividendsEstimateRepository.Find(queryOptions)
	if err != nil {
		return nil, err
	}

	dividendsOrderRepository, _ := repository.NewDividendsOrderRepository(ptr.transactionContext)
	dividendsReturnedOrderRepository, _ := repository.NewDividendsReturnedOrderRepository(ptr.transactionContext)
	for i := range estimates {
		item := estimates[i]
		result := response{
			DividendsEstimateId:      item.DividendsEstimateId,
			DividendsAccountStatus:   item.DividendsAccountStatus,
			DividendsAmount:          item.DividendsAmount,
			DividendsParticipateType: item.DividendsParticipateType,
			DividendsType:            item.DividendsType,
			DividendsUser:            item.DividendsUser,
		}
		// TODO:优化查询 统一查询到map里面
		if item.DividendsType == 1 {
			order, err := dividendsOrderRepository.FindOne(map[string]interface{}{
				"dividendsOrderNumber": item.OrderOrReturnedOrderNum,
				"orgId":                request.OrgId,
			})
			if err != nil {
				return nil, err
			}
			result.DividendsOrderAmount = order.DividendsOrderAmount
			result.CustomerName = order.CustomerName
		} else if item.DividendsType == 2 {
			order, err := dividendsReturnedOrderRepository.FindOne(map[string]interface{}{"dividendsReturnedOrderNumber": item.OrderOrReturnedOrderNum, "orgId": request.OrgId})
			if err != nil {
				return nil, err
			}
			result.DividendsOrderAmount = order.DividendsReturnedOrderRefund
			result.CustomerName = order.DividendsReturnedCustomerName
		}
		results = append(results, result)
	}
	return results, nil
}

// 账期结算单统计
func (ptr *CooperationStatisticsService) CreditAccountStatistics(queryOptions map[string]interface{}) (interface{}, error) {
	// 参数验证
	var request = struct {
		OrgId      int64     `json:"orgId"`
		UserBaseId int64     `json:"userBaseId"`
		BeginTime  time.Time `json:"beginTime"`
		EndTime    time.Time `json:"endTime"`
		// 合约编号列表
		CooperationContractNumbers []string `json:"cooperationContractNumbers"`
	}{}
	if err := LoadQueryObject(queryOptions, &request); err != nil {
		return nil, err
	}
	queryOptions = tool_funs.SimpleStructToMap(&request)

	creditAccountDao, _ := dao.NewCreditAccountDao(ptr.transactionContext)

	var allDividends = &CreditAccountStatisticsResponse{}
	if err := creditAccountDao.DividendsStatistics(queryOptions, allDividends); err != nil {
		return nil, err
	}
	allDividends.Accounting = allDividends.Total - allDividends.Accounted
	return allDividends, nil
}

type CreditAccountStatisticsResponse struct {
	Total      float64 `json:"total"`
	Accounting float64 `json:"accounting"`
	Accounted  float64 `json:"accounted"`
	Paid       float64 `json:"paid"`
	Unpaid     float64 `json:"unpaid"`
}

//getRelevantContracts  获取相关人的合约
func (ptr *CooperationStatisticsService) RelevantCooperationContractNumbers(queryOptions map[string]interface{}) (interface{}, error) {
	var request = struct {
		//企业
		CompanyId int64 `json:"companyId" valid:"Required"`
		OrgId     int64 `json:"orgId" valid:"Required"`
		UserId    int64 `json:"userId" valid:"Required"`
		//个人
		//UserBaseId int64 `json:"userBaseId"`
		//Offset     int   `json:"offset"`
		//Limit      int   `json:"limit"`
	}{}
	if err := LoadQueryObject(queryOptions, &request); err != nil {
		return nil, err
	}
	queryOptions = tool_funs.SimpleStructToMap(&request)
	queryOptions["limit"] = 999
	contractNumbers, err := ptr.getRelevantContracts(queryOptions)
	return contractNumbers, err
}