temp: everything auth related (TODO: don't hardcode)

Author: heyzec

package.json

@@ -31,6 +31,10 @@
       {
         "command": "source-academy.eval-editor",
         "title": "Source Academy: Run Code in Editor"
+      },
+      {
+        "command": "source-academy.login",
+        "title": "Source Academy: Login"
       }
     ],
     "configuration": {

src/commands/index.tsx

@@ -2,6 +2,7 @@ import * as vscode from "vscode";
 import { runLanguagePicker } from "./language";
 import { evalEditor } from "./evalEditor";
 import { showPanel } from "./showPanel";
+import { login } from "./login";
 
 const EXTENSION_ID = "source-academy";
 
@@ -12,6 +13,7 @@ const EXTENSION_ID = "source-academy";
 const commands = (context: vscode.ExtensionContext) => ({
   pick: () => runLanguagePicker(context),
   "show-panel": () => showPanel(context),
+  login: () => login(context),
   "eval-editor": () => evalEditor(context),
 });
 

src/commands/login.ts

@@ -0,0 +1,11 @@
+import * as vscode from "vscode";
+
+export async function login(context: vscode.ExtensionContext) {
+  vscode.env.openExternal(
+    vscode.Uri.parse(
+      // "http://localhost:4000/sso/auth/signin/student?target_url=http://localhost:4000/v2/auth/saml_redirect?provider=test_saml_vscode",
+      "http://localhost:4000/sso/auth/signin/student?target_url=http://localhost:4000/v2/auth/saml_redirect_vscode?provider=test_saml",
+    ),
+  );
+  // vscode.env.openExternal(vscode.Uri.parse('http://localhost:4000/sso/auth/signin/student?target_url=http://example.com'));
+}

src/commands/showPanel.tsx

@@ -36,7 +36,23 @@ async function handleMessage(
     console.log(`${Date.now()} Beginning handleMessage: ${message.type}`);
     switch (message.type) {
       case MessageTypeNames.ExtensionPing:
-        sendToFrontend(panel, Messages.ExtensionPong(null));
+        let state = context.globalState.get("token") ?? null;
+
+        const token = state ? JSON.stringify(state) : null;
+
+        console.log(`WebviewStarted: Obtain token from VSC: ${token}`);
+
+        // if (token !== null && Object.keys(token).length === 0) {
+        //   token = null
+        // }
+        // if (token !== null) {
+        //   token = JSON.stringify(token);
+        // }
+
+        panel!.webview.postMessage(Messages.ExtensionPong(token));
+        if (token) {
+          context.globalState.update("token", undefined);
+        }
         break;
       case MessageTypeNames.NewEditor:
         activeEditor = await Editor.create(
@@ -56,8 +72,10 @@ async function handleMessage(
           }
           if (editor !== activeEditor) {
             console.log(
-              `EXTENSION: Editor ${editor.assessmentName}_${editor.questionId} is no longer active, skipping onChange`,
+              `EXTENSION: Editor is no longer active, skipping onChange`,
+              `User edited ${editor.assessmentName}_${editor.questionId}, active is ${activeEditor?.assessmentName}_${activeEditor?.questionId}`,
             );
+            return;
           }
           const message = Messages.Text(workspaceLocation, code);
           console.log(`Sending message: ${JSON.stringify(message)}`);
@@ -78,7 +96,8 @@ async function handleMessage(
   handling = false;
 }
 
-export async function showPanel(context: vscode.ExtensionContext) {
+export async function showPanel(context: vscode.ExtensionContext, url: string) {
+  vscode.window.showInformationMessage(`showPanel caleld with ${url}`);
   let language: string | undefined = context.workspaceState.get("language");
   if (!language) {
     language = LANGUAGES.SOURCE_1;
@@ -117,7 +136,8 @@ export async function showPanel(context: vscode.ExtensionContext) {
     >
       <iframe
         id={FRONTEND_ELEMENT_ID}
-        src={frontendUrl}
+        src={url || frontendUrl}
+        // src={"vscode://source-academy.source-academy/sso"}
         width="100%"
         height="100%"
         // @ts-ignore

src/extension.ts

@@ -2,9 +2,9 @@
 // Import the module and reference it with the alias vscode in your code below
 import * as vscode from "vscode";
 import { setupStatusBar } from "./statusbar/status";
-import { evalEditor } from "./commands/evalEditor";
 import { registerAllCommands } from "./commands";
 import { activateLspClient, deactivateLspClient } from "./lsp/client";
+import { showPanel } from "./commands/showPanel";
 
 // This method is called when your extension is activated
 // Your extension is activated the very first time the command is executed
@@ -14,6 +14,22 @@ export function activate(context: vscode.ExtensionContext) {
   context.subscriptions.push(setupStatusBar(context));
 
   activateLspClient(context);
+  vscode.window.registerUriHandler({
+    handleUri(uri: vscode.Uri) {
+      const searchParams = new URLSearchParams(uri.query);
+      const code = searchParams.get("code");
+      vscode.window.showInformationMessage(`Code: ${code}`);
+      const provider = searchParams.get("provider");
+      vscode.window.showInformationMessage(`Provider: ${provider}`);
+
+      // context.globalState.update("token", {
+      //   accessToken: code,
+      //   refreshToken: provider,
+      // });
+      const url = `http://localhost:4000/v2/auth/exchange/?code=${code}&provider=${provider}`;
+      showPanel(context, url);
+    },
+  });
 }
 
 // This method is called when your extension is deactivated