package home

import (
	"context"
	contDao "dashoo.cn/micro/app/dao/contract"
	projDao "dashoo.cn/micro/app/dao/proj"
	projSrv "dashoo.cn/micro/app/service/proj"
	"dashoo.cn/opms_libary/myerrors"
	"database/sql"
	"fmt"

	"dashoo.cn/micro/app/model/home"
	"dashoo.cn/micro/app/service"
	contractService "dashoo.cn/micro/app/service/contract"
	"dashoo.cn/opms_libary/micro_srv"
	"github.com/gogf/gf/database/gdb"
	"github.com/gogf/gf/os/gtime"
	"github.com/gogf/gf/util/gconv"
)

// getPersonalContractReportData 获取个人数据
// dataType：collection为回款；其他为合同
func getPersonalContractReportData(ctx context.Context, dataType string, params *map[string]interface{}) (*home.ReportData, error) {
	var reportData home.ReportData
	targetMap := make(map[int]gdb.Record, 0)
	realMap := make(map[int][]*gdb.Record, 0)
	targetField := "sales_target"
	realField := "contract_amount"
	year := gtime.Now().Format("Y")
	if params != nil && (*params)["year"] != nil {
		year = gconv.String((*params)["year"])
	}
	// 统计字段
	if dataType == "COLLECTION" {
		targetField = "collection_target"
		realField = "collection_amount"
	}

	srv, err := contractService.NewCtrContractService(ctx)
	if err != nil {
		return nil, err
	}
	// 获取用户信息
	userInfo, err := micro_srv.GetUserInfo(ctx)
	if err != nil {
		return nil, fmt.Errorf("获取用户信息异常：%s", err.Error())
	}

	targets, err := srv.Dao.DB.Model("rpt_sales_target").Where("sale_user_id", userInfo.Id).Order("monthly ASC").FindAll()
	if err != nil && err != sql.ErrNoRows {
		return nil, err
	}
	for index, target := range targets {
		targetMap[target["monthly"].Int()] = targets[index]
	}
	// 统计12个月份的数据
	// 统计目标值
	for index := 1; index <= 12; index++ {
		reportData.XData = append(reportData.XData, fmt.Sprintf("%v月", index))
		if target, ok := targetMap[index]; ok {
			reportData.YDataTarget = append(reportData.YDataTarget, target[targetField].Float64())
		} else {
			reportData.YDataTarget = append(reportData.YDataTarget, 0)
		}
	}
	// 统计合同值
	contractModel := srv.Dao.DB.Model("ctr_contract").Where("ctr_contract.incharge_id", userInfo.Id)
	if dataType != "COLLECTION" {
		contracts, err := contractModel.Where("ctr_contract.contract_start_time LIKE ?", year+"-%").Order("ctr_contract.contract_start_time ASC").FindAll()
		if err != nil && err != sql.ErrNoRows {
			return nil, err
		}
		for index, contract := range contracts {
			realMap[contract["contract_start_time"].GTime().Month()] = append(realMap[contract["contract_start_time"].GTime().Month()], &contracts[index])
		}
	} else {
		// 回款数据统计
		collections, err := contractModel.InnerJoin("ctr_contract_collection", "ctr_contract.id=ctr_contract_collection.contract_id").Where("ctr_contract_collection.appro_status", "20").Where("ctr_contract_collection.collection_datetime LIKE ?", year+"-%").Order("ctr_contract_collection.collection_datetime ASC").Fields("ctr_contract_collection.*").FindAll()
		if err != nil && err != sql.ErrNoRows {
			return nil, err
		}
		for index, collection := range collections {
			realMap[collection["collection_datetime"].GTime().Month()] = append(realMap[collection["collection_datetime"].GTime().Month()], &collections[index])
		}
	}
	// 统计12个月份的数据
	// 统计实际值
	for index := 1; index <= 12; index++ {
		if realData, ok := realMap[index]; ok {
			realAmount := float64(0)
			for _, data := range realData {
				realAmount += (*data)[realField].Float64()
			}
			reportData.YDataReal = append(reportData.YDataReal, realAmount)
		} else {
			reportData.YDataReal = append(reportData.YDataReal, 0)
		}
		if reportData.YDataTarget[index-1] == 0 {
			reportData.PercentData = append(reportData.PercentData, 0)
		} else {
			reportData.PercentData = append(reportData.PercentData, reportData.YDataReal[index-1]*100/reportData.YDataTarget[index-1])
		}
	}

	return &reportData, nil
}

