Shrlalgo.RvKits/ShrlAlgoToolkit.Revit/Extensions/ExtensibleStorageExtensions.cs

using System.IO;
using System.Windows;

using Autodesk.Revit.DB;
using Autodesk.Revit.DB.ExtensibleStorage;

namespace ShrlAlgoToolkit.Revit.Assists;

/// <summary>
///     可扩展存储工具Schema框架->由字段Field组成，是定义
///     由Schema可以生成、构建Entity（实体）对象，Entity是Schema的实例化对象，包含了Schema中定义的字段Field的值。
///     层级关系:Entity->（Schema/Field定义）/Field属性-值
///     元素阵列不会复制扩展数据
/// </summary>
/// <example>
/// <c>
/// IList<string> list = new List<string> { "我的v1", "我的v2", "我的v3", "我的v4" };
/// IDictionary<string, string> dict = new Dictionary<string, string>
///        {
///            { "p1", "v1" },
///            { "p2", "v2" },
///            { "p3", "v3" },
///            { "p4", "v4" }
///        };
/// var builder = CreateSchemaBuilder(schemaName);
/// //添加字段定义
/// builder.AddSimpleField("simple", typeof(string));
/// builder.AddArrayField("errorModels", typeof(string));
/// builder.AddMapField("dict", typeof(string), typeof(string));
/// builder.AddSingleField<string>("simple");
///	builder.AddListField<string>("errorModels");
///	builder.AddDictField<string, string>("dict");
///	//创建Schema
///	var schema = builder.CreateSchema();
/// var entity = e.CreateEntity(schema);
/// e.SetFieldValue("simple", "简单");
///	e.SetFieldValue("errorModels", list);
///	e.SetFieldValue("dict", dict);</c>
/// </example>
public static class ExtensibleStorageExtensions
{
    #region Schema
    /// <summary>
    ///    创建 Schema
    /// </summary>
    /// <param name="schemaName"></param>
    /// <param name="accessLevel"></param>
    /// <param name="description"></param>
    /// <returns></returns>
    public static Schema CreateSchemaBuilder(string schemaName, AccessLevel accessLevel = AccessLevel.Public, string description = null)
    {
        if (schemaName is null)
        {
            throw new ArgumentNullException(nameof(schemaName));
        }

        SchemaBuilder builder = new(Guid.NewGuid());
        builder.SetReadAccessLevel(accessLevel);
        builder.SetWriteAccessLevel(accessLevel);
        builder.SetSchemaName(schemaName);
        description ??= schemaName;
        builder.SetDocumentation(description);
#if USE_FORGETYPEID
		//roleFieldBuilder.SetSpec(SpecTypeId.HCLight);//设置字段值的单位，并不是所有类型都需要单位

#else
        //roleFieldBuilder.SetUnitType(UnitType.UT_Undefined);//设置字段值的单位，并不是所有类型都需要单位
        //roleFieldBuilder1.SetUnitType(UnitType.UT_Undefined);//设置字段值的单位，并不是所有类型都需要单位
        //roleFieldBuilder2.SetUnitType(UnitType.UT_Undefined);//设置字段值的单位，并不是所有类型都需要单位

#endif
        return builder?.Finish();
    }

    /// <summary>
    ///     通过字段名称获取Schema
    /// </summary>
    /// <param name="elm"></param>
    /// <param name="fieldName"></param>
    /// <returns></returns>
    public static Schema GetSchemaByFieldName(this Element elm, string fieldName)
    {
        if (elm is null)
        {
            throw new ArgumentNullException(nameof(elm));
        }

        if (fieldName is null)
        {
            throw new ArgumentNullException(nameof(fieldName));
        }

        var schemaGuids = elm.GetEntitySchemaGuids();

        return schemaGuids.Select(Schema.Lookup).FirstOrDefault(f => f.GetField(fieldName) != null);
    }

