Initial code commit

1 files changed, 115 insertions, 0 deletions

diff --git a/src/WixToolset.Core/Bind/Databases/ProcessUncompressedFilesCommand.cs b/src/WixToolset.Core/Bind/Databases/ProcessUncompressedFilesCommand.cs
new file mode 100644
index 00000000..dd7b85b7
--- /dev/null
+++ b/src/WixToolset.Core/Bind/Databases/ProcessUncompressedFilesCommand.cs
@@ -0,0 +1,115 @@
+// 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.Bind.Databases
+{
+    using System;
+    using System.Collections;
+    using System.Collections.Generic;
+    using System.IO;
+    using WixToolset.Data;
+    using WixToolset.Data.Rows;
+    using WixToolset.Msi;
+    using WixToolset.Core.Native;
+
+    /// <summary>
+    /// Defines the file transfers necessary to layout the uncompressed files.
+    /// </summary>
+    internal class ProcessUncompressedFilesCommand : ICommand
+    {
+        public string DatabasePath { private get; set; }
+
+        public IEnumerable<FileFacade> FileFacades { private get; set; }
+
+        public RowDictionary<MediaRow> MediaRows { private get; set; }
+
+        public string LayoutDirectory { private get; set; }
+
+        public bool Compressed { private get; set; }
+
+        public bool LongNamesInImage { private get; set; }
+
+        public Func<MediaRow, string, string, string> ResolveMedia { private get; set; }
+
+        public Table WixMediaTable { private get; set; }
+
+        public IEnumerable<FileTransfer> FileTransfers { get; private set; }
+
+        public void Execute()
+        {
+            List<FileTransfer> fileTransfers = new List<FileTransfer>();
+
+            Hashtable directories = new Hashtable();
+
+            RowDictionary<WixMediaRow> wixMediaRows = new RowDictionary<WixMediaRow>(this.WixMediaTable);
+
+            using (Database db = new Database(this.DatabasePath, OpenDatabase.ReadOnly))
+            {
+                using (View directoryView = db.OpenExecuteView("SELECT `Directory`, `Directory_Parent`, `DefaultDir` FROM `Directory`"))
+                {
+                    while (true)
+                    {
+                        using (Record directoryRecord = directoryView.Fetch())
+                        {
+                            if (null == directoryRecord)
+                            {
+                                break;
+                            }
+
+                            string sourceName = Installer.GetName(directoryRecord.GetString(3), true, this.LongNamesInImage);
+
+                            directories.Add(directoryRecord.GetString(1), new ResolvedDirectory(directoryRecord.GetString(2), sourceName));
+                        }
+                    }
+                }
+
+                using (View fileView = db.OpenView("SELECT `Directory_`, `FileName` FROM `Component`, `File` WHERE `Component`.`Component`=`File`.`Component_` AND `File`.`File`=?"))
+                {
+                    using (Record fileQueryRecord = new Record(1))
+                    {
+                        // for each file in the array of uncompressed files
+                        foreach (FileFacade facade in this.FileFacades)
+                        {
+                            MediaRow mediaRow = this.MediaRows.Get(facade.WixFile.DiskId);
+                            string relativeFileLayoutPath = null;
+
+                            WixMediaRow wixMediaRow = null;
+                            string mediaLayoutFolder = null;
+
+                            if (wixMediaRows.TryGetValue(mediaRow.GetKey(), out wixMediaRow))
+                            {
+                                mediaLayoutFolder = wixMediaRow.Layout;
+                            }
+
+                            string mediaLayoutDirectory = this.ResolveMedia(mediaRow, mediaLayoutFolder, this.LayoutDirectory);
+
+                            // setup up the query record and find the appropriate file in the
+                            // previously executed file view
+                            fileQueryRecord[1] = facade.File.File;
+                            fileView.Execute(fileQueryRecord);
+
+                            using (Record fileRecord = fileView.Fetch())
+                            {
+                                if (null == fileRecord)
+                                {
+                                    throw new WixException(WixErrors.FileIdentifierNotFound(facade.File.SourceLineNumbers, facade.File.File));
+                                }
+
+                                relativeFileLayoutPath = Binder.GetFileSourcePath(directories, fileRecord[1], fileRecord[2], this.Compressed, this.LongNamesInImage);
+                            }
+
+                            // finally put together the base media layout path and the relative file layout path
+                            string fileLayoutPath = Path.Combine(mediaLayoutDirectory, relativeFileLayoutPath);
+                            FileTransfer transfer;
+                            if (FileTransfer.TryCreate(facade.WixFile.Source, fileLayoutPath, false, "File", facade.File.SourceLineNumbers, out transfer))
+                            {
+                                fileTransfers.Add(transfer);
+                            }
+                        }
+                    }
+                }
+            }
+
+            this.FileTransfers = fileTransfers;
+        }
+    }
+}