(GH-36) Return PublicKeyToken as string

chocolatey · Jan 25, 2015 · ce3a0a4 · ce3a0a4
1 parent 749058e
commit ce3a0a4
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 0 deletions.
diff --git a/src/chocolatey/AssemblyExtensions.cs b/src/chocolatey/AssemblyExtensions.cs
@@ -16,6 +16,7 @@
 namespace chocolatey
 {
     using System.IO;
+    using System.Linq;
     using infrastructure.adapters;
 
     /// <summary>
@@ -59,5 +60,22 @@ public static Stream get_manifest_stream(this IAssembly assembly, string manifes
         {
             return assembly.GetManifestResourceStream(manifestResourceStreamLocation);
         }
+
+        /// <summary>
+        ///   Gets the public key token of an assembly.
+        /// </summary>
+        /// <param name="assembly">The assembly.</param>
+        /// <returns></returns>
+        /// <remarks>Borrowed heavily from http://dhvik.blogspot.com/2009/05/assemblynamegetpublickeytoken-tostring.html </remarks>
+        public static string get_public_key_token(this IAssembly assembly)
+        {
+            if (assembly == null) return string.Empty;
+
+            byte[] publicKeyToken = assembly.GetName().GetPublicKeyToken();
+
+            if (publicKeyToken == null || publicKeyToken.Length == 0) return string.Empty;
+
+            return publicKeyToken.Select(x => x.ToString("x2")).Aggregate((x, y) => x + y);
+        }
     }
 }
diff --git a/src/chocolatey/infrastructure/adapters/Assembly.cs b/src/chocolatey/infrastructure/adapters/Assembly.cs
@@ -18,6 +18,7 @@ namespace chocolatey.infrastructure.adapters
     using System;
     using System.ComponentModel;
     using System.IO;
+    using System.Reflection;
 
     // ReSharper disable InconsistentNaming
 
@@ -66,6 +67,10 @@ public static IAssembly GetAssembly(Type type)
             return new Assembly(System.Reflection.Assembly.GetAssembly(type));
             //return System.Reflection.Assembly.GetAssembly(type);
         }
+        public AssemblyName GetName()
+        {
+            return _assembly.GetName();
+        }
 
         public static IAssembly GetExecutingAssembly()
         {

diff --git a/src/chocolatey/infrastructure/adapters/IAssembly.cs b/src/chocolatey/infrastructure/adapters/IAssembly.cs
@@ -18,6 +18,7 @@ namespace chocolatey.infrastructure.adapters
     using System;
     using System.ComponentModel;
     using System.IO;
+    using System.Reflection;
 
     // ReSharper disable InconsistentNaming
 
@@ -64,6 +65,8 @@ public interface IAssembly
         /// </returns>
         string[] GetManifestResourceNames();
 
+        AssemblyName GetName();
+
         /// <summary>
         ///   Loads the specified manifest resource from this assembly.
         /// </summary>