package data_manage

import (
	"eta/eta_api/controllers"
	"eta/eta_api/models"
	"eta/eta_api/models/data_manage"
	"eta/eta_api/services/data"
	"eta/eta_api/utils"
	"github.com/rdlucklib/rdluck_tools/paging"
)

// EdbInfoRelationController 指标引用管理
type EdbInfoRelationController struct {
	controllers.BaseAuthController
}

// RelationEdbList
// @Title 获取被引用的指标列表接口
// @Description 获取被引用的指标列表接口
// @Param   Source   query   int  true       "来源:2:wind,34:钢联化工"
// @Param   ClassifyId   query   string  false             "分类ID，支持多选，用英文,隔开"
// @Param   SysUserId   query   string  false       "创建人，支持多选，用英文,隔开"
// @Param   Frequency   query   string  false       "频度，支持多选，用英文,隔开"
// @Param   Keyword   query   string  false       "关键词"
// @Param   SortParam   query   string  false       "排序字段参数，用来排序的字段， 枚举值：'RelationTime':引用日期 'RelationNum' 引用次数"
// @Param   SortType   query   string  true       "如何排序，是正序还是倒序，枚举值：`asc 正序`,`desc 倒叙`"
// @Success 200 {object} data_manage.BaseRelationEdbInfoResp
// @router /edb_info/relation/edb_list [get]
func (c *EdbInfoRelationController) RelationEdbList() {
	br := new(models.BaseResponse).Init()

	defer func() {
		c.Data["json"] = br
		c.ServeJSON()
	}()

	source, _ := c.GetInt("Source")
	classifyId := c.GetString("ClassifyId")
	sysUserId := c.GetString("SysUserId")
	frequency := c.GetString("Frequency")
	keyword := c.GetString("Keyword")
	status := c.GetString("Status")

	sortParam := c.GetString("SortParam")
	sortType := c.GetString("SortType")

	switch sortParam {
	case "RelationTime":
		sortParam = " r.relation_time "
	case "RelationNum":
		sortParam = " r.relation_num "
	default:
		sortParam = " r.relation_time "
	}
	switch sortType {
	case "desc", "asc":
	default:
		sortType = "desc"
	}

	pageSize, _ := c.GetInt("PageSize")
	currentIndex, _ := c.GetInt("CurrentIndex")
	var startSize int

	if pageSize <= 0 {
		pageSize = utils.PageSize20
	}
	if currentIndex <= 0 {
		currentIndex = 1
	}
	startSize = utils.StartIndex(currentIndex, pageSize)

	total, list, err := data.GetEdbRelationList(source, classifyId, sysUserId, frequency, keyword, status, startSize, pageSize, sortParam, sortType)
	if err != nil && err.Error() != utils.ErrNoRow() {
		br.Msg = "获取失败"
		br.ErrMsg = "获取数据失败，Err:" + err.Error()
		return
	}

	page := paging.GetPaging(currentIndex, pageSize, total)

	resp := data_manage.BaseRelationEdbInfoResp{
		Paging: page,
		List:   list,
	}

	br.Ret = 200
	br.Success = true
	br.Msg = "获取成功"
	br.Data = resp
}

// RelationEdbListDetail
// @Title 获取引用详情接口
// @Description 获取引用详情接口
// @Param   EdbInfoId   query   int  true       "指标ID"
// @Success 200 {object} data_manage.RefreshBaseEdbInfoResp
// @router /edb_info/relation/detail [get]
func (c *EdbInfoRelationController) RelationEdbListDetail() {
	br := new(models.BaseResponse).Init()

	defer func() {
		c.Data["json"] = br
		c.ServeJSON()
	}()

	edbInfoId, _ := c.GetInt("EdbInfoId")
	pageSize, _ := c.GetInt("PageSize")
	currentIndex, _ := c.GetInt("CurrentIndex")
	var startSize int

	if pageSize <= 0 {
		pageSize = utils.PageSize20
	}
	if currentIndex <= 0 {
		currentIndex = 1
	}
	startSize = utils.StartIndex(currentIndex, pageSize)

	total, relationList, err := data_manage.GetEdbInfoRelationDetailList(edbInfoId, startSize, pageSize)
	if err != nil && err.Error() != utils.ErrNoRow() {
		br.Msg = "获取失败"
		br.ErrMsg = "获取数据失败，Err:" + err.Error()
		return
	}

	list := make([]*data_manage.EdbInfoRelationDetail, 0)
	for _, v := range relationList {
		tmp := &data_manage.EdbInfoRelationDetail{
			EdbInfoRelationId:  v.EdbInfoRelationId,
			EdbInfoId:          v.EdbInfoId,
			ReferObjectId:      v.ReferObjectId,
			ReferObjectType:    v.ReferObjectType,
			ReferObjectSubType: v.ReferObjectSubType,
			RelationTime:       v.ModifyTime.Format(utils.FormatDateTime),
		}
		switch v.ReferObjectType {
		case utils.EDB_RELATION_SANDBOX:
			tmp.ReferObjectTypeName = "逻辑图"
		case utils.EDB_RELATION_CALENDAR:
			tmp.ReferObjectTypeName = "事件日历"
		case utils.EDB_RELATION_CHART:
			switch v.ReferObjectSubType {
			case utils.CHART_SOURCE_DEFAULT:
				tmp.ReferObjectTypeName = "图库"
			case utils.CHART_SOURCE_CORRELATION, utils.CHART_SOURCE_ROLLING_CORRELATION:
				tmp.ReferObjectTypeName = "相关性分析"
			case utils.CHART_SOURCE_LINE_EQUATION:
				tmp.ReferObjectTypeName = "拟合方程曲线"
			case utils.CHART_SOURCE_LINE_FEATURE_STANDARD_DEVIATION, utils.CHART_SOURCE_LINE_FEATURE_PERCENTILE, utils.CHART_SOURCE_LINE_FEATURE_FREQUENCY:
				tmp.ReferObjectTypeName = "统计特征"
			case utils.CHART_SOURCE_CROSS_HEDGING:
				tmp.ReferObjectTypeName = "跨品种分析"
			case utils.CHART_SOURCE_FUTURE_GOOD, utils.CHART_SOURCE_FUTURE_GOOD_PROFIT:
				tmp.ReferObjectTypeName = "商品价格曲线"
			}
		}
		list = append(list, tmp)
	}
	page := paging.GetPaging(currentIndex, pageSize, total)

	resp := data_manage.BaseRelationEdbInfoDetailResp{
		Paging: page,
		List:   list,
	}

	br.Ret = 200
	br.Success = true
	br.Msg = "获取成功"
	br.Data = resp
}