Add IBurnBackendHelper and TryAddTupleToDataManifest.

1 files changed, 23 insertions, 1 deletions

diff --git a/src/WixToolset.Extensibility/BaseBurnBackendExtension.cs b/src/WixToolset.Extensibility/BaseBurnBackendExtension.cs
index 58f96b31..5dc36715 100644
--- a/src/WixToolset.Extensibility/BaseBurnBackendExtension.cs
+++ b/src/WixToolset.Extensibility/BaseBurnBackendExtension.cs
@@ -6,7 +6,7 @@ namespace WixToolset.Extensibility
     using WixToolset.Extensibility.Data;
     using WixToolset.Extensibility.Services;
 
-    public class BaseBurnBackendExtension : IBurnBackendExtension
+    public abstract class BaseBurnBackendExtension : IBurnBackendExtension
     {
         /// <summary>
         /// Context for use by the extension.
@@ -18,10 +18,20 @@ namespace WixToolset.Extensibility
         /// </summary>
         protected IMessaging Messaging { get; private set; }
 
+        /// <summary>
+        /// Backend helper for use by the extension.
+        /// </summary>
+        protected IBurnBackendHelper BackendHelper { get; private set; }
+
         public virtual void BundleFinalize()
         {
         }
 
+        public virtual bool IsTupleForExtension(IntermediateTuple tuple)
+        {
+            return false;
+        }
+
         public virtual void PostBackendBind(IBindResult result)
         {
         }
@@ -30,6 +40,7 @@ namespace WixToolset.Extensibility
         {
             this.Context = context;
             this.Messaging = context.ServiceProvider.GetService<IMessaging>();
+            this.BackendHelper = context.ServiceProvider.GetService<IBurnBackendHelper>();
         }
 
         public virtual IResolveFileResult ResolveRelatedFile(string source, string relatedSource, string type, SourceLineNumber sourceLineNumbers, BindStage bindStage)
@@ -41,5 +52,16 @@ namespace WixToolset.Extensibility
         {
             return null;
         }
+
+        public virtual bool TryAddTupleToDataManifest(IntermediateSection section, IntermediateTuple tuple)
+        {
+            if (this.IsTupleForExtension(tuple) && tuple.HasTag(WixToolset.Data.Burn.BurnConstants.BootstrapperApplicationDataTupleDefinitionTag))
+            {
+                this.BackendHelper.AddBootstrapperApplicationData(tuple);
+                return true;
+            }
+
+            return false;
+        }
     }
 }