Add special case for ExcelHotkeyAnnotation in the AttributeAnnotation…

…Provider
rubberduck-vba · Aug 22, 2019 · 73a3ffd · 73a3ffd
1 parent 9a094fa
commit 73a3ffd
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/Rubberduck.Parsing/Annotations/AttributeAnnotationProvider.cs b/Rubberduck.Parsing/Annotations/AttributeAnnotationProvider.cs
@@ -33,6 +33,11 @@ public class AttributeAnnotationProvider : IAttributeAnnotationProvider
             var flexibleValueAttributeAnnotation = FirstMatchingFlexibleAttributeValueAnnotation(annotationTypes, attributeName, attributeValues.Count);
             if (flexibleValueAttributeAnnotation != default)
             {
+                // FIXME special cased bodge for ExcelHotKeyAnnotation to deal with the value transformation:
+                if (flexibleValueAttributeAnnotation == AnnotationType.ExcelHotKey)
+                {
+                    return (flexibleValueAttributeAnnotation, attributeValues.Select(keySpec => keySpec.Substring(0, 1)).ToList());
+                }
                 return (flexibleValueAttributeAnnotation, attributeValues);
             }
 

diff --git a/Rubberduck.Parsing/Annotations/VBAParserAnnotationFactory.cs b/Rubberduck.Parsing/Annotations/VBAParserAnnotationFactory.cs
@@ -33,6 +33,7 @@ public VBAParserAnnotationFactory()
             _creators.Add(AnnotationType.ModuleAttribute.ToString().ToUpperInvariant(), typeof(ModuleAttributeAnnotation));
             _creators.Add(AnnotationType.MemberAttribute.ToString().ToUpperInvariant(), typeof(MemberAttributeAnnotation));
             _creators.Add(AnnotationType.ModuleDescription.ToString().ToUpperInvariant(), typeof(ModuleDescriptionAnnotation));
+            _creators.Add(AnnotationType.ExcelHotKey.ToString().ToUpperInvariant(), typeof(ExcelHotKeyAnnotation));
         }
 
         public IAnnotation Create(VBAParser.AnnotationContext context, QualifiedSelection qualifiedSelection)