Fixing logic on API function to use correct syntax for type of token

PSModule · Sep 19, 2023 · da9114d · da9114d
1 parent cb7a3c0
commit da9114d
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/src/GitHub/public/Core/Invoke-GitHubAPI.ps1 b/src/GitHub/public/Core/Invoke-GitHubAPI.ps1
@@ -27,12 +27,23 @@
         [switch] $UseWebRequest
     )
 
+    if ([string]::IsNullOrEmpty($Token)) {
+        throw 'Token is required'
+    }
+
+    if ($Token.StartsWith('ghp_')) {
+        $authorization = "token $Token"
+    }
+    if ($Token.StartsWith('ghu_')) {
+        $authorization = "Bearer $Token"
+    }
+
     $APICall = @{
         Uri     = "$script:APIBaseURI$($APIEndpoint.Replace('\', '/').Replace('//', '/'))"
         Method  = $Method
         Body    = [string]::IsNullOrEmpty($Data) ? ($Body | ConvertTo-Json -Depth 100) : $Data
         Headers = @{
-            Authorization          = "token $Token"
+            Authorization          = $authorization
             'Content-Type'         = $ContentType
             'X-GitHub-Api-Version' = $Version
         }