1 files changed, 80 insertions, 0 deletions

diff --git a/src/tools/heat/IIsHeatExtension.cs b/src/tools/heat/IIsHeatExtension.cs
new file mode 100644
index 00000000..be998370
--- /dev/null
+++ b/src/tools/heat/IIsHeatExtension.cs
@@ -0,0 +1,80 @@
+// 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 WixToolset.Harvesters.Data;
+    using WixToolset.Harvesters.Extensibility;
+
+    /// <summary>
+    /// An IIS harvesting extension for the WiX Toolset Harvester application.
+    /// </summary>
+    public sealed class IIsHeatExtension : BaseHeatExtension
+    {
+        /// <summary>
+        /// Gets the supported command line types for this extension.
+        /// </summary>
+        /// <value>The supported command line types for this extension.</value>
+        public override HeatCommandLineOption[] CommandLineTypes
+        {
+            get
+            {
+                return new HeatCommandLineOption[]
+                {
+                    new HeatCommandLineOption("website", "harvest an IIS web site"),
+                };
+            }
+        }
+
+        /// <summary>
+        /// Parse the command line options for this extension.
+        /// </summary>
+        /// <param name="type">The active harvester type.</param>
+        /// <param name="args">The option arguments.</param>
+        public override void ParseOptions(string type, string[] args)
+        {
+            bool active = false;
+            IHarvesterExtension harvesterExtension = null;
+            IIsHarvesterMutator iisHarvesterMutator = new IIsHarvesterMutator();
+
+            // select the harvester
+            switch (type)
+            {
+                case "website":
+                    harvesterExtension = new IIsWebSiteHarvester();
+                    active = true;
+                    break;
+            }
+
+            // set default settings
+            iisHarvesterMutator.SetUniqueIdentifiers = true;
+
+            // parse the options
+            foreach (string arg in args)
+            {
+                if (null == arg || 0 == arg.Length) // skip blank arguments
+                {
+                    continue;
+                }
+
+                if ('-' == arg[0] || '/' == arg[0])
+                {
+                    string parameter = arg.Substring(1);
+
+                    if ("suid" == parameter)
+                    {
+                        iisHarvesterMutator.SetUniqueIdentifiers = false;
+                    }
+                }
+            }
+
+            // set the appropriate harvester extension
+            if (active)
+            {
+                this.Core.Harvester.Extension = harvesterExtension;
+                this.Core.Mutator.AddExtension(iisHarvesterMutator);
+                this.Core.Mutator.AddExtension(new IIsFinalizeHarvesterMutator());
+                this.Core.Mutator.AddExtension(new UtilFinalizeHarvesterMutator());
+            }
+        }
+    }
+}