123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277
import { border, IBorder } from "./border.ts";import { Cell, Direction } from "./cell.ts";import { TableLayout } from "./layout.ts";import { IDataRow, IRow, Row } from "./row.ts";
/** Border characters settings. */export type IBorderOptions = Partial<IBorder>;
/** Table options. */export interface ITableOptions {  indent?: number;  border?: boolean;  align?: Direction;  maxColWidth?: number | number[];  minColWidth?: number | number[];  padding?: number | number[];  chars?: IBorderOptions;}
/** Table settings. */export interface ITableSettings extends Required<Omit<ITableOptions, "align">> {  chars: IBorder;  align?: Direction;}
/** Table type. */export type ITable<T extends IRow = IRow> = T[] | Table<T>;
/** Table representation. */export class Table<T extends IRow = IRow> extends Array<T> {  protected static _chars: IBorder = { ...border };  protected options: ITableSettings = {    indent: 0,    border: false,    maxColWidth: Infinity,    minColWidth: 0,    padding: 1,    chars: { ...Table._chars },  };  private headerRow?: Row;
  /**   * Create a new table. If rows is a table, all rows and options of the table   * will be copied to the new table.   * @param rows   */  public static from<T extends IRow>(rows: ITable<T>): Table<T> {    const table = new this(...rows);    if (rows instanceof Table) {      table.options = { ...rows.options };      table.headerRow = rows.headerRow ? Row.from(rows.headerRow) : undefined;    }    return table;  }
  /**   * Create a new table from an array of json objects. An object represents a   * row and each property a column.   * @param rows Array of objects.   */  public static fromJson(rows: IDataRow[]): Table {    return new this().fromJson(rows);  }
  /**   * Set global default border characters.   * @param chars Border options.   */  public static chars(chars: IBorderOptions): typeof Table {    Object.assign(this._chars, chars);    return this;  }
  /**   * Write table or rows to stdout.   * @param rows Table or rows.   */  public static render<T extends IRow>(rows: ITable<T>): void {    Table.from(rows).render();  }
  /**   * Read data from an array of json objects. An object represents a   * row and each property a column.   * @param rows Array of objects.   */  public fromJson(rows: IDataRow[]): this {    this.header(Object.keys(rows[0]));    this.body(rows.map((row) => Object.values(row) as T));    return this;  }
  /**   * Set table header.   * @param header Header row or cells.   */  public header(header: IRow): this {    this.headerRow = header instanceof Row ? header : Row.from(header);    return this;  }
  /**   * Set table body.   * @param rows Table rows.   */  public body(rows: T[]): this {    this.length = 0;    this.push(...rows);    return this;  }
  /** Clone table recursively with header and options. */  public clone(): Table {    const table = new Table(      ...this.map((row: T) =>        row instanceof Row ? row.clone() : Row.from(row).clone()      ),    );    table.options = { ...this.options };    table.headerRow = this.headerRow?.clone();    return table;  }
  /** Generate table string. */  public toString(): string {    return new TableLayout(this, this.options).toString();  }
  /** Write table to stdout. */  public render(): this {    console.log(this.toString());    return this;  }
  /**   * Set max col with.   * @param width     Max col width.   * @param override  Override existing value.   */  public maxColWidth(width: number | number[], override = true): this {    if (override || typeof this.options.maxColWidth === "undefined") {      this.options.maxColWidth = width;    }    return this;  }
  /**   * Set min col width.   * @param width     Min col width.   * @param override  Override existing value.   */  public minColWidth(width: number | number[], override = true): this {    if (override || typeof this.options.minColWidth === "undefined") {      this.options.minColWidth = width;    }    return this;  }
  /**   * Set table indentation.   * @param width     Indent width.   * @param override  Override existing value.   */  public indent(width: number, override = true): this {    if (override || typeof this.options.indent === "undefined") {      this.options.indent = width;    }    return this;  }
  /**   * Set cell padding.   * @param padding   Cell padding.   * @param override  Override existing value.   */  public padding(padding: number | number[], override = true): this {    if (override || typeof this.options.padding === "undefined") {      this.options.padding = padding;    }    return this;  }
  /**   * Enable/disable cell border.   * @param enable    Enable/disable cell border.   * @param override  Override existing value.   */  public border(enable: boolean, override = true): this {    if (override || typeof this.options.border === "undefined") {      this.options.border = enable;    }    return this;  }
  /**   * Align table content.   * @param direction Align direction.   * @param override  Override existing value.   */  public align(direction: Direction, override = true): this {    if (override || typeof this.options.align === "undefined") {      this.options.align = direction;    }    return this;  }
  /**   * Set border characters.   * @param chars Border options.   */  public chars(chars: IBorderOptions): this {    Object.assign(this.options.chars, chars);    return this;  }
  /** Get table header. */  public getHeader(): Row | undefined {    return this.headerRow;  }
  /** Get table body. */  public getBody(): T[] {    return [...this];  }
  /** Get mac col widrth. */  public getMaxColWidth(): number | number[] {    return this.options.maxColWidth;  }
  /** Get min col width. */  public getMinColWidth(): number | number[] {    return this.options.minColWidth;  }
  /** Get table indentation. */  public getIndent(): number {    return this.options.indent;  }
  /** Get cell padding. */  public getPadding(): number | number[] {    return this.options.padding;  }
  /** Check if table has border. */  public getBorder(): boolean {    return this.options.border === true;  }
  /** Check if header row has border. */  public hasHeaderBorder(): boolean {    const hasBorder = this.headerRow?.hasBorder();    return hasBorder === true || (this.getBorder() && hasBorder !== false);  }
  /** Check if table bordy has border. */  public hasBodyBorder(): boolean {    return this.getBorder() ||      this.some((row) =>        row instanceof Row          ? row.hasBorder()          : row.some((cell) => cell instanceof Cell ? cell.getBorder : false)      );  }
  /** Check if table header or body has border. */  public hasBorder(): boolean {    return this.hasHeaderBorder() || this.hasBodyBorder();  }
  /** Get table alignment. */  public getAlign(): Direction {    return this.options.align ?? "left";  }}