using EpicGames.UHT.Types;
using System.Collections.Generic;
using System.Linq;
using UnrealSharpScriptGenerator.Utilities;

namespace UnrealSharpScriptGenerator;

public static class InclusionLists
{
    private static readonly IDictionary<string, HashSet<string>> BannedProperties = new Dictionary<string, HashSet<string>>();
    private static readonly HashSet<string> BannedEquality = new HashSet<string>();
    private static readonly HashSet<string> BannedArithmetic = new HashSet<string>();

    public static void BanProperty(string structName, string propertyName)
    {
        if (!BannedProperties.TryGetValue(structName, out var propertySet))
        {
            propertySet = new HashSet<string>();
            BannedProperties[structName] = propertySet;
        }

        propertySet.Add(propertyName);
    }
    
    public static bool HasBannedProperty(UhtProperty property)
    {
        return BannedProperties.TryGetValue(property.Outer!.SourceName, out var propertySet) && propertySet.Contains(property.SourceName);
    }

    public static void BanEquality(string structName)
    {
        BannedEquality.Add(structName);
    }

    public static bool HasBannedEquality(UhtStruct scriptStruct)
    {
        return BannedEquality.Contains(scriptStruct.GetStructName());
    }

    public static void BanArithmetic(string structName)
    {
        BannedArithmetic.Add(structName);
    }

    public static bool HasBannedArithmetic(UhtStruct scriptStruct)
    {
        return BannedArithmetic.Contains(scriptStruct.GetStructName());
    }
}