Auto-GPT/frontend/lib/viewmodels/task_viewmodel.dart

import 'package:auto_gpt_flutter_client/models/task.dart';
import 'package:auto_gpt_flutter_client/models/task_response.dart';
import 'package:flutter/foundation.dart';
import 'package:collection/collection.dart';
import 'package:auto_gpt_flutter_client/services/task_service.dart';
import 'package:auto_gpt_flutter_client/models/task_request_body.dart';

class TaskViewModel with ChangeNotifier {
  final TaskService _taskService;
  List<Task> _tasks = [];
  Task? _selectedTask; // This will store the currently selected task

  TaskViewModel(this._taskService);

  /// Returns the list of tasks.
  List<Task> get tasks => _tasks;

  /// Returns the currently selected task.
  Task? get selectedTask => _selectedTask;

  /// Adds a task and returns its ID.
  Future<String> createTask(String title) async {
    final newTask = TaskRequestBody(input: title);
    // Add to data source
    final createdTask = await _taskService.createTask(newTask);
    // Create a Task object from the created task response
    final newTaskObject =
        Task(id: createdTask['task_id'], title: createdTask['input']);

    // Update local tasks list and notify listeners
    _tasks.add(newTaskObject);
    notifyListeners();

    return newTaskObject.id; // Return the ID of the new task
  }

  /// Deletes a task.
  void deleteTask(String taskId) {
    _taskService.saveDeletedTask(taskId);
    tasks.removeWhere((task) => task.id == taskId);
    notifyListeners();
    print("Tasks deleted successfully!");
  }

  /// Fetches tasks from the data source.
  void fetchTasks() async {
    try {
      final TaskResponse tasksResponse = await _taskService.listAllTasks();
      final tasksFromApi = tasksResponse.tasks;
      _tasks = tasksFromApi
          .where((task) => !_taskService.isTaskDeleted(task.id))
          .toList();

      notifyListeners();
      print("Tasks fetched successfully!");
    } catch (error) {
      print("Error fetching tasks: $error");
    }
  }

  /// Handles the selection of a task by its ID.
  void selectTask(String id) {
    final task = _tasks.firstWhereOrNull((t) => t.id == id);

    if (task != null) {
      _selectedTask = task;
      print("Selected task with ID: ${task.id} and Title: ${task.title}");
      notifyListeners(); // Notify listeners to rebuild UI
    } else {
      final errorMessage =
          "Error: Attempted to select a task with ID: $id that does not exist in the data source.";
      print(errorMessage);
      throw ArgumentError(errorMessage);
    }
  }

  /// Deselects the currently selected task.
  void deselectTask() {
    _selectedTask = null;
    print("Deselected the current task.");
    notifyListeners(); // Notify listeners to rebuild UI
  }
}