using Autodesk.Revit.DB;
using Autodesk.Revit.DB.Events;
using Autodesk.Revit.DB.Plumbing;
using Autodesk.Revit.UI;
using Autodesk.Revit.UI.Events;
using DevExpress.Utils.Extensions;
using HStation.RevitDev.Model.ModelEnum;
using HStation.RevitDev.RevitDataExport.Common;
using HStation.RevitDev.RevitDataExport.Service;
using System;
using System.Collections.Generic;
using System.Linq;

namespace HStation.RevitDev.RevitDataExport.Utility
{
    public static class DocumentUtil
    {
        public static void RegistDocumentEvent(ExternalCommandData commandData)
        {
            commandData.Application.Application.DocumentChanged -= App_DocumentChanged;
            commandData.Application.Application.DocumentChanged += App_DocumentChanged;

            GlobalResource.CurrentUIDocument = commandData.Application.ActiveUIDocument;
            RegistViewEvent(commandData);
        }

        private static void App_DocumentChanged(object sender, DocumentChangedEventArgs e)
        {
            //当创建一个新文档时，先执行documentChanged事件，在执行documentCreated事件
            var doc = e.GetDocument();
            GlobalResource.CurrentDocument = doc;
            try
            {
                var addedIds = e.GetAddedElementIds();
                var removeIds = e.GetDeletedElementIds();

                AddElements(doc, addedIds);
                RemoveElements(removeIds);

                AddUnhandledElements();
                GlobalResource.InstancePanel.UpdateForm();
            }
            catch (Exception ex)
            {
                TaskDialog.Show("错误", ex.Message);
            }
        }

        //private static void UpdateLinkerInfo(Document doc)
        //{
        //    if (!doc.IsValidObject) { return; }

        //    foreach (var pair in GlobalResource.RevitModels)
        //    {
        //        foreach (var id in pair.Value)
        //        {
        //            if (id == null) { continue; }

        //            var element = doc.GetElement(new ElementId(int.Parse(id)));
        //            var linkedElems = element.GetConnectElements();
        //            var linkedIds = linkedElems.Select(x => x.Id.IntegerValue.ToString());
        //            model.LinkIds = string.Join(",", linkedIds);
        //        }
        //    }
        //}

        private static void RemoveElements(ICollection<ElementId> removeIds)
        {
            foreach (var id in removeIds)
            {
                GlobalResource.RevitModels.Remove(id.IntegerValue.ToString());
            }
            return;
        }

        private static void AddElements(Document doc, ICollection<ElementId> addedIds)
        {
            if (GlobalResource.PlacingType == RevitType.RFT_Pipe)
            {
                foreach (var id in addedIds)
                {
                    var elem = doc.GetElement(id);
                    if (elem is FamilyInstance fi)
                    {
                        AddAutoCreated(fi);
                    }
                    if (elem is Pipe)
                    {
                        GlobalResource.RevitModels.Add(elem, RevitType.RFT_Pipe);
                    }
                }
            }
            else if (GlobalResource.PlacingType != RevitType.RFT_Unknown)
            {
                foreach (var id in addedIds)
                {
                    var elem = doc.GetElement(id);
                    if (elem is FamilyInstance)
                    {
                        GlobalResource.RevitModels.Add(elem, GlobalResource.PlacingType);
                    }
                }
            }
            else
            {
                bool isPumpSystemElement = false;
                foreach (var id in addedIds)
                {
                    //手动布置的构件如消火栓、热换器等基本不能自动加入到系统中，能自动布置并连接
                    //系统的基本都是三通、四通、弯头、过渡件、管帽这些
                    if (IsConnectedWithPumpSystem(doc, id))
                    {
                        isPumpSystemElement = true;
                        break;
                    }
                }

                if (isPumpSystemElement)
                {
                    foreach (var id in addedIds)
                    {
                        AddToPumpSystem(doc, id);
                    }
                }
            }
            return;
        }

