aethex-forge/client/contexts/DiscordActivityContext.tsx

import React, { createContext, useContext, useEffect, useState } from "react";

interface DiscordUser {
  id: string;
  discord_id: string;
  full_name: string | null;
  username: string | null;
  avatar_url: string | null;
  bio: string | null;
  user_type: string | null;
  primary_arm: string | null;
}

interface DiscordActivityContextType {
  isActivity: boolean;
  isLoading: boolean;
  user: DiscordUser | null;
  error: string | null;
  discordSdk: any | null;
}

const DiscordActivityContext = createContext<DiscordActivityContextType>({
  isActivity: false,
  isLoading: false,
  user: null,
  error: null,
  discordSdk: null,
});

export const useDiscordActivity = () => {
  const context = useContext(DiscordActivityContext);
  if (!context) {
    throw new Error(
      "useDiscordActivity must be used within DiscordActivityProvider",
    );
  }
  return context;
};

interface DiscordActivityProviderProps {
  children: React.ReactNode;
}

export const DiscordActivityProvider: React.FC<
  DiscordActivityProviderProps
> = ({ children }) => {
  const [isActivity, setIsActivity] = useState(false);
  const [isLoading, setIsLoading] = useState(false);
  const [user, setUser] = useState<DiscordUser | null>(null);
  const [error, setError] = useState<string | null>(null);
  const [discordSdk, setDiscordSdk] = useState<any>(null);

  useEffect(() => {
    const initializeActivity = async () => {
      // Check if we're running inside a Discord Activity
      // Discord passes frame_id as a query parameter when launching an Activity
      const searchParams = new URLSearchParams(window.location.search);
      const frameId = searchParams.get("frame_id");
      const isInDiscordActivity = frameId !== null;

      if (isInDiscordActivity) {
        try {
          setIsActivity(true);
          setIsLoading(true);

          // Import the Discord SDK dynamically
          const { DiscordSDK } = await import("@discord/embedded-app-sdk");

          const clientId =
            import.meta.env.VITE_DISCORD_CLIENT_ID || "578971245454950421";

          const sdk = new DiscordSDK({
            clientId,
          });

          setDiscordSdk(sdk);

          // Wait for SDK to be ready
          await sdk.ready();

          // Get the current user from the SDK
          const currentUser = await sdk.user.getUser();

          if (!currentUser) {
            // User not authenticated, authorize them
            const { access_token } = await sdk.commands.authorize({
              scopes: ["identify", "guilds"],
            });

            // Exchange access token for user data via our proxy
            const response = await fetch("/api/discord/activity-auth", {
              method: "POST",
              headers: {
                "Content-Type": "application/json",
              },
              body: JSON.stringify({ access_token }),
            });

            if (!response.ok) {
              const errorData = await response.json();
              setError(errorData.error || "Failed to authenticate");
              setIsLoading(false);
              return;
            }

            const data = await response.json();
            if (data.success && data.user) {
              setUser(data.user);
              setError(null);
            } else {
              setError("Authentication failed");
            }
          } else {
            // User already authenticated, just fetch via our proxy
            const response = await fetch("/api/discord/activity-auth", {
              method: "POST",
              headers: {
                "Content-Type": "application/json",
              },
              body: JSON.stringify({
                access_token: currentUser.access_token || "",
              }),
            });

            if (!response.ok) {
              const errorData = await response.json();
              setError(errorData.error || "Failed to fetch user data");
              setIsLoading(false);
              return;
            }

            const data = await response.json();
            if (data.success && data.user) {
              setUser(data.user);
              setError(null);
            } else {
              setError("Failed to load user data");
            }
          }
        } catch (err: any) {
          console.error("Discord Activity initialization error:", err);
          setError(err?.message || "Failed to initialize Discord Activity");
        } finally {
          setIsLoading(false);
        }
      } else {
        // Not in a Discord iframe
        setIsActivity(false);
        setIsLoading(false);
      }
    };

    initializeActivity();
  }, []);

  return (
    <DiscordActivityContext.Provider
      value={{
        isActivity,
        isLoading,
        user,
        error,
        discordSdk,
      }}
    >
      {children}
    </DiscordActivityContext.Provider>
  );
};

export default DiscordActivityContext;