diff --git a/scripts/Get-CopilotToken.ps1 b/scripts/Get-CopilotToken.ps1 index f1954f9..dea35a4 100644 --- a/scripts/Get-CopilotToken.ps1 +++ b/scripts/Get-CopilotToken.ps1 @@ -1,6 +1,12 @@ +param( + [string]$ClientId = "Iv1.b507a08c87ecfe98" # GitHub Copilot Plugin by GitHub +) + function getToken { + param( + [string]$ClientId = "Iv1.b507a08c87ecfe98" + ) $Timeout = 10 - $ClientId = "Iv1.b507a08c87ecfe98" # GitHub Copilot Plugin by GitHub # get login info $url = "https://github.com/login/device/code" $body = @{ @@ -37,7 +43,7 @@ function getToken { } try { - $token = getToken + $token = getToken -ClientId $ClientId } catch { Write-Host $_.Exception.GetType().Name $_.Exception.Message diff --git a/scripts/get_copilot_token.py b/scripts/get_copilot_token.py index 071e0ea..1196706 100644 --- a/scripts/get_copilot_token.py +++ b/scripts/get_copilot_token.py @@ -1,12 +1,13 @@ import http.client import json +import sys import time TIMEOUT = 10 -CLIENT_ID = "Iv1.b507a08c87ecfe98" # GitHub Copilot Plugin by GitHub +DEFAULT_CLIENT_ID = "Iv1.b507a08c87ecfe98" # GitHub Copilot Plugin by GitHub -def get_token(): +def get_token(client_id=DEFAULT_CLIENT_ID): # get login info conn = http.client.HTTPSConnection("github.com", timeout=TIMEOUT) headers = { @@ -14,7 +15,7 @@ def get_token(): "content-type": "application/json" } body = { - "client_id": CLIENT_ID, + "client_id": client_id, "scope": "read:user" } conn.request("POST", "/login/device/code", json.dumps(body), headers) @@ -26,7 +27,7 @@ def get_token(): print(f"Please open {login_info['verification_uri']} in browser and enter {login_info['user_code']} to login.") body = { - "client_id": CLIENT_ID, + "client_id": client_id, "device_code": login_info['device_code'], "grant_type": "urn:ietf:params:oauth:grant-type:device_code" } diff --git a/scripts/get_copilot_token.sh b/scripts/get_copilot_token.sh index f4d84ce..2e1a2f6 100644 --- a/scripts/get_copilot_token.sh +++ b/scripts/get_copilot_token.sh @@ -1,6 +1,7 @@ #!/bin/bash -CLIENT_ID='Iv1.b507a08c87ecfe98' # GitHub Copilot Plugin by GitHub +DEFAULT_CLIENT_ID='Iv1.b507a08c87ecfe98' # GitHub Copilot Plugin by GitHub +CLIENT_ID="${1:-$DEFAULT_CLIENT_ID}" if command -v jq >/dev/null 2>&1; then json_value() { #https://jqlang.github.io/jq/