123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378
import type { KeyCode } from "../keycode/key_code.ts";import {  GenericInput,  GenericInputKeys,  GenericInputPromptOptions,  GenericInputPromptSettings,} from "./_generic_input.ts";import { blue, bold, dim, stripColor, yellow } from "./deps.ts";import { Figures } from "./figures.ts";import { distance } from "../_utils/distance.ts";
/** Select key options. */export interface GenericListKeys extends GenericInputKeys {  previous?: string[];  next?: string[];  previousPage?: string[];  nextPage?: string[];}
/** Generic list option options. */export interface GenericListOption {  value: string;  name?: string;  disabled?: boolean;}
/** Generic list option settings. */export interface GenericListOptionSettings extends GenericListOption {  name: string;  value: string;  disabled: boolean;}
export type GenericListValueOptions = (string | GenericListOption)[];export type GenericListValueSettings = GenericListOptionSettings[];
type UnsupportedInputOptions = "suggestions" | "list";
/** Generic list prompt options. */export interface GenericListOptions<T, V>  extends Omit<GenericInputPromptOptions<T, V>, UnsupportedInputOptions> {  options: GenericListValueOptions;  keys?: GenericListKeys;  indent?: string;  listPointer?: string;  searchIcon?: string;  maxRows?: number;  searchLabel?: string;  search?: boolean;  info?: boolean;}
/** Generic list prompt settings. */export interface GenericListSettings<T, V>  extends GenericInputPromptSettings<T, V> {  options: GenericListValueSettings;  keys?: GenericListKeys;  indent: string;  listPointer: string;  maxRows: number;  searchLabel: string;  search?: boolean;  info?: boolean;}
/** Generic list prompt representation. */export abstract class GenericList<T, V, S extends GenericListSettings<T, V>>  extends GenericInput<T, V, S> {  protected options: S["options"] = this.settings.options;  protected listIndex: number = this.getListIndex();  protected listOffset: number = this.getPageOffset(this.listIndex);
  /**   * Create list separator.   * @param label Separator label.   */  public static separator(label = "------------"): GenericListOption {    return { value: label, disabled: true };  }
  /**   * Set list option defaults.   * @param option List option.   */  protected static mapOption(    option: GenericListOption,  ): GenericListOptionSettings {    return {      value: option.value,      name: typeof option.name === "undefined" ? option.value : option.name,      disabled: !!option.disabled,    };  }
  constructor(settings: S) {    super({      ...settings,      keys: {        previous: settings.search ? ["up"] : ["up", "u", "8"],        next: settings.search ? ["down"] : ["down", "d", "2"],        previousPage: ["pageup"],        nextPage: ["pagedown"],        ...(settings.keys ?? {}),      },    });  }
  protected match(): void {    const input: string = this.getCurrentInputValue().toLowerCase();    if (!input.length) {      this.options = this.settings.options.slice();    } else {      this.options = this.settings.options        .filter((option: GenericListOptionSettings) =>          match(option.name) ||          (option.name !== option.value && match(option.value))        )        .sort((a: GenericListOptionSettings, b: GenericListOptionSettings) =>          distance(a.name, input) - distance(b.name, input)        );    }    this.listIndex = Math.max(      0,      Math.min(this.options.length - 1, this.listIndex),    );    this.listOffset = Math.max(      0,      Math.min(        this.options.length - this.getListHeight(),        this.listOffset,      ),    );
    function match(value: string): boolean {      return stripColor(value)        .toLowerCase()        .includes(input);    }  }
  protected message(): string {    let message = `${this.settings.indent}${yellow("?")} ` +      bold(this.settings.message) +      this.defaults();    if (this.settings.search) {      message += " " + this.settings.searchLabel + " ";    }    this.cursor.x = stripColor(message).length + this.inputIndex + 1;    return message + this.input();  }
  /** Render options. */  protected body(): string | Promise<string> {    return this.getList() + this.getInfo();  }
  protected getInfo(): string {    if (!this.settings.info) {      return "";    }    const selected: number = this.listIndex + 1;    const actions: Array<[string, Array<string>]> = [      ["Next", [Figures.ARROW_DOWN]],      ["Previous", [Figures.ARROW_UP]],      ["Next Page", [Figures.PAGE_DOWN]],      ["Previous Page", [Figures.PAGE_UP]],      ["Submit", [Figures.ENTER]],    ];
    return "\n" + this.settings.indent + blue(Figures.INFO) +      bold(` ${selected}/${this.options.length} `) +      actions        .map((cur) => `${cur[0]}: ${bold(cur[1].join(" "))}`)        .join(", ");  }
  /** Render options list. */  protected getList(): string {    const list: Array<string> = [];    const height: number = this.getListHeight();    for (let i = this.listOffset; i < this.listOffset + height; i++) {      list.push(        this.getListItem(          this.options[i],          this.listIndex === i,        ),      );    }    if (!list.length) {      list.push(        this.settings.indent + dim("  No matches..."),      );    }    return list.join("\n");  }
  /**   * Render option.   * @param item        Option.   * @param isSelected  Set to true if option is selected.   */  protected abstract getListItem(    item: GenericListOptionSettings,    isSelected?: boolean,  ): string;
  /** Get options row height. */  protected getListHeight(): number {    return Math.min(      this.options.length,      this.settings.maxRows || this.options.length,    );  }
  protected getListIndex(value?: string) {    return typeof value === "undefined"      ? this.options.findIndex((item: GenericListOptionSettings) =>        !item.disabled      ) || 0      : this.options.findIndex((item: GenericListOptionSettings) =>        item.value === value      ) || 0;  }
  protected getPageOffset(index: number) {    if (index === 0) {      return 0;    }    const height: number = this.getListHeight();    return Math.floor(index / height) * height;  }
  /**   * Find option by value.   * @param value Value of the option.   */  protected getOptionByValue(    value: string,  ): GenericListOptionSettings | undefined {    return this.options.find((option) => option.value === value);  }
  /** Read user input. */  protected read(): Promise<boolean> {    if (!this.settings.search) {      this.tty.cursorHide();    }    return super.read();  }
  /**   * Handle user input event.   * @param event Key event.   */  protected async handleEvent(event: KeyCode): Promise<void> {    switch (true) {      case this.isKey(this.settings.keys, "previous", event):        this.selectPrevious();        break;      case this.isKey(this.settings.keys, "next", event):        this.selectNext();        break;      case this.isKey(this.settings.keys, "nextPage", event):        this.selectNextPage();        break;      case this.isKey(this.settings.keys, "previousPage", event):        this.selectPreviousPage();        break;      default:        await super.handleEvent(event);    }  }
  protected moveCursorLeft(): void {    if (this.settings.search) {      super.moveCursorLeft();    }  }
  protected moveCursorRight(): void {    if (this.settings.search) {      super.moveCursorRight();    }  }
  protected deleteChar(): void {    if (this.settings.search) {      super.deleteChar();    }  }
  protected deleteCharRight(): void {    if (this.settings.search) {      super.deleteCharRight();      this.match();    }  }
  protected addChar(char: string): void {    if (this.settings.search) {      super.addChar(char);      this.match();    }  }
  /** Select previous option. */  protected selectPrevious(): void {    if (this.options.length < 2) {      return;    }    if (this.listIndex > 0) {      this.listIndex--;      if (this.listIndex < this.listOffset) {        this.listOffset--;      }      if (this.options[this.listIndex].disabled) {        this.selectPrevious();      }    } else {      this.listIndex = this.options.length - 1;      this.listOffset = this.options.length - this.getListHeight();      if (this.options[this.listIndex].disabled) {        this.selectPrevious();      }    }  }
  /** Select next option. */  protected selectNext(): void {    if (this.options.length < 2) {      return;    }    if (this.listIndex < this.options.length - 1) {      this.listIndex++;      if (this.listIndex >= this.listOffset + this.getListHeight()) {        this.listOffset++;      }      if (this.options[this.listIndex].disabled) {        this.selectNext();      }    } else {      this.listIndex = this.listOffset = 0;      if (this.options[this.listIndex].disabled) {        this.selectNext();      }    }  }
  /** Select previous page. */  protected selectPreviousPage(): void {    if (this.options?.length) {      const height: number = this.getListHeight();      if (this.listOffset >= height) {        this.listIndex -= height;        this.listOffset -= height;      } else if (this.listOffset > 0) {        this.listIndex -= this.listOffset;        this.listOffset = 0;      }    }  }
  /** Select next page. */  protected selectNextPage(): void {    if (this.options?.length) {      const height: number = this.getListHeight();      if (this.listOffset + height + height < this.options.length) {        this.listIndex += height;        this.listOffset += height;      } else if (this.listOffset + height < this.options.length) {        const offset = this.options.length - height;        this.listIndex += offset - this.listOffset;        this.listOffset = offset;      }    }  }}