Simple.ts

import { AsyncDuckDB, AsyncDuckDBConnection } from "@duckdb/duckdb-wasm"
import { Connection, Database } from "duckdb"

export default class Simple {
    /** A flag indicating whether debugging information should be logged. Defaults to false. @category Properties */
    debug: boolean
    /** The number of rows to log. Defaults to 10. @category Properties */
    nbRowsToLog: number
    /** The number of characters to log for text cells. By default, the whole text is logged. @category Properties */
    nbCharactersToLog: number | undefined
    /** A DuckDB database. @category Properties */
    db!: AsyncDuckDB | Database
    /** A connection to a DuckDB database. @category Properties */
    connection!: AsyncDuckDBConnection | Connection
    /** A worker to make DuckDB WASM work. @category Properties */
    worker!: Worker | null
    /** A flag for SimpleDB. Default is true. When data is retrieved from the database as an array of objects, BIGINT values are automatically converted to integers, which are easier to work with in JavaScript. If you want actual bigint values, set this option to false. @category Properties */
    bigIntToInt: boolean | undefined
    /** A number used when creating new tables. @category Properties */
    tableIncrement: number
    /** A flag to know if the name of the table has been attributed by default. @category Properties */
    defaultTableName: boolean
    /**
     * For internal use only. If you want to run a SQL query, use the customQuery method. @category Properties
     */
    runQuery!: (
        query: string,
        connection: AsyncDuckDBConnection | Connection,
        returnDataFromQuery: boolean,
        options: {
            debug: boolean
            method: string | null
            parameters: { [key: string]: unknown } | null
            bigIntToInt?: boolean
        }
    ) => Promise<
        | {
              [key: string]: number | string | Date | boolean | null
          }[]
        | null
    >

    constructor(
        runQuery: (
            query: string,
            connection: AsyncDuckDBConnection | Connection,
            returnDataFromQuery: boolean,
            options: {
                debug: boolean
                method: string | null
                parameters: { [key: string]: unknown } | null
                bigIntToInt?: boolean
            }
        ) => Promise<
            | {
                  [key: string]: number | string | Date | boolean | null
              }[]
            | null
        >,
        options: {
            debug?: boolean
            nbRowsToLog?: number
            nbCharactersToLog?: number
        } = {}
    ) {
        this.nbRowsToLog = options.nbRowsToLog ?? 10
        this.nbCharactersToLog = options.nbCharactersToLog
        this.debug = options.debug ?? false
        this.worker = null
        this.tableIncrement = 1
        this.defaultTableName = false
        this.runQuery = runQuery
    }
}