authClient.ts

import { PostHogAPIClient } from "@renderer/api/posthogClient";
import { trpcClient } from "@renderer/trpc/client";
import { getCloudUrlFromRegion } from "@shared/utils/urls";
import { useMemo } from "react";
import {
  type AuthState,
  fetchAuthState,
  useAuthStateValue,
} from "./authQueries";

async function getValidAccessToken(): Promise<string> {
  const { accessToken } = await trpcClient.auth.getValidAccessToken.query();
  return accessToken;
}

async function refreshAccessToken(): Promise<string> {
  const { accessToken } = await trpcClient.auth.refreshAccessToken.mutate();
  return accessToken;
}

export function createAuthenticatedClient(
  authState: AuthState | null | undefined,
): PostHogAPIClient | null {
  if (authState?.status !== "authenticated" || !authState.cloudRegion) {
    return null;
  }

  const client = new PostHogAPIClient(
    getCloudUrlFromRegion(authState.cloudRegion),
    getValidAccessToken,
    refreshAccessToken,
    authState.projectId ?? undefined,
  );

  if (authState.projectId) {
    client.setTeamId(authState.projectId);
  }

  return client;
}

export async function getAuthenticatedClient(): Promise<PostHogAPIClient | null> {
  return createAuthenticatedClient(await fetchAuthState());
}

export function useOptionalAuthenticatedClient(): PostHogAPIClient | null {
  const authState = useAuthStateValue((state) => state);

  return useMemo(
    () => createAuthenticatedClient(authState),
    [authState.cloudRegion, authState.projectId, authState.status, authState],
  );
}

export function useAuthenticatedClient(): PostHogAPIClient {
  const client = useOptionalAuthenticatedClient();

  if (!client) {
    throw new Error("Not authenticated");
  }

  return client;
}