add the user defined compilation arguments to the symbol table

rubberduck-vba · Jul 12, 2018 · 3f08f25 · 3f08f25
1 parent 774aed7
commit 3f08f25
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/Rubberduck.Parsing/Preprocessing/VBAPreprocessorVisitor.cs b/Rubberduck.Parsing/Preprocessing/VBAPreprocessorVisitor.cs
@@ -41,6 +41,7 @@ public sealed class VBAPreprocessorVisitor : VBAConditionalCompilationParserBase
             _tokenStream = tokenStream;
             _symbolTable = symbolTable;
             AddPredefinedConstantsToSymbolTable(predefinedConstants);
+            AddUserDefinedConstantsToSymbolTable(userDefinedConstants);
         }
 
         private void AddPredefinedConstantsToSymbolTable(VBAPredefinedCompilationConstants predefinedConstants)
@@ -53,6 +54,14 @@ private void AddPredefinedConstantsToSymbolTable(VBAPredefinedCompilationConstan
             _symbolTable.Add(VBAPredefinedCompilationConstants.MAC_NAME, new BoolValue(predefinedConstants.Mac));
         }
 
+        private void AddUserDefinedConstantsToSymbolTable(Dictionary<string, short> userDefinedConstants)
+        {
+            foreach (var constant in userDefinedConstants)
+            {
+                _symbolTable.Add(constant.Key, new DecimalValue(constant.Value));
+            }
+        }
+
         public override IExpression VisitCompilationUnit([NotNull] VBAConditionalCompilationParser.CompilationUnitContext context)
         {
             return Visit(context.ccBlock());