123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122
import type { RowDescription } from "./connection.ts";import type { Connection } from "./connection.ts";import { encode, EncodedArg } from "./encode.ts";
import { decode } from "./decode.ts";
const commandTagRegexp = /^([A-Za-z]+)(?: (\d+))?(?: (\d+))?/;
type CommandType = (  | "INSERT"  | "DELETE"  | "UPDATE"  | "SELECT"  | "MOVE"  | "FETCH"  | "COPY");
export interface QueryConfig {  text: string;  args?: Array<unknown>;  name?: string;  encoder?: (arg: unknown) => EncodedArg;}
export class QueryResult {  public rowDescription!: RowDescription;  private _done = false;  // deno-lint-ignore no-explicit-any  public rows: any[] = []; // actual results  public rowCount?: number;  public command!: CommandType;
  constructor(public query: Query) {}
  handleRowDescription(description: RowDescription) {    this.rowDescription = description;  }
  // deno-lint-ignore no-explicit-any  private _parseDataRow(dataRow: any[]): any[] {    const parsedRow = [];
    for (let i = 0, len = dataRow.length; i < len; i++) {      const column = this.rowDescription.columns[i];      const rawValue = dataRow[i];
      if (rawValue === null) {        parsedRow.push(null);      } else {        parsedRow.push(decode(rawValue, column));      }    }
    return parsedRow;  }
  // deno-lint-ignore no-explicit-any  handleDataRow(dataRow: any[]): void {    if (this._done) {      throw new Error("New data row, after result if done.");    }
    const parsedRow = this._parseDataRow(dataRow);    this.rows.push(parsedRow);  }
  handleCommandComplete(commandTag: string): void {    const match = commandTagRegexp.exec(commandTag);    if (match) {      this.command = match[1] as CommandType;      if (match[3]) {        // COMMAND OID ROWS        this.rowCount = parseInt(match[3], 10);      } else {        // COMMAND ROWS        this.rowCount = parseInt(match[2], 10);      }    }  }
  rowsOfObjects() {    return this.rows.map((row) => {      // deno-lint-ignore no-explicit-any      const rv: { [key: string]: any } = {};      this.rowDescription.columns.forEach((column, index) => {        rv[column.name] = row[index];      });
      return rv;    });  }
  done() {    this._done = true;  }}
export class Query {  public text: string;  public args: EncodedArg[];  public result: QueryResult;
  // TODO: can we use more specific type for args?  constructor(text: string | QueryConfig, ...args: unknown[]) {    let config: QueryConfig;    if (typeof text === "string") {      config = { text, args };    } else {      config = text;    }    this.text = config.text;    this.args = this._prepareArgs(config);    this.result = new QueryResult(this);  }
  private _prepareArgs(config: QueryConfig): EncodedArg[] {    const encodingFn = config.encoder ? config.encoder : encode;    return (config.args || []).map(encodingFn);  }}