1 files changed, 31 insertions, 0 deletions

diff --git a/src/WixToolset.Core.Burn/BurnBackendFactory.cs b/src/WixToolset.Core.Burn/BurnBackendFactory.cs
new file mode 100644
index 00000000..8e2b3ce2
--- /dev/null
+++ b/src/WixToolset.Core.Burn/BurnBackendFactory.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.Core.Burn
+{
+    using System;
+    using System.IO;
+    using WixToolset.Extensibility;
+    using WixToolset.Extensibility.Services;
+
+    internal class BurnBackendFactory : IBackendFactory
+    {
+        public bool TryCreateBackend(string outputType, string outputFile, IBindContext context, out IBackend backend)
+        {
+            if (String.IsNullOrEmpty(outputType))
+            {
+                outputType = Path.GetExtension(outputFile);
+            }
+
+            switch (outputType.ToLowerInvariant())
+            {
+                case "bundle":
+                case ".exe":
+                    backend = new BundleBackend();
+                    return true;
+            }
+
+            backend = null;
+            return false;
+        }
+    }
+}