    /// <summary>
    ///     通过Schema名称获取Schema
    /// </summary>
    /// <param name="elm"></param>
    /// <param name="schemaName"></param>
    /// <returns></returns>
    public static Schema GetSchemaByName(this Element elm, string schemaName)
    {
        if (elm is null)
        {
            throw new ArgumentNullException(nameof(elm));
        }

        if (schemaName is null)
        {
            throw new ArgumentNullException(nameof(schemaName));
        }

        var schemaGuids = elm.GetEntitySchemaGuids();

        return schemaGuids.Select(Schema.Lookup).FirstOrDefault(f => f.SchemaName == schemaName);
    }
    /// <summary>
    ///     通过guid名称获取Schema
    /// </summary>
    /// <param name="elm"></param>
    /// <param name="schemaName"></param>
    /// <returns></returns>
    public static Schema GetSchemaByGuid(Guid schemaGuid)
    {
        return Schema.Lookup(schemaGuid);
    }
    #endregion
    #region Entity
    public static Entity GetEntity(this Element element, Schema schema)
    {
        if (element is null)
        {
            return null;
        }

        var entity = element.GetEntity(schema);
        if (entity.Schema is not null)
        {
            return entity;
        }

        entity = new Entity(schema);
        return entity;
    }

    /// <summary>
    ///     创建元素的扩展数据Entity
    /// </summary>
    /// <param name="element"></param>
    /// <param name="schema"></param>
    /// <returns></returns>
    public static Entity CreateEntity(this Element element, Schema schema)
    {
        Entity entity = new(schema);
        element.SetEntity(entity);
        return entity;
    }

    /// <summary>
    ///     删除扩展数据Entity
    /// </summary>
    /// <param name="element">元素</param>
    /// <param name="schemaName">表名</param>
    public static void DeleteEntity(this Element element, string schemaName)
    {
        var schema = element.GetSchemaByName(schemaName);
        if (schema != null)
        {
            element.DeleteEntity(schema);
        }
    }
    /// <summary>
    ///     从元素中读取Entity中存储的值
    /// </summary>
    /// <param name="element">存储数据的元素</param>
    /// <param name="schema">Existing schema</param>
    /// <param name="fieldName">The Field name</param>
    /// <typeparam name="T">schema中要存储的数据类型。该类型必须与schemabuilder中指定的数据类型相匹配。</typeparam>
    /// <returns>存储在元素中的数据。如果字段不存在或数据尚未保存，则返回空值</returns>
    /// <example>
    ///     <code>
    ///         document.ProjectInformation.LoadEntity&lt;string&lt;(schema, "schemaField")
    ///     </code>
    /// </example>
    public static T LoadEntity<T>(this Element element, Schema schema, string fieldName)
    {
        var field = schema.GetField(fieldName);
        var entity = element.GetEntity(schema);
        return entity is null || field is null ? default : entity.Get<T>(field);
    }
    /// <summary>
    ///    在元素中存储数据。现有数据将被覆盖
    /// </summary>
    /// <param name="element">存储数据的元素</param>
    /// <param name="schema">Existing schema</param>
    /// <param name="data">Type of data</param>
    /// <param name="fieldName">The Field name</param>
    /// <typeparam name="T">The type of data to be stored in the schema. The type must match the type of data specified in the SchemaBuilder</typeparam>
    /// <returns>True if entity save succeeded</returns>
    /// <example>
    ///     <code>
    ///         document.ProjectInformation.SaveEntity(schema, "data", "schemaField")
    ///     </code>
    /// </example>
    /// <summary>
    public static bool SaveEntity<T>(this Element element, Schema schema, T data, string fieldName)
    {
        var field = schema.GetField(fieldName);
        if (field is null)
        {
            return false;
        }

        var entity = element.GetEntity(schema);
        if (entity is null)
        {
            return false;
        }

        entity.Set(field, data);
        element.SetEntity(entity);
        return true;
    }
    #endregion
    #region Field
    /// <summary>
    ///     创建字典类型的字段
    /// </summary>
    /// <typeparam name="TK">key：Boolean, Byte, Int16, Int32, Float, Double, ElementId,GUID, String, XYZ, UV and Entity</typeparam>
    /// <typeparam name="TV">value：Boolean, Byte, Int16, Int32, Float, Double, ElementId,GUID, String, XYZ, UV and Entity</typeparam>
    /// <param name="schemaBuilder"></param>
    /// <param name="fieldName"></param>
    /// <param name="description">描述</param>
    /// <returns></returns>
    public static FieldBuilder AddDictField<TK, TV>(this SchemaBuilder schemaBuilder, string fieldName, string description = "")
    {
        if (schemaBuilder is null)
        {
            throw new ArgumentNullException(nameof(schemaBuilder));
        }

        if (fieldName is null)
        {
            throw new ArgumentNullException(nameof(fieldName));
        }

        //if (!typeof(TK).IsPrimitive)
        //    throw new NotSupportedException(nameof(TK));

        //if (!typeof(TV).IsPrimitive)
        //    throw new NotSupportedException(nameof(TV));
        if (description == string.Empty)
        {
            description = fieldName;
        }

        var result = schemaBuilder.AddMapField(fieldName, typeof(TK), typeof(TV));

        result.SetDocumentation(description);
        return result;
    }