// getCompanyContractReportData 获取总部数据
// dataType：collection为回款；其他为合同
func getCompanyContractReportData(ctx context.Context, dataType string, params *map[string]interface{}) (*home.ReportData, error) {
	var reportData home.ReportData
	realMap := make(map[int]gdb.Record, 0)
	targetField := "sales_target"
	realField := "contract_amount"
	dateWhere1 := "" // 查询条件
	dateWhere2 := "" // 查询条件
	dateWhere3 := ""
	date := gtime.Now()
	if params != nil && (*params)["date"] != nil {
		date = gconv.GTime((*params)["date"])
	}
	dateWhere1 = fmt.Sprintf("ctr_contract.contract_start_time LIKE '%v'", date.Format("Y")+"-%")
	dateWhere2 = fmt.Sprintf("ctr_contract_collection.collection_datetime LIKE '%v'", date.Format("Y")+"-%")
	if params != nil && (*params)["searchType"] != nil {
		searchType := gconv.String((*params)["searchType"])
		if searchType == "month" {
			dateWhere1 = fmt.Sprintf("ctr_contract.contract_start_time LIKE '%v'", date.Format("Y-m")+"-%")
			dateWhere2 = fmt.Sprintf("ctr_contract_collection.collection_datetime LIKE '%v'", date.Format("Y-m")+"-%")
			dateWhere3 = fmt.Sprintf("monthly=%v", date.Month())
		} else if searchType == "quarter" {
			dateWhere1 = getQuarterWhere(date, "ctr_contract.contract_start_time")
			dateWhere2 = getQuarterWhere(date, "ctr_contract_collection.collection_datetime")
			dateWhere3 = getQuarterMonthWhere(date, "monthly")
		}
	}
	// 统计字段
	if dataType == "COLLECTION" {
		targetField = "collection_target"
		realField = "collection_amount"
	}

	srv, err := contractService.NewCtrContractService(ctx)
	if err != nil {
		return nil, err
	}

	targets, err := srv.Dao.DB.Model("rpt_sales_target").Where(dateWhere3).Order("sale_user_id ASC").Group("sale_user_id").Fields(fmt.Sprintf("sale_user_id, sale_user_name, SUM(%v) %v", targetField, targetField)).FindAll()
	if err != nil && err != sql.ErrNoRows {
		return nil, err
	}
	// 统计合同值
	contractModel := srv.Dao.DB.Model("ctr_contract").Group("ctr_contract.incharge_id").Order("ctr_contract.incharge_id ASC")
	if dataType != "COLLECTION" {
		contracts, err := contractModel.Where(dateWhere1).Fields("ctr_contract.incharge_id, ctr_contract.incharge_name, SUM(ctr_contract.contract_amount) contract_amount").FindAll()
		if err != nil && err != sql.ErrNoRows {
			return nil, err
		}
		for index, contract := range contracts {
			realMap[contract["incharge_id"].Int()] = contracts[index]
		}
	} else {
		// 回款数据统计
		collections, err := contractModel.InnerJoin("ctr_contract_collection", "ctr_contract.id=ctr_contract_collection.contract_id").Where("ctr_contract_collection.appro_status", "20").Where(dateWhere2).Fields("ctr_contract.incharge_id, ctr_contract.incharge_name, SUM(ctr_contract_collection.collection_amount) collection_amount").FindAll()
		if err != nil && err != sql.ErrNoRows {
			return nil, err
		}
		for index, collection := range collections {
			realMap[collection["incharge_id"].Int()] = collections[index]
		}
	}

	// 统计目标值和实际值
	for index, target := range targets {
		reportData.XData = append(reportData.XData, target["sale_user_name"].String())
		reportData.YDataTarget = append(reportData.YDataTarget, target[targetField].Float64())
		if realData, ok := realMap[target["sale_user_id"].Int()]; ok {
			reportData.YDataReal = append(reportData.YDataReal, realData[realField].Float64())
		} else {
			reportData.YDataReal = append(reportData.YDataReal, 0)
		}
		if reportData.YDataTarget[index] == 0 {
			reportData.PercentData = append(reportData.PercentData, 0)
		} else {
			reportData.PercentData = append(reportData.PercentData, reportData.YDataReal[index]*100/reportData.YDataTarget[index])
		}
	}

	return &reportData, nil
}

