From 95f2f4425b900374c7d7b583ae810b096121b3c4 Mon Sep 17 00:00:00 2001
From: Rob Mensching <rob@firegiant.com>
Date: Sat, 2 Dec 2017 00:46:11 -0800
Subject: Implement support for IExtensionCommandLine and
 IPreprocessorExtension

---
 src/WixToolset.Core/WixToolsetServiceProvider.cs | 11 +++++++++++
 1 file changed, 11 insertions(+)

(limited to 'src/WixToolset.Core/WixToolsetServiceProvider.cs')

diff --git a/src/WixToolset.Core/WixToolsetServiceProvider.cs b/src/WixToolset.Core/WixToolsetServiceProvider.cs
index 8693461b..d31d7355 100644
--- a/src/WixToolset.Core/WixToolsetServiceProvider.cs
+++ b/src/WixToolset.Core/WixToolsetServiceProvider.cs
@@ -12,6 +12,7 @@ namespace WixToolset.Core
     {
         private ExtensionManager extensionManager;
         private ParseHelper parseHelper;
+        private PreprocessHelper preprocessHelper;
         private TupleDefinitionCreator tupleDefinitionCreator;
 
         public object GetService(Type serviceType)
@@ -19,6 +20,11 @@ namespace WixToolset.Core
             if (serviceType == null) throw new ArgumentNullException(nameof(serviceType));
 
             // Transients.
+            if (serviceType == typeof(IPreprocessContext))
+            {
+                return new PreprocessContext(this);
+            }
+
             if (serviceType == typeof(ICompileContext))
             {
                 return new CompileContext(this);
@@ -65,6 +71,11 @@ namespace WixToolset.Core
                 return this.parseHelper = this.parseHelper ?? new ParseHelper(this);
             }
 
+            if (serviceType == typeof(IPreprocessHelper))
+            {
+                return this.preprocessHelper = this.preprocessHelper ?? new PreprocessHelper(this);
+            }
+
             throw new ArgumentException($"Unknown service type: {serviceType.Name}", nameof(serviceType));
         }
     }
-- 
cgit v1.2.3-55-g6feb