package data_source

import (
	"eta/eta_api/utils"
	"github.com/beego/beego/v2/client/orm"
	"github.com/rdlucklib/rdluck_tools/paging"
	"time"
)

type BaseFromIcpiIndex struct {
	BaseFromIcpiIndexId    int       `orm:"column(base_from_icpi_index_id);pk"`
	BaseFromIcpiClassifyId int       `description:"分类id"`
	IndexCode              string    `description:"指标编码"`
	IndexName              string    `description:"指标名称"`
	Frequency              string    `description:"频度"`
	StartDate              time.Time `description:"开始日期"`
	EndDate                time.Time `description:"结束日期"`
	CreateTime             time.Time `description:"创建时间"`
	ModifyTime             time.Time `description:"修改时间"`
	Unit                   string    `description:"单位"`
	LatestValue            float64   `description:"最新值"`
}

func (obj *BaseFromIcpiIndex) toView() *BaseFromIcpiIndexView {
	return &BaseFromIcpiIndexView{
		BaseFromIcpiIndexId:    obj.BaseFromIcpiIndexId,
		BaseFromIcpiClassifyId: obj.BaseFromIcpiClassifyId,
		IndexCode:              obj.IndexCode,
		IndexName:              obj.IndexName,
		Frequency:              obj.Frequency,
		Unit:                   obj.Unit,
		StartDate:              obj.StartDate.Format(utils.FormatDate),
		EndDate:                obj.EndDate.Format(utils.FormatDate),
		CreateTime:             obj.CreateTime.Format(utils.FormatDateTime),
		ModifyTime:             obj.ModifyTime.Format(utils.FormatDateTime),
	}
}

type BaseFromIcpiData struct {
	BaseFromIcpiDataId  int       `orm:"column(base_from_icpi_data_id);pk"`
	BaseFromIcpiIndexId int       `description:"指标id"`
	IndexCode           string    `description:"指标编码"`
	DataTime            time.Time `description:"日期"`
	Value               string    `description:"值"`
	CreateTime          time.Time `description:"创建时间"`
	ModifyTime          time.Time `description:"修改时间"`
}

type BaseFromIcpiClassify struct {
	BaseFromIcpiClassifyId int    `orm:"column(base_from_icpi_classify_id);pk"`
	ClassifyName           string `description:"分类名称"`
	ClassifyNameEn         string `description:"英文名称"`
	ParentId               int    `description:"上级id"`
	CreateTime             string `description:"创建时间"`
	ModifyTime             string `description:"修改时间"`
}

// [2025-zsh-时间类型修复-chenhan]
type BaseFromIcpiClassifyOrm struct {
	BaseFromIcpiClassifyId int       `orm:"column(base_from_icpi_classify_id);pk"`
	ClassifyName           string    `description:"分类名称"`
	ClassifyNameEn         string    `description:"英文名称"`
	ParentId               int       `description:"上级id"`
	CreateTime             time.Time `description:"创建时间"`
	ModifyTime             time.Time `description:"修改时间"`
}

// [2025-zsh-时间类型修复-chenhan]
func (obj *BaseFromIcpiClassifyOrm) toView() *BaseFromIcpiClassify {
	return &BaseFromIcpiClassify{
		BaseFromIcpiClassifyId: obj.BaseFromIcpiClassifyId,
		ClassifyName:           obj.ClassifyName,
		ClassifyNameEn:         obj.ClassifyNameEn,
		ParentId:               obj.ParentId,
		CreateTime:             obj.CreateTime.Format(utils.FormatDateTime),
		ModifyTime:             obj.ModifyTime.Format(utils.FormatDateTime),
	}
}

func toClassifyList(list []*BaseFromIcpiClassifyOrm) (itemList []*BaseFromIcpiClassify) {
	for _, item := range list {
		itemList = append(itemList, item.toView())
	}
	return
}
func (obj *BaseFromIcpiIndex) GetBaseFromIcpiClassifyAll() (list []*BaseFromIcpiClassify, err error) {
	o := orm.NewOrmUsingDB("data")
	sql := `SELECT * FROM base_from_icpi_classify `
	var ormList []*BaseFromIcpiClassifyOrm
	_, err = o.Raw(sql).QueryRows(&ormList)
	if err != nil {
		return
	}
	list = toClassifyList(ormList)
	return
}

type BaseFromIcpiDataView struct {
	BaseFromIcpiDataId  int       `orm:"column(base_from_icpi_data_id);pk"`
	BaseFromIcpiIndexId int       `description:"指标id"`
	IndexCode           string    `description:"指标编码"`
	DataTime            string    `description:"日期"`
	Value               string    `description:"值"`
	CreateTime          time.Time `description:"创建时间"`
	ModifyTime          time.Time `description:"修改时间"`
}

