import 'package:auto_gpt_flutter_client/models/task_request_body.dart';
import 'package:auto_gpt_flutter_client/utils/rest_api_utility.dart';

/// Service class for performing task-related operations.
class TaskService {
  final RestApiUtility api;

  TaskService(this.api);

  /// Creates a new task.
  ///
  /// [taskRequestBody] is a Map representing the request body for creating a task.
  Future<Map<String, dynamic>> createTask(
      TaskRequestBody taskRequestBody) async {
    try {
      return await api.post('agent/tasks', taskRequestBody.toJson());
    } catch (e) {
      throw Exception('Failed to create a new task: $e');
    }
  }

  /// Lists all tasks.
  ///
  /// [currentPage] and [pageSize] are optional pagination parameters.
  ///
  Future<List<dynamic>> listAllTasks(
      {int currentPage = 1, int pageSize = 10}) async {
    try {
      return await api
          .getList('agent/tasks?current_page=$currentPage&page_size=$pageSize');
    } catch (e) {
      throw Exception('Failed to list all tasks: $e');
    }
  }

  /// Gets details about a specific task.
  ///
  /// [taskId] is the ID of the task.
  Future<Map<String, dynamic>> getTaskDetails(String taskId) async {
    try {
      return await api.get('agent/tasks/$taskId');
    } catch (e) {
      throw Exception('Failed to get task details: $e');
    }
  }

  /// Lists all artifacts for a specific task.
  ///
  /// [taskId] is the ID of the task.
  /// [currentPage] and [pageSize] are optional pagination parameters.
  Future<List<dynamic>> listTaskArtifacts(String taskId,
      {int currentPage = 1, int pageSize = 10}) async {
    try {
      return await api.getList(
          'agent/tasks/$taskId/artifacts?current_page=$currentPage&page_size=$pageSize');
    } catch (e) {
      throw Exception('Failed to list task artifacts: $e');
    }
  }
}