using UnityEngine;
using UnityEditor;
using GraphProcessor;
using System;
using System.Collections.Generic;
using System.Linq;

public class FoldGraphView : BaseGraphView
{
    public FoldGraphView(EditorWindow window) : base(window) {}

    public override IEnumerable<(string path, Type type)> FilterCreateNodeMenuEntries()
    {
      return NodeProvider.GetNodeMenuEntries(graph)
        .Where(entry =>
            typeof(BaseFoldNode).IsAssignableFrom(entry.type) ||
            typeof(KeyframeNode).IsAssignableFrom(entry.type) ||
            // Whitelist a few built in nodes that we use.
            typeof(FloatNode).IsAssignableFrom(entry.type) ||
            typeof(GameObjectNode).IsAssignableFrom(entry.type) ||
            typeof(VectorNode).IsAssignableFrom(entry.type)
            );
    }
}