func (obj *BaseFromIcpiData) toView() *BaseFromIcpiDataView {
	return &BaseFromIcpiDataView{
		BaseFromIcpiDataId:  obj.BaseFromIcpiDataId,
		BaseFromIcpiIndexId: obj.BaseFromIcpiIndexId,
		IndexCode:           obj.IndexCode,
		DataTime:            obj.DataTime.Format(utils.FormatDate),
		Value:               obj.Value,
		CreateTime:          obj.CreateTime,
		ModifyTime:          obj.ModifyTime,
	}
}

type BaseFromIcpiIndexView struct {
	BaseFromIcpiIndexId    int    `description:"指标id"`
	BaseFromIcpiClassifyId int    `description:"分类id"`
	IndexCode              string `description:"指标编码"`
	IndexName              string `description:"指标名称"`
	Frequency              string `description:"频度"`
	Unit                   string `description:"单位"`
	StartDate              string `description:"开始日期"`
	EndDate                string `description:"结束日期"`
	CreateTime             string `description:"创建时间"`
	ModifyTime             string `description:"修改时间"`
	DataList               []*BaseFromIcpiDataView
	Paging                 *paging.PagingItem `description:"分页数据"`
}

func toIcpiIndexViewList(list []*BaseFromIcpiIndex) (itemList []*BaseFromIcpiIndexView) {
	for _, item := range list {
		itemList = append(itemList, item.toView())
	}
	return
}

// [2025-zsh-时间类型修复-chenhan]
func toIndexDataViewList(list []*BaseFromIcpiData) (itemList []*BaseFromIcpiDataView) {
	for _, item := range list {
		itemList = append(itemList, item.toView())
	}
	return
}
func (obj *BaseFromIcpiIndex) GetIcpiIndex(condition string, pars interface{}) (items []*BaseFromIcpiIndexView, err error) {
	o := orm.NewOrmUsingDB("data")
	sql := ` SELECT * FROM base_from_icpi_index WHERE 1=1  `
	if condition != "" {
		sql += condition
	}
	sql += `ORDER BY base_from_icpi_index_id ASC `
	// [2025-zsh-时间类型修复-chenhan]
	var ormList []*BaseFromIcpiIndex
	_, err = o.Raw(sql, pars).QueryRows(&items)
	if err != nil {
		return
	}
	items = toIcpiIndexViewList(ormList)
	return
}

func (obj *BaseFromIcpiIndex) GetIcpiIndexDataCount(indexCode string) (count int, err error) {
	o := orm.NewOrmUsingDB("data")
	sql := ` SELECT COUNT(1) AS count  FROM base_from_icpi_data WHERE index_code=? `
	err = o.Raw(sql, indexCode).QueryRow(&count)
	return
}

func (obj *BaseFromIcpiIndex) GetIcpiIndexData(indexCode string, startSize, pageSize int) (items []*BaseFromIcpiDataView, err error) {
	o := orm.NewOrmUsingDB("data")
	sql := ` SELECT *  FROM base_from_icpi_data WHERE index_code=? ORDER BY data_time DESC LIMIT ?,? `
	// [2025-zsh-时间类型修复-chenhan]
	var ormList []*BaseFromIcpiData
	_, err = o.Raw(sql, indexCode, startSize, pageSize).QueryRows(&ormList)
	if err != nil {
		return
	}
	items = toIndexDataViewList(ormList)
	return
}

func (obj *BaseFromIcpiIndex) GetIcpiDataMaxCount(classifyId int) (count int, err error) {
	o := orm.NewOrmUsingDB("data")
	sql := `SELECT MAX(t.num) AS count FROM (
				SELECT COUNT(1) AS num  FROM base_from_icpi_index AS a
				INNER JOIN base_from_icpi_data AS b ON a.base_from_icpi_index_id=b.base_from_icpi_index_id
				WHERE a.base_from_icpi_classify_id=?
				GROUP BY a.base_from_icpi_index_id
			)AS t `
	err = o.Raw(sql, classifyId).QueryRow(&count)
	return
}

func (obj *BaseFromIcpiIndex) GetIcpiIndexDataByCode(indexCode string) (items []*BaseFromIcpiDataView, err error) {
	o := orm.NewOrmUsingDB("data")
	sql := ` SELECT *  FROM base_from_icpi_data WHERE index_code=? ORDER BY data_time DESC  `
	// [2025-zsh-时间类型修复-chenhan]
	var ormList []*BaseFromIcpiData
	// [2025-zsh-时间类型修复-chenhan]
	_, err = o.Raw(sql, indexCode).QueryRows(&ormList)
	if err != nil {
		return
	}
	items = toIndexDataViewList(ormList)
	return
}