Preview support for MG CLI

Author: pschaeflein

src/MagicTooltips/Dtos/ProviderKeys.cs

@@ -2,6 +2,7 @@
 {
   public class ProviderKeys
   {
+    public static string MGCLI = "mgcli";  // Probably merge w/PWSH. Review after release
     public static string MicrosoftGraph = "mg";
     public static string M365 = "m365";
     public static string Kubernetes = "kubernetes";

src/MagicTooltips/Providers/AzureProvider.cs

@@ -32,7 +32,7 @@ public string GetValue()
           }
           azProfilePath = Path.Combine(azConfigDir, "azureProfile.json");
         }
-        currentFileHash = CalculateMd5(azProfilePath);
+        currentFileHash = Md5Utility.CalculateMd5(azProfilePath);
       }
       catch (Exception ex)
       {
@@ -55,17 +55,5 @@ public string GetValue()
 
       return azureAccount;
     }
-
-    private static string CalculateMd5(string filePath)
-    {
-      using (var md5 = MD5.Create())
-      {
-        using (var stream = File.OpenRead(filePath))
-        {
-          var hash = md5.ComputeHash(stream);
-          return BitConverter.ToString(hash);
-        }
-      }
-    }
   }
 }

src/MagicTooltips/Providers/M365Provider.cs

@@ -27,7 +27,7 @@ public string GetValue()
         {
           m365ConnectionInfoFilePath = Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.UserProfile), ".cli-m365-tokens.json");
         }
-        currentFileHash = CalculateMd5(m365ConnectionInfoFilePath);
+        currentFileHash = Md5Utility.CalculateMd5(m365ConnectionInfoFilePath);
       }
       catch (Exception ex)
       {
@@ -55,16 +55,5 @@ public string GetValue()
       return m365Account.Trim('"');
     }
 
-    private static string CalculateMd5(string filePath)
-    {
-      using (var md5 = MD5.Create())
-      {
-        using (var stream = File.OpenRead(filePath))
-        {
-          var hash = md5.ComputeHash(stream);
-          return BitConverter.ToString(hash);
-        }
-      }
-    }
   }
 }

src/MagicTooltips/Providers/Md5Utility.cs

@@ -0,0 +1,32 @@
+using System;
+using System.Collections.Generic;
+using System.IO;
+using System.Security.Cryptography;
+using System.Text;
+
+namespace MagicTooltips.Providers
+{
+  static class Md5Utility
+  {
+    internal static string CalculateMd5(string filePath)
+    {
+      if (File.Exists(filePath))
+      {
+
+      using (var md5 = MD5.Create())
+      {
+        using (var stream = File.OpenRead(filePath))
+        {
+          var hash = md5.ComputeHash(stream);
+          return BitConverter.ToString(hash);
+        }
+      }
+      }
+      else
+      {
+        return null;
+      }
+    }
+
+  }
+}

src/MagicTooltips/Providers/MicrosoftGraphCLIProvider.cs

@@ -0,0 +1,93 @@
+using MagicTooltips.Services;
+using System;
+using System.Collections.Generic;
+using System.IO;
+using System.Security.Cryptography;
+using System.Text;
+using System.Text.Json;
+
+namespace MagicTooltips.Providers
+{
+  public class MicrosoftGraphCLIProvider : IProvider
+  {
+    public string ProviderKey => "mgcli";
+    public string DefaultCommands => "mg";  // macOS will get a new name...
+    public string DefaultNounPrefixes => "";
+    public string DefaultFgColor => "#32A5E6";
+    public string DefaultBgColor => "";
+    public string DefaultTemplate => "\uf871 {value}";
+
+    private static string fileHash = null;
+    private static string mgAccount = null;
+    private static string mgRecordFilePath = null;
+
+    public string GetValue()
+    {
+      string currentFileHash = null;
+      try
+      {
+        if (string.IsNullOrWhiteSpace(mgRecordFilePath))
+        {
+          var folderPath = Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.UserProfile), ".mg");
+          mgRecordFilePath = Path.Combine(folderPath, "record.txt");
+          LoggingService.LogDebug($"mgRecordFilePath: {mgRecordFilePath}");
+        }
+
+        if (!string.IsNullOrEmpty(mgRecordFilePath))
+        {
+          currentFileHash = Md5Utility.CalculateMd5(mgRecordFilePath) ?? "null";
+        }
+        else
+        {
+          currentFileHash = "null";
+        }
+      }
+      catch (Exception ex)
+      {
+        LoggingService.LogDebug(ex.ToString());
+      }
+
+      if (currentFileHash == "null")
+      {
+        LoggingService.LogDebug("No mgRecordFile found");
+        mgAccount = null;
+      }
+      if (currentFileHash != fileHash)
+      {
+        LoggingService.LogDebug("mgRecordFile has changed, clearing cache");
+        fileHash = currentFileHash;
+        mgAccount = null;
+      }
+
+      /*
+       *  For now, there is no command to get the current user
+       * 
+       *  So I'm going to read & parse the record.txt file
+       */
+      if (string.IsNullOrWhiteSpace(mgAccount))
+      {
+        //var script = "(Get-MgContext).Account";
+        //mgAccount = PowershellInvoker.InvokeScript(script);
+
+        if (File.Exists(mgRecordFilePath))
+        {
+          var content = File.ReadAllText(mgRecordFilePath);
+          using (var contentDoc = JsonDocument.Parse(content))
+          {
+            mgAccount = contentDoc.RootElement.GetProperty("username").GetString();
+          }
+        }
+      }
+
+      if (string.IsNullOrEmpty(mgAccount))
+      {
+        return "Not connected";
+      }
+      else
+      {
+        return mgAccount.Trim('"');
+      }
+    }
+
+  }
+}

src/MagicTooltips/Providers/MicrosoftGraphPowerShellProvider.cs

@@ -35,7 +35,7 @@ public string GetValue()
 
         if (!string.IsNullOrEmpty(mgCacheFilePath))
         {
-          currentFileHash = CalculateMd5(mgCacheFilePath);
+          currentFileHash = Md5Utility.CalculateMd5(mgCacheFilePath);
         }
         else
         {
@@ -74,17 +74,5 @@ public string GetValue()
         return mgAccount.Trim('"');
       }
     }
-
-    private static string CalculateMd5(string filePath)
-    {
-      using (var md5 = MD5.Create())
-      {
-        using (var stream = File.OpenRead(filePath))
-        {
-          var hash = md5.ComputeHash(stream);
-          return BitConverter.ToString(hash);
-        }
-      }
-    }
   }
 }

src/MagicTooltips/Services/ProviderFactory.cs

@@ -30,7 +30,8 @@ public static Dictionary<string, IProvider> GetAllProviders()
           { ProviderKeys.Azure, new AzureProvider() },
           { ProviderKeys.Aws, new AwsProvider() },
           { ProviderKeys.M365, new M365Provider() },
-          { ProviderKeys.MicrosoftGraph, new MicrosoftGraphPowerShellProvider() }
+          { ProviderKeys.MicrosoftGraph, new MicrosoftGraphPowerShellProvider() },
+          { ProviderKeys.MGCLI, new MicrosoftGraphCLIProvider() }
         };
       }
       return AllProviders;