update files

dist/main.js

@@ -252,12 +252,37 @@ if (storedOAuth) {
     oauthResult = null;
   }
 }
+async function getOrganizationInfo(accessToken) {
+  const response = await fetch("https://huggingface.co/api/whoami-v2", {
+    headers: {
+      Authorization: `Bearer ${accessToken}`
+    }
+  });
+  if (!response.ok) {
+    throw new Error(
+      `Failed to fetch organization info: ${response.statusText}`
+    );
+  }
+  return response.json();
+}
 var init = async () => {
   oauthResult ||= await oauthHandleRedirectIfPresent();
   if (oauthResult !== null && oauthResult !== false) {
-    const preElement = document.querySelector("pre");
-    if (preElement) {
-      preElement.textContent = JSON.stringify(oauthResult, null, 2);
+    try {
+      const orgInfo = await getOrganizationInfo(oauthResult.accessToken);
+      const preElement = document.querySelector("pre");
+      if (preElement) {
+        preElement.textContent = JSON.stringify(
+          {
+            oauth: oauthResult,
+            user: orgInfo
+          },
+          null,
+          2
+        );
+      }
+    } catch (error) {
+      console.error("Failed to fetch organization info:", error);
     }
     localStorage.setItem("oauth", JSON.stringify(oauthResult));
     const signoutButton = document.getElementById("signout");

src/main.ts

@@ -8,6 +8,23 @@ interface HuggingFaceWindow extends Window {
   };
 }
 
+interface OrganizationInfo {
+  type: string;
+  id: string;
+  name: string;
+  role: string;
+}
+
+interface WhoAmIResponse {
+  type: string;
+  id: string;
+  name: string;
+  email?: string;
+  fullname?: string;
+  avatarUrl?: string;
+  organizations: OrganizationInfo[];
+}
+
 declare const window: HuggingFaceWindow;
 
 console.log("huggingface env", window.huggingface);
@@ -25,13 +42,44 @@ if (storedOAuth) {
   }
 }
 
+async function getOrganizationInfo(
+  accessToken: string
+): Promise<WhoAmIResponse> {
+  const response = await fetch("https://huggingface.co/api/whoami-v2", {
+    headers: {
+      Authorization: `Bearer ${accessToken}`,
+    },
+  });
+
+  if (!response.ok) {
+    throw new Error(
+      `Failed to fetch organization info: ${response.statusText}`
+    );
+  }
+
+  return response.json();
+}
+
 const init = async (): Promise<void> => {
   oauthResult ||= await oauthHandleRedirectIfPresent();
 
   if (oauthResult !== null && oauthResult !== false) {
-    const preElement = document.querySelector("pre");
-    if (preElement) {
-      preElement.textContent = JSON.stringify(oauthResult, null, 2);
+    // Get organization info after successful authentication
+    try {
+      const orgInfo = await getOrganizationInfo(oauthResult.accessToken);
+      const preElement = document.querySelector("pre");
+      if (preElement) {
+        preElement.textContent = JSON.stringify(
+          {
+            oauth: oauthResult,
+            user: orgInfo,
+          },
+          null,
+          2
+        );
+      }
+    } catch (error) {
+      console.error("Failed to fetch organization info:", error);
     }
 
     localStorage.setItem("oauth", JSON.stringify(oauthResult));