using System;
using System.Collections.Generic;
using System.Linq;
using MessagePack;
using Unity.Mathematics;
using static Unity.Mathematics.math;

[MessagePackObject]
public class SavedGame
{
    [Key(0)]
    public SavedZone[] Zones;
    
    [Key(1)]
    public Guid[] Factions;
    
    [Key(2)]
    public Dictionary<int, int> HomeZones;
    
    [Key(3)]
    public Dictionary<int, int> BossZones;
    
    [Key(4)]
    public int Entrance;
    
    [Key(5)]
    public int Exit;
    
    [Key(6)]
    public int CurrentZone;
    
    [Key(7)]
    public int CurrentZoneEntity;

    [Key(8)]
    public SectorBackgroundSettings Background;

    [Key(9)]
    public int[] DiscoveredZones;

    [Key(10)]
    public SavedActionBarBinding[] ActionBarBindings;

    [Key(11)]
    public bool IsTutorial;

    [Key(12)]
    public FactionRelationship[] Relationships;
    
    public SavedGame() { }

    public SavedGame(Sector sector, Zone currentZone, Entity currentEntity)
    {
        DiscoveredZones = sector.DiscoveredZones.Select(dz => Array.IndexOf(sector.Zones, dz)).ToArray();
        Background = sector.Background;
        Factions = sector.HomeZones.Keys.Select(f => f.ID).ToArray();
        Relationships = sector.Factions.Select(f => sector.FactionRelationships[f]).ToArray();
        
        HomeZones = sector.HomeZones.ToDictionary(
            x => Array.IndexOf(Factions, x.Key.ID),
            x => Array.IndexOf(sector.Zones, x.Value));
        BossZones = sector.BossZones.ToDictionary(
            x => Array.IndexOf(Factions, x.Key.ID),
            x => Array.IndexOf(sector.Zones, x.Value));
        
        Zones = sector.Zones.Select(zone => new SavedZone
        {
            Name = zone.Name,
            Position = zone.Position,
            AdjacentZones = zone.AdjacentZones.Select(az=> Array.IndexOf(sector.Zones, az)).ToArray(),
            Factions = zone.Factions.Select(f=> Array.IndexOf(Factions, f.ID)).ToArray(),
            Contents = zone.Contents?.PackZone(),
            Owner = zone.Owner == null ? -1 : Array.IndexOf(Factions, zone.Owner.ID)
        }).ToArray();

        CurrentZone = Array.FindIndex(sector.Zones, zone => zone.Contents == currentZone);
        CurrentZoneEntity = currentZone.Entities.IndexOf(currentEntity);
        
        Entrance = Array.IndexOf(sector.Zones, sector.Entrance);
        Exit = Array.IndexOf(sector.Zones, sector.Exit);
    }
}

[MessagePackObject]
public class SavedZone
{
    [Key(0)]
    public string Name;
    
    [Key(1)]
    public float2 Position;
    
    [Key(2)]
    public int[] AdjacentZones;
    
    [Key(3)]
    public int[] Factions;
    
    [Key(4)]
    public int Owner;
    
    [Key(5)]
    public ZonePack Contents;
}

[MessagePackObject,
 Union(0, typeof(SavedActionBarConsumableBinding)),
 Union(1, typeof(SavedActionBarGearBinding)),
 Union(2, typeof(SavedActionBarWeaponGroupBinding))
]
public abstract class SavedActionBarBinding
{
}

[MessagePackObject]
public class SavedActionBarConsumableBinding : SavedActionBarBinding
{
    [Key(0)] public DatabaseLink<ConsumableItemData> Target;
}

[MessagePackObject]
public class SavedActionBarGearBinding : SavedActionBarBinding
{
    [Key(0)] public int EquipmentIndex;
    [Key(1)] public int BehaviorIndex;
}

[MessagePackObject]
public class SavedActionBarWeaponGroupBinding : SavedActionBarBinding
{
    [Key(0)] public int Group;
}