package controller

import (
	"context"

	"rag-local/kb/model/dto"
	"rag-local/kb/service"
)

type parseTask struct{}

var ParseTask = &parseTask{}

func (c *parseTask) List(ctx context.Context, req *dto.ListParseTaskReq) (*dto.ListParseTaskRes, error) {
	list, total, err := service.ParseTaskService.List(ctx, req.Page, req.PageSize)
	if err != nil {
		return nil, err
	}
	return &dto.ListParseTaskRes{
		List:     list,
		Total:    total,
		Page:     req.Page,
		PageSize: req.PageSize,
	}, nil
}

func (c *parseTask) Retry(ctx context.Context, req *dto.RetryParseTaskReq) (*dto.RetryParseTaskRes, error) {
	if err := service.ParseTaskService.Retry(ctx, req.Id); err != nil {
		return nil, err
	}
	return &dto.RetryParseTaskRes{}, nil
}