Update to WiX Intermediate Representation

1 files changed, 52 insertions, 0 deletions

diff --git a/src/WixToolset.Core/TupleDefinitionCreator.cs b/src/WixToolset.Core/TupleDefinitionCreator.cs
new file mode 100644
index 00000000..8c9b9d29
--- /dev/null
+++ b/src/WixToolset.Core/TupleDefinitionCreator.cs
@@ -0,0 +1,52 @@
+// 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.Core
+{
+    using System;
+    using System.Collections.Generic;
+    using WixToolset.Data;
+    using WixToolset.Extensibility;
+    using WixToolset.Extensibility.Services;
+
+    internal class TupleDefinitionCreator : ITupleDefinitionCreator
+    {
+        public TupleDefinitionCreator(IServiceProvider serviceProvider)
+        {
+            this.ServiceProvider = serviceProvider;
+        }
+
+        private IServiceProvider ServiceProvider { get; }
+
+        private IEnumerable<IExtensionData> ExtensionData { get; set; }
+
+        public bool TryGetTupleDefinitionByName(string name, out IntermediateTupleDefinition tupleDefinition)
+        {
+            tupleDefinition = TupleDefinitions.ByName(name);
+
+            if (tupleDefinition == null)
+            {
+                if (this.ExtensionData == null)
+                {
+                    this.LoadExtensionData();
+                }
+
+                foreach (var data in this.ExtensionData)
+                {
+                    if (data.TryGetTupleDefinitionByName(name, out tupleDefinition))
+                    {
+                        break;
+                    }
+                }
+            }
+
+            return tupleDefinition != null;
+        }
+
+        private void LoadExtensionData()
+        {
+            var extensionManager = (IExtensionManager)this.ServiceProvider.GetService(typeof(IExtensionManager));
+
+            this.ExtensionData = extensionManager.Create<IExtensionData>();
+        }
+    }
+}