package dao

import (
	"context"
	"rag/consts/public"
	"rag/model/dto"
	"rag/model/entity"

	"gitea.com/red-future/common/db/gfdb"
	"github.com/gogf/gf/v2/util/gconv"
)

var Task = new(taskDao)

type taskDao struct{}

// Insert 创建任务
func (d *taskDao) Insert(ctx context.Context, req *dto.CreateTaskReq) (id int64, err error) {
	var res *entity.Task
	if err = gconv.Struct(req, &res); err != nil {
		return
	}
	r, err := gfdb.DB(ctx, public.DbNameKnowledge).Model(ctx, public.TableNameTask).Data(&res).Insert()
	if err != nil {
		return
	}
	return r.LastInsertId()
}

// Update 更新任务
func (d *taskDao) Update(ctx context.Context, req *dto.UpdateTaskReq) (rows int64, err error) {
	model := gfdb.DB(ctx, public.DbNameKnowledge).Model(ctx, public.TableNameTask).OmitEmpty()
	r, err := model.Data(&req).Where(entity.TaskCol.Id, req.Id).Where(entity.TaskCol.TaskId, req.TaskId).Update()
	if err != nil {
		return
	}
	return r.RowsAffected()
}

func (d *taskDao) Get(ctx context.Context, req *dto.GetTaskReq) (res []*entity.Task, total int, err error) {
	r, total, err := gfdb.DB(ctx, public.DbNameKnowledge).Model(ctx, public.TableNameTask).OmitEmpty().
		Where(entity.TaskCol.Id, req.Id).
		Where(entity.TaskCol.TaskId, req.TaskId).
		Where(entity.TaskCol.TaskType, req.TaskType).AllAndCount(false)
	if err != nil {
		return
	}
	err = r.Structs(&res)
	return
}

func (d *taskDao) DeleteByTaskId(ctx context.Context, req *dto.DeleteTaskByTaskIdReq) (rows int64, err error) {
	r, err := gfdb.DB(ctx, public.DbNameKnowledge).Model(ctx, public.TableNameTask).Where(entity.TaskCol.TaskId, req.TaskId).Delete()
	if err != nil {
		return
	}
	return r.RowsAffected()
}