CIMTool for Windows Management Instrumentation - Part 1

using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Linq;

namespace System.Windows.Controls.WpfPropertyGrid
{
  /// <summary>
  /// Holds custom attributes collection and provides facilities accessing attribute values from UI.
  /// </summary>
  public class AttributesContainer
  {
    private readonly AttributeCollection _attributes;
    private readonly Dictionary<string, Type> _keys = new Dictionary<string, Type>();
    private const string AttributeSuffix = "Attribute";

    /// <summary>
    /// Initializes a new instance of the <see cref="AttributesContainer"/> class.
    /// </summary>
    /// <param name="attributes">The collection of attributes.</param>
    public AttributesContainer(AttributeCollection attributes)
    {
      if (attributes == null) throw new ArgumentNullException("attributes");
      
      _attributes = attributes;

      foreach (var type in from Attribute attr in _attributes select attr.GetType())
        RegisterAttribute(type.Name, type);
    }

    /// <summary>
    /// Registers the attribute within the container.
    /// </summary>
    /// <param name="name">The public name of the attribute. The "Attribute" suffix will be automatically removed.</param>
    /// <param name="attributeType">Type of the attribute.</param>
    /// <returns>true is attribute was successfully registered, otherwise false</returns>
    public bool RegisterAttribute(string name, Type attributeType)
    {
      if (string.IsNullOrEmpty(name)) return false;
      if (attributeType == null) return false;

      string attributeName = name.EndsWith(AttributeSuffix, StringComparison.Ordinal)
        ? name.Remove(name.Length - AttributeSuffix.Length, AttributeSuffix.Length)
        : name;

      if (attributeName.Length == 0) return false;

      if (!_keys.ContainsKey(attributeName))
      {
        _keys.Add(attributeName, attributeType);
        return true;
      }
      return false;
    }

    /// <summary>
    /// Gets the attribute with the specified key.
    /// </summary>
    /// <value>Attribute with the specified key.</value>
    public object this[string key]
    {
      get
      {
        if (_attributes != null)
        {
          Type type;
          if (_keys.TryGetValue(key, out type))
            return _attributes[type];
        }
        return null;
      }
    }
  }
}