1 files changed, 204 insertions, 0 deletions

diff --git a/src/internal/WixInternal.MSTestSupport/Builder.cs b/src/internal/WixInternal.MSTestSupport/Builder.cs
new file mode 100644
index 00000000..62f2891b
--- /dev/null
+++ b/src/internal/WixInternal.MSTestSupport/Builder.cs
@@ -0,0 +1,204 @@
+// 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 WixInternal.MSTestSupport
+{
+    using System;
+    using System.Collections.Generic;
+    using System.IO;
+
+    public class Builder
+    {
+        public Builder(string sourceFolder, Type extensionType = null, string[] bindPaths = null, string outputFile = null)
+        {
+            this.SourceFolder = sourceFolder;
+            if (extensionType != null)
+            {
+                this.ExtensionTypes = new Type[] { extensionType };
+            }
+            else
+            {
+                this.ExtensionTypes = new Type[] { };
+            }
+            this.BindPaths = bindPaths;
+            this.OutputFile = outputFile ?? "test.msi";
+        }
+
+        public Builder(string sourceFolder, Type[] extensionTypes, string[] bindPaths = null, string outputFile = null)
+        {
+            this.SourceFolder = sourceFolder;
+            this.ExtensionTypes = extensionTypes;
+            this.BindPaths = bindPaths;
+            this.OutputFile = outputFile ?? "test.msi";
+        }
+
+        public string[] BindPaths { get; set; }
+
+        public Type[] ExtensionTypes { get; set; }
+
+        public string OutputFile { get; set; }
+
+        public string SourceFolder { get; }
+
+        public string[] BuildAndQuery(Action<string[]> buildFunc, params string[] tables)
+        {
+            return this.BuildAndQuery(buildFunc, validate: false, tables);
+        }
+
+        public string[] BuildAndQuery(Action<string[]> buildFunc, bool validate, params string[] tables)
+        {
+            var sourceFiles = Directory.GetFiles(this.SourceFolder, "*.wxs");
+            var wxlFiles = Directory.GetFiles(this.SourceFolder, "*.wxl");
+
+            using (var fs = new DisposableFileSystem())
+            {
+                var intermediateFolder = fs.GetFolder();
+                var outputPath = Path.Combine(intermediateFolder, "bin", this.OutputFile);
+
+                var args = new List<string>
+                {
+                    "build",
+                    "-o", outputPath,
+                    "-intermediateFolder", intermediateFolder,
+                };
+
+                foreach (var ext in this.ExtensionTypes)
+                {
+                    args.Add("-ext");
+                    args.Add(Path.GetFullPath(ext.Assembly.Location));
+                }
+
+                args.AddRange(sourceFiles);
+
+                foreach (var wxlFile in wxlFiles)
+                {
+                    args.Add("-loc");
+                    args.Add(wxlFile);
+                }
+
+                foreach (var bindPath in this.BindPaths)
+                {
+                    args.Add("-bindpath");
+                    args.Add(bindPath);
+                }
+
+                buildFunc(args.ToArray());
+
+                if (validate)
+                {
+                    args = new List<string>
+                    {
+                        "msi",
+                        "validate",
+                        "-intermediateFolder", intermediateFolder,
+                        outputPath,
+                    };
+
+                    buildFunc(args.ToArray());
+                }
+
+                return Query.QueryDatabase(outputPath, tables);
+            }
+        }
+
+        public void BuildAndDecompileAndBuild(Action<string[]> buildFunc, Action<string[]> decompileFunc, string decompilePath, bool validate = false)
+        {
+            var sourceFiles = Directory.GetFiles(this.SourceFolder, "*.wxs");
+            var wxlFiles = Directory.GetFiles(this.SourceFolder, "*.wxl");
+
+            using (var fs = new DisposableFileSystem())
+            {
+                var intermediateFolder = fs.GetFolder();
+                var outputFolder = Path.Combine(intermediateFolder, "bin");
+                var decompileExtractFolder = Path.Combine(intermediateFolder, "decompiled", "extract");
+                var decompileIntermediateFolder = Path.Combine(intermediateFolder, "decompiled", "obj");
+                var decompileBuildFolder = Path.Combine(intermediateFolder, "decompiled", "bin");
+                var outputPath = Path.Combine(outputFolder, this.OutputFile);
+                var decompileBuildPath = Path.Combine(decompileBuildFolder, this.OutputFile);
+
+                // First build.
+                var firstBuildArgs = new List<string>
+                {
+                    "build",
+                    "-o", outputPath,
+                    "-intermediateFolder", intermediateFolder,
+                };
+
+                foreach (var ext in this.ExtensionTypes)
+                {
+                    firstBuildArgs.Add("-ext");
+                    firstBuildArgs.Add(Path.GetFullPath(ext.Assembly.Location));
+                }
+
+                firstBuildArgs.AddRange(sourceFiles);
+
+                foreach (var wxlFile in wxlFiles)
+                {
+                    firstBuildArgs.Add("-loc");
+                    firstBuildArgs.Add(wxlFile);
+                }
+
+                foreach (var bindPath in this.BindPaths)
+                {
+                    firstBuildArgs.Add("-bindpath");
+                    firstBuildArgs.Add(bindPath);
+                }
+
+                buildFunc(firstBuildArgs.ToArray());
+
+                if (validate)
+                {
+                    firstBuildArgs = new List<string>
+                    {
+                        "msi",
+                        "validate",
+                        "-intermediateFolder", intermediateFolder,
+                        outputPath,
+                    };
+
+                    buildFunc(firstBuildArgs.ToArray());
+                }
+
+                // Decompile built output.
+                var decompileArgs = new List<string>
+                {
+                    "msi", "decompile",
+                    outputPath,
+                    "-intermediateFolder", decompileIntermediateFolder,
+                    "-x", decompileExtractFolder,
+                    "-o", decompilePath
+                };
+
+                foreach (var ext in this.ExtensionTypes)
+                {
+                    decompileArgs.Add("-ext");
+                    decompileArgs.Add(Path.GetFullPath(ext.Assembly.Location));
+                }
+
+                decompileFunc(decompileArgs.ToArray());
+
+                // Build decompiled output.
+                var secondBuildArgs = new List<string>
+                {
+                    "build",
+                    decompilePath,
+                    "-o", decompileBuildPath,
+                    "-intermediateFolder", decompileIntermediateFolder
+                };
+
+                foreach (var ext in this.ExtensionTypes)
+                {
+                    secondBuildArgs.Add("-ext");
+                    secondBuildArgs.Add(Path.GetFullPath(ext.Assembly.Location));
+                }
+
+                secondBuildArgs.Add("-bindpath");
+                secondBuildArgs.Add(outputFolder);
+
+                secondBuildArgs.Add("-bindpath");
+                secondBuildArgs.Add(decompileExtractFolder);
+
+                buildFunc(secondBuildArgs.ToArray());
+            }
+        }
+    }
+}