using System.Collections.Generic;

using SolidWorks.Interop.sldworks;
using SolidWorks.Interop.swconst;

namespace PdmSwPlugin.Common.Util
{
    /// <summary>
    /// 为 SolidWorks 文档得到或设置自定义属性的类
    /// </summary>
    public static class CustomerPropertiesWrapper
    {
        /// <summary>
        /// 从指定 SolidWorks 文档得到当前配置里的自定义属性管理器
        /// </summary>
        /// <param name="currentModelDoc">SolidWorks 文档</param>
        /// <returns>自定义属性管理器</returns>
        public static CustomPropertyManager GetCustomPropertyManager(ModelDoc2 currentModelDoc)
        {
            if (currentModelDoc == null)
                return null;
            try
            {
                if (currentModelDoc.GetType() == (int)swDocumentTypes_e.swDocDRAWING)
                {
                    return currentModelDoc.Extension.CustomPropertyManager[""];
                }
                else
                {
                    Configuration activeConfig = currentModelDoc.GetActiveConfiguration() as Configuration;
                    if (activeConfig == null)
                        return null;
                    return currentModelDoc.Extension.CustomPropertyManager[activeConfig.Name];
                }
            }
            catch
            {
            }
            return null;
        }

        /// <summary>
        /// 得到指定 SolidWorks 文档的当前配置里的属性的值
        /// </summary>
        /// <param name="currentModelDoc">SolidWorks 文档</param>
        /// <param name="propertiesNames">属性名称</param>
        /// <param name="fullyResolve">完全还原属性的值吗？指的是如果属性值为 $PRP:SW - FileName，是否要还原成不带后缀的文件名</param>
        /// <returns>属性的值, key 是名称，value 是值</returns>
        public static Dictionary<string, string> Get(ModelDoc2 currentModelDoc, IEnumerable<string> propertiesNames, bool fullyResolve)
        {
            if (currentModelDoc == null || propertiesNames == null)
                return null;
            try
            {
                CustomPropertyManager customPropertyManager = GetCustomPropertyManager(currentModelDoc);
                if (customPropertyManager == null)
                {
                    return null;
                }
                string valueOut = string.Empty;
                string resolvedValueOut = string.Empty;
                bool resolved = false;
                Dictionary<string, string> propertiesValue = new Dictionary<string, string>();
                foreach (var v in propertiesNames)
                {
                    if (customPropertyManager.Get5(v, false, out valueOut, out resolvedValueOut, out resolved) != (int)swCustomInfoGetResult_e.swCustomInfoGetResult_NotPresent)
                    {
                       if (valueOut == "$PRP:\"SW-File Name\"") // 如果是表达式，需要 resolve
                        {
                            if (fullyResolve)
                            {
                                //propertiesValue[v] = resolvedValueOut;
                                string fileNameWithoutExt = System.IO.Path.GetFileNameWithoutExtension(currentModelDoc.GetPathName());
                                propertiesValue[v] = fileNameWithoutExt;
                            }
                            else
                                propertiesValue[v] = valueOut;
                        }
                        else if (resolved) // 不是表达式的，都要还原
                        {
                            propertiesValue[v] = resolvedValueOut;
                        }
                        else
                            propertiesValue[v] = valueOut;
                    }
                }
                return propertiesValue;
            }
            catch
            {
            }
            return null;
        }

        /// <summary>
        /// 得到指定 SolidWorks 文档的当前配置里的属性的值
        /// </summary>
        /// <param name="currentModelDoc">SolidWorks 文档</param>
        /// <param name="propertyName">属性名称</param>
        /// <param name="fullyResolve">完全还原属性的值吗？指的是如果属性值为 $PRP:SW - FileName，是否要还原成不带后缀的文件名</param>
        /// <returns>属性的值</returns>
        public static string GetSingle(ModelDoc2 currentModelDoc, string propertyName, bool fullyResolve)
        {
            List<string> properties = new List<string>();
            properties.Add(propertyName);
            Dictionary<string, string> keyValuePairs = Get(currentModelDoc, properties, fullyResolve);
            if (keyValuePairs == null || !keyValuePairs.ContainsKey(propertyName))
                return string.Empty;
            return keyValuePairs[propertyName];
        }

