package dao

import (
	"youngee_b_api/app/entity"
	"youngee_b_api/app/vo"
)

type LocalLifeTaskInfoDao struct{}

// 获取指定违约类型的本地生活子任务数量
func (d LocalLifeTaskInfoDao) CountByDefaultType(localId string, defaultType int64) int64 {
	var total int64
	Db.Model(&entity.LocalLifeTaskInfo{}).Where("local_id = ? AND cur_default_type = ?", localId, defaultType).Count(&total)
	return total
}

// 获取指定任务阶段的本地生活子任务数量
func (d LocalLifeTaskInfoDao) CountByTaskStage(localId string, taskStage int64) int64 {
	var total int64
	Db.Model(&entity.LocalLifeTaskInfo{}).Where("local_id = ? AND task_stage = ?", localId, taskStage).Count(&total)
	return total
}

// 根据openid获取报名的本地生活子任务阶段
func (d LocalLifeTaskInfoDao) CountAllByOpenid(openid string) int64 {
	var total int64
	Db.Model(&entity.LocalLifeTaskInfo{}).Where("open_id = ?", openid).Count(&total)
	return total
}

// 根据openid获取指定任务阶段的本地生活子任务阶段
func (d LocalLifeTaskInfoDao) CountByOpenid(openid string, taskStage int) int64 {
	var total int64
	Db.Model(&entity.LocalLifeTaskInfo{}).Where("open_id = ? AND task_stage = ?", openid, taskStage).Count(&total)
	return total
}

// 根据openid获取执行中的本地生活子任务阶段
func (d LocalLifeTaskInfoDao) CountExcuteNumByOpenid(openid string) int64 {
	var total int64
	Db.Model(&entity.LocalLifeTaskInfo{}).
		Where("open_id = ? AND task_stage NOT IN (?)", openid, []int{1, 3, 15, 16, 17}).
		Count(&total)
	return total
}

// 获取指定任务阶段的本地生活子任务
func (d LocalLifeTaskInfoDao) GetListByTaskStage(localId string, taskStage int64, time string, page int, pageSize int) ([]*entity.LocalLifeTaskInfo, int64, error) {
	var taskInfos []*entity.LocalLifeTaskInfo
	var total int64
	query := Db.Debug().Model(&entity.LocalLifeTaskInfo{}).Where("local_id = ? AND task_stage = ?", localId, taskStage)
	// 计算偏移量
	offset := (page - 1) * pageSize
	var err error
	if taskStage == 4 { // 待预约探店
		query.Count(&total)
		err = query.Order("select_date asc").Offset(offset).Limit(pageSize).Find(&taskInfos).Error
	} else if taskStage == 5 { // 待确认探店
		if time != "" {
			query = query.Where("DATE(reserve_time) = ?", time)
		}
		query.Count(&total)
		err = query.Order("reserve_time desc").Offset(offset).Limit(pageSize).Find(&taskInfos).Error
	}
	if err != nil {
		return nil, 0, err
	}
	return taskInfos, total, nil
}

// 获取指定探店状态的本地生活子任务
func (d LocalLifeTaskInfoDao) GetListByBookStatus(localId string, bookStatus int64, time string, page int, pageSize int) ([]*entity.LocalLifeTaskInfo, int64, error) {
	var taskInfos []*entity.LocalLifeTaskInfo
	var total int64
	query := Db.Model(&entity.LocalLifeTaskInfo{}).Where("local_id = ? AND book_status = ?", localId, bookStatus)
	// 计算偏移量
	offset := (page - 1) * pageSize
	var err error
	if bookStatus == 5 { // 待探店
		if time != "" {
			query = query.Where("DATE(explore_time) = ?", time)
		}
		query.Count(&total)
		err = query.Order("explore_time desc").Offset(offset).Limit(pageSize).Find(&taskInfos).Error
	} else if bookStatus == 6 { // 已探店
		if time != "" {
			query = query.Where("DATE(finish_explore_time) = ?", time)
		}
		query.Count(&total)
		err = query.Order("finish_explore_time desc").Offset(offset).Limit(pageSize).Find(&taskInfos).Error
	}
	if err != nil {
		return nil, 0, err
	}
	return taskInfos, total, nil
}

