Reorganize data into Extensibility.Data namespace

1 files changed, 10 insertions, 0 deletions

diff --git a/src/WixToolset.Extensibility/BaseLayoutExtension.cs b/src/WixToolset.Extensibility/BaseLayoutExtension.cs
index bc80d432..78ec8d0a 100644
--- a/src/WixToolset.Extensibility/BaseLayoutExtension.cs
+++ b/src/WixToolset.Extensibility/BaseLayoutExtension.cs
@@ -2,6 +2,9 @@
 
 namespace WixToolset.Extensibility
 {
+    using WixToolset.Extensibility.Data;
+    using WixToolset.Extensibility.Services;
+
     /// <summary>
     /// Base class for creating a resolver extension.
     /// </summary>
@@ -13,11 +16,18 @@ namespace WixToolset.Extensibility
         protected ILayoutContext Context { get; private set; }
 
         /// <summary>
+        /// Messaging for use by the extension.
+        /// </summary>
+        protected IMessaging Messaging { get; private set; }
+
+        /// <summary>
         /// Called at the beginning of layout.
         /// </summary>
         public virtual void PreLayout(ILayoutContext context)
         {
             this.Context = context;
+
+            this.Messaging = context.ServiceProvider.GetService<IMessaging>();
         }
 
         public virtual bool CopyFile(string source, string destination)