func getQuarterGoalReportData(ctx context.Context, dataType string, params *map[string]interface{}) (*home.ReportData, error) {
	if params == nil {
		return nil, fmt.Errorf("请输入年度")
	}
	p := *params
	year := gconv.Int(p["year"])
	quarter := gconv.Int(p["quarter"])
	if year == 0 {
		return nil, fmt.Errorf("请输入年度")
	}

	srv, err := contractService.NewCtrContractService(ctx)
	if err != nil {
		return nil, err
	}

	goal := map[string]float64{}
	goaldao := srv.GoalDao.Where("year = ?", year).Where("goal_type = ?", dataType)
	if quarter != 0 {
		goaldao = goaldao.Where("quarter = ?", quarter)
	} else {
		goaldao = goaldao.Where("quarter = 1 || quarter = 2 || quarter = 3 || quarter = 4")
	}
	goalent, err := goaldao.All()
	if err != nil {
		return nil, err
	}
	for _, ent := range goalent {
		goal[ent.ProductLine] += ent.Amount
	}

	got := map[string]float64{}
	ctrdao := srv.Dao.Where("year(created_time) = ?", year)
	if quarter != 0 {
		if quarter == 1 {
			ctrdao = ctrdao.Where("month(created_time) in (1,2,3)")
		}
		if quarter == 2 {
			ctrdao = ctrdao.Where("month(created_time) in (4,5,6)")
		}
		if quarter == 3 {
			ctrdao = ctrdao.Where("month(created_time) in (7,8,9)")
		}
		if quarter == 4 {
			ctrdao = ctrdao.Where("month(created_time) in (10,11,12)")
		}
	}
	ctrent, err := ctrdao.All()
	if err != nil {
		return nil, err
	}
	for _, ent := range ctrent {
		if dataType == "10" {
			got[ent.ProductLine] += ent.ContractAmount
		}
		if dataType == "20" {
			got[ent.ProductLine] += ent.CollectedAmount
		}
	}

	productLine, err := service.GetDictDataByType(ctx, "sys_product_line")
	if err != nil {
		return nil, err
	}
	productLineCode := []string{}
	productLineName := []string{}
	for k, v := range productLine {
		productLineCode = append(productLineCode, k)
		productLineName = append(productLineName, v)
	}

	var reportData home.ReportData
	for _, c := range productLineCode {
		reportData.XData = append(reportData.XData, productLine[c])
		reportData.YDataTarget = append(reportData.YDataTarget, goal[c])
		reportData.YDataReal = append(reportData.YDataReal, got[c]/10000)
	}
	return &reportData, nil
}

func getQuarterWhere(date *gtime.Time, field string) string {
	if date.Month() >= 1 && date.Month() <= 3 {
		return fmt.Sprintf("%v >= '%v' AND %v < '%v'", field, date.Format("Y-")+"01-01 00:00:00", field, date.Format("Y-")+"04-01 00:00:00")
	} else if date.Month() >= 4 && date.Month() <= 6 {
		return fmt.Sprintf("%v >= '%v' AND %v < '%v'", field, date.Format("Y-")+"04-01 00:00:00", field, date.Format("Y-")+"07-01 00:00:00")
	} else if date.Month() >= 7 && date.Month() <= 9 {
		return fmt.Sprintf("%v >= '%v' AND %v < '%v'", field, date.Format("Y-")+"07-01 00:00:00", field, date.Format("Y-")+"10-01 00:00:00")
	} else if date.Month() >= 10 && date.Month() <= 12 {
		return fmt.Sprintf("%v >= '%v' AND %v <= '%v'", field, date.Format("Y-")+"10-01 00:00:00", field, date.Format("Y-")+"12-31 23:59:59")
	}
	return ""
}

func getQuarterMonthWhere(date *gtime.Time, field string) string {
	if date.Month() >= 1 && date.Month() <= 3 {
		return fmt.Sprintf("%v >= %v AND %v < %v", field, 1, field, 4)
	} else if date.Month() >= 4 && date.Month() <= 6 {
		return fmt.Sprintf("%v >= %v AND %v < %v", field, 4, field, 7)
	} else if date.Month() >= 7 && date.Month() <= 9 {
		return fmt.Sprintf("%v >= %v AND %v < %v", field, 7, field, 10)
	} else if date.Month() >= 10 && date.Month() <= 12 {
		return fmt.Sprintf("%v >= %v AND %v <= %v", field, 10, field, 12)
	}
	return ""
}

type BusCount struct {
	NboType     string `json:"nboType"`
	ProductLine string `json:"productLine"`
	Count       int    `json:"count"`
	Remark      string `json:"remark"`
}

