123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211
import { PostgrestBuilder } from './types.ts'import { PostgrestFilterBuilder } from './PostgrestFilterBuilder.ts'
export class PostgrestQueryBuilder<T> extends PostgrestBuilder<T> {  constructor(    url: string,    { headers = {}, schema }: { headers?: { [key: string]: string }; schema?: string } = {}  ) {    super({} as PostgrestBuilder<T>)    this.url = new URL(url)    this.headers = { ...headers }    this.schema = schema  }
  /**   * Performs vertical filtering with SELECT.   *   * @param columns  The columns to retrieve, separated by commas.   * @param head  When set to true, select will void data.   * @param count  Count algorithm to use to count rows in a table.   */  select(    columns = '*',    {      head = false,      count = null,    }: {      head?: boolean      count?: null | 'exact' | 'planned' | 'estimated'    } = {}  ): PostgrestFilterBuilder<T> {    this.method = 'GET'    // Remove whitespaces except when quoted    let quoted = false    const cleanedColumns = columns      .split('')      .map((c) => {        if (/\s/.test(c) && !quoted) {          return ''        }        if (c === '"') {          quoted = !quoted        }        return c      })      .join('')    this.url.searchParams.set('select', cleanedColumns)    if (count) {      this.headers['Prefer'] = `count=${count}`    }    if (head) {      this.method = 'HEAD'    }    return new PostgrestFilterBuilder(this)  }
  /**   * Performs an INSERT into the table.   *   * @param values  The values to insert.   * @param returning  By default the new record is returned. Set this to 'minimal' if you don't need this value.   * @param count  Count algorithm to use to count rows in a table.   */  insert(    values: Partial<T> | Partial<T>[],    options?: {      returning?: 'minimal' | 'representation'      count?: null | 'exact' | 'planned' | 'estimated'    }  ): PostgrestFilterBuilder<T>  /**   * @deprecated Use `upsert()` instead.   */  insert(    values: Partial<T> | Partial<T>[],    options?: {      upsert?: boolean      onConflict?: string      returning?: 'minimal' | 'representation'      count?: null | 'exact' | 'planned' | 'estimated'    }  ): PostgrestFilterBuilder<T>  insert(    values: Partial<T> | Partial<T>[],    {      upsert = false,      onConflict,      returning = 'representation',      count = null,    }: {      upsert?: boolean      onConflict?: string      returning?: 'minimal' | 'representation'      count?: null | 'exact' | 'planned' | 'estimated'    } = {}  ): PostgrestFilterBuilder<T> {    this.method = 'POST'
    const prefersHeaders = [`return=${returning}`]    if (upsert) prefersHeaders.push('resolution=merge-duplicates')
    if (upsert && onConflict !== undefined) this.url.searchParams.set('on_conflict', onConflict)    this.body = values    if (count) {      prefersHeaders.push(`count=${count}`)    }
    this.headers['Prefer'] = prefersHeaders.join(',')
    if (Array.isArray(values)) {      const columns = values.reduce((acc, x) => acc.concat(Object.keys(x)), [] as string[])      if (columns.length > 0) {        const uniqueColumns = [...new Set(columns)].map((column) => `"${column}"`)        this.url.searchParams.set('columns', uniqueColumns.join(','))      }    }
    return new PostgrestFilterBuilder(this)  }
  /**   * Performs an UPSERT into the table.   *   * @param values  The values to insert.   * @param onConflict  By specifying the `on_conflict` query parameter, you can make UPSERT work on a column(s) that has a UNIQUE constraint.   * @param returning  By default the new record is returned. Set this to 'minimal' if you don't need this value.   * @param count  Count algorithm to use to count rows in a table.   * @param ignoreDuplicates  Specifies if duplicate rows should be ignored and not inserted.   */  upsert(    values: Partial<T> | Partial<T>[],    {      onConflict,      returning = 'representation',      count = null,      ignoreDuplicates = false,    }: {      onConflict?: string      returning?: 'minimal' | 'representation'      count?: null | 'exact' | 'planned' | 'estimated'      ignoreDuplicates?: boolean    } = {}  ): PostgrestFilterBuilder<T> {    this.method = 'POST'
    const prefersHeaders = [      `resolution=${ignoreDuplicates ? 'ignore' : 'merge'}-duplicates`,      `return=${returning}`,    ]
    if (onConflict !== undefined) this.url.searchParams.set('on_conflict', onConflict)    this.body = values    if (count) {      prefersHeaders.push(`count=${count}`)    }
    this.headers['Prefer'] = prefersHeaders.join(',')
    return new PostgrestFilterBuilder(this)  }
  /**   * Performs an UPDATE on the table.   *   * @param values  The values to update.   * @param returning  By default the updated record is returned. Set this to 'minimal' if you don't need this value.   * @param count  Count algorithm to use to count rows in a table.   */  update(    values: Partial<T>,    {      returning = 'representation',      count = null,    }: {      returning?: 'minimal' | 'representation'      count?: null | 'exact' | 'planned' | 'estimated'    } = {}  ): PostgrestFilterBuilder<T> {    this.method = 'PATCH'    const prefersHeaders = [`return=${returning}`]    this.body = values    if (count) {      prefersHeaders.push(`count=${count}`)    }    this.headers['Prefer'] = prefersHeaders.join(',')    return new PostgrestFilterBuilder(this)  }
  /**   * Performs a DELETE on the table.   *   * @param returning  If `true`, return the deleted row(s) in the response.   * @param count  Count algorithm to use to count rows in a table.   */  delete({    returning = 'representation',    count = null,  }: {    returning?: 'minimal' | 'representation'    count?: null | 'exact' | 'planned' | 'estimated'  } = {}): PostgrestFilterBuilder<T> {    this.method = 'DELETE'    const prefersHeaders = [`return=${returning}`]    if (count) {      prefersHeaders.push(`count=${count}`)    }    this.headers['Prefer'] = prefersHeaders.join(',')    return new PostgrestFilterBuilder(this)  }}