// 获取未传初稿的本地生活子任务数据
func (d LocalLifeTaskInfoDao) GetListBySketchDefault(param *vo.DefaultSearchParam) ([]entity.LocalLifeTaskInfo, int64, error) {
	localTaskInfos := []entity.LocalLifeTaskInfo{}
	var total int64
	query := Db.Model(&entity.LocalLifeTaskInfo{}).Where("local_id = ? AND cur_default_type = ?", param.TaskId, 4)
	query.Count(&total)
	query = query.Select("task_id, talent_id, settle_amount, draft_fee, sketch_missing_time, open_id")
	offset := (param.Page - 1) * param.PageSize
	if err := query.Order("sketch_missing_time desc").Offset(offset).Limit(param.PageSize).Find(&localTaskInfos).Error; err != nil {
		return nil, 0, err
	}
	return localTaskInfos, total, nil
}

// 获取未发作品的本地生活子任务数据
func (d LocalLifeTaskInfoDao) GetListByLinkDefault(param *vo.DefaultSearchParam) ([]entity.LocalLifeTaskInfo, int64, error) {
	localTaskInfos := []entity.LocalLifeTaskInfo{}
	var total int64
	query := Db.Model(&entity.LocalLifeTaskInfo{}).Where("local_id = ? AND cur_default_type = ?", param.TaskId, 6)
	query.Count(&total)
	query = query.Select("task_id, talent_id, settle_amount, draft_fee, link_missing_time, open_id")
	offset := (param.Page - 1) * param.PageSize
	if err := query.Order("link_missing_time desc").Offset(offset).Limit(param.PageSize).Find(&localTaskInfos).Error; err != nil {
		return nil, 0, err
	}
	return localTaskInfos, total, nil
}

// 获取未传数据的本地生活子任务数据
func (d LocalLifeTaskInfoDao) GetListByDataDefault(param *vo.DefaultSearchParam) ([]entity.LocalLifeTaskInfo, int64, error) {
	localTaskInfos := []entity.LocalLifeTaskInfo{}
	var total int64
	query := Db.Model(&entity.LocalLifeTaskInfo{}).Where("local_id = ? AND cur_default_type = ?", param.TaskId, 8)
	query.Count(&total)
	query = query.Select("task_id, talent_id, settle_amount, draft_fee, data_missing_time, open_id")
	offset := (param.Page - 1) * param.PageSize
	if err := query.Order("data_missing_time desc").Offset(offset).Limit(param.PageSize).Find(&localTaskInfos).Error; err != nil {
		return nil, 0, err
	}
	return localTaskInfos, total, nil
}

// 获取终止合作的本地生活子任务数据
func (d LocalLifeTaskInfoDao) GetListByTerminateDefault(param *vo.DefaultSearchParam) ([]entity.LocalLifeTaskInfo, int64, error) {
	localTaskInfos := []entity.LocalLifeTaskInfo{}
	var total int64
	query := Db.Debug().Model(&entity.LocalLifeTaskInfo{}).Where("local_id = ? AND task_stage = ?", param.TaskId, 17)
	query.Count(&total)
	query = query.Select("task_id, talent_id, settle_amount, draft_fee, terminate_time, terminate_reason, terminate_operator_type, terminate_operator, open_id")
	offset := (param.Page - 1) * param.PageSize
	if err := query.Order("terminate_time desc").Offset(offset).Limit(param.PageSize).Find(&localTaskInfos).Error; err != nil {
		return nil, 0, err
	}
	return localTaskInfos, total, nil
}

// 获取已解约的本地生活子任务数据
func (d LocalLifeTaskInfoDao) GetListByCancelDefault(param *vo.DefaultSearchParam) ([]entity.LocalLifeTaskInfo, int64, error) {
	localTaskInfos := []entity.LocalLifeTaskInfo{}
	var total int64
	query := Db.Debug().Model(&entity.LocalLifeTaskInfo{}).Where("local_id = ? AND task_stage = ?", param.TaskId, 16)
	query.Count(&total)
	query = query.Select("task_id, talent_id, settle_amount, draft_fee, cancel_time, cancel_reason, cancel_operator_type, cancel_operator, open_id")
	offset := (param.Page - 1) * param.PageSize
	if err := query.Order("cancel_time desc").Offset(offset).Limit(param.PageSize).Find(&localTaskInfos).Error; err != nil {
		return nil, 0, err
	}
	return localTaskInfos, total, nil
}

