using UnityEngine;
using System;
using System.Collections.Generic;

namespace Unheimlich.Data.DetectiveBoard
{
    [Serializable]
    public class PostItNodeData
    {
        public string NodeID;
        public Vector2 Position;
        public bool IsEliminated;
        public List<string> ConnectedNodeIDs = new List<string>();
        
        public PostItNodeData(string id, Vector2 position)
        {
            NodeID = id;
            Position = position;
            IsEliminated = false;
        }
    }
    
    [Serializable]
    public class DetectiveBoardSaveData
    {
        public List<PostItNodeData> Nodes = new List<PostItNodeData>();
        
        public PostItNodeData GetNodeData(string nodeID)
        {
            return Nodes.Find(n => n.NodeID == nodeID);
        }
        
        public void AddOrUpdateNode(PostItNodeData nodeData)
        {
            int index = Nodes.FindIndex(n => n.NodeID == nodeData.NodeID);
            if (index >= 0)
            {
                Nodes[index] = nodeData;
            }
            else
            {
                Nodes.Add(nodeData);
            }
        }
    }
}