using Autodesk.Revit.DB;
using Autodesk.Revit.UI;

namespace Szmedi.Toolkit.RvAssists
{
    public class EnableOnPlanOrSection : IExternalCommandAvailability
    {
        public bool IsCommandAvailable(UIApplication applicationData, CategorySet selectedCategories)
        {
            var view = applicationData?.ActiveUIDocument?.ActiveView;
            return view is ViewPlan or ViewSection;
        }
    }
}