1 files changed, 46 insertions, 0 deletions

diff --git a/src/samples/Dtf/Inventory/msiutils.cs b/src/samples/Dtf/Inventory/msiutils.cs
new file mode 100644
index 00000000..a345e194
--- /dev/null
+++ b/src/samples/Dtf/Inventory/msiutils.cs
@@ -0,0 +1,46 @@
+// 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.Collections;
+using WixToolset.Dtf.WindowsInstaller;
+
+
+namespace WixToolset.Dtf.Samples.Inventory
+{
+        public class MsiUtils
+        {
+                private static Hashtable productCodesToNames = new Hashtable();
+                private static Hashtable productNamesToCodes = new Hashtable();
+
+                public static string GetProductName(string productCode)
+                {
+                        string productName = (string) productCodesToNames[productCode];
+                        if(productName == null)
+                        {
+                productName = new ProductInstallation(productCode).ProductName;
+                                productName = productName.Replace('\\', ' ');
+                                if(productNamesToCodes.Contains(productName))
+                                {
+                                        string modifiedProductName = null;
+                                        for(int i = 2; i < Int32.MaxValue; i++)
+                                        {
+                                                modifiedProductName = productName + " [" + i + "]";
+                                                if(!productNamesToCodes.Contains(modifiedProductName)) break;
+                                        }
+                                        productName = modifiedProductName;
+                                }
+                                productCodesToNames[productCode] = productName;
+                                productNamesToCodes[productName] = productCode;
+                        }
+                        return productName;
+                }
+
+                // Assumes GetProductName() has already been called for this product.
+                public static string GetProductCode(string productName)
+                {
+                        return (string) productNamesToCodes[productName];
+                }
+
+                private MsiUtils() { }
+        }
+}