package youngee_talent_service

import (
	"youngmini_server/app/dao"
	"youngmini_server/app/model/youngee_talent_model"
	"youngmini_server/app/utils"

	"github.com/gogf/gf/frame/g"
	"github.com/gogf/gf/net/ghttp"
)

func OnPostTalentInfo(r *ghttp.Request) *TalentHttpResult {
	tid, err := utils.SessionTalentInfo.GetTalentIdFromSession(r)
	if err != nil {
		return &TalentHttpResult{Code: -1, Msg: "Get talent info failed"}
	}
	var infos *youngee_talent_model.TalentSelfInputInfo
	err = r.ParseForm(&infos)
	if err != nil {
		return &TalentHttpResult{Code: -2, Msg: err.Error()}
	}
	infos.IsBindInfo = 1
	_, err = g.DB().Model(dao.YoungeeTalentInfo.Table).Update(infos, "id", tid)
	if err != nil {
		return &TalentHttpResult{Code: -4, Msg: "update failed"}
	}

	return &TalentHttpResult{Code: 0, Msg: "success"}
}

func OnUpdateTalentInfo(r *ghttp.Request) *TalentHttpResult {
	tid, err := utils.SessionTalentInfo.GetTalentIdFromSession(r)
	if err != nil {
		return &TalentHttpResult{Code: -1, Msg: "Get talent info failed"}
	}
	var infos *youngee_talent_model.TalentSelfInputInfo
	err = r.ParseForm(&infos)
	if err != nil {
		return &TalentHttpResult{Code: -2, Msg: err.Error()}
	}
	_, err = g.DB().Model(dao.YoungeeTalentInfo.Table).Data(g.Map{"talent_wx_nickname": infos.TalentWxNickname, "avatar": infos.Avatar}).Where("id = ?", tid).Update()
	if err != nil {
		return &TalentHttpResult{Code: -4, Msg: "update failed"}
	}

	return &TalentHttpResult{Code: 0, Msg: "success"}
}

func GetTalentInfo(r *ghttp.Request) *TalentHttpResult {
	tid, err := utils.SessionTalentInfo.GetTalentIdFromSession(r)
	if err != nil {
		return &TalentHttpResult{Code: -1, Msg: "Get talent info failed"}
	}

	talentInfo := youngee_talent_model.TalentInfo{}
	err = g.DB().Model("youngee_talent_info").WithAll().Where("id", tid).Scan(&talentInfo)

	if err != nil {
		return &TalentHttpResult{Code: -2, Msg: "Get talent info failed"}
	}
	return &TalentHttpResult{Code: 0, Msg: "success", Data: talentInfo}
}

// 查询任务数量、young之团数量、选品任务数量
func GetMyInfoNum(r *ghttp.Request) *TalentHttpResult {
	tid, err := utils.SessionTalentInfo.GetTalentIdFromSession(r)
	if err != nil {
		return &TalentHttpResult{Code: -1, Msg: "Get talent info failed"}
	}
	talentInfoNum := youngee_talent_model.TalentInfoNum{}
	// 查询任务相关数量
	var endTaskStageList = [3]int{3, 5, 16}
	whereCondition := g.Map{
		dao.YoungeeTaskInfo.Columns.TalentId: tid,
	}

	talentInfoNum.AllTaskNum, err = g.DB().Model(dao.YoungeeTaskInfo.Table).Where(whereCondition).Count()
	if err != nil {
		return &TalentHttpResult{Code: -2, Msg: "Get task num failed"}
	}
	talentInfoNum.ApplyTaskNum, err = g.DB().Model(dao.YoungeeTaskInfo.Table).Where("talent_id = ? and task_stage = 1", tid).Count()
	if err != nil {
		return &TalentHttpResult{Code: -2, Msg: "Get task num failed"}
	}
	talentInfoNum.ExeTaskNum, err = g.DB().Model(dao.YoungeeTaskInfo.Table).Where("talent_id = ? and task_stage >= 4 and task_stage <= 14", tid).Count()
	if err != nil {
		return &TalentHttpResult{Code: -2, Msg: "Get task num failed"}
	}
	talentInfoNum.EndTaskNum, err = g.DB().Model(dao.YoungeeTaskInfo.Table).Where("talent_id = ? and task_stage in (?)", tid, endTaskStageList).Count()
	if err != nil {
		return &TalentHttpResult{Code: -2, Msg: "Get task num failed"}
	}

	// 查询young之团相关数量
	talentInfoNum.ExeTeamNum, err = g.DB().Model(dao.YounggeeTalentTeam.Table).Where("talent_id = ? and team_status = 2", tid).Count()
	if err != nil {
		return &TalentHttpResult{Code: -2, Msg: "Get task num failed"}
	}
	talentInfoNum.EndTeamNum, err = g.DB().Model(dao.YounggeeTalentTeam.Table).Where("talent_id = ? and team_status = 3", tid).Count()
	if err != nil {
		return &TalentHttpResult{Code: -2, Msg: "Get task num failed"}
	}

	// 查询选品任务相关数量
	var signSecTaskStageList = []int{2, 3}
	var exeSecTaskStageList = []int{6, 7, 8}
	var endSecTaskStageList = []int{5, 9, 10}
	talentInfoNum.AllSecTaskNum, err = g.DB().Model(dao.YounggeeSecTaskInfo.Table).Where("talent_id = ?", tid).Count()
	if err != nil {
		return &TalentHttpResult{Code: -2, Msg: "Get task num failed"}
	}
	talentInfoNum.ApplySecTaskNum, err = g.DB().Model(dao.YounggeeSecTaskInfo.Table).Where("talent_id = ? and task_stage in (?)", tid, signSecTaskStageList).Count()
	if err != nil {
		return &TalentHttpResult{Code: -2, Msg: "Get task num failed"}
	}
	talentInfoNum.ExeSecTaskNum, err = g.DB().Model(dao.YounggeeSecTaskInfo.Table).Where("talent_id = ? and task_stage in (?)", tid, exeSecTaskStageList).Count()
	if err != nil {
		return &TalentHttpResult{Code: -2, Msg: "Get task num failed"}
	}
	talentInfoNum.EndSecTaskNum, err = g.DB().Model(dao.YounggeeSecTaskInfo.Table).Where("talent_id = ? and task_stage in (?) ", tid, endSecTaskStageList).Count()
	if err != nil {
		return &TalentHttpResult{Code: -2, Msg: "Get task num failed"}
	}

	return &TalentHttpResult{Code: 0, Msg: "success", Data: talentInfoNum}
}