Edits history of script submission #22558 for ' Fetch data (supabase)'

• bun

  Ap­pro­ved version

  Edit in Windmill

  Embed

  import { createClient, SupabaseClient } from "@supabase/[email protected]"
  
  /**
   * @param token Supabase `access_token` and `refresh_token`. `expires_at` (optional) is a UNIX
   * timestamp in seconds.
   *
   * @param count Count algorithm to use to count rows in the table or view.
   * `"exact"`: Exact but slow count algorithm. Performs a `COUNT(*)` under the hood.
   * `"planned"`: Approximated but fast count algorithm. Uses the Postgres statistics under the hood.
   * `"estimated"`: Uses exact count for low numbers and planned count for high numbers.
   *
   * @param head When set to `true`, `data` will not be returned.
   * Useful if you only need the count.
   *
   * @param filter Learn more at https://supabase.com/docs/reference/javascript/filter
   *
   * @param order Learn more at https://supabase.com/docs/reference/javascript/order
   *
   * @param limit Learn more at https://supabase.com/docs/reference/javascript/limit
   */
  type Supabase = {
    url: string;
    key: string;
  };
  export async function main(
    auth: Supabase,
    table: string,
    columns?: string,
    token?: {
      access: string;
      refresh: string;
      expires_at?: number;
    },
    count?: "exact" | "planned" | "estimated",
    head?: boolean,
    filter?: {
      column: string;
      operator: string;
      value: any;
    },
    order?: {
      column: string;
      foreignTable: string;
      ascending?: boolean;
      nullsFirst?: boolean;
    },
    limit?: {
      count: number;
      foreignTable?: string;
    },
  ) {
    return await refreshAndRetryIfExpired(auth, token, async (client) => {
      const options = head || count ? { head, count } : undefined;
      let query = client.from(table).select(columns || undefined, options);
      if (filter?.column) {
        query = query.filter(filter.column, filter.operator, filter.value);
      }
      if (order?.column) {
        const { column, ...options } = order;
        query = query.order(column, options);
      }
      if (limit?.count) {
        const { count, foreignTable } = limit;
        query = query.limit(count, foreignTable ? { foreignTable } : undefined);
      }
  
      return query;
    });
  }
  
  async function refreshAndRetryIfExpired(
    auth: { url: string; key: string },
    token: { access: string; refresh: string; expires_at?: number } | undefined,
    fn: (client: SupabaseClient) => Promise<{ data: any; error?: any }>,
  ): Promise<{ data: any; error?: any; token?: { access: string; refresh: string; expires_at?: number } }> {
    const makeClient = async (autoRefreshToken: boolean) => {
      const client = createClient(auth.url, auth.key, {
        auth: { autoRefreshToken, persistSession: false },
      })
      if (token) {
        await client.auth.setSession({ access_token: token.access, refresh_token: token.refresh })
      }
      return client
    }
    try {
      let result = await fn(await makeClient(false))
      if (result?.error?.code === "PGRST301" && token) {
        const client = await makeClient(true)
        result = await fn(client)
        const { data } = await client.auth.getSession()
        if (data?.session) {
          token = {
            access: data.session.access_token,
            refresh: data.session.refresh_token,
            expires_at: data.session.expires_at,
          }
        }
      }
      return { ...result, token }
    } catch (error) {
      return { data: null, error }
    }
  }
  

  Submitted by hugo989 19 days ago