        /// <summary>
        /// 设置自定义属性的值
        /// </summary>
        /// <param name="currentModelDoc">SolidWorks 文档</param>
        /// <param name="propertiesNamesAndValues">属性的名称与值, key 是名称，value 是值</param>
        /// <returns>任何失败都将返回 false, 全部成功则返回 true</returns>
        public static bool Set(ModelDoc2 currentModelDoc, IDictionary<string, string> propertiesNamesAndValues)
        {
            if (currentModelDoc == null || propertiesNamesAndValues == null)
                return false;
            try
            {
                CustomPropertyManager customPropertyManager = GetCustomPropertyManager(currentModelDoc);
                if (customPropertyManager == null)
                    return false;
                foreach (KeyValuePair<string, string> v in propertiesNamesAndValues)
                {
                    /*
                    swCustomInfoSetResult_e setResult = (swCustomInfoSetResult_e)customPropertyManager.Set2(v.Key, v.Value);
                    switch (setResult)
                    {
                        case swCustomInfoSetResult_e.swCustomInfoSetResult_NotPresent: // 如果设置失败的原因是不存在该属性，那就增加
                            if (customPropertyManager.Add3(v.Key, (int)swCustomInfoType_e.swCustomInfoText, v.Value, 1) != (int)swCustomInfoAddResult_e.swCustomInfoAddResult_AddedOrChanged)
                                return false;
                            else
                                break;
                        case swCustomInfoSetResult_e.swCustomInfoSetResult_OK:
                            break;
                        default:
                            return false; // 失败了
                    }*/ // Set2 竟然返回类型不匹配
                    swCustomInfoDeleteResult_e resultOfDelete = (swCustomInfoDeleteResult_e)customPropertyManager.Delete2(v.Key);
                    return customPropertyManager.Add3(v.Key, (int)swCustomInfoType_e.swCustomInfoText, v.Value, 1) == (int)swCustomInfoAddResult_e.swCustomInfoAddResult_AddedOrChanged;

                }
                return true;
            }
            catch
            {
            }
            return false;
        }

        /// <summary>
        /// 设置自定义属性的值
        /// </summary>
        /// <param name="currentModelDoc">SolidWorks 文档</param>
        /// <param name="propertyName">属性的名称</param>
        /// <param name="propertyValue">属性的值</param>
        /// <returns>任何失败都将返回 false, 全部成功则返回 true</returns>
        public static bool SetSingle(ModelDoc2 currentModelDoc, string propertyName, string propertyValue)
        {
            if (string.IsNullOrEmpty(propertyName))
                return false;
            Dictionary<string, string> name_value = new Dictionary<string, string>
            {
                [propertyName] = propertyValue
            };
            return Set(currentModelDoc, name_value);
        }

        /// <summary>
        /// 设置自定义属性的值
        /// </summary>
        /// <param name="currentModelDoc">SolidWorks 文档</param>
        /// <param name="propertiesNamesAndValues">属性的名称与值, key 是名称，value 是值</param>
        /// <returns>任何失败都将返回 false, 全部成功则返回 true</returns>
        public static bool Delete(ModelDoc2 currentModelDoc, string propertyName)
        {
            if (currentModelDoc == null || string.IsNullOrEmpty(propertyName))
                return false;
            try
            {
                CustomPropertyManager customPropertyManager = GetCustomPropertyManager(currentModelDoc);
                if (customPropertyManager == null)
                    return false;
                return customPropertyManager.Delete2(propertyName) == (int)swCustomInfoDeleteResult_e.swCustomInfoDeleteResult_OK;
            }
            catch
            {
            }
            return false;
        }
    }
}