// 报表数据   三大产品线，新增以及转化项目(C转B、B转A、A转签约、C转A、C转签约、B转签约、储备转A/B/C/签约)数量， 按周和月
// params 10:周 20:月
func (s *HomeService) getNewAndConvertBusiness(productLine []string, params *map[string]interface{}) (interface{}, error) {
	if params == nil {
		return nil, myerrors.TipsError("请求参数传递不正确")
	}
	searchType, ok := (*params)["searchType"]
	if !ok {
		return nil, myerrors.TipsError("请求查询类型参数传递错误")
	}
	currentTime := gtime.Now()
	weekStart, weekEnd := currentTime.StartOfWeek(), currentTime.EndOfWeek()
	monthStart, monthEnd := currentTime.StartOfMonth(), currentTime.EndOfMonth()

	businessDao := projDao.NewProjBusinessDao(s.Tenant)
	busDynamicsDao := projDao.NewProjBusinessDynamicsDao(s.Tenant)
	contractDao := contDao.NewCtrContractDao(s.Tenant)

	// 获取三大产品线新增项目
	getAddBusCount := func(searchType string) (addCount []BusCount, err error) {
		commonDb := businessDao.DataScope(s.Ctx, "sale_id").Group(businessDao.C.ProductLine).OrderAsc(businessDao.C.ProductLine).
			Fields("product_line, count(id) as count")
		if searchType == "week" {
			err = commonDb.WhereGTE(businessDao.C.FilingTime, weekStart).WhereLTE(businessDao.C.FilingTime, weekEnd).Scan(&addCount)
		}
		if searchType == "month" {
			err = commonDb.WhereGTE(businessDao.C.FilingTime, monthStart).WhereLTE(businessDao.C.FilingTime, monthEnd).Scan(&addCount)
		}
		return addCount, err
	}
	addCount, err := getAddBusCount(searchType.(string))
	if err != nil {
		return nil, err
	}

	// 获取三大产品线签约项目
	getSignBusCount := func(searchType string) (signCount []BusCount, err error) {
		commonDb := contractDao.As("contract").DataScope(s.Ctx, "incharge_id").LeftJoin(businessDao.Table, "bus", "bus.id=contract.nbo_id").
			Fields("bus.product_line, bus.nbo_type, count(contract.id) as count").
			Group("bus.product_line, bus.nbo_type").Order("bus.product_line ASC, bus.nbo_type ASC")
		if searchType == "week" {
			err = commonDb.WhereGTE("contract."+contractDao.C.CreatedTime, weekStart).WhereLTE("contract."+contractDao.C.CreatedTime, weekEnd).Scan(&signCount)
		}
		if searchType == "month" {
			err = commonDb.WhereGTE("contract."+contractDao.C.CreatedTime, monthStart).WhereLTE("contract."+contractDao.C.CreatedTime, monthEnd).Scan(&signCount)
		}
		return signCount, err
	}
	signCount, err := getSignBusCount(searchType.(string))
	if err != nil {
		return nil, err
	}

	// 获取三大产品线转化项目(C转B、B转A、A转签约、C转A、C转签约、B转签约、储备转A/B/C/签约)数量
	getConvertBusData := func(searchType string) (convertData []BusCount, err error) {
		commonDb := busDynamicsDao.As("dy").LeftJoin(businessDao.Table, "bus", "bus.id=dy.bus_id").DataScope(s.Ctx, "sale_id", "bus").
			Where("dy."+busDynamicsDao.C.OpnType, projSrv.OpnUpgradeApproval).WhereNot("dy."+busDynamicsDao.C.Remark, "").
			Fields("bus.product_line, dy.remark").OrderAsc("bus.product_line")
		if searchType == "week" {
			err = commonDb.WhereGTE("dy."+busDynamicsDao.C.CreatedTime, weekStart).WhereLTE("dy."+busDynamicsDao.C.CreatedTime, weekEnd).Scan(&convertData)
		}
		if searchType == "month" {
			err = commonDb.WhereGTE("dy."+busDynamicsDao.C.CreatedTime, monthStart).WhereLTE("dy."+busDynamicsDao.C.CreatedTime, monthEnd).Scan(&convertData)
		}
		return convertData, err
	}
	convertBusData, err := getConvertBusData(searchType.(string))
	if err != nil {
		return nil, err
	}

	// 处理数据
	result := make([][]int, len(productLine))
	for i, pl := range productLine {
		result[i] = make([]int, 8)
		for _, item := range addCount {
			if item.ProductLine == pl {
				result[i][0] = item.Count
			}
		}
		for _, item := range signCount {
			if item.ProductLine != pl {
				continue
			}
			switch item.NboType {
			case projSrv.StatusA:
				result[i][3] += item.Count
			case projSrv.StatusB:
				result[i][6] += item.Count
			case projSrv.StatusC:
				result[i][5] += item.Count
			case projSrv.StatusReserve:
				result[i][7] += item.Count
			}
		}
		for _, item := range convertBusData {
			if item.ProductLine != pl {
				continue
			}
			info := gconv.Map(item.Remark)
			if info["origNboType"] == projSrv.StatusC && info["nboType"] == projSrv.StatusB {
				result[i][1] += 1
			}
			if info["origNboType"] == projSrv.StatusB && info["nboType"] == projSrv.StatusA {
				result[i][2] += 1
			}
			if info["origNboType"] == projSrv.StatusC && info["nboType"] == projSrv.StatusA {
				result[i][4] += 1
			}
			if info["origNboType"] == projSrv.StatusReserve {
				result[i][7] += 1
			}
		}
	}
	return result, nil
}

