Allow BA to update feed source

Fixes wixtoolset/issues#5568

3 files changed, 20 insertions, 0 deletions

diff --git a/src/WixToolset.Mba.Core/Engine.cs b/src/WixToolset.Mba.Core/Engine.cs
index d5c43a53..e07ecd8b 100644
--- a/src/WixToolset.Mba.Core/Engine.cs
+++ b/src/WixToolset.Mba.Core/Engine.cs
@@ -247,6 +247,12 @@ namespace WixToolset.Mba.Core
         }
 
         /// <inheritdoc/>
+        public void SetUpdateSource(string url)
+        {
+            this.engine.SetUpdateSource(url);
+        }
+
+        /// <inheritdoc/>
         public void SetLocalSource(string packageOrContainerId, string payloadId, string path)
         {
             this.engine.SetLocalSource(packageOrContainerId, payloadId, path);
diff --git a/src/WixToolset.Mba.Core/IBootstrapperEngine.cs b/src/WixToolset.Mba.Core/IBootstrapperEngine.cs
index af3101dc..78753a42 100644
--- a/src/WixToolset.Mba.Core/IBootstrapperEngine.cs
+++ b/src/WixToolset.Mba.Core/IBootstrapperEngine.cs
@@ -278,6 +278,14 @@ namespace WixToolset.Mba.Core
             );
 
         /// <summary>
+        /// Sets the URL to the update feed.
+        /// </summary>
+        /// <param name="url">URL of the update feed.</param>
+        void SetUpdateSource(
+            [MarshalAs(UnmanagedType.LPWStr)] string url
+            );
+
+        /// <summary>
         /// See <see cref="IEngine.CompareVersions(string, string)"/>.
         /// </summary>
         /// <param name="wzVersion1"></param>
diff --git a/src/WixToolset.Mba.Core/IEngine.cs b/src/WixToolset.Mba.Core/IEngine.cs
index 0899ec43..3e636961 100644
--- a/src/WixToolset.Mba.Core/IEngine.cs
+++ b/src/WixToolset.Mba.Core/IEngine.cs
@@ -145,6 +145,12 @@ namespace WixToolset.Mba.Core
         void SetUpdate(string localSource, string downloadSource, long size, UpdateHashType hashType, byte[] hash);
 
         /// <summary>
+        /// Sets the URL to the update feed.
+        /// </summary>
+        /// <param name="url">URL of the update feed.</param>
+        void SetUpdateSource(string url);
+
+        /// <summary>
         /// Set the local source for a package or container.
         /// </summary>
         /// <param name="packageOrContainerId">The id that uniquely identifies the package or container.</param>