import 'package:auto_gpt_flutter_client/models/skill_tree/skill_tree_edge.dart';
import 'package:auto_gpt_flutter_client/models/skill_tree/skill_tree_node.dart';
import 'package:flutter/foundation.dart';
import 'package:graphview/GraphView.dart';

class SkillTreeViewModel extends ChangeNotifier {
  List<SkillTreeNode> _skillTreeNodes = [];
  List<SkillTreeEdge> _skillTreeEdges = [];
  SkillTreeNode? _selectedNode;

  SkillTreeNode? get selectedNode => _selectedNode;

  final Graph graph = Graph()..isTree = true;
  BuchheimWalkerConfiguration builder = BuchheimWalkerConfiguration();

  void initializeSkillTree() {
    _skillTreeNodes = [];
    _skillTreeEdges = [];
    _selectedNode = null;

    graph.nodes.clear();
    graph.edges.clear();

    // Add nodes to _skillTreeNodes
    _skillTreeNodes.addAll([
      SkillTreeNode(color: 'red', id: 1),
      SkillTreeNode(color: 'blue', id: 2),
      SkillTreeNode(color: 'green', id: 3),
      SkillTreeNode(color: 'yellow', id: 4),
      SkillTreeNode(color: 'orange', id: 5),
      SkillTreeNode(color: 'purple', id: 6),
      SkillTreeNode(color: 'brown', id: 7),
      SkillTreeNode(color: 'pink', id: 8),
      SkillTreeNode(color: 'grey', id: 9),
      SkillTreeNode(color: 'cyan', id: 10),
      SkillTreeNode(color: 'magenta', id: 11),
      SkillTreeNode(color: 'lime', id: 12)
    ]);

    // Add edges to _skillTreeEdges
    _skillTreeEdges.addAll([
      SkillTreeEdge(id: '1_to_2', from: '1', to: '2', arrows: 'to'),
      SkillTreeEdge(id: '1_to_3', from: '1', to: '3', arrows: 'to'),
      SkillTreeEdge(id: '1_to_4', from: '1', to: '4', arrows: 'to'),
      SkillTreeEdge(id: '2_to_5', from: '2', to: '5', arrows: 'to'),
      SkillTreeEdge(id: '2_to_6', from: '2', to: '6', arrows: 'to'),
      SkillTreeEdge(id: '6_to_7', from: '6', to: '7', arrows: 'to'),
      SkillTreeEdge(id: '6_to_8', from: '6', to: '8', arrows: 'to'),
      SkillTreeEdge(id: '4_to_9', from: '4', to: '9', arrows: 'to'),
      SkillTreeEdge(id: '4_to_10', from: '4', to: '10', arrows: 'to'),
      SkillTreeEdge(id: '4_to_11', from: '4', to: '11', arrows: 'to'),
      SkillTreeEdge(id: '11_to_12', from: '11', to: '12', arrows: 'to')
    ]);

    builder
      ..siblingSeparation = (100)
      ..levelSeparation = (150)
      ..subtreeSeparation = (150)
      ..orientation = (BuchheimWalkerConfiguration.ORIENTATION_LEFT_RIGHT);

    notifyListeners();
  }

  void toggleNodeSelection(int nodeId) {
    if (_selectedNode?.id == nodeId) {
      // Unselect the node if it's already selected
      _selectedNode = null;
    } else {
      // Select the new node
      _selectedNode = _skillTreeNodes.firstWhere((node) => node.id == nodeId);
    }
    notifyListeners();
  }

  // Getter to expose nodes for the View
  List<SkillTreeNode> get skillTreeNodes => _skillTreeNodes;

  // Getter to expose edges for the View
  List<SkillTreeEdge> get skillTreeEdges => _skillTreeEdges;
}