youngee_mc_api/app/service/youngee_task_service/task_data.go

package youngee_task_service

import (
	"fmt"
	"github.com/gogf/gf/frame/g"
	"github.com/gogf/gf/net/ghttp"
	"github.com/gogf/gf/os/gtime"
	"github.com/gogf/gf/util/gconv"
	"time"
	"youngmini_server/app/dao"
	"youngmini_server/app/model"
	"youngmini_server/app/model/youngee_talent_model"
	"youngmini_server/app/utils"
)

// 添加数据service
func AddTaskData(r *ghttp.Request) *TalentHttpResult {
	var DataInfoReq *youngee_talent_model.AddTaskDataRequest
	err := r.ParseForm(&DataInfoReq)
	if err != nil {
		return &TalentHttpResult{Code: -1, Msg: err.Error()}
	}
	taskDataInfo := []model.YounggeeDataInfo{}
	err = g.DB().Model(model.YounggeeDataInfo{}).Where("task_id = ? ", DataInfoReq.TaskId).OrderDesc("create_at").Scan(&taskDataInfo)

	var autoAgreeAt *gtime.Time
	//获取种草和本地生活的task表
	taskInfo := youngee_talent_model.YoungeeTaskInfo{}
	localTaskInfo := youngee_talent_model.YoungeeLocalTaskInfo{}

	//查找auto_task_info表 AutoAgreeAt= task的创建时间 + auto_task_info的task_finish_auto
	if DataInfoReq.TaskType == 1 {
		err = g.DB().Model("youngee_task_info").WithAll().Where("task_id = ?", DataInfoReq.TaskId).Scan(&taskInfo)
		if err != nil {
			return &TalentHttpResult{Code: -2, Msg: "YoungeeTaskInfo find failed", Data: err.Error()}
		}
		autoTaskInfo, err := g.DB().Model("info_auto_task").Where("auto_task_id = ?", taskInfo.ProjectDetail.AutoTaskId).One()
		if err != nil {
			return &TalentHttpResult{Code: -2, Msg: "InfoAutoTask find failed", Data: err.Error()}
		}
		if autoTaskInfo != nil {
			autoAgreeAt = taskInfo.CreateDate.Add(time.Duration(autoTaskInfo["task_finish_auto"].Int64()) * time.Hour)
		} else {
			return &TalentHttpResult{Code: -2, Msg: "auto_task_id not found"}
		}
	} else if DataInfoReq.TaskType == 2 {
		err = g.DB().Model("youngee_local_task_info").WithAll().Where("task_id = ?", DataInfoReq.TaskId).Scan(localTaskInfo)
		if err != nil {
			return &TalentHttpResult{Code: -2, Msg: "YoungeeLocalTaskInfo find failed", Data: err.Error()}
		}
		autoTaskInfo, err := g.DB().Model("info_auto_task").WithAll().Where("auto_task_id = ?", localTaskInfo.LocalInfoDetail.AutoTaskId).One()
		if err != nil {
			return &TalentHttpResult{Code: -2, Msg: "InfoAutoTask find failed", Data: err.Error()}
		}
		if autoTaskInfo != nil {
			autoAgreeAt = localTaskInfo.CreateDate.Add(time.Duration(autoTaskInfo["task_finish_auto"].Int64()) * time.Hour)
		} else {
			return &TalentHttpResult{Code: -2, Msg: "auto_task_id not found"}
		}
	}
	//最新data数据
	DataInfo := model.YounggeeDataInfo{
		TaskId:        DataInfoReq.TaskId,
		LikeNumber:    DataInfoReq.LikeNumber,
		PlayNumber:    DataInfoReq.PlayNumber,
		CommentNumber: DataInfoReq.CommentNumber,
		CollectNumber: DataInfoReq.CollectNumber,
		PhotoUrl:      DataInfoReq.PhotoUrl,
		ForwardNumber: DataInfoReq.ForwardNumber,
		CreateAt:      gtime.Now(),
		IsReview:      0,
		IsSubmit:      0,
		IsOk:          0,
		AutoAgreeAt:   autoAgreeAt,
	}

	//上传过但是被拒了
	var condition1 bool = len(taskDataInfo) != 0 && taskDataInfo[0].IsReview == 1 && taskDataInfo[0].IsOk == 0
	//没有上传过初稿
	var condition2 bool = len(taskDataInfo) == 0

	if condition1 || condition2 {
		//插入新数据
		_, err := g.DB().Model(model.YounggeeDataInfo{}).Data(DataInfo).InsertAndGetId()
		if err != nil {
			return &TalentHttpResult{Code: -3, Msg: "YounggeeLinkInfo insert failed"}
		}
	} else {
		return &TalentHttpResult{Code: -1, Msg: "数据已提交，请勿重复提交"}
	}
	if DataInfoReq.TaskType == 1 { //种草
		DataStatus, err := g.DB().Model("youngee_task_info").Fields("data_status").Where("task_id = ?", DataInfoReq.TaskId).Value()
		if err != nil {
			return &TalentHttpResult{Code: -5, Msg: "Get task info failed"}
		}
		if DataStatus.Int64() == 1 {
			_, err = g.Model("youngee_task_info").Where("task_id = ?", DataInfoReq.TaskId).Update(g.Map{"data_status": 2})
			if err != nil {
				return &TalentHttpResult{Code: -6, Msg: "YoungeeTaskInfo update failed"}
			}
		} else if DataStatus.Int64() == 3 {
			_, err = g.Model("youngee_task_info").Where("task_id = ?", DataInfoReq.TaskId).Update(g.Map{"data_status": 4})
			if err != nil {
				return &TalentHttpResult{Code: -6, Msg: "YoungeeTaskInfo update failed"}
			}
		}

		taskInfo := youngee_talent_model.YoungeeTaskInfo{}
		// 查询是否处于违约状态
		err = g.DB().Model("youngee_task_info").Where("task_id = ?", DataInfoReq.TaskId).Scan(&taskInfo)
		//修改task表的属性
		err1 := g.DB().Model("youngee_task_info").Where("task_id = ?", DataInfoReq.TaskId).Scan(&taskInfo)
		if err1 != nil {
			return &TalentHttpResult{Code: -1, Msg: "YoungeeTaskInfo find failed"}
		}
		if taskInfo.CurDefaultType == 7 || taskInfo.CurDefaultType == 9 {
			// 若处于违约状态则解除并更新企业应支付金额
			_, err1 = g.DB().Model("youngee_task_info").Data(g.Map{"cur_default_type": "0", "real_payment": taskInfo.AllPayment, "err_break_rate": 0}).Where("task_id = ?", DataInfoReq.TaskId).Update()
			if err1 != nil {
				return &TalentHttpResult{Code: -2, Msg: "YoungeeTaskInfo update failed"}
			}

			// 更新违约记录表
			_, err1 = g.DB().Model("youngee_task_info").Data(g.Map{"default_status": 2}).Where("task_id = ? and default_status in (?)", DataInfoReq.TaskId, g.Slice{1, 3, 4}).Update()
			if err1 != nil {
				return &TalentHttpResult{Code: -2, Msg: "YoungeeContractInfo update failed"}
			}
		}

		//设置为数据待审
		_, err = g.DB().Model("youngee_task_info").Data(g.Map{"task_stage": "14"}).Where("task_id = ?", DataInfoReq.TaskId).Update()
		if err != nil {
			return &TalentHttpResult{Code: -4, Msg: "YoungeeTaskInfo update failed"}
		}
		// 记录任务日志-上传数据
		taskLog := model.YounggeeTaskLog{
			TaskId:  DataInfoReq.TaskId,
			Content: "上传数据",
			LogAt:   gtime.Now(),
		}
		_, err = g.DB().Model(dao.YounggeeTaskLog.Table).Data(&taskLog).Insert()
		if err != nil {
			return &TalentHttpResult{Code: -5, Msg: "YounggeeTaskLog insert failed"}
		}
	} else if DataInfoReq.TaskType == 2 { //本地生活
		DataStatus, err := g.DB().Model("youngee_local_task_info").Fields("data_status").Where("task_id = ?", DataInfoReq.TaskId).Value()
		if err != nil {
			return &TalentHttpResult{Code: -5, Msg: "Get task info failed"}
		}
		if DataStatus.Int64() == 1 {
			_, err = g.Model("youngee_local_task_info").Where("task_id = ?", DataInfoReq.TaskId).Update(g.Map{"data_status": 2})
			if err != nil {
				return &TalentHttpResult{Code: -6, Msg: "YoungeeTaskInfo update failed"}
			}
		} else if DataStatus.Int64() == 3 {
			_, err = g.Model("youngee_local_task_info").Where("task_id = ?", DataInfoReq.TaskId).Update(g.Map{"data_status": 4})
			if err != nil {
				return &TalentHttpResult{Code: -6, Msg: "YoungeeTaskInfo update failed"}
			}
		}

		taskInfo := youngee_talent_model.YoungeeLocalTaskInfo{}
		// 查询是否处于违约状态
		err = g.DB().Model(model.YoungeeTaskInfo{}).Where("task_id = ?", DataInfoReq.TaskId).Scan(&taskInfo)
		//修改task表的属性
		err1 := g.DB().Model(model.YoungeeTaskInfo{}).Where("task_id = ?", DataInfoReq.TaskId).Scan(&taskInfo)
		if err1 != nil {
			return &TalentHttpResult{Code: -1, Msg: "YoungeeTaskInfo find failed"}
		}
		if taskInfo.CurDefaultType == 7 || taskInfo.CurDefaultType == 9 {
			// 若处于违约状态则解除并更新企业应支付金额
			_, err1 = g.DB().Model("youngee_local_task_info").Data(g.Map{"cur_default_type": "0", "real_payment": taskInfo.AllPayment, "err_break_rate": 0}).Where("task_id = ?", DataInfoReq.TaskId).Update()
			if err1 != nil {
				return &TalentHttpResult{Code: -2, Msg: "YoungeeTaskInfo update failed"}
			}

			// 更新违约记录表
			_, err1 = g.DB().Model(model.YoungeeContractInfo{}).Data(g.Map{"default_status": 2}).Where("task_id = ? and default_status in (?)", DataInfoReq.TaskId, g.Slice{1, 3, 4}).Update()
			if err1 != nil {
				return &TalentHttpResult{Code: -2, Msg: "YoungeeContractInfo update failed"}
			}
		}

		//设置为数据待审
		_, err = g.DB().Model("youngee_local_task_info").Data(g.Map{"task_stage": "14"}).Where("task_id = ?", DataInfoReq.TaskId).Update()
		if err != nil {
			return &TalentHttpResult{Code: -4, Msg: "YoungeeTaskInfo update failed"}
		}
		// 记录任务日志-上传数据
		taskLog := youngee_talent_model.YounggeeLocalTaskLog{
			TaskId:  DataInfoReq.TaskId,
			Content: "上传数据",
			LogAt:   gtime.Now(),
		}
		_, err = g.DB().Model("younggee_local_task_log").Data(&taskLog).Insert()
		if err != nil {
			return &TalentHttpResult{Code: -5, Msg: "YounggeeTaskLog insert failed", Data: err.Error()}
		}

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

// 提交数据service
//func SubmitTaskData(r *ghttp.Request) *TalentHttpResult {
//	taskId, _ := r.Get("task_id").(string)
//	taskInfo := model.YoungeeTaskInfo{}
//	// 查询是否处于违约状态
//	err1 := g.DB().Model(model.YoungeeTaskInfo{}).Where("task_id = ?", taskId).Scan(&taskInfo)
//	if err1 != nil {
//		return &TalentHttpResult{Code: -1, Msg: "YoungeeTaskInfo find failed"}
//	}
//	if taskInfo.CurDefaultType == 7 || taskInfo.CurDefaultType == 9 {
//		// 若处于违约状态则解除并更新企业应支付金额
//		_, err1 = g.DB().Model(model.YoungeeTaskInfo{}).Data(g.Map{"cur_default_type": "0", "real_payment": taskInfo.AllPayment, "err_break_rate": 0}).Where("task_id = ?", taskId).Update()
//		if err1 != nil {
//			return &TalentHttpResult{Code: -2, Msg: "YoungeeTaskInfo update failed"}
//		}
//
//		// 更新违约记录表
//		_, err1 = g.DB().Model(model.YoungeeContractInfo{}).Data(g.Map{"default_status": 2}).Where("task_id = ? and default_status in (?)", taskId, g.Slice{1, 3, 4}).Update()
//		if err1 != nil {
//			return &TalentHttpResult{Code: -2, Msg: "YoungeeContractInfo update failed"}
//		}
//	}
//
//	// 查询该任务是否有已添加或已修改数据
//	res, err := g.DB().Model(model.YounggeeDataInfo{}).Where("task_id = ? and is_submit = 0", taskId).Count()
//	if err != nil {
//		return &TalentHttpResult{Code: -1, Msg: "YounggeeDataInfo find failed"}
//	}
//	if res == 1 && (taskInfo.DataStatus == 2 || taskInfo.DataStatus == 4) {
//		_, err = g.DB().Model(model.YounggeeDataInfo{}).Data(g.Map{"is_submit": "1", "submit_at": gtime.Now()}).Where("task_id = ? and is_submit = 0", taskId).Update()
//		if err != nil {
//			return &TalentHttpResult{Code: -3, Msg: "YounggeeDataInfo update failed"}
//		}
//		_, err = g.DB().Model(model.YoungeeTaskInfo{}).Data(g.Map{"task_stage": "14"}).Where("task_id = ?", taskId).Update()
//		if err != nil {
//			return &TalentHttpResult{Code: -4, Msg: "YoungeeTaskInfo update failed"}
//		}
//		// 记录任务日志-上传数据
//		taskLog := model.YounggeeTaskLog{
//			TaskId:  taskId,
//			Content: "上传数据",
//			LogAt:   gtime.Now(),
//		}
//		_, err = g.DB().Model(dao.YounggeeTaskLog.Table).Data(&taskLog).Insert()
//		if err != nil {
//			return &TalentHttpResult{Code: -5, Msg: "YounggeeTaskLog insert failed"}
//		}
//		projectInfo := model.ProjectInfo{}
//		err1 := g.DB().Model(model.ProjectInfo{}).Where("project_id = ?", taskInfo.ProjectId).Scan(&projectInfo)
//		if err1 != nil {
//			return &TalentHttpResult{Code: -8, Msg: "ProjectInfo find failed"}
//		}
//		messageInfo := model.YounggeeMessageInfo{
//			MessageId:   13,
//			MessageType: 2,
//			CreatedAt:   gtime.Now(),
//			TalentId:    taskInfo.TalentId,
//			ProjectName: projectInfo.ProjectName,
//			IsReaded:    0,
//			IsDeleted:   0,
//		}
//		_, err = g.DB().Model(dao.YounggeeMessageInfo.Table).Data(&messageInfo).Insert()
//		if err != nil {
//			return &TalentHttpResult{Code: -9, Msg: "YounggeeMessageInfo insert failed"}
//		}
//	}
//
//	return &TalentHttpResult{Code: 0, Msg: "success"}
//}

// 查询数据提交审阅记录service
func GetTaskData(r *ghttp.Request) *TalentHttpResult {
	taskId := r.Get("task_id")

	res, err := g.DB().Model(dao.YounggeeDataInfo.Table).Where("is_review = 1 and task_id = ?", taskId).OrderAsc("submit_at").All()
	if err != nil {
		return &TalentHttpResult{Code: -1, Msg: err.Error()}
	}

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

// 查询未提交数据service
func GetUnSubmitTaskData(r *ghttp.Request) *TalentHttpResult {
	taskId := r.Get("task_id")

	res, err := g.DB().Model(dao.YounggeeDataInfo.Table).Where("is_review = 0 and is_ok = 0 and task_id = ?", taskId).One()
	if err != nil {
		return &TalentHttpResult{Code: -1, Msg: err.Error()}
	}

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

// 收藏种草项目
func ProjectCollection(r *ghttp.Request) *TalentHttpResult {
	tId, _ := utils.SessionTalentInfo.GetTalentIdFromSession(r)
	pId := r.GetQueryString("project_id", "")
	if pId == "" {
		return &TalentHttpResult{Code: -1, Msg: "参数传递有误"}
	}
	record, err := g.DB().Model("younggee_project_collect_info").Where("talent_id = ? and project_id = ?", tId, pId).One()
	if err != nil {
		return &TalentHttpResult{Code: -1, Msg: err.Error()}
	}
	if record == nil {
		collectionInfo := youngee_talent_model.ProjectCollection{
			TalentId:   tId,
			ProjectId:  pId,
			CreateTime: gtime.Now(),
			Deleted:    0,
		}

		_, err = g.DB().Model("younggee_project_collect_info").Data(collectionInfo).Insert()

	} else {
		// 如果记录存在，更新 Delete 字段的值 和 收藏时间
		fmt.Println("*******", record)
		currentDeleteValue := gconv.Int(record["deleted"])
		newDeleteValue := 1 - currentDeleteValue // 0 改成 1，1 改成 0
		_, err = g.DB().Model("younggee_project_collect_info").
			Where("talent_id = ? and project_id = ?", tId, pId).
			Data(g.Map{
				"deleted":     newDeleteValue,
				"create_time": gtime.Now(),
			}).Update()
		if err != nil {
			return &TalentHttpResult{Code: -2, Msg: err.Error()}
		}
	}
	return &TalentHttpResult{Code: 0, Msg: "collection success"}
}

// 收藏带货项目
//func SelectionCollection(r *ghttp.Request) *TalentHttpResult {
//	tId, _ := utils.SessionTalentInfo.GetTalentIdFromSession(r)
//	selectionId := r.GetQueryString("selection_id")
//	productionId := r.GetQueryString("production_id")
//	record, err := g.DB().Model("younggee_selection_collect_info").Where("talent_id = ? and selection_id = ?", tId, selectionId).One()
//	if err != nil {
//		return &TalentHttpResult{Code: -1, Msg: err.Error()}
//	}
//	if record == nil {
//		collectionInfo := youngee_talent_model.SelectionCollection{
//			TalentId:    tId,
//			SelectionId: selectionId,
//			ProductId:   productionId,
//			CreateTime:  gtime.Now(),
//			Deleted:     0,
//		}
//		_, err = g.DB().Model("younggee_selection_collect_info").Data(collectionInfo).Insert()
//
//	} else {
//		// 如果记录存在，更新 Delete 字段的值 和 收藏时间
//		fmt.Println("*******", record)
//		currentDeleteValue := gconv.Int(record["deleted"])
//		newDeleteValue := 1 - currentDeleteValue // 0 改成 1，1 改成 0
//		_, err = g.DB().Model("younggee_selection_collect_info").
//			Where("talent_id = ? and selection_id = ?", tId, selectionId).
//			Data(g.Map{
//				"deleted":     newDeleteValue,
//				"create_time": gtime.Now(),
//			}).Update()
//		if err != nil {
//			return &TalentHttpResult{Code: -1, Msg: err.Error()}
//		}
//	}
//	return &TalentHttpResult{Code: 0, Data: record["deleted"], Msg: "collection success"}
//}

func SelectionCollection(r *ghttp.Request) *TalentHttpResult {
	tId, _ := utils.SessionTalentInfo.GetTalentIdFromSession(r)
	selectionId := r.GetQueryString("selection_id")
	productionId := r.GetQueryString("production_id")
	collectionRes := []youngee_talent_model.SelectionCollection{}
	err := g.DB().Model("younggee_selection_collect_info").Where("talent_id = ? and selection_id = ?", tId, selectionId).Scan(&collectionRes)
	if err != nil {
		return &TalentHttpResult{Code: -1, Msg: err.Error()}
	}
	// 1. 没有数据，插入新记录
	if len(collectionRes) == 0 {
		collectInfo := &youngee_talent_model.SelectionCollection{
			TalentId:    tId,
			SelectionId: selectionId,
			ProductId:   productionId,
			CreateTime:  gtime.Now(),
			Deleted:     0, // 默认未删除状态
		}
		_, err := g.DB().Model("younggee_selection_collect_info").Data(collectInfo).Insert()
		if err != nil {
			return &TalentHttpResult{Code: -1, Msg: err.Error()}
		}
		return &TalentHttpResult{Code: 0, Msg: "收藏成功", Data: 0} //收藏
	}

	// 2. 有数据，且 Deleted == 0，更新 Deleted 为 1（取消收藏）
	if len(collectionRes) != 0 && collectionRes[0].Deleted == 0 {
		_, err := g.DB().Model("younggee_selection_collect_info").Where("collection_id = ?", collectionRes[0].CollectionId).Data(g.Map{
			"Deleted":    1,
			"UpdateTime": gtime.Now(),
		}).Update()
		if err != nil {
			return &TalentHttpResult{Code: -1, Msg: err.Error()}
		}
		return &TalentHttpResult{Code: 0, Msg: "取消收藏成功", Data: 1} //取消收藏
	}

	// 3. 有数据，且 Deleted == 1，更新 Deleted 为 0（重新收藏）
	if len(collectionRes) != 0 && collectionRes[0].Deleted == 1 {
		_, err := g.DB().Model("younggee_selection_collect_info").Where("collection_id = ?", collectionRes[0].CollectionId).Data(g.Map{
			"Deleted":    0,
			"UpdateTime": gtime.Now(),
		}).Update()
		if err != nil {
			return &TalentHttpResult{Code: -1, Msg: err.Error()}
		}
		return &TalentHttpResult{Code: 0, Msg: "重新收藏成功"}
	}

	return &TalentHttpResult{Code: 0, Msg: "操作成功"}

}