feat: changes for OAuth flow

Author: heyzec

src/extension.ts

@@ -51,6 +51,20 @@ export function activate(context: vscode.ExtensionContext) {
       "*.js": "source",
     });
   }
+
+  vscode.window.registerUriHandler({
+    handleUri(uri: vscode.Uri) {
+      const searchParams = new URLSearchParams(uri.query);
+      // The following two params are available when logging in via OAuth providers
+      const code = searchParams.get("code");
+      const clientRequestId = searchParams.get("client-request-id");
+
+      vscode.commands.executeCommand(
+        "source-academy.show-panel",
+        `/login/vscode_callback?code=${code}&client-request-id=${clientRequestId}`,
+      );
+    },
+  });
 }
 
 // This method is called when your extension is deactivated

src/utils/messageHandler.tsx

@@ -173,6 +173,9 @@ export class MessageHandler {
             this.panel?.reveal(vscode.ViewColumn.Two);
           }
           break;
+      case MessageTypeNames.LoginWithBrowser:
+        const { route } = message;
+        vscode.env.openExternal(vscode.Uri.parse(route));
       }
       console.log(`${Date.now()} Finish handleMessage: ${message.type}`);
     }

src/utils/messages.ts

@@ -89,6 +89,9 @@ const Messages = createMessages({
     questionId,
     choice,
   }),
+  LoginWithBrowser: (route: string) => ({
+    route,
+  }),
 });
 
 export default Messages;