// 报表数据   三大产品线，AB类项目出货金额
func (s *HomeService) getShipmentAmount(productLine, nboType []string) (interface{}, error) {
	businessDao := projDao.NewProjBusinessDao(s.Tenant)
	data, err := businessDao.Fields("product_line, nbo_type, SUM(est_trans_price) as est_trans_price").
		WhereIn("product_line", productLine).WhereIn("nbo_type", nboType).
		Group("product_line, nbo_type").Order("product_line ASC, nbo_type ASC").DataScope(s.Ctx, "sale_id").All()
	if err != nil {
		return nil, err
	}
	result := make([][]float64, len(productLine))
	for i, pl := range productLine {
		result[i] = make([]float64, len(nboType))
		for j, t := range nboType {
			for _, item := range data {
				if item.ProductLine == pl && item.NboType == t {
					result[i][j] = item.EstTransPrice
				}
			}
		}
	}

	return arrayReversal(result), nil
}

// 报表数据   三大产品线，当月和累计的签约、回款额
func (s *HomeService) getProductLineSignedAndBackAmount(productLine []string) (interface{}, error) {
	currentTime := gtime.Now()
	monthStart := currentTime.StartOfMonth()
	monthEnd := currentTime.EndOfMonth()

	contractDao := contDao.NewCtrContractDao(s.Tenant)
	commonDao := contractDao.DataScope(s.Ctx, "incharge_id").WhereIn(contractDao.C.ProductLine, productLine).
		Group(contractDao.C.ProductLine).OrderAsc(contractDao.C.ProductLine)

	//累计签约合同金额 contract_amount
	allContractAmount, err := commonDao.Fields("product_line, SUM(contract_amount) as contract_amount").All()
	if err != nil {
		return nil, err
	}
	//累计回款金额 collected_amount
	allCollectedAmount, err := commonDao.Fields("product_line, SUM(collected_amount) as collected_amount").All()
	if err != nil {
		return nil, err
	}
	//当月签约合同金额（维度：月）
	monthContractAmount, err := commonDao.WhereGTE(contractDao.C.CreatedTime, monthStart).WhereLTE(contractDao.C.CreatedTime, monthEnd).
		Fields("product_line, SUM(contract_amount) as contract_amount").All()
	if err != nil {
		return nil, err
	}
	//当月回款金额（维度：月）
	monthCollectedAmount, err := commonDao.WhereGTE(contractDao.C.CreatedTime, monthStart).WhereLTE(contractDao.C.CreatedTime, monthEnd).
		Fields("product_line, SUM(collected_amount) as collected_amount").All()
	if err != nil {
		return nil, err
	}
	result := make([][]float64, len(productLine))
	for i, pl := range productLine {
		result[i] = make([]float64, 4)
		for _, item := range monthContractAmount {
			if item.ProductLine == pl {
				result[i][0] = item.ContractAmount
			}
		}
		for _, item := range allContractAmount {
			if item.ProductLine == pl {
				result[i][1] = item.ContractAmount
			}
		}
		for _, item := range monthCollectedAmount {
			if item.ProductLine == pl {
				result[i][2] = item.CollectedAmount
			}
		}
		for _, item := range allCollectedAmount {
			if item.ProductLine == pl {
				result[i][3] = item.CollectedAmount
			}
		}
	}

	return arrayReversal(result), nil
}

// 二维数组反转
func arrayReversal(result [][]float64) [][]float64 {
	if len(result) == 0 {
		return [][]float64{}
	}
	data := make([][]float64, len(result[0]))
	for k, v := range result {
		for m, n := range v {
			if k == 0 {
				data[m] = make([]float64, len(result))
			}
			data[m][k] = n
		}
	}
	return data
}