Support windows with Invoke-WebRequest + HttpIt helper function

Author: DanBradbury

plugin/copilot_chat.vim

@@ -3,6 +3,14 @@ scriptencoding utf-8
 let s:plugin_dir = expand('<sfile>:p:h:h')
 let s:device_token_file = s:plugin_dir . "/.device_token"
 let s:chat_token_file = s:plugin_dir . "/.chat_token"
+let s:token_headers = [
+  \ 'Accept: application/json',
+  \ 'User-Agent: GithubCopilot/1.155.0',
+  \ 'Accept-Encoding: gzip,deflate,br',
+  \ 'Editor-Plugin-Version: copilot.vim/1.16.0',
+  \ 'Editor-Version: Neovim/0.6.1',
+  \ 'Content-Type: application/json',
+  \ ]
 
 function! CopilotChat()
   " Open a new split window for the chat
@@ -31,78 +39,93 @@ function! SubmitChatMessage()
   call AsyncRequest(l:message)
 endfunction
 
-function! GetBearerToken()
-  let l:token_url = 'https://github.com/login/device/code'
-
-  let l:token_headers = [
-        \ 'Accept: application/json',
-        \ 'User-Agent: GithubCopilot/1.155.0',
-        \ 'Accept-Encoding: gzip,deflate,br',
-        \ 'Editor-Plugin-Version: copilot.vim/1.16.0',
-        \ 'Editor-Version: Neovim/0.6.1',
-        \ 'Content-Type: application/json',
-        \ ]
-  let l:token_data = json_encode({
-        \ 'client_id': 'Iv1.b507a08c87ecfe98',
-        \ 'scope': 'read:user'
-        \ })
-
-  " Construct the curl command for token setup
-  let l:curl_cmd = 'curl -s -X POST --compressed '
-  for header in l:token_headers
-    let l:curl_cmd .= '-H "' . header . '" '
-  endfor
-  let l:curl_cmd .= "-d '" . l:token_data . "' " . l:token_url
-
-  " Execute the curl command
-  let l:response = system(l:curl_cmd)
-
-  " Check for errors in the response
-  if v:shell_error != 0
-    echom 'Error: ' . v:shell_error
-    return ''
+function HttpIt(method, url, headers, body)
+  if has("win32")
+    let l:ps_cmd = 'powershell -Command "'
+    let l:ps_cmd .= '$headers = @{'
+    for header in a:headers
+      let [key, value] = split(header, ": ")
+      let l:ps_cmd .= "'" . key . "'='" . value . "';"
+    endfor
+    let l:ps_cmd .= "};"
+    if a:method != "GET"
+      let l:ps_cmd .= '$body = ConvertTo-Json @{'
+      for obj in keys(a:body)
+        let l:ps_cmd .= obj . "='" . a:body[obj] . "';"
+      endfor
+      let l:ps_cmd .= "};"
+    endif
+    let l:ps_cmd .= "Invoke-WebRequest -Uri '" . a:url . "' -Method " .a:method . " -Headers $headers -Body $body -ContentType 'application/json' | Select-Object -ExpandProperty Content"
+    let l:ps_cmd .= '"'
+    let l:response = system(l:ps_cmd)
+  else
+    let l:token_headers = [
+      \ 'Accept: application/json',
+      \ 'User-Agent: GithubCopilot/1.155.0',
+      \ 'Accept-Encoding: gzip,deflate,br',
+      \ 'Editor-Plugin-Version: copilot.vim/1.16.0',
+      \ 'Editor-Version: Neovim/0.6.1',
+      \ 'Content-Type: application/json',
+      \ ]
+    let l:token_data = json_encode(a:body)
+
+    " Construct the curl command for token setup
+    let l:curl_cmd = 'curl -s -X POST --compressed '
+    for header in a:headers
+      let l:curl_cmd .= '-H "' . header . '" '
+    endfor
+    let l:curl_cmd .= "-d '" . l:token_data . "' " . l:token_url
+
+    let l:response = system(l:curl_cmd)
+
+    " Check for errors in the response
+    if v:shell_error != 0
+      echom 'Error: ' . v:shell_error
+      return ''
+    endif
   endif
