Move DifxApp.wixext into ext

10 files changed, 480 insertions, 0 deletions

diff --git a/src/ext/DifxApp/wixext/DifxAppCompiler.cs b/src/ext/DifxApp/wixext/DifxAppCompiler.cs
new file mode 100644
index 00000000..e056988f
--- /dev/null
+++ b/src/ext/DifxApp/wixext/DifxAppCompiler.cs
@@ -0,0 +1,161 @@
+// 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.DifxApp
+{
+    using System;
+    using System.Collections.Generic;
+    using System.Xml.Linq;
+    using WixToolset.Data;
+    using WixToolset.DifxApp.Symbols;
+    using WixToolset.Extensibility;
+
+    /// <summary>
+    /// The compiler for the WiX Toolset Driver Install Frameworks for Applications Extension.
+    /// </summary>
+    public sealed class DifxAppCompiler : BaseCompilerExtension
+    {
+        private HashSet<string> components;
+
+        public override XNamespace Namespace => "http://wixtoolset.org/schemas/v4/wxs/difxapp";
+        /// <summary>
+        /// Instantiate a new DifxAppCompiler.
+        /// </summary>
+        public DifxAppCompiler()
+        {
+            this.components = new HashSet<string>();
+        }
+
+        /// <summary>
+        /// Processes an element for the Compiler.
+        /// </summary>
+        /// <param name="sourceLineNumbers">Source line number for the parent element.</param>
+        /// <param name="parentElement">Parent element of element to process.</param>
+        /// <param name="element">Element to process.</param>
+        /// <param name="contextValues">Extra information about the context in which this element is being parsed.</param>
+        public override void ParseElement(Intermediate intermediate, IntermediateSection section, XElement parentElement, XElement element, IDictionary<string, string> context)
+        {
+            switch (parentElement.Name.LocalName)
+            {
+                case "Component":
+                    var componentId = context["ComponentId"];
+                    var directoryId = context["DirectoryId"];
+                    var componentWin64 = Boolean.Parse(context["Win64"]);
+
+                    switch (element.Name.LocalName)
+                    {
+                        case "Driver":
+                            this.ParseDriverElement(intermediate, section, element, componentId, componentWin64);
+                            break;
+                        default:
+                            this.ParseHelper.UnexpectedElement(parentElement, element);
+                            break;
+                    }
+                    break;
+                default:
+                    this.ParseHelper.UnexpectedElement(parentElement, element);
+                    break;
+            }
+        }
+
+        /// <summary>
+        /// Parses a Driver element.
+        /// </summary>
+        /// <param name="node">Element to parse.</param>
+        /// <param name="componentId">Identifier for parent component.</param>
+        private void ParseDriverElement(Intermediate intermediate, IntermediateSection section, XElement node, string componentId, bool win64)
+        {
+            var sourceLineNumbers = this.ParseHelper.GetSourceLineNumbers(node);
+            int attributes = 0;
+            var sequence = CompilerConstants.IntegerNotSet;
+
+            // check the number of times a Driver element has been nested under this Component element
+            if (null != componentId)
+            {
+                if (this.components.Contains(componentId))
+                {
+                    this.Messaging.Write(ErrorMessages.TooManyElements(sourceLineNumbers, "Component", node.Name.LocalName, 1));
+                }
+                else
+                {
+                    this.components.Add(componentId);
+                }
+            }
+
+            foreach (var attrib in node.Attributes())
+            {
+                if (String.IsNullOrEmpty(attrib.Name.NamespaceName) || this.Namespace == attrib.Name.Namespace)
+                {
+                    switch (attrib.Name.LocalName)
+                    {
+                    case "AddRemovePrograms":
+                        if (YesNoType.No == this.ParseHelper.GetAttributeYesNoValue(sourceLineNumbers, attrib))
+                        {
+                            attributes |= 0x4;
+                        }
+                        break;
+                    case "DeleteFiles":
+                        if (YesNoType.Yes == this.ParseHelper.GetAttributeYesNoValue(sourceLineNumbers, attrib))
+                        {
+                            attributes |= 0x10;
+                        }
+                        break;
+                    case "ForceInstall":
+                        if (YesNoType.Yes == this.ParseHelper.GetAttributeYesNoValue(sourceLineNumbers, attrib))
+                        {
+                            attributes |= 0x1;
+                        }
+                        break;
+                    case "Legacy":
+                        if (YesNoType.Yes == this.ParseHelper.GetAttributeYesNoValue(sourceLineNumbers, attrib))
+                        {
+                            attributes |= 0x8;
+                        }
+                        break;
+                    case "PlugAndPlayPrompt":
+                        if (YesNoType.No == this.ParseHelper.GetAttributeYesNoValue(sourceLineNumbers, attrib))
+                        {
+                            attributes |= 0x2;
+                        }
+                        break;
+                    case "Sequence":
+                        sequence = this.ParseHelper.GetAttributeIntegerValue(sourceLineNumbers, attrib, 0, int.MaxValue);
+                        break;
+                    default:
+                        this.ParseHelper.UnexpectedAttribute(node, attrib);
+                        break;
+                    }
+                }
+                else
+                {
+                    this.ParseHelper.ParseExtensionAttribute(this.Context.Extensions, intermediate, section, node, attrib);
+                }
+            }
+
+            this.ParseHelper.ParseForExtensionElements(this.Context.Extensions, intermediate, section, node);
+
+            if (!this.Messaging.EncounteredError)
+            {
+                switch (this.Context.Platform)
+                {
+                    case Platform.X86:
+                        this.ParseHelper.CreateSimpleReference(section, sourceLineNumbers, SymbolDefinitions.CustomAction, "MsiProcessDrivers");
+                        break;
+                    case Platform.X64:
+                        this.ParseHelper.CreateSimpleReference(section, sourceLineNumbers, SymbolDefinitions.CustomAction, "MsiProcessDrivers_x64");
+                        break;
+                }
+
+                var symbol = section.AddSymbol(new MsiDriverPackagesSymbol(sourceLineNumbers)
+                {
+                    ComponentRef = componentId,
+                    Flags = attributes,
+                });
+
+                if (CompilerConstants.IntegerNotSet != sequence)
+                {
+                    symbol.Sequence = sequence;
+                }
+            }
+        }
+    }
+}
diff --git a/src/ext/DifxApp/wixext/DifxAppDecompiler.cs b/src/ext/DifxApp/wixext/DifxAppDecompiler.cs
new file mode 100644
index 00000000..e41d8b98
--- /dev/null
+++ b/src/ext/DifxApp/wixext/DifxAppDecompiler.cs
@@ -0,0 +1,98 @@
+// 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.DifxApp
+{
+#if TODO_CONSIDER_DECOMPILER
+    using System;
+    using System.Collections;
+    using System.Globalization;
+    using WixToolset.Data;
+    using WixToolset.Extensibility;
+    using DifxApp = WixToolset.Extensions.Serialize.DifxApp;
+    using Wix = WixToolset.Data.Serialize;
+
+    /// <summary>
+    /// The decompiler for the WiX Toolset Driver Install Frameworks for Applications Extension.
+    /// </summary>
+    public sealed class DifxAppDecompiler : DecompilerExtension
+    {
+        /// <summary>
+        /// Creates a decompiler for Gaming Extension.
+        /// </summary>
+        public DifxAppDecompiler()
+        {
+            this.TableDefinitions = DifxAppExtensionData.GetExtensionTableDefinitions();
+        }
+
+        /// <summary>
+        /// Decompiles an extension table.
+        /// </summary>
+        /// <param name="table">The table to decompile.</param>
+        public override void DecompileTable(Table table)
+        {
+            switch (table.Name)
+            {
+                case "MsiDriverPackages":
+                    this.DecompileMsiDriverPackagesTable(table);
+                    break;
+                default:
+                    base.DecompileTable(table);
+                    break;
+            }
+        }
+
+        /// <summary>
+        /// Decompile the MsiDriverPackages table.
+        /// </summary>
+        /// <param name="table">The table to decompile.</param>
+        private void DecompileMsiDriverPackagesTable(Table table)
+        {
+            foreach (Row row in table.Rows)
+            {
+                DifxApp.Driver driver = new DifxApp.Driver();
+
+                int attributes = (int)row[1];
+                if (0x1 == (attributes & 0x1))
+                {
+                    driver.ForceInstall = DifxApp.YesNoType.yes;
+                }
+
+                if (0x2 == (attributes & 0x2))
+                {
+                    driver.PlugAndPlayPrompt = DifxApp.YesNoType.no;
+                }
+
+                if (0x4 == (attributes & 0x4))
+                {
+                    driver.AddRemovePrograms = DifxApp.YesNoType.no;
+                }
+
+                if (0x8 == (attributes & 0x8))
+                {
+                    driver.Legacy = DifxApp.YesNoType.yes;
+                }
+
+                if (0x10 == (attributes & 0x10))
+                {
+                    driver.DeleteFiles = DifxApp.YesNoType.yes;
+                }
+
+                if (null != row[2])
+                {
+                    driver.Sequence = (int)row[2];
+                }
+
+                Wix.Component component = (Wix.Component)this.Core.GetIndexedElement("Component", (string)row[0]);
+                if (null != component)
+                {
+                    component.AddChild(driver);
+                }
+                else
+                {
+                    this.Core.OnMessage(WixWarnings.ExpectedForeignRow(row.SourceLineNumbers, table.Name, row.GetPrimaryKey(DecompilerConstants.PrimaryKeyDelimiter), "Component", (string)row[0], "Component"));
+                }
+            }
+        }
+    }
+#endif
+}
diff --git a/src/ext/DifxApp/wixext/DifxAppExtensionData.cs b/src/ext/DifxApp/wixext/DifxAppExtensionData.cs
new file mode 100644
index 00000000..31a95b8e
--- /dev/null
+++ b/src/ext/DifxApp/wixext/DifxAppExtensionData.cs
@@ -0,0 +1,23 @@
+// 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.DifxApp
+{
+    using WixToolset.Data;
+    using WixToolset.Extensibility;
+
+    public sealed class DifxAppExtensionData : BaseExtensionData
+    {
+        public override string DefaultCulture => "en-US";
+
+        public override bool TryGetSymbolDefinitionByName(string name, out IntermediateSymbolDefinition symbolDefinition)
+        {
+            symbolDefinition = DifxAppSymbolDefinitions.ByName(name);
+            return symbolDefinition != null;
+        }
+
+        public override Intermediate GetLibrary(ISymbolDefinitionCreator symbolDefinitions)
+        {
+            return Intermediate.Load(typeof(DifxAppExtensionData).Assembly, "WixToolset.DifxApp.difxapp.wixlib", symbolDefinitions);
+        }
+    }
+}
diff --git a/src/ext/DifxApp/wixext/DifxAppExtensionFactory.cs b/src/ext/DifxApp/wixext/DifxAppExtensionFactory.cs
new file mode 100644
index 00000000..3932685d
--- /dev/null
+++ b/src/ext/DifxApp/wixext/DifxAppExtensionFactory.cs
@@ -0,0 +1,18 @@
+// 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.DifxApp
+{
+    using System;
+    using System.Collections.Generic;
+    using WixToolset.Extensibility;
+
+    public class DifxAppExtensionFactory : BaseExtensionFactory
+    {
+        protected override IReadOnlyCollection<Type> ExtensionTypes => new[]
+        {
+            typeof(DifxAppCompiler),
+            typeof(DifxAppExtensionData),
+            typeof(DifxAppWindowsInstallerBackendBinderExtension),
+        };
+    }
+}
diff --git a/src/ext/DifxApp/wixext/DifxAppTableDefinitions.cs b/src/ext/DifxApp/wixext/DifxAppTableDefinitions.cs
new file mode 100644
index 00000000..a6c26444
--- /dev/null
+++ b/src/ext/DifxApp/wixext/DifxAppTableDefinitions.cs
@@ -0,0 +1,26 @@
+// 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.DifxApp
+{
+    using WixToolset.Data.WindowsInstaller;
+
+    public static class DifxAppTableDefinitions
+    {
+        public static readonly TableDefinition MsiDriverPackages = new TableDefinition(
+            "MsiDriverPackages",
+            DifxAppSymbolDefinitions.MsiDriverPackages,
+            new[]
+            {
+                new ColumnDefinition("Component", ColumnType.String, 72, primaryKey: true, nullable: false, ColumnCategory.Identifier, keyTable: "Component", keyColumn: 1, description: "Name of the component that represents the driver package", modularizeType: ColumnModularizeType.Column),
+                new ColumnDefinition("Flags", ColumnType.Number, 4, primaryKey: false, nullable: false, ColumnCategory.Unknown, minValue: 0, maxValue: 31, description: "Flags for installing and uninstalling driver packages"),
+                new ColumnDefinition("Sequence", ColumnType.Number, 4, primaryKey: false, nullable: true, ColumnCategory.Unknown, minValue: 0, description: "Order in which the driver packages are processed"),
+            },
+            symbolIdIsPrimaryKey: false
+        );
+
+        public static readonly TableDefinition[] All = new[]
+        {
+            MsiDriverPackages,
+        };
+    }
+}
diff --git a/src/ext/DifxApp/wixext/DifxAppWindowsInstallerBackendBinderExtension.cs b/src/ext/DifxApp/wixext/DifxAppWindowsInstallerBackendBinderExtension.cs
new file mode 100644
index 00000000..41f01175
--- /dev/null
+++ b/src/ext/DifxApp/wixext/DifxAppWindowsInstallerBackendBinderExtension.cs
@@ -0,0 +1,13 @@
+// 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.DifxApp
+{
+    using System.Collections.Generic;
+    using WixToolset.Data.WindowsInstaller;
+    using WixToolset.Extensibility;
+
+    public class DifxAppWindowsInstallerBackendBinderExtension : BaseWindowsInstallerBackendBinderExtension
+    {
+        public override IReadOnlyCollection<TableDefinition> TableDefinitions => DifxAppTableDefinitions.All;
+    }
+}
diff --git a/src/ext/DifxApp/wixext/Symbols/DifxAppSymbolDefinitions.cs b/src/ext/DifxApp/wixext/Symbols/DifxAppSymbolDefinitions.cs
new file mode 100644
index 00000000..76f4d88f
--- /dev/null
+++ b/src/ext/DifxApp/wixext/Symbols/DifxAppSymbolDefinitions.cs
@@ -0,0 +1,39 @@
+// 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.DifxApp
+{
+    using System;
+    using WixToolset.Data;
+
+    public enum DifxAppSymbolDefinitionType
+    {
+        MsiDriverPackages,
+    }
+
+    public static partial class DifxAppSymbolDefinitions
+    {
+        public static readonly Version Version = new Version("4.0.0");
+
+        public static IntermediateSymbolDefinition ByName(string name)
+        {
+            if (!Enum.TryParse(name, out DifxAppSymbolDefinitionType type))
+            {
+                return null;
+            }
+
+            return ByType(type);
+        }
+
+        public static IntermediateSymbolDefinition ByType(DifxAppSymbolDefinitionType type)
+        {
+            switch (type)
+            {
+                case DifxAppSymbolDefinitionType.MsiDriverPackages:
+                    return DifxAppSymbolDefinitions.MsiDriverPackages;
+
+                default:
+                    throw new ArgumentOutOfRangeException(nameof(type));
+            }
+        }
+    }
+}
diff --git a/src/ext/DifxApp/wixext/Symbols/MsiDriverPackagesSymbol.cs b/src/ext/DifxApp/wixext/Symbols/MsiDriverPackagesSymbol.cs
new file mode 100644
index 00000000..2fd91bc8
--- /dev/null
+++ b/src/ext/DifxApp/wixext/Symbols/MsiDriverPackagesSymbol.cs
@@ -0,0 +1,63 @@
+// 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.DifxApp
+{
+    using WixToolset.Data;
+    using WixToolset.DifxApp.Symbols;
+
+    public static partial class DifxAppSymbolDefinitions
+    {
+        public static readonly IntermediateSymbolDefinition MsiDriverPackages = new IntermediateSymbolDefinition(
+            DifxAppSymbolDefinitionType.MsiDriverPackages.ToString(),
+            new[]
+            {
+                new IntermediateFieldDefinition(nameof(MsiDriverPackagesSymbolFields.ComponentRef), IntermediateFieldType.String),
+                new IntermediateFieldDefinition(nameof(MsiDriverPackagesSymbolFields.Flags), IntermediateFieldType.Number),
+                new IntermediateFieldDefinition(nameof(MsiDriverPackagesSymbolFields.Sequence), IntermediateFieldType.Number),
+            },
+            typeof(MsiDriverPackagesSymbol));
+    }
+}
+
+namespace WixToolset.DifxApp.Symbols
+{
+    using WixToolset.Data;
+
+    public enum MsiDriverPackagesSymbolFields
+    {
+        ComponentRef,
+        Flags,
+        Sequence,
+    }
+
+    public class MsiDriverPackagesSymbol : IntermediateSymbol
+    {
+        public MsiDriverPackagesSymbol() : base(DifxAppSymbolDefinitions.MsiDriverPackages, null, null)
+        {
+        }
+
+        public MsiDriverPackagesSymbol(SourceLineNumber sourceLineNumber, Identifier id = null) : base(DifxAppSymbolDefinitions.MsiDriverPackages, sourceLineNumber, id)
+        {
+        }
+
+        public IntermediateField this[MsiDriverPackagesSymbolFields index] => this.Fields[(int)index];
+
+        public string ComponentRef
+        {
+            get => this.Fields[(int)MsiDriverPackagesSymbolFields.ComponentRef].AsString();
+            set => this.Set((int)MsiDriverPackagesSymbolFields.ComponentRef, value);
+        }
+
+        public int Flags
+        {
+            get => this.Fields[(int)MsiDriverPackagesSymbolFields.Flags].AsNumber();
+            set => this.Set((int)MsiDriverPackagesSymbolFields.Flags, value);
+        }
+
+        public int? Sequence
+        {
+            get => this.Fields[(int)MsiDriverPackagesSymbolFields.Sequence].AsNullableNumber();
+            set => this.Set((int)MsiDriverPackagesSymbolFields.Sequence, value);
+        }
+    }
+}
\ No newline at end of file
diff --git a/src/ext/DifxApp/wixext/WixToolset.DifxApp.wixext.csproj b/src/ext/DifxApp/wixext/WixToolset.DifxApp.wixext.csproj
new file mode 100644
index 00000000..50d3debf
--- /dev/null
+++ b/src/ext/DifxApp/wixext/WixToolset.DifxApp.wixext.csproj
@@ -0,0 +1,28 @@
+<?xml version="1.0" encoding="utf-8"?>
+<!-- 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. -->
+
+<Project Sdk="Microsoft.NET.Sdk">
+  <PropertyGroup>
+    <TargetFramework>netstandard2.0</TargetFramework>
+    <RootNamespace>WixToolset.DifxApp</RootNamespace>
+    <Description>WiX Toolset DIFxApp Extension</Description>
+    <Title>WiX Toolset DIFxApp Extension</Title>
+    <IsTool>true</IsTool>
+    <ContentTargetFolders>build</ContentTargetFolders>
+  </PropertyGroup>
+  <ItemGroup>
+    <Content Include="$(MSBuildThisFileName).targets" />
+    <EmbeddedResource Include="$(OutputPath)..\difxapp.wixlib" />
+  </ItemGroup>
+  <ItemGroup>
+    <PackageReference Include="WixToolset.Extensibility" Version="4.0.*" PrivateAssets="all" />
+  </ItemGroup>
+
+  <ItemGroup>
+    <ProjectReference Include="..\wixlib\difxapp.wixproj" ReferenceOutputAssembly="false" Condition=" '$(NCrunch)'=='' " />
+  </ItemGroup>
+
+  <ItemGroup>
+    <PackageReference Include="Nerdbank.GitVersioning" Version="3.3.37" PrivateAssets="all" />
+  </ItemGroup>
+</Project>
diff --git a/src/ext/DifxApp/wixext/WixToolset.DifxApp.wixext.targets b/src/ext/DifxApp/wixext/WixToolset.DifxApp.wixext.targets
new file mode 100644
index 00000000..b01d4116
--- /dev/null
+++ b/src/ext/DifxApp/wixext/WixToolset.DifxApp.wixext.targets
@@ -0,0 +1,11 @@
+<?xml version="1.0" encoding="utf-8"?>
+<!-- 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. -->
+
+<Project xmlns="http://schemas.microsoft.com/developer/msbuild/2003" ToolsVersion="4.0">
+  <PropertyGroup>
+    <WixToolsetDifxAppWixextPath Condition=" '$(WixToolsetDifxAppWixextPath)' == '' ">$(MSBuildThisFileDirectory)..\tools\WixToolset.DifxApp.wixext.dll</WixToolsetDifxAppWixextPath>
+  </PropertyGroup>
+  <ItemGroup>
+    <WixExtension Include="$(WixToolsetDifxAppWixextPath)" />
+  </ItemGroup>
+</Project>