tweb-i2p/src/components/chat/commandsHelper.ts

/*
 * https://github.com/morethanwords/tweb
 * Copyright (C) 2019-2021 Eduard Kuzmenko
 * https://github.com/morethanwords/tweb/blob/master/LICENSE
 */

import type ChatInput from "./input";
import type { BotInfo, ChatFull, UserFull } from "../../layer";
import AutocompleteHelperController from "./autocompleteHelperController";
import AutocompletePeerHelper from "./autocompletePeerHelper";
import SearchIndex from "../../lib/searchIndex";
import { AppManagers } from "../../lib/appManagers/managers";

export function processPeerFullForCommands(peerId: PeerId, full: ChatFull.chatFull | ChatFull.channelFull | UserFull.userFull, query?: string) {
  const botInfos: BotInfo.botInfo[] = [].concat(full.bot_info);
  let index: SearchIndex<string>; 
  
  if(query !== undefined) {
    index = new SearchIndex<string>({
      ignoreCase: true
    });
  }
  
  type T = {peerId: PeerId, name: string, description: string, index: number, command: string};
  const commands: Map<string, T> = new Map();
  botInfos.forEach((botInfo) => {
    botInfo.commands.forEach(({command, description}, idx) => {
      const c = '/' + command;
      commands.set(command, {
        peerId: botInfo.user_id ? botInfo.user_id.toPeerId(false) : peerId, 
        command: command, 
        name: c, 
        description: description,
        index: idx
      });

      if(index) {
        index.indexObject(command, c);
      }
    });
  });

  let out: T[];
  if(!index) {
    out = [...commands.values()];
  } else {
    const found = index.search(query);
    out = Array.from(found).map((command) => commands.get(command));
  }

  out = out.sort((a, b) => commands.get(a.command).index - commands.get(b.command).index);
  
  return out;
}

export default class CommandsHelper extends AutocompletePeerHelper {
  constructor(
    appendTo: HTMLElement, 
    controller: AutocompleteHelperController, 
    chatInput: ChatInput, 
    private managers: AppManagers
  ) {
    super(appendTo, 
      controller,
      'commands-helper',
      (target) => {
        const innerHTML = target.querySelector(`.${AutocompletePeerHelper.BASE_CLASS_LIST_ELEMENT}-name`).innerHTML;
        return chatInput.getReadyToSend(() => {
          chatInput.messageInput.innerHTML = innerHTML;
          chatInput.sendMessage(true);
        });
      }
    );
  }

  public async checkQuery(query: string, peerId: PeerId) {
    if(!(await this.managers.appUsersManager.isBot(peerId))) {
      return false;
    }

    const middleware = this.controller.getMiddleware();
    this.managers.appProfileManager.getProfileByPeerId(peerId).then((full) => {
      if(!middleware()) {
        return;
      }

      const filtered = processPeerFullForCommands(peerId, full, query);
      this.render(filtered);
      // console.log('found commands', found, filtered);
    });

    return true;
  }
}