Add CreateCustomActionReference method

...to simplify creating custom action references following naming conventions.

2 files changed, 79 insertions, 0 deletions

diff --git a/src/WixToolset.Core/ExtensibilityServices/CustomActionPlatforms.cs b/src/WixToolset.Core/ExtensibilityServices/CustomActionPlatforms.cs
new file mode 100644
index 00000000..17f381ac
--- /dev/null
+++ b/src/WixToolset.Core/ExtensibilityServices/CustomActionPlatforms.cs
@@ -0,0 +1,31 @@
+// Copyright (c) .NET Foundation and contributors. All rights reserved. Licensed under the Microsoft Reciprocal License. See LICENSE.TXT file in the project root for full license information.
+
+namespace WixToolset.Core.ExtensibilityServices
+{
+    using System;
+
+    /// <summary>
+    /// Platforms supported by custom actions.
+    /// </summary>
+    [Flags]
+    public enum CustomActionPlatforms
+    {
+        /// <summary>Not specified.</summary>
+        None = 0,
+
+        /// <summary>x86.</summary>
+        X86 = 0x1,
+
+        /// <summary>x64.</summary>
+        X64 = 0x2,
+
+        /// <summary>ia64.</summary>
+        IA64 = 0x4,
+
+        /// <summary>arm.</summary>
+        ARM = 0x8,
+
+        /// <summary>arm64.</summary>
+        ARM64 = 0x10
+    }
+}
diff --git a/src/WixToolset.Core/ExtensibilityServices/ParseHelper.cs b/src/WixToolset.Core/ExtensibilityServices/ParseHelper.cs
index 8fbfdd87..271e5697 100644
--- a/src/WixToolset.Core/ExtensibilityServices/ParseHelper.cs
+++ b/src/WixToolset.Core/ExtensibilityServices/ParseHelper.cs
@@ -892,6 +892,54 @@ namespace WixToolset.Core.ExtensibilityServices
             return actionTuple;
         }
 
+        /// <summary>
+        /// Create a reference in the specified section for a custom action specialized for specific platforms,
+        /// given standard prefixes for naming and suffixes for platforms.
+        /// </summary>
+        /// <param name="sourceLineNumbers">Source line information.</param>
+        /// <param name="customAction">The custom action base name.</param>
+        /// <param name="supportedPlatforms">The platforms for which there are specialized custom actions.</param>
+        public void CreateCustomActionReference(SourceLineNumber sourceLineNumbers, IntermediateSection section, string customAction, Platform platform, CustomActionPlatforms supportedPlatforms)
+        {
+            if (!this.Messaging.EncounteredError)
+            {
+                var name = String.Concat("Wix4", customAction);
+
+                switch (platform)
+                {
+                    case Platform.X86:
+                        if ((supportedPlatforms & CustomActionPlatforms.X86) == CustomActionPlatforms.X86)
+                        {
+                            name = String.Concat(customAction, "_X86");
+                        }
+                        break;
+                    case Platform.X64:
+                        if ((supportedPlatforms & CustomActionPlatforms.X64) == CustomActionPlatforms.X64)
+                        {
+                            name = String.Concat(customAction, "_X64");
+                        }
+                        break;
+                    case Platform.ARM:
+                        if ((supportedPlatforms & CustomActionPlatforms.ARM) == CustomActionPlatforms.ARM)
+                        {
+                            name = String.Concat(customAction, "_A32");
+                        }
+                        break;
+                    case Platform.ARM64:
+                        if ((supportedPlatforms & CustomActionPlatforms.ARM64) == CustomActionPlatforms.ARM64)
+                        {
+                            name = String.Concat(customAction, "_A64");
+                        }
+                        break;
+                    case Platform.IA64:
+                        // yeah, no
+                        break;
+                }
+
+                this.CreateSimpleReference(section, sourceLineNumbers, nameof(TupleDefinitionType.CustomAction), name);
+            }
+        }
+
         public void UnexpectedAttribute(XElement element, XAttribute attribute)
         {
             var sourceLineNumbers = Preprocessor.GetSourceLineNumbers(element);