package data_manage

import (
	"eta/eta_api/utils"
	"time"

	"gorm.io/gorm"

	"database/sql"
	"eta/eta_api/global"

	"github.com/rdlucklib/rdluck_tools/paging"
)

type BaseFromPurangIndex struct {
	BaseFromPurangIndexId int `orm:"column(base_from_purang_index_id);pk"`
	ClassifyId             int
	IndexCode              string
	IndexName              string
	Frequency              string
	Unit                   string
	Sort                   int
	StartDate              time.Time `description:"开始日期"`
	EndDate                time.Time `description:"结束日期"`
	EndValue               float64
	CreateTime             time.Time
	ModifyTime             time.Time
}

type BaseFromPurangIndexList struct {
	BaseFromPurangIndexId int `orm:"column(base_from_purang_index_id);pk"`
	ClassifyId             int
	Interface              string
	EdbInfoId              int
	EdbUniqueCode          string `description:"指标库唯一编码"`
	EdbClassifyId          int    `description:"指标库分类ID"`
	StartDate              string
	EndDate                string
	EndValue               float64
	IndexCode              string
	IndexName              string
	Frequency              string
	Unit                   string
	Sort                   int
	CreateTime             string
	ModifyTime             string
	EdbExist               int                    `description:"指标库是否已添加：0-否；1-是"`
	DataList               []*BaseFromPurangData `gorm:"-"`
	Paging                 *paging.PagingItem     `description:"分页数据" gorm:"-"`
}

func (baseFromPurangIndexList *BaseFromPurangIndexList) AfterFind(tx *gorm.DB) (err error) {
	baseFromPurangIndexList.CreateTime = utils.GormDateStrToDateTimeStr(baseFromPurangIndexList.CreateTime)
	baseFromPurangIndexList.ModifyTime = utils.GormDateStrToDateTimeStr(baseFromPurangIndexList.ModifyTime)
	baseFromPurangIndexList.StartDate = utils.GormDateStrToDateStr(baseFromPurangIndexList.StartDate)
	baseFromPurangIndexList.EndDate = utils.GormDateStrToDateStr(baseFromPurangIndexList.EndDate)
	return
}

type BaseFromPurangIndexSearchList struct {
	List   []*BaseFromPurangIndexList
	Paging *paging.PagingItem `description:"分页数据"`
}

type PurangSingleDataResp struct {
	BaseFromPurangIndexId int
	ClassifyId             int
	EdbInfoId              int
	IndexCode              string
	IndexName              string
	Frequency              string
	Unit                   string
	StartTime              string
	CreateTime             string
	ModifyTime             string
	EdbExist               int                  `description:"指标库是否已添加：0-否；1-是"`
	Data                   []*PurangSingleData `gorm:"-"`
	Paging                 *paging.PagingItem   `description:"分页数据" gorm:"-"`
}

type PurangSingleData struct {
	Value    string `orm:"column(value)" description:"日期"`
	DataTime string `orm:"column(data_time)" description:"值"`
}

func GetPurangIndexByClassifyId(classifyId int) (items []*BaseFromPurangIndex, err error) {
	sql := ` SELECT base_from_purang_index_id, classify_id, index_code, index_name FROM base_from_purang_index WHERE classify_id=? ORDER BY sort ASC, base_from_purang_index_id ASC `
	err = global.DbMap[utils.DbNameIndex].Raw(sql, classifyId).Find(&items).Error
	return
}

func GetPurangIndex(condition string, pars []interface{}) (items []*BaseFromPurangIndexList, err error) {
	sql := ` SELECT * FROM base_from_purang_index WHERE 1=1  `
	if condition != "" {
		sql += condition
	}
	sql += ` ORDER BY sort ASC, base_from_purang_index_id asc`
	err = global.DbMap[utils.DbNameIndex].Raw(sql, pars...).Find(&items).Error
	return
}

func GetPurangIndexPage(condition string, pars []interface{}, startSize, pageSize int) (items []*BaseFromPurangIndexList, err error) {
	sql := ` SELECT * FROM base_from_purang_index WHERE 1=1  `
	if condition != "" {
		sql += condition
	}
	sql += ` ORDER BY sort ASC, base_from_purang_index_id asc LIMIT ?,?`
	pars = append(pars, startSize, pageSize)
	err = global.DbMap[utils.DbNameIndex].Raw(sql, pars...).Find(&items).Error
	return
}

