Backyard-CTF/Backyard CTF/Assets/Scripts/Grid/GridUnit.cs

using UnityEngine;

public enum Team { Blue, Red }

public class GridUnit
{
    public int UnitId;
    public Team Team;
    public Vector2Int GridPosition;
    public int ConsecutiveDefenseMoves;
    public bool IsTaggedOut;
    public int RespawnTurnsRemaining;

    // Visual representation
    public GameObject GameObject;
    public SpriteRenderer SpriteRenderer;

    public GridUnit(int unitId, Team team, Vector2Int position, GameObject go)
    {
        UnitId = unitId;
        Team = team;
        GridPosition = position;
        ConsecutiveDefenseMoves = 0;
        IsTaggedOut = false;
        RespawnTurnsRemaining = 0;
        GameObject = go;
        SpriteRenderer = go.GetComponent<SpriteRenderer>();
    }

    public void SetWorldPosition(Vector2 worldPos)
    {
        if (GameObject != null)
        {
            GameObject.transform.position = new Vector3(worldPos.x, worldPos.y, 0);
        }
    }
}