1 files changed, 160 insertions, 0 deletions

diff --git a/src/tools/heat/IIsFinalizeHarvesterMutator.cs b/src/tools/heat/IIsFinalizeHarvesterMutator.cs
new file mode 100644
index 00000000..cfe90272
--- /dev/null
+++ b/src/tools/heat/IIsFinalizeHarvesterMutator.cs
@@ -0,0 +1,160 @@
+// 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.Harvesters
+{
+    using System;
+    using System.Collections;
+    using System.Collections.Specialized;
+    using WixToolset.Harvesters.Data;
+    using WixToolset.Harvesters.Extensibility;
+    using Wix = WixToolset.Harvesters.Serialize;
+    using IIs = Serialize.IIs;
+
+    /// <summary>
+    /// The finalize harvester mutator for the WiX Toolset Internet Information Services Extension.
+    /// </summary>
+    public sealed class IIsFinalizeHarvesterMutator : BaseMutatorExtension
+    {
+        private Hashtable directoryPaths;
+        private Hashtable filePaths;
+        private ArrayList webFilters;
+        private ArrayList webSites;
+        private ArrayList webVirtualDirs;
+
+        /// <summary>
+        /// Instantiate a new IIsFinalizeHarvesterMutator.
+        /// </summary>
+        public IIsFinalizeHarvesterMutator()
+        {
+            this.directoryPaths = CollectionsUtil.CreateCaseInsensitiveHashtable();
+            this.filePaths = CollectionsUtil.CreateCaseInsensitiveHashtable();
+            this.webFilters = new ArrayList();
+            this.webSites = new ArrayList();
+            this.webVirtualDirs = new ArrayList();
+        }
+
+        /// <summary>
+        /// Gets the sequence of this mutator extension.
+        /// </summary>
+        /// <value>The sequence of this mutator extension.</value>
+        public override int Sequence
+        {
+            get { return 1900; }
+        }
+
+        /// <summary>
+        /// Mutate a WiX document.
+        /// </summary>
+        /// <param name="wix">The Wix document element.</param>
+        public override void Mutate(Wix.Wix wix)
+        {
+            this.directoryPaths.Clear();
+            this.filePaths.Clear();
+            this.webFilters.Clear();
+            this.webSites.Clear();
+            this.webVirtualDirs.Clear();
+
+            this.IndexElement(wix);
+
+            this.MutateWebFilters();
+            this.MutateWebSites();
+            this.MutateWebVirtualDirs();
+        }
+
+        /// <summary>
+        /// Index an element.
+        /// </summary>
+        /// <param name="element">The element to index.</param>
+        private void IndexElement(Wix.ISchemaElement element)
+        {
+            if (element is IIs.WebFilter)
+            {
+                this.webFilters.Add(element);
+            }
+            else if (element is IIs.WebSite)
+            {
+                this.webSites.Add(element);
+            }
+            else if (element is IIs.WebVirtualDir)
+            {
+                this.webVirtualDirs.Add(element);
+            }
+            else if (element is Wix.Directory)
+            {
+                Wix.Directory directory = (Wix.Directory)element;
+
+                if (null != directory.Id && null != directory.FileSource)
+                {
+                    this.directoryPaths.Add(directory.FileSource, directory.Id);
+                }
+            }
+            else if (element is Wix.File)
+            {
+                Wix.File file = (Wix.File)element;
+
+                if (null != file.Id && null != file.Source)
+                {
+                    this.filePaths[file.Source] = String.Concat("[#", file.Id, "]");
+                }
+            }
+
+            // index the child elements
+            if (element is Wix.IParentElement)
+            {
+                foreach (Wix.ISchemaElement childElement in ((Wix.IParentElement)element).Children)
+                {
+                    this.IndexElement(childElement);
+                }
+            }
+        }
+
+        /// <summary>
+        /// Mutate the WebFilters.
+        /// </summary>
+        private void MutateWebFilters()
+        {
+            foreach (IIs.WebFilter webFilter in this.webFilters)
+            {
+                webFilter.Path = (string)this.filePaths[webFilter.Path];
+            }
+        }
+
+        /// <summary>
+        /// Mutate the WebSites.
+        /// </summary>
+        private void MutateWebSites()
+        {
+            foreach (IIs.WebSite webSite in this.webSites)
+            {
+                string path = (string)this.directoryPaths[webSite.Directory];
+                if (null == path)
+                {
+                    this.Core.Messaging.Write(HarvesterWarnings.EncounteredNullDirectoryForWebSite(path));
+                }
+                else
+                {
+                    webSite.Directory = path;
+                }
+            }
+        }
+
+        /// <summary>
+        /// Mutate the WebVirtualDirs.
+        /// </summary>
+        private void MutateWebVirtualDirs()
+        {
+            foreach (IIs.WebVirtualDir webVirtualDir in this.webVirtualDirs)
+            {
+                string path = (string)this.directoryPaths[webVirtualDir.Directory];
+                if (null == path)
+                {
+                    this.Core.Messaging.Write(HarvesterWarnings.EncounteredNullDirectoryForWebSite(path));
+                }
+                else
+                {
+                    webVirtualDir.Directory = path;
+                }
+            }
+        }
+    }
+}