func GetPurangIndexPageCount(condition string, pars []interface{}) (count int, err error) {
	sql := ` SELECT COUNT(1) AS count  FROM base_from_purang_index WHERE 1=1  `
	if condition != "" {
		sql += condition
	}
	err = global.DbMap[utils.DbNameIndex].Raw(sql, pars...).Scan(&count).Error
	return
}

func GetPurangIndexDataCount(indexCode string) (count int, err error) {
	sql := ` SELECT COUNT(1) AS count  FROM base_from_purang_data WHERE index_code=? `
	err = global.DbMap[utils.DbNameIndex].Raw(sql, indexCode).Scan(&count).Error
	return
}

func GetPurangIndexDataByDataTime(indexCodes []string, startDate, endDate string) (items []*BaseFromPurangData, err error) {
	sql := ` SELECT *  FROM base_from_purang_data WHERE  index_code in (` + utils.GetOrmInReplace(len(indexCodes)) + `) and data_time >=? and data_time <? ORDER BY data_time DESC `
	err = global.DbMap[utils.DbNameIndex].Raw(sql, indexCodes, startDate, endDate).Find(&items).Error
	return
}

func GetPurangIndexDataTimePageByCodes(indexCodes []string, startSize, pageSize int) (dataTimes []string, err error) {
	sql := ` SELECT data_time FROM base_from_purang_data WHERE index_code in ? GROUP BY data_time ORDER BY data_time DESC LIMIT ?,? `
	err = global.DbMap[utils.DbNameIndex].Raw(sql, indexCodes, startSize, pageSize).Find(&dataTimes).Error
	return
}

func GetPurangIndexDataTimePageCount(indexCodes []string) (count int, err error) {
	sql := ` SELECT COUNT(DISTINCT data_time) AS count  FROM base_from_purang_data WHERE index_code in (` + utils.GetOrmInReplace(len(indexCodes)) + `) `
	err = global.DbMap[utils.DbNameIndex].Raw(sql, indexCodes).Scan(&count).Error
	return
}

func GetPurangIndexDataByCodes(indexCode []string) (items []*BaseFromPurangData, err error) {
	if len(indexCode) == 0 {
		return
	}
	sql := ` SELECT *  FROM base_from_purang_data WHERE index_code in (` + utils.GetOrmInReplace(len(indexCode)) + `) ORDER BY data_time DESC  `
	err = global.DbMap[utils.DbNameIndex].Raw(sql, indexCode).Find(&items).Error
	return
}

// GetPurangByConditionAndFrequency 根据条件获取普瑞指标列表
func GetPurangByConditionAndFrequency(condition, frequency string, pars []interface{}) (items []*BaseFromPurangIndex, err error) {
	sql := ` SELECT * FROM base_from_purang_index WHERE 1=1 `

	if condition != "" {
		sql += condition
	}
	sql += ` AND frequency=?`
	sql += ` ORDER BY sort ASC, base_from_purang_index_id ASC`
	err = global.DbMap[utils.DbNameIndex].Raw(sql, pars, frequency).Find(&items).Error
	return
}

func GetPurangFrequencyByCondition(condition string, pars []interface{}) (items []string, err error) {
	sql := `SELECT DISTINCT frequency FROM base_from_purang_index WHERE 1=1 `
	if condition != "" {
		sql += condition
	}
	sql += ` ORDER BY FIELD(frequency,'日度','周度','旬度','月度','季度','半年度','年度') `
	err = global.DbMap[utils.DbNameIndex].Raw(sql, pars...).Find(&items).Error
	return
}

// GetPurangDataDataTimeByIndexId 根据指标id获取指标数据的日期列表
func GetPurangDataDataTimeByIndexId(indexIdList []int) (items []string, err error) {
	if len(indexIdList) == 0 {
		return
	}
	sql := ` SELECT DISTINCT data_time FROM base_from_purang_data WHERE base_from_purang_index_id IN (` + utils.GetOrmInReplace(len(indexIdList)) + `) ORDER BY data_time DESC`
	err = global.DbMap[utils.DbNameIndex].Raw(sql, indexIdList).Find(&items).Error
	for i, item := range items {
		items[i] = utils.GormDateStrToDateStr(item)
	}
	return
}

