Add StringPrivateKeySource

adriangodong · Aug 2, 2018 · 28cc702 · 28cc702
1 parent 9bad63b
commit 28cc702
Show file tree

Hide file tree

Showing 3 changed files with 43 additions and 13 deletions.
diff --git a/GitHubJwt/EnvironmentVariablePrivateKeySource.cs b/GitHubJwt/EnvironmentVariablePrivateKeySource.cs
@@ -1,6 +1,5 @@
 using System;
 using System.IO;
-using System.Text;
 
 namespace GitHubJwt
 {
@@ -15,20 +14,9 @@ public EnvironmentVariablePrivateKeySource(string environmentVariableName)
 
         public TextReader GetPrivateKeyReader()
         {
-            var privateKeyPem = HydrateEnvVarPem(
-                Environment.GetEnvironmentVariable(environmentVariableName));
+            var privateKeyPem = Environment.GetEnvironmentVariable(environmentVariableName).HydrateRsaVariable();
             return new StringReader(privateKeyPem);
         }
 
-        private static string HydrateEnvVarPem(string input)
-        {
-            var stringBuilder = new StringBuilder();
-
-            stringBuilder.AppendLine("-----BEGIN RSA PRIVATE KEY-----");
-            stringBuilder.AppendLine(input);
-            stringBuilder.AppendLine("-----END RSA PRIVATE KEY-----");
-
-            return stringBuilder.ToString();
-        }
     }
 }
diff --git a/GitHubJwt/StringExtensions.cs b/GitHubJwt/StringExtensions.cs
@@ -0,0 +1,16 @@
+using System.Text;
+
+namespace GitHubJwt
+{
+    internal static class StringExtensions
+    {
+        public static string HydrateRsaVariable(this string input)
+        {
+            StringBuilder stringBuilder = new StringBuilder();
+            stringBuilder.AppendLine("-----BEGIN RSA PRIVATE KEY-----");
+            stringBuilder.AppendLine(input);
+            stringBuilder.AppendLine("-----END RSA PRIVATE KEY-----");
+            return stringBuilder.ToString();
+        }
+    }
+}
diff --git a/GitHubJwt/StringPrivateKeySource.cs b/GitHubJwt/StringPrivateKeySource.cs
@@ -0,0 +1,26 @@
+using System;
+using System.IO;
+using System.Text;
+
+namespace GitHubJwt
+{
+    public class StringPrivateKeySource : IPrivateKeySource
+    {
+        protected readonly string Key;
+
+        public StringPrivateKeySource(string key)
+        {
+            if (string.IsNullOrEmpty(key))
+            {
+                throw new ArgumentNullException(nameof(key));
+            }
+
+            Key = key;
+        }
+
+        public TextReader GetPrivateKeyReader()
+        {
+            return new StringReader(Key.HydrateRsaVariable());
+        }
+    }
+}