main.ts

import { createClient as _createClient, type SupabaseClient } from "@supabase/supabase-js";
import type {
  SupabaseClientOptions,
} from "@supabase/supabase-js";
import type {
  CreateFirebaseAuthClientProps,
} from "./auth.d.ts";
import type {
  Database,
} from "./types.d.ts";


/**
 * 
 * @param args `CreateFirebaseAuthClientProps` props
 * @returns A function that creates a Supabase client with Firebase Auth of the currently logged in user.
 * @see https://supabase.com/docs/guides/auth/third-party/firebase-auth
 */
export const createFirebaseAuthSupabaseClient = <D extends Database = Database>(
    ...args: CreateFirebaseAuthClientProps<D>
): <DBR extends D = D, S extends string & keyof D = "public">(schemaName?: S) => SupabaseClient<DBR, S> => {
    const [{auth, accessToken, jwtToken}, {supabaseUrl, supabaseAnonKey}] = args;
    
  const createClient = <
      DB extends D = D,
      S extends string & keyof D = "public",
  >(
      schemaName: S = "public" as S,
  ) => {
      if (!auth && !accessToken && !jwtToken) throw new Error("No auth, accessToken, or jwtToken provided");

      const options: SupabaseClientOptions<S> = { db: { schema: schemaName } };
      if (auth || accessToken) {
          options.accessToken = accessToken ??  (async () => ((await auth?.currentUser?.getIdToken(false)) ?? null) as string);
      } else if (jwtToken) {
          options.global = {
          headers: { Authorization: `Bearer ${jwtToken}` },
          };
      } else console.warn("No JWT token provided, this will probably fail.");
    
      return _createClient<DB, S>(supabaseUrl, supabaseAnonKey, options);
  };
  return createClient;
};

/**
 * Get the JWT token from the Request headers.
 * This is required to create a Supabase client with the correct permissions in Edge functions.
 * @param headers Edge function's Request headers
 * @returns The JWT token if it exists, otherwise null
 */
export const getJwtToken = (headers: Headers): string | null => headers.get("Authorization")?.replace("Bearer ", "") ?? null;