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

/// Service class for performing task-related operations.
class TaskService {
  final RestApiUtility api;
  List<String> _deletedTaskIds = [];

  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<TaskResponse> listAllTasks(
      {int currentPage = 1, int pageSize = 10}) async {
    try {
      final response = await api
          .get('agent/tasks?current_page=$currentPage&page_size=$pageSize');
      return TaskResponse.fromJson(response);
    } 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<Map<String, dynamic>> listTaskArtifacts(String taskId,
      {int currentPage = 1, int pageSize = 10}) async {
    try {
      return await api.get(
          'agent/tasks/$taskId/artifacts?current_page=$currentPage&page_size=$pageSize');
    } catch (e) {
      throw Exception('Failed to list task artifacts: $e');
    }
  }

  Future<void> loadDeletedTasks() async {
    _deletedTaskIds =
        (await SharedPreferences.getInstance()).getStringList('deletedTasks') ??
            [];
    print("Deleted tasks fetched successfully!");
  }

  void saveDeletedTask(String taskId) {
    _deletedTaskIds.add(taskId);
    SharedPreferences.getInstance()
        .then((prefs) => prefs.setStringList('deletedTasks', _deletedTaskIds));
  }

  bool isTaskDeleted(String taskId) {
    return _deletedTaskIds.contains(taskId);
  }
}