import constants from "./constants"; import Mousetrap from "mousetrap"; import { Strategy, Textcomplete, StrategyProps } from "@textcomplete/core"; import { TextareaEditor } from "@textcomplete/textarea"; import fuzzy from "fuzzy"; import { store } from "./store"; export default enableAutocomplete; const nicksStrategy: StrategyProps = { id: "nicks", match: /(^|\s)(@([a-zA-Z_[\]\\^{}|`@][a-zA-Z0-9_[\]\\^{}|`-]*)?)$/, search(term: string, callback: (matches: string[] | string[][]) => void) { term = term.slice(1); if (term[0] === "@") { // TODO: type // eslint-disable-next-line @typescript-eslint/restrict-plus-operands callback( completeNicks(term.slice(1), true).map((val) => [ "@" + val[0], "@" + val[1], ]) ); } else { callback(completeNicks(term, true)); } }, template([string]: [string, string]) { return string; }, replace([, original]: [string, string]) { return "$1" + replaceNick(original); }, index: 2, }; const chanStrategy: StrategyProps = { id: "chans", match: /(^|\s)((?:#|\+|&|![A-Z0-9]{5})(?:[^\s]+)?)$/, search(term: string, callback: (matches: string[][]) => void) { callback(completeChans(term)); }, template([string]: [string, string]) { return string; }, replace([, original]: [string, string]) { return "$1" + original; }, index: 2, }; const commandStrategy: StrategyProps = { id: "commands", match: /^\/(\w*)$/, search(term: string, callback: (matches: string[][]) => void) { callback(completeCommands("/" + term)); }, template([string]: [string, string]) { return string; }, replace([, original]: [string, string]) { return original; }, index: 1, }; const foregroundColorStrategy: StrategyProps = { id: "foreground-colors", match: /\x03(\d{0,2}|[A-Za-z ]{0,10})$/, search(term: string, callback: (matches: string[][]) => void) { term = term.toLowerCase(); const matchingColorCodes = constants.colorCodeMap .filter((i) => fuzzy.test(term, i[0]) || fuzzy.test(term, i[1])) .map((i) => { if (fuzzy.test(term, i[1])) { return [ i[0], fuzzy.match(term, i[1], { pre: "", post: "", }).rendered, ]; } return i; }); callback(matchingColorCodes); }, template(value: string[]) { return `${ value[1] }`; }, replace(value: string) { return "\x03" + value[0]; }, index: 1, }; const backgroundColorStrategy: StrategyProps = { id: "background-colors", match: /\x03(\d{2}),(\d{0,2}|[A-Za-z ]{0,10})$/, search( term: string, callback: (matchingColorCodes: string[][]) => void, match: string[] ) { term = term.toLowerCase(); const matchingColorCodes = constants.colorCodeMap .filter((i) => fuzzy.test(term, i[0]) || fuzzy.test(term, i[1])) .map((pair) => { if (fuzzy.test(term, pair[1])) { return [ pair[0], fuzzy.match(term, pair[1], { pre: "", post: "", }).rendered, ]; } return pair; }) .map((pair) => pair.concat(match[1])); // Needed to pass fg color to `template`... callback(matchingColorCodes); }, template(value: string[]) { return `${value[1]}`; }, replace(value: string[]) { return "\x03$1," + value[0]; }, index: 2, }; function enableAutocomplete(input: HTMLTextAreaElement) { let tabCount = 0; let lastMatch = ""; let currentMatches: string[] | string[][] = []; input.addEventListener("input", (e) => { if ((e as CustomEvent).detail === "autocomplete") { return; } tabCount = 0; currentMatches = []; lastMatch = ""; }); Mousetrap(input).bind( "tab", (e) => { if (store.state.isAutoCompleting) { return; } e.preventDefault(); const text = input.value; if (tabCount === 0) { lastMatch = text.substring(0, input.selectionStart).split(/\s/).pop() || ""; if (lastMatch.length === 0) { return; } currentMatches = completeNicks(lastMatch, false); if (currentMatches.length === 0) { return; } } const position = input.selectionStart - lastMatch.length; const newMatch = replaceNick( // TODO: type this properly String(currentMatches[tabCount % currentMatches.length]), position ); const remainder = text.substring(input.selectionStart); input.value = text.substr(0, position) + newMatch + remainder; input.selectionStart -= remainder.length; input.selectionEnd = input.selectionStart; // Propagate change to Vue model input.dispatchEvent( new CustomEvent("input", { detail: "autocomplete", }) ); lastMatch = newMatch; tabCount++; }, "keydown" ); const strategies = [ nicksStrategy, chanStrategy, commandStrategy, foregroundColorStrategy, backgroundColorStrategy, ]; const editor = new TextareaEditor(input); const textcomplete = new Textcomplete(editor, strategies, { dropdown: { className: "textcomplete-menu", placement: "top", }, }); textcomplete.on("show", () => { store.commit("isAutoCompleting", true); }); textcomplete.on("hidden", () => { store.commit("isAutoCompleting", false); }); return { hide() { textcomplete.hide(); }, destroy() { textcomplete.destroy(); store.commit("isAutoCompleting", false); }, }; } function replaceNick(original: string, position = 1) { // If no postfix specified, return autocompleted nick as-is if (!store.state.settings.nickPostfix) { return original; } // If there is whitespace in the input already, append space to nick if ( position > 0 && /\s/.test(store.state.activeChannel?.channel.pendingMessage || "") ) { return original + " "; } // If nick is first in the input, append specified postfix return original + store.state.settings.nickPostfix; } function fuzzyGrep(term: string, array: Array) { const results = fuzzy.filter(term, array, { pre: "", post: "", }); return results.map((el) => [el.string, el.original]); } function rawNicks() { if (!store.state.activeChannel) { return []; } if (store.state.activeChannel.channel.users.length > 0) { const users = store.state.activeChannel.channel.users.slice(); return users .sort((a, b) => b.lastMessage - a.lastMessage) .map((u) => u.nick); } const me = store.state.activeChannel.network.nick; const otherUser = store.state.activeChannel.channel.name; // If this is a query, add their name to autocomplete if ( me !== otherUser && store.state.activeChannel.channel.type === "query" ) { return [otherUser, me]; } // Return our own name by default for anything that isn't a channel or query return [me]; } function completeNicks(word: string, isFuzzy: boolean) { const users = rawNicks(); word = word.toLowerCase(); if (isFuzzy) { return fuzzyGrep(word, users); } return users.filter((w) => !w.toLowerCase().indexOf(word)); } function getCommands() { let cmds = constants.commands.slice(); if (!store.state.settings.searchEnabled) { cmds = cmds.filter((c) => c !== "/search"); } return cmds; } function completeCommands(word: string) { const commands = getCommands(); return fuzzyGrep(word, commands); } function completeChans(word: string) { const words: string[] = []; if (store.state.activeChannel) { for (const channel of store.state.activeChannel.network.channels) { // Push all channels that start with the same CHANTYPE if (channel.type === "channel" && channel.name[0] === word[0]) { words.push(channel.name); } } } return fuzzyGrep(word, words); }