Add support for loading Intermediates from extensions

1 files changed, 39 insertions, 0 deletions

diff --git a/src/test/TestData/Example.Extension/ExampleExtensionFactory.cs b/src/test/TestData/Example.Extension/ExampleExtensionFactory.cs
new file mode 100644
index 00000000..b91d06e9
--- /dev/null
+++ b/src/test/TestData/Example.Extension/ExampleExtensionFactory.cs
@@ -0,0 +1,39 @@
+// 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 Example.Extension
+{
+    using System;
+    using WixToolset.Extensibility;
+
+    public class ExampleExtensionFactory : IExtensionFactory
+    {
+        private ExamplePreprocessorExtensionAndCommandLine preprocessorExtension;
+
+        public bool TryCreateExtension(Type extensionType, out object extension)
+        {
+            if (extensionType == typeof(IExtensionCommandLine) || extensionType == typeof(IPreprocessorExtension))
+            {
+                if (preprocessorExtension == null)
+                {
+                    preprocessorExtension = new ExamplePreprocessorExtensionAndCommandLine();
+                }
+
+                extension = preprocessorExtension;
+            }
+            else if (extensionType == typeof(ICompilerExtension))
+            {
+                extension = new ExampleCompilerExtension();
+            }
+            else if (extensionType == typeof(IExtensionData))
+            {
+                extension = new ExampleExtensionData();
+            }
+            else
+            {
+                extension = null;
+            }
+            
+            return extension != null;
+        }
+    }
+}