        private static bool IsConnectedWithPumpSystem(Document doc, ElementId id)
        {
            var elem = doc.GetElement(id);
            if (elem is Pipe pipe) //直接点击水泵的连接件创建管道
            {
                if (pipe.ConnectWithPumpSystem())
                {
                    return true;
                }
            }
            else if (elem is FamilyInstance fi)
            {
                if (fi.ConnectWithPumpSystem()) //直接点击管道连接点创建管道时，Revit会自动创建弯头，都需要加入到缓存
                {
                    return true;
                }
            }
            return false;
        }

        private static void AddToPumpSystem(Document doc, ElementId id)
        {
            var elem = doc.GetElement(id);
            if (elem is Pipe)
            {
                GlobalResource.RevitModels.Add(elem, RevitType.RFT_Pipe);
            }
            else if (elem is FamilyInstance fi)
            {
                AddAutoCreated(fi);
            }
            return;
        }

        private static void AddAutoCreated(FamilyInstance fi)
        {
            if (fi.IsPipeHat())
            {
                GlobalResource.RevitModels.Add(fi, RevitType.RFT_Blocker);
            }
            else if (fi.IsWanTou())
            {
                GlobalResource.RevitModels.Add(fi, RevitType.RFT_Elbow);
            }
            else if (fi.IsConverter())
            {
                GlobalResource.RevitModels.Add(fi, RevitType.RFT_Converter);
            }
            else if (fi.IsSanTong())
            {
                GlobalResource.RevitModels.Add(fi, RevitType.RFT_ThreeJoint);
            }
            else if (fi.IsSiTong())
            {
                GlobalResource.RevitModels.Add(fi, RevitType.RFT_FourJoint);
            }
            else
            {
                TaskDialog.Show("警告", $"未知类型被添加！id:{fi.Id.IntegerValue}");
                //报错
            }
        }

        public static void RegistViewEvent(ExternalCommandData commandData)
        {
            commandData.Application.ViewActivated -= Application_ViewActivated;
            commandData.Application.ViewActivated += Application_ViewActivated;
        }

        private static void Application_ViewActivated(object sender, ViewActivatedEventArgs e)
        {
            //切换文档时
            if (GlobalResource.CurrentDocument == null ||
                GlobalResource.CurrentDocument.GetHashCode() != e.CurrentActiveView.Document.GetHashCode())
            {
                GlobalResource.CurrentDocument = e.CurrentActiveView.Document;
                GlobalResource.InstancePanel.UpdateForm();
            }
        }

        public static List<Element> GetModelElements(this Document doc)
        {
            if (!doc.IsValidObject)
            {
                return new List<Element>();
            }

            var collector1 = new FilteredElementCollector(doc).
                OfCategory(BuiltInCategory.OST_PipeCurves).
                WhereElementIsNotElementType();

            var collector2 = new FilteredElementCollector(doc).
                OfClass(typeof(FamilyInstance)).
                WhereElementIsNotElementType();

           

            var elements1 = collector1.ToElements()?.ToList();
            var elements2 = collector2.ToElements()?.ToList();
            elements1.AddRange(elements2);

            elements1 = elements1.Where(x => x.Category.CategoryType == CategoryType.Model).ToList();
            elements1 = elements1.Where(x => x.get_BoundingBox(null) != null).ToList();

            return elements1;
        }

        public static void AddUnhandledElements()
        {
            var elements = GlobalResource.CurrentDocument.GetModelElements();
            foreach (var element in elements)
            {
                var id = element.Id.IntegerValue.ToString();
                if (GlobalResource.RevitModels.Contains(id))
                {
                    continue;
                }
                RevitMepCategoryService.AddToCache(element);
            }
            //UpdateLinkerInfo(GlobalResource.CurrentDocument);
        }

        public static void UpdataDocumentCache(this Document doc)
        {
            if (!doc.IsValidObject)
            {
                return;
            }
            if (!GlobalResource.CurrentDocument.IsValidObject || doc.GetHashCode() != GlobalResource.CurrentDocument.GetHashCode())
            {
                GlobalResource.CurrentDocument = doc;
                //GlobalResource.RevitModels = new List<Tuple<string, Dictionary<RevitType, List<string>>>>();
            }
            AddUnhandledElements();
        }
    }
}