1 files changed, 107 insertions, 0 deletions

diff --git a/src/tools/Dtf/Inventory/Features.cs b/src/tools/Dtf/Inventory/Features.cs
new file mode 100644
index 00000000..9d2747ba
--- /dev/null
+++ b/src/tools/Dtf/Inventory/Features.cs
@@ -0,0 +1,107 @@
+// 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.
+
+using System;
+using System.IO;
+using System.Data;
+using System.Collections;
+using System.Collections.Generic;
+using System.Globalization;
+using System.Windows.Forms;
+using WixToolset.Dtf.WindowsInstaller;
+
+namespace WixToolset.Dtf.Tools.Inventory
+{
+        /// <summary>
+        /// Provides inventory data about features of products installed on the system.
+        /// </summary>
+        public class FeaturesInventory : IInventoryDataProvider
+        {
+        private static object syncRoot = new object();
+
+                public FeaturesInventory()
+                {
+                }
+
+                public string Description
+                {
+                        get { return "Features of installed products"; }
+                }
+
+                public string[] GetNodes(InventoryDataLoadStatusCallback statusCallback)
+                {
+            statusCallback(0, @"Products\...\Features");
+            ArrayList nodes = new ArrayList();
+                        foreach (ProductInstallation product in ProductInstallation.AllProducts)
+                        {
+                nodes.Add(String.Format(@"Products\{0}\Features", MsiUtils.GetProductName(product.ProductCode)));
+                        }
+                        statusCallback(nodes.Count, String.Empty);
+                        return (string[]) nodes.ToArray(typeof(string));
+                }
+
+                public bool IsNodeSearchable(string searchRoot, string searchNode)
+                {
+                        return true;
+                }
+
+                public DataView GetData(string nodePath)
+                {
+                        string[] path = nodePath.Split('\\');
+
+                        if(path.Length == 3 && path[0] == "Products" && path[2] == "Features")
+                        {
+                                return GetProductFeaturesData(MsiUtils.GetProductCode(path[1]));
+                        }
+                        return null;
+                }
+
+                public DataView GetProductFeaturesData(string productCode)
+                {
+                        DataTable table = new DataTable("ProductFeatures");
+                        table.Locale = CultureInfo.InvariantCulture;
+                        table.Columns.Add("ProductFeaturesFeatureTitle", typeof(string));
+                        table.Columns.Add("ProductFeaturesFeatureName", typeof(string));
+                        table.Columns.Add("ProductFeaturesInstallState", typeof(string));
+
+                        try
+                        {
+                                IntPtr hWnd = IntPtr.Zero;
+                Installer.SetInternalUI(InstallUIOptions.Silent, ref hWnd);
+                                lock(syncRoot)  // Only one Installer session can be active at a time
+                                {
+                                        using(Session session = Installer.OpenProduct(productCode))
+                                        {
+                                                session.DoAction("CostInitialize");
+                                                session.DoAction("FileCost");
+                                                session.DoAction("CostFinalize");
+
+                                                IList<string> featuresAndTitles = session.Database.ExecuteStringQuery(
+                                                        "SELECT `Title`, `Feature` FROM `Feature`");
+
+                                                for(int i = 0; i < featuresAndTitles.Count; i += 2)
+                                                {
+                            InstallState featureState = session.Features[featuresAndTitles[i + 1]].CurrentState;
+                                                        table.Rows.Add(new object[] { featuresAndTitles[i], featuresAndTitles[i+1],
+                                                                                                                        (featureState == InstallState.Advertised ? "Advertised" : featureState.ToString()) });
+                                                }
+                                        }
+                                }
+                                return new DataView(table, "", "ProductFeaturesFeatureTitle ASC", DataViewRowState.CurrentRows);
+                        }
+                        catch(InstallerException) { }
+                        catch(IOException) { }
+                        return null;
+                }
+
+                public string GetLink(string nodePath, DataRow row)
+                {
+                        string[] path = nodePath.Split('\\');
+
+                        if(path.Length == 3 && path[0] == "Products" && path[2] == "Features")
+                        {
+                                return String.Format(@"Products\{0}\Features\{1}", path[1], row["ProductFeaturesFeatureName"]);
+                        }
+                        return null;
+                }
+        }
+}