Use ExePayloadRef for PrimaryPayloadId and SecondaryPayloadId

src/ext/Bal/test/WixToolsetTest.BootstrapperApplications/BalExtensionFixture.cs

@@ -3,12 +3,14 @@
 namespace WixToolsetTest.BootstrapperApplications
 {
     using System;
+    using System.Collections.Generic;
     using System.IO;
     using System.Linq;
-    using System.Xml;
-    using WixToolset.BootstrapperApplications;
     using WixInternal.Core.TestPackage;
     using WixInternal.TestSupport;
+    using WixToolset.Core;
+    using WixToolset.Extensibility.Data;
+    using WixToolset.Extensibility.Services;
     using Xunit;
 
     public class BalExtensionFixture
@@ -51,6 +53,52 @@ public void CanBuildUsingDisplayInternalUICondition()
             }
         }
 
+[Fact]
+        public void CanBuildUsingBootstrapperApplicationId()
+        {
+            using (var fs = new DisposableFileSystem())
+            {
+                var baseFolder = fs.GetFolder();
+                var bundleFile = Path.Combine(baseFolder, "bin", "test.exe");
+                var bundleSourceFolder = TestData.Get(@"TestData\WixStdBa");
+                var intermediateFolder = Path.Combine(baseFolder, "obj");
+                var baFolderPath = Path.Combine(baseFolder, "ba");
+                var extractFolderPath = Path.Combine(baseFolder, "extract");
+
+                var serviceProvider = WixToolsetServiceProviderFactory.CreateServiceProvider();
+                var parseHelper = serviceProvider.GetService<IParseHelper>();
+                var compileContext = serviceProvider.GetService<ICompileContext>();
+                var wixStdBaId = parseHelper.CreateIdentifierValueFromPlatform( "WixStandardBootstrapperApplication", compileContext.Platform, BurnPlatforms.X86 | BurnPlatforms.X64 | BurnPlatforms.ARM64);
+
+                var compileResult = WixRunner.Execute(new[]
+                {
+                    "build",
+                    Path.Combine(bundleSourceFolder, "BootstrapperApplicationId.wxs"),
+                    "-ext", TestData.Get(@"WixToolset.BootstrapperApplications.wixext.dll"),
+                    "-intermediateFolder", intermediateFolder,
+                    "-bindpath", Path.Combine(bundleSourceFolder, "data"),
+                    "-o", bundleFile,
+                });
+                compileResult.AssertSuccess();
+
+                Assert.True(File.Exists(bundleFile));
+
+                var extractResult = BundleExtractor.ExtractBAContainer(null, bundleFile, baFolderPath, extractFolderPath);
+                extractResult.AssertSuccess();
+
+                var ignoreAttributesByElementName = new Dictionary<string, List<string>>
+                {
+                    { "Payload", new List<string> { "SourcePath" } },
+                };
+
+                var wixStdBaPayloadInfo = extractResult.GetManifestTestXmlLines("/burn:BurnManifest/burn:UX/burn:Payload[@FilePath='wixstdba.exe']", ignoreAttributesByElementName);
+                WixAssert.CompareLineByLine(new string[]
+                {
+                    $@"<Payload Id='{wixStdBaId}' FilePath='wixstdba.exe' SourcePath='*' />"
+                }, wixStdBaPayloadInfo);
+            }
+        }
+
         [Fact]
         public void CanBuildUsingOverridable()
         {

src/ext/Bal/test/WixToolsetTest.BootstrapperApplications/TestData/WixStdBa/BootstrapperApplicationId.wxs

@@ -0,0 +1,12 @@
+<?xml version="1.0" encoding="utf-8"?>
+<Wix xmlns="http://wixtoolset.org/schemas/v4/wxs"
+     xmlns:bal="http://wixtoolset.org/schemas/v4/wxs/bal">
+    <Bundle Name="WixStdBa" Version="1.0.0.0" Manufacturer="Example Corporation" UpgradeCode="75D5D534-E177-4689-AAE9-CAC1C39002C2">
+        <BootstrapperApplication Id="Custom">
+            <bal:WixStandardBootstrapperApplication LicenseUrl="http://wixtoolset.org/about/license/" Theme="hyperlinkLicense" />
+        </BootstrapperApplication>
+        <Chain>
+            <MsiPackage SourceFile="test.msi" />
+        </Chain>
+    </Bundle>
+</Wix>

src/wix/WixToolset.Core.Burn/Bundles/CreateBurnManifestCommand.cs

@@ -168,11 +168,11 @@ public void Execute()
                 // write the UX element
                 writer.WriteStartElement("UX");
 
-                writer.WriteAttributeString("PrimaryPayloadId", this.PrimaryBundleApplicationSymbol.Id.Id);
+                writer.WriteAttributeString("PrimaryPayloadId", this.PrimaryBundleApplicationSymbol.ExePayloadRef);
 
-                if (!String.IsNullOrEmpty(this.SecondaryBundleApplicationSymbol?.Id.Id))
+                if (!String.IsNullOrEmpty(this.SecondaryBundleApplicationSymbol?.ExePayloadRef))
                 {
-                    writer.WriteAttributeString("SecondaryPayloadId", this.SecondaryBundleApplicationSymbol.Id.Id);
+                    writer.WriteAttributeString("SecondaryPayloadId", this.SecondaryBundleApplicationSymbol.ExePayloadRef);
                 }
 
                 // write the UX allPayloads...