    /// <summary>
    ///     创建集合类型的字段
    /// </summary>
    /// <typeparam name="T">Boolean, Byte, Int16, Int32, Float, Double, ElementId,GUID, String, XYZ, UV and Entity</typeparam>
    /// <param name="schemaBuilder"></param>
    /// <param name="fieldName"></param>
    /// <param name="description">描述</param>
    public static FieldBuilder AddListField<T>(this SchemaBuilder schemaBuilder, string fieldName, string description = "")
    {
        if (schemaBuilder is null)
        {
            throw new ArgumentNullException(nameof(schemaBuilder));
        }

        if (fieldName is null)
        {
            throw new ArgumentNullException(nameof(fieldName));
        }

        //if (!typeof(Command).IsPrimitive)
        //    throw new NotSupportedException(nameof(Command));

        var result = schemaBuilder.AddArrayField(fieldName, typeof(T));
        if (description == string.Empty)
        {
            description = fieldName;
        }

        result.SetDocumentation(description);
        return result;
    }

    /// <summary>
    ///     创建简单类型的字段
    /// </summary>
    /// <typeparam name="T">Boolean, Byte, Int16, Int32, Float, Double, ElementId,GUID, String, XYZ, UV and Entity</typeparam>
    /// <param name="schemaBuilder"></param>
    /// <param name="fieldName"></param>
    /// <param name="description">描述</param>
    /// <returns></returns>
    public static FieldBuilder AddSingleField<T>(this SchemaBuilder schemaBuilder, string fieldName, string description = "")
    {
        if (schemaBuilder is null)
        {
            throw new ArgumentNullException(nameof(schemaBuilder));
        }

        if (fieldName is null)
        {
            throw new ArgumentNullException(nameof(fieldName));
        }

        //if (!typeof(Command).IsPrimitive)
        //    throw new NotSupportedException(nameof(Command));

        var result = schemaBuilder.AddSimpleField(fieldName, typeof(T));
        //result.SetUnitType(unitType);

        if (description == string.Empty)
        {
            description = fieldName;
        }

        result.SetDocumentation(description);
        return result;
    }
    /// <summary>
    ///     设置可扩展参数的值：简单类型、集合、字典(这一步骤涉及对文档的修改，需在事务中进行)
    /// </summary>
    /// <param name="element">元素</param>
    /// <param name="fieldName">字段</param>
    /// <param name="value">字段值</param>
    public static void SetFieldValue<T>(this Element element, string fieldName, T value)
    {
        Schema schema = element.GetSchemaByFieldName(fieldName);
        var entity = element.GetEntity(schema);
        Field field = schema.GetField(fieldName);
        entity.Set(field, value);
        element.SetEntity(entity);
    }

    /// <summary>
    ///     设置可扩展参数的值：简单类型、集合、字典(这一步骤涉及对文档的修改，需在事务中进行)
    /// </summary>
    /// <param name="element">元素</param>
    /// <param name="schema"></param>
    /// <param name="fieldName">字段</param>
    /// <param name="value">字段值</param>
    public static void SetFieldValue<T>(this Element element, Schema schema, string fieldName, T value)
    {
        var entity = element.GetEntity(schema);
        Field field = schema.GetField(fieldName);
        entity.Set(field, value);
        element.SetEntity(entity);
    }
    /// <summary>
    ///     获取字段参数值
    /// </summary>
    /// <param name="element">元素</param>
    /// <param name="fieldName">字段名</param>
    public static T GetFieldValue<T>(this Element element, string fieldName)
        where T : new()
    {
        if (!typeof(T).IsPrimitive)
        {
            throw new NotSupportedException(nameof(T));
        }

        T t = new();
        try
        {
            Schema schema = element.GetSchemaByFieldName(fieldName);
            Entity entity = element.GetEntity(schema);
            if (!entity.IsValid())
            {
                return t;
            }

            Field f = schema.GetField(fieldName);
            return entity.Get<T>(f);
        }
        catch
        {
            return t;
        }
    }

