1 files changed, 31 insertions, 0 deletions

diff --git a/src/api/wix/WixToolset.Data/SimpleSymbolDefinitionCreator.cs b/src/api/wix/WixToolset.Data/SimpleSymbolDefinitionCreator.cs
new file mode 100644
index 00000000..e22d53e7
--- /dev/null
+++ b/src/api/wix/WixToolset.Data/SimpleSymbolDefinitionCreator.cs
@@ -0,0 +1,31 @@
+// Copyright (c) .NET Foundation and contributors. All rights reserved. Licensed under the Microsoft Reciprocal License. See LICENSE.TXT file in the project root for full license information.
+
+namespace WixToolset.Data
+{
+    using System.Collections.Generic;
+
+    internal class SimpleSymbolDefinitionCreator : ISymbolDefinitionCreator
+    {
+        private Dictionary<string, IntermediateSymbolDefinition> CustomDefinitionByName { get; } = new Dictionary<string, IntermediateSymbolDefinition>();
+
+        public void AddCustomSymbolDefinition(IntermediateSymbolDefinition definition)
+        {
+            if (!this.CustomDefinitionByName.TryGetValue(definition.Name, out var existing) || definition.Revision > existing.Revision)
+            {
+                this.CustomDefinitionByName[definition.Name] = definition;
+            }
+        }
+
+        public bool TryGetSymbolDefinitionByName(string name, out IntermediateSymbolDefinition symbolDefinition)
+        {
+            symbolDefinition = SymbolDefinitions.ByName(name);
+
+            if (symbolDefinition == null)
+            {
+                symbolDefinition = this.CustomDefinitionByName.GetValueOrDefault(name);
+            }
+
+            return symbolDefinition != null;
+        }
+    }
+}