+  return l:response
+endfunction
+
+function! GetDeviceToken()
+  let l:token_url = 'https://github.com/login/device/code'
+  let l:headers = [
+    \ 'Accept: application/json',
+    \ 'User-Agent: GithubCopilot/1.155.0',
+    \ 'Accept-Encoding: gzip,deflate,br',
+    \ 'Editor-Plugin-Version: copilot.vim/1.16.0',
+    \ 'Editor-Version: Neovim/0.6.1',
+    \ 'Content-Type: application/json',
+    \ ]
+  let l:data = {
+    \ 'client_id': 'Iv1.b507a08c87ecfe98',
+    \ 'scope': 'read:user'
+    \ }
+
+  return HttpIt("POST", l:token_url, l:headers, l:data)
+endfunction
 
-  " Parse the response to get the device code and user code
+function! GetBearerToken()
+  let l:response = GetDeviceToken()
   let l:json_response = json_decode(l:response)
   let l:device_code = l:json_response.device_code
   let l:user_code = l:json_response.user_code
   let l:verification_uri = l:json_response.verification_uri
 
-  " Display the user code and verification URI to the user
   echo 'Please visit ' . l:verification_uri . ' and enter the code: ' . l:user_code
   call input("Press Enter to continue...\n")
 
-  " Poll the token endpoint to get the Bearer token
   let l:token_poll_url = 'https://github.com/login/oauth/access_token'
-  let l:token_poll_data = json_encode({
-        \ 'client_id': 'Iv1.b507a08c87ecfe98',
-        \ 'device_code': l:device_code,
-        \ 'grant_type': 'urn:ietf:params:oauth:grant-type:device_code'
-        \ })
-
-  " Construct the curl command for token polling
-  let l:curl_cmd = 'curl -s -X POST --compressed '
-  for header in l:token_headers
-    let l:curl_cmd .= '-H "' . header . '" '
-  endfor
-  let l:curl_cmd .= "-d '" . l:token_poll_data . "' " . l:token_poll_url
-
-  " Execute the curl command
-  let l:response = system(l:curl_cmd)
-
-  " Check for errors in the response
-  if v:shell_error != 0
-    echom 'Error: ' . v:shell_error
-    return ''
-  endif
-
-  " Parse the response to get the Bearer token
-  let l:json_response = json_decode(l:response)
+  let l:token_poll_data = {
+    \ 'client_id': 'Iv1.b507a08c87ecfe98',
+    \ 'device_code': l:device_code,
+    \ 'grant_type': 'urn:ietf:params:oauth:grant-type:device_code'
+    \ }
+  let l:access_token_response = HttpIt("POST", l:token_poll_url, s:token_headers, l:token_poll_data)
+  let l:json_response = json_decode(l:access_token_response)
   let l:bearer_token = l:json_response.access_token
   call writefile([l:bearer_token], s:device_token_file)
 
-  " Return the Bearer token
   return l:bearer_token
 endfunction
 
@@ -113,29 +136,11 @@ function! GetChatToken(bearer_token)
         \ 'Editor-Version: vscode/1.80.1',
         \ 'Authorization: token ' . a:bearer_token,
         \ ]
-  let l:token_data = json_encode({
+  let l:token_data = {
         \ 'client_id': 'Iv1.b507a08c87ecfe98',
         \ 'scope': 'read:user'
-        \ })
-
-  " Construct the curl command for token setup
-  let l:curl_cmd = 'curl -s -X GET '
-  for header in l:token_headers
-    let l:curl_cmd .= '-H "' . header . '" '
-  endfor
-  let l:curl_cmd .= "-d '" . l:token_data . "' " . l:token_url
-
-  " Execute the curl command
-  let l:response = system(l:curl_cmd)
-  "echo l:response
-
-  " Check for errors in the response
-  if v:shell_error != 0
-    echom 'Error: ' . v:shell_error
-    return ''
-  endif
-
-  " Parse the response to get the device code and user code
+        \ }
+  let l:response = HttpIt("GET", l:token_url, l:token_headers, l:token_data)
   let l:json_response = json_decode(l:response)
   return l:json_response.token
 endfunction