type BaseFromPurangData struct {
	BaseFromPurangDataId  int `orm:"column(base_from_purang_data_id);pk"`
	BaseFromPurangIndexId int
	IndexCode              string
	DataTime               string
	Value                  string
	CreateTime             string
	ModifyTime             string
	DataTimestamp          int64
}

func (baseFromPurangData *BaseFromPurangData) AfterFind(tx *gorm.DB) (err error) {
	baseFromPurangData.ModifyTime = utils.GormDateStrToDateTimeStr(baseFromPurangData.ModifyTime)
	baseFromPurangData.CreateTime = utils.GormDateStrToDateTimeStr(baseFromPurangData.CreateTime)
	baseFromPurangData.DataTime = utils.GormDateStrToDateStr(baseFromPurangData.DataTime)
	return
}

type BaseFromPurangIndexSearchItem struct {
	BaseFromPurangIndexId int `orm:"column(base_from_purang_index_id);pk"`
	ClassifyId             int
	ParentClassifyId       int
	IndexCode              string
	IndexName              string
}

// BatchCheckPurangEdbReq 指标数据结构体
type BatchCheckPurangEdbReq struct {
	Frequencies   string `description:"频度；枚举值：日度、周度、月度、季度、半年度、年度"`
	Keyword       string `description:"关键字"`
	ClassifyIds   string `description:"所选品种id列表"`
	ListAll       bool   `form:"ListAll" json:"ListAll" description:"列表全选"`
	TradeCodeList string `form:"TradeCodeList" json:"TradeCodeList" description:"全选为false时, 该数组为选中; 全选为true时, 该数组为不选的指标"`
}

// GetPurangItemList 模糊查询Purang数据库指标列表
func GetPurangItemList(condition string) (items []*BaseFromPurangIndexSearchItem, err error) {
	sql := "SELECT * FROM base_from_purang_index  WHERE 1=1"
	if condition != "" {
		sql += condition
	}
	err = global.DbMap[utils.DbNameIndex].Raw(sql).Find(&items).Error
	return
}

func GetPurangIndexDataByCode(indexCode string, pageIndex, pageSize int) (list []*BaseFromPurangData, err error) {
	sql := `SELECT * FROM base_from_purang_data WHERE index_code=? order by data_time desc limit ?,?`
	err = global.DbMap[utils.DbNameIndex].Raw(sql, indexCode, pageIndex, pageSize).Find(&list).Error
	return
}

func GetPurangIndexDataTotalByCode(indexCode string) (total int64, err error) {
	sqlStr := `SELECT count(*) FROM base_from_purang_data WHERE index_code=?`
	var totalNull sql.NullInt64
	err = global.DbMap[utils.DbNameIndex].Raw(sqlStr, indexCode).Scan(&totalNull).Error
	if !totalNull.Valid {
		total = 0
	} else {
		total = totalNull.Int64
	}
	return
}

func GetBaseFromPurangIndexByIndexCode(indexCode string) (list *BaseFromPurangIndex, err error) {
	sql := ` SELECT * FROM base_from_purang_index WHERE index_code=? `
	err = global.DbMap[utils.DbNameIndex].Raw(sql, indexCode).First(&list).Error
	return
}

type BaseFromPurangIndexType struct {
	Type2 string `gorm:"column:type_2"`
	Type3 string `gorm:"column:type_3"`
}

// Update 更新Purang指标基础信息
func (item *BaseFromPurangIndex) Update(cols []string) (err error) {
	err = global.DbMap[utils.DbNameIndex].Select(cols).Updates(item).Error
	return
}

// EditPurangIndexInfoResp 新增指标的返回
type EditPurangIndexInfoResp struct {
	BaseFromPurangIndexId int    `description:"指标ID"`
	IndexCode              string `description:"指标code"`
}

type PurangIndexSource2EdbReq struct {
	EdbCode       string
	EdbName       string
	Frequency     string
	Unit          string
	ClassifyId    int
	AdminId       int
	AdminRealName string
}

func GetPurangFrequencyByClassifyId(classifyId int) (items []*GlFrequency, err error) {
	sql := ` SELECT frequency FROM base_from_purang_index WHERE classify_id = ? `
	sql += ` GROUP BY frequency ORDER BY frequency ASC `
	err = global.DbMap[utils.DbNameIndex].Raw(sql, classifyId).Find(&items).Error
	return
}