    /// <summary>
    ///     获取元素的所有字段值
    /// </summary>
    /// <typeparam name="T"></typeparam>
    /// <param name="e"></param>
    /// <returns></returns>
    public static List<T> GetFiledValues<T>(this Element e)
    {
        if (!typeof(T).IsPrimitive)
        {
            throw new NotSupportedException(nameof(T));
        }

        List<T> storages = new();
        //方法一
        var schemaGuids = e.GetEntitySchemaGuids();
        foreach (var schemaGuid in schemaGuids)
        {
            Schema schema = Schema.Lookup(schemaGuid);
            Entity ent = e.GetEntity(schema);
            if (!ent.IsValid())
            {
                return storages;
            }

            IList<Field> fields = schema.ListFields();
            foreach (var field in fields)
            {
                var f = ent.Get<T>(field);
                storages.Add(f);
            }
        }

        //if (storages.Count == 0)
        //{
        //    //方法二
        //    //从内存中获取 Schema 列表
        //    IList<Schema> schemas = Schema.ListSchemas();
        //    //从 Schema 中获取 Field 列表
        //    foreach (var schema in schemas)
        //    {
        //        IList<Field> fields = schema.ListFields();
        //        Entity ent = e.GetEntity(schema);
        //        foreach (var field in fields)
        //        {
        //            var f = ent.Get<Command>(field);
        //            storages.Add(f);
        //        }
        //    }
        //}

        return storages;
    }
    #endregion

    #region DataStorage
    /// <summary>
    ///     获取文件级数据存储
    /// </summary>
    /// <param name="doc"></param>
    /// <param name="schemaName"></param>
    /// <param name="fieldName"></param>
    /// <param name="fileName"></param>
    public static void GetDateStorage(Document doc, string schemaName, string fieldName, string fileName)
    {
        DataStorage storage = new FilteredElementCollector(doc).OfClass(typeof(DataStorage)).Cast<DataStorage>().First();
        Schema sch = Schema.ListSchemas().First(o => o.SchemaName == schemaName);
        Entity ent = storage.GetEntity(sch);
        byte[] bytes = ent.Get<byte[]>(sch.GetField(fieldName)).ToArray();
        using (FileStream fs = new(fileName, FileMode.Create, FileAccess.Write))
        {
            fs.Write(bytes, 0, bytes.Length);
        }

        MessageBox.Show("文件提取完毕!", "提示", MessageBoxButton.OK, MessageBoxImage.Information);
    }


    /// <summary>
    ///     设置文件级别数据存储
    /// </summary>
    /// <param name="doc"></param>
    /// <param name="fileName"></param>
    public static void SetDateStorage(Document doc, string fileName)
    {
        Transaction tran = new(doc, "BIM");
        tran.Start();
        //项目文档中使用扩展数据，可以使用静态方法创建
        DataStorage store = DataStorage.Create(doc);
        Schema schema;
        using (SchemaBuilder schemaBuilder = new(new Guid(Guid.NewGuid().ToString("D"))))
        {
            schemaBuilder.SetReadAccessLevel(AccessLevel.Public);
            schemaBuilder.SetWriteAccessLevel(AccessLevel.Public);
            schemaBuilder.SetSchemaName("LeiFile");
            schemaBuilder.SetDocumentation("DataStorageFile");

            FieldBuilder arrayField = schemaBuilder.AddArrayField("data", typeof(byte));
            arrayField.SetDocumentation("Store file data");

            schema = schemaBuilder.Finish();
        }

        byte[] data;
        using (FileStream stream = new(fileName, FileMode.Open, FileAccess.Read))
        {
            data = new byte[(int)stream.Length];
            stream.Read(data, 0, data.Length);
            stream.Close();
        }

        Entity entity = new(schema);
        Field field = schema.GetField("data");
        entity.Set(field, data);
        store.SetEntity(entity);
        MessageBox.Show("文件存储成功！", "提示", MessageBoxButton.OK, MessageBoxImage.Information);
        tran.Commit();
    }
    #endregion


}