123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403
import type { KeyCode } from "../keycode/key_code.ts";import {  GenericInput,  GenericInputKeys,  GenericInputPromptOptions,  GenericInputPromptSettings,} from "./_generic_input.ts";import { blue, bold, dim, stripColor, underline } from "./deps.ts";import { Figures } from "./figures.ts";import { distance } from "../_utils/distance.ts";
interface LocalStorage {  getItem(key: string): string | null;  removeItem(key: string): void;  setItem(key: string, value: string): void;}
/** Input keys options. */export interface GenericSuggestionsKeys extends GenericInputKeys {  complete?: string[];  next?: string[];  previous?: string[];  nextPage?: string[];  previousPage?: string[];}
/** Generic input prompt options. */export interface GenericSuggestionsOptions<T, V>  extends GenericInputPromptOptions<T, V> {  keys?: GenericSuggestionsKeys;  id?: string;  suggestions?: Array<string | number>;  list?: boolean;  info?: boolean;  listPointer?: string;  maxRows?: number;}
/** Generic input prompt settings. */export interface GenericSuggestionsSettings<T, V>  extends GenericInputPromptSettings<T, V> {  keys?: GenericSuggestionsKeys;  id?: string;  suggestions?: Array<string | number>;  list?: boolean;  info?: boolean;  listPointer: string;  maxRows: number;}
/** Generic input prompt representation. */export abstract class GenericSuggestions<  T,  V,  S extends GenericSuggestionsSettings<T, V>,> extends GenericInput<T, V, S> {  protected suggestionsIndex = -1;  protected suggestionsOffset = 0;  protected suggestions: Array<string | number> = [];
  /**   * Prompt constructor.   * @param settings Prompt settings.   */  protected constructor(settings: S) {    super({      ...settings,      keys: {        complete: ["tab"],        next: ["up"],        previous: ["down"],        nextPage: ["pageup"],        previousPage: ["pagedown"],        ...(settings.keys ?? {}),      },    });    const suggestions: Array<string | number> = this.loadSuggestions();    if (suggestions.length || this.settings.suggestions) {      this.settings.suggestions = [        ...suggestions,        ...this.settings.suggestions ?? [],      ].filter(uniqueSuggestions);    }  }
  protected get localStorage(): LocalStorage | null {    // Keep support for deno < 1.10.    if (this.settings.id && "localStorage" in window) {      try {        // deno-lint-ignore no-explicit-any        return (window as any).localStorage;      } catch (_) {        // Ignore error if --location is not set.      }    }    return null;  }
  protected loadSuggestions(): Array<string | number> {    if (this.settings.id) {      const json = this.localStorage?.getItem(this.settings.id);      const suggestions: Array<string | number> = json ? JSON.parse(json) : [];      if (!Array.isArray(suggestions)) {        return [];      }      return suggestions;    }    return [];  }
  protected saveSuggestions(...suggestions: Array<string | number>): void {    if (this.settings.id) {      this.localStorage?.setItem(        this.settings.id,        JSON.stringify([          ...suggestions,          ...this.loadSuggestions(),        ].filter(uniqueSuggestions)),      );    }  }
  protected render(): Promise<void> {    this.match();    return super.render();  }
  protected match(): void {    if (!this.settings.suggestions?.length) {      return;    }    const input: string = this.getCurrentInputValue().toLowerCase();    if (!input.length) {      this.suggestions = this.settings.suggestions.slice();    } else {      this.suggestions = this.settings.suggestions        .filter((value: string | number) =>          stripColor(value.toString())            .toLowerCase()            .startsWith(input)        )        .sort((a: string | number, b: string | number) =>          distance((a || a).toString(), input) -          distance((b || b).toString(), input)        );    }    this.suggestionsIndex = Math.max(      this.getCurrentInputValue().trim().length === 0 ? -1 : 0,      Math.min(this.suggestions.length - 1, this.suggestionsIndex),    );    this.suggestionsOffset = Math.max(      0,      Math.min(        this.suggestions.length - this.getListHeight(),        this.suggestionsOffset,      ),    );  }
  protected input(): string {    return super.input() + dim(this.getSuggestion());  }
  protected getSuggestion(): string {    return this.suggestions[this.suggestionsIndex]?.toString()      .substr(        this.getCurrentInputValue().length,      ) ?? "";  }
  protected body(): string | Promise<string> {    return this.getList() + this.getInfo();  }
  protected getInfo(): string {    if (!this.settings.info) {      return "";    }    const selected: number = this.suggestionsIndex + 1;    const matched: number = this.suggestions.length;    const actions: Array<[string, Array<string>]> = [];
    if (this.settings.suggestions?.length) {      if (this.settings.list) {        actions.push(          ["Next", [Figures.ARROW_DOWN]],          ["Previous", [Figures.ARROW_UP]],          ["Next Page", [Figures.PAGE_DOWN]],          ["Previous Page", [Figures.PAGE_UP]],        );      } else {        actions.push(          ["Next", [Figures.ARROW_UP]],          ["Previous", [Figures.ARROW_DOWN]],        );      }      actions.push(        ["Complete", [Figures.TAB_RIGHT, dim(" or"), Figures.ARROW_RIGHT]],      );    }    actions.push(      ["Submit", [Figures.ENTER]],    );
    let info = this.settings.indent;    if (this.settings.suggestions?.length) {      info += (blue(Figures.INFO) + bold(` ${selected}/${matched} `));    }    info += actions      .map((cur) => `${cur[0]}: ${bold(cur[1].join(" "))}`)      .join(", ");
    return info;  }
  protected getList(): string {    if (!this.settings.suggestions?.length || !this.settings.list) {      return "";    }    const list: Array<string> = [];    const height: number = this.getListHeight();    for (      let i = this.suggestionsOffset;      i < this.suggestionsOffset + height;      i++    ) {      list.push(        this.getListItem(          this.suggestions[i],          this.suggestionsIndex === i,        ),      );    }    if (list.length && this.settings.info) {      list.push("");    }    return list.join("\n");  }
  /**   * Render option.   * @param value        Option.   * @param isSelected  Set to true if option is selected.   */  protected getListItem(    value: string | number,    isSelected?: boolean,  ): string {    let line = this.settings.indent ?? "";    line += isSelected ? `${this.settings.listPointer} ` : "  ";    if (isSelected) {      line += underline(this.highlight(value));    } else {      line += this.highlight(value);    }    return line;  }
  /** Get suggestions row height. */  protected getListHeight(    suggestions: Array<string | number> = this.suggestions,  ): number {    return Math.min(      suggestions.length,      this.settings.maxRows || suggestions.length,    );  }
  /**   * Handle user input event.   * @param event Key event.   */  protected async handleEvent(event: KeyCode): Promise<void> {    switch (true) {      case this.isKey(this.settings.keys, "next", event):        if (this.settings.list) {          this.selectPreviousSuggestion();        } else {          this.selectNextSuggestion();        }        break;      case this.isKey(this.settings.keys, "previous", event):        if (this.settings.list) {          this.selectNextSuggestion();        } else {          this.selectPreviousSuggestion();        }        break;      case this.isKey(this.settings.keys, "nextPage", event):        if (this.settings.list) {          this.selectPreviousSuggestionsPage();        } else {          this.selectNextSuggestionsPage();        }        break;      case this.isKey(this.settings.keys, "previousPage", event):        if (this.settings.list) {          this.selectNextSuggestionsPage();        } else {          this.selectPreviousSuggestionsPage();        }        break;      case this.isKey(this.settings.keys, "complete", event):        this.complete();        break;      case this.isKey(this.settings.keys, "moveCursorRight", event):        if (this.inputIndex < this.inputValue.length) {          this.moveCursorRight();        } else {          this.complete();        }        break;      default:        await super.handleEvent(event);    }  }
  /** Delete char right. */  protected deleteCharRight(): void {    if (this.inputIndex < this.inputValue.length) {      super.deleteCharRight();      if (!this.getCurrentInputValue().length) {        this.suggestionsIndex = -1;        this.suggestionsOffset = 0;      }    }  }
  protected complete(): void {    if (this.suggestions.length && this.suggestions[this.suggestionsIndex]) {      this.inputValue = this.suggestions[this.suggestionsIndex].toString();      this.inputIndex = this.inputValue.length;      this.suggestionsIndex = 0;      this.suggestionsOffset = 0;    }  }
  /** Select previous suggestion. */  protected selectPreviousSuggestion(): void {    if (this.suggestions?.length) {      if (this.suggestionsIndex > -1) {        this.suggestionsIndex--;        if (this.suggestionsIndex < this.suggestionsOffset) {          this.suggestionsOffset--;        }      }    }  }
  /** Select next suggestion. */  protected selectNextSuggestion(): void {    if (this.suggestions?.length) {      if (this.suggestionsIndex < this.suggestions.length - 1) {        this.suggestionsIndex++;        if (          this.suggestionsIndex >=            this.suggestionsOffset + this.getListHeight()        ) {          this.suggestionsOffset++;        }      }    }  }
  /** Select previous suggestions page. */  protected selectPreviousSuggestionsPage(): void {    if (this.suggestions?.length) {      const height: number = this.getListHeight();      if (this.suggestionsOffset >= height) {        this.suggestionsIndex -= height;        this.suggestionsOffset -= height;      } else if (this.suggestionsOffset > 0) {        this.suggestionsIndex -= this.suggestionsOffset;        this.suggestionsOffset = 0;      }    }  }
  /** Select next suggestions page. */  protected selectNextSuggestionsPage(): void {    if (this.suggestions?.length) {      const height: number = this.getListHeight();      if (this.suggestionsOffset + height + height < this.suggestions.length) {        this.suggestionsIndex += height;        this.suggestionsOffset += height;      } else if (this.suggestionsOffset + height < this.suggestions.length) {        const offset = this.suggestions.length - height;        this.suggestionsIndex += offset - this.suggestionsOffset;        this.suggestionsOffset = offset;      }    }  }}
function uniqueSuggestions(  value: unknown,  index: number,  self: Array<unknown>,) {  return typeof value !== "undefined" && value !== "" &&    self.indexOf(value) === index;}