using System;
using System.Collections.Generic;
using Oxide.Core;
using Oxide.Core.Plugins;
using UnityEngine;

namespace Oxide.Plugins
{
    [Info("SupplyLock", "Ghosty", "1.0.4")]
    [Description("Locks Supplydrops to a player or team.")]
    public class SupplyLock : RustPlugin
    {
        Queue<Tuple<ulong, Vector3, DateTime>> supplySignals = new Queue<Tuple<ulong, Vector3, DateTime>>();
        Dictionary<string, ulong> supplyOwners = new Dictionary<string, ulong>();

        private void Init()
        {
            Puts("SupplyLock has been loaded!");
        }

        void OnExplosiveThrown(BasePlayer player, BaseEntity entity, ThrownWeapon item)
        {
            if (entity is SupplySignal)
            {
                if (entity.net == null || entity.net.ID == null || player == null)
                {
                    return;
                }
                supplySignals.Enqueue(Tuple.Create(player.userID.Get(), entity.transform.position, DateTime.Now));
            }
        }

        void OnEntitySpawned(BaseNetworkable entity)
        {
            if (entity is SupplyDrop drop)
            {  
                drop.MakeLootable(); //Im doing this as a fix for cargo, Yes i know its bad but i ran out of ideas..
                string dropIdStr = drop.net.ID.ToString();

                if (supplySignals.Count > 0)
                {
                    var signal = supplySignals.Dequeue();
                    supplyOwners[dropIdStr] = signal.Item1;
                }
            }
        }

        object CanLootEntity(BasePlayer player, StorageContainer container)
        {
            if (container is SupplyDrop drop)
            {
                string dropID = drop.net.ID.ToString();

                if (supplyOwners.TryGetValue(dropID, out ulong ownerID))
                {
                    if (ownerID == player.userID)
                    {
                        return null;
                    }

                    RelationshipManager.PlayerTeam team = RelationshipManager.ServerInstance.FindPlayersTeam(ownerID);
                    if (team != null && team.members.Contains(player.userID))
                    {
                        return null;
                    }
                    player.ChatMessage("<size=20><color=red>Supply</color>Lock</size>\n<size=15><color=white>This supply drop is locked!\nYou must be in a team with this player to loot the drop.</size>");
                    return false; // forgot to add a return here, whoops.
                }
            }

            return null;
        }
    }
}