1 files changed, 12 insertions, 0 deletions

diff --git a/src/ext/UI/wixext/UICompiler.cs b/src/ext/UI/wixext/UICompiler.cs
index 1d845237..44ab4cdc 100644
--- a/src/ext/UI/wixext/UICompiler.cs
+++ b/src/ext/UI/wixext/UICompiler.cs
@@ -55,6 +55,7 @@ namespace WixToolset.UI
         {
             var sourceLineNumbers = this.ParseHelper.GetSourceLineNumbers(element);
             string id = null;
+            string installDirectory = null;
 
             foreach (var attrib in element.Attributes())
             {
@@ -65,6 +66,9 @@ namespace WixToolset.UI
                         case "Id":
                             id = this.ParseHelper.GetAttributeIdentifierValue(sourceLineNumbers, attrib);
                             break;
+                        case "InstallDirectory":
+                            installDirectory = this.ParseHelper.GetAttributeIdentifierValue(sourceLineNumbers, attrib);
+                            break;
                         default:
                             this.ParseHelper.UnexpectedAttribute(element, attrib);
                             break;
@@ -111,6 +115,14 @@ namespace WixToolset.UI
                     IgnoreResult = true,
                     ExecutionType = CustomActionExecutionType.Immediate,
                 });
+
+                if (installDirectory != null)
+                {
+                    section.AddSymbol(new PropertySymbol(sourceLineNumbers, new Identifier(AccessModifier.Global, "WIXUI_INSTALLDIR"))
+                    {
+                        Value = installDirectory
+                    });
+                }
             }
 
             this.ParseHelper.ParseForExtensionElements(this.Context.Extensions, intermediate, section, element);