// 更新字段
func (d LocalLifeTaskInfoDao) UpdateField(taskId string, updateData map[string]interface{}) error {
	err := Db.Model(&entity.LocalLifeTaskInfo{}).Where("task_id = ?", taskId).Updates(updateData).Error
	if err != nil {
		return err
	}
	return nil
}

// 批量更新字段
func (d LocalLifeTaskInfoDao) UpdateFieldBatch(taskIds []string, updateData map[string]interface{}) error {
	err := Db.Model(&entity.LocalLifeTaskInfo{}).Where("task_id IN ?", taskIds).Updates(updateData).Error
	if err != nil {
		return err
	}
	return nil
}

// 批量更新字段
func (d LocalLifeTaskInfoDao) UpdateLocalStatus(taskIds []string, localLifeTaskInfo entity.LocalLifeTaskInfo) error {
	err := Db.Debug().Model(&entity.LocalLifeTaskInfo{}).Where("task_id IN ?", taskIds).Updates(localLifeTaskInfo).Error
	if err != nil {
		return err
	}
	return nil
}

func (d LocalLifeTaskInfoDao) GetLocalLifeInfoByTalentId(talentid string, others string, sortField []string, sortOrder []string, page int, pageSize int, reltype int, enterpriseid string, platform int) ([]*entity.LocalLifeTaskInfo, int64, error) {
	projectTaskInfos := []*entity.LocalLifeTaskInfo{}
	var total int64

	query := Db.Model(&entity.LocalLifeTaskInfo{}).
		Joins("JOIN younggee_local_life_info ylli ON ylli.local_id = youngee_local_task_info.local_id").
		Where("youngee_local_task_info.talent_id = ? AND youngee_local_task_info.task_stage = ? AND youngee_local_task_info.platform_id = ?", talentid, 15, platform)

	// 搜索条件（local_name）
	if others != "" {
		query = query.Where("ylli.local_name LIKE ?", "%"+others+"%")
	}

	if reltype == 2 {
		query = query.Where("ylli.enterprise_id = ?", enterpriseid)
	}

	// 排序处理
	if len(sortField) > 0 && len(sortOrder) > 0 && len(sortField) == len(sortOrder) {
		for i := 0; i < len(sortField); i++ {
			sortfield := sortField[i]
			sortorder := sortOrder[i]
			switch sortfield {
			case "commit_avg":
				if sortorder == "asc" {
					query = query.Order("youngee_local_task_info.commit_avg asc")
				} else {
					query = query.Order("youngee_local_task_info.commit_avg desc")
				}
			case "collect_num":
				if sortorder == "asc" {
					query = query.Order("youngee_local_task_info.collect_num asc")
				} else {
					query = query.Order("youngee_local_task_info.collect_num desc")
				}
			case "vote_avg":
				if sortorder == "asc" {
					query = query.Order("youngee_local_task_info.vote_avg asc")
				} else {
					query = query.Order("youngee_local_task_info.vote_avg desc")
				}

			}
		}
	}
	// 获取总数
	if err := query.Count(&total).Error; err != nil {
		return nil, 0, err
	}

	//// 选择字段（明确指定表名）
	query = query.Select(`
	   youngee_local_task_info.local_id,
	   youngee_local_task_info.task_id,
	   ylli.local_name,
	   youngee_local_task_info.commit_avg,
		youngee_local_task_info.collect_num,
		youngee_local_task_info.vote_avg`)

	// 分页查询
	offset := (page - 1) * pageSize
	err := query.Offset(offset).Limit(pageSize).Find(&projectTaskInfos).Error
	if err != nil {
		return nil, 0, err
	}

	return projectTaskInfos, total, nil

}