pokemon-showdown/server/chat-plugins/usersearch.ts

import {Utils, FS} from '../../lib';

export const nameList: string[] = JSON.parse(
	FS('config/chat-plugins/usersearch.json').readIfExistsSync() || "[]"
);

const ONLINE_SYMBOL = ` \u25C9 `;
const OFFLINE_SYMBOL = ` \u25CC `;

function getPunishmentHTML(userid: ID, target: string) {
	return [
		'Forcerename', 'Namelock', 'Weeknamelock',
	].map((cmd) => (
		`<button class="button" name="send" value="/msgroom staff,/${toID(cmd)} ${userid}&#10;/uspage ${target}">${cmd}</button>`
	)).join(' | ');
}

function searchUsernames(target: string, page = false) {
	const results: {offline: string[], online: string[]} = {
		offline: [],
		online: [],
	};
	for (const curUser of Users.users.values()) {
		if (!curUser.id.includes(target) || curUser.id.startsWith('guest')) continue;
		if (curUser.connected) {
			results.online.push(`${!page ? ONLINE_SYMBOL : ''} ${curUser.name}`);
		} else {
			results.offline.push(`${!page ? OFFLINE_SYMBOL : ''} ${curUser.name}`);
		}
	}
	for (const k in results) {
		Utils.sortBy(results[k as keyof typeof results], result => toID(result));
	}
	let buf = ``;
	if (!page) {
		buf = `Users with a name matching '${target}':<br />`;
		if (!results.offline.length && !results.online.length) {
			buf += `No users found.`;
		} else {
			buf += results.online.join('; ');
			if (results.offline.length) {
				if (results.online.length) buf += `<br /><br />`;
				buf += results.offline.join('; ');
			}
		}
	} else {
		buf += `<div class="pad"><h3>Usernames containing "${target}"</h3>`;
		if (!results.offline.length && !results.online.length) {
			buf += `<p>No results found.</p>`;
		} else {
			if (!results.offline.length && !results.online.length) {
				buf += `<p>No users found.</p>`;
			} else {
				buf += `<div class="ladder pad"><h3>Online users</h3><table><tr><th>Username</th><th>Punish</th></tr>`;
				for (const username of results.online) {
					buf += `<tr><td><username>${Utils.escapeHTML(username)}</username></td>`;
					buf += `<td>${getPunishmentHTML(toID(username), target)}</td></tr>`;
				}
				buf += `</table></div>`;
				if (results.offline.length && results.online.length) {
					buf += `<hr />`;
				}
				if (results.offline.length) {
					buf += `<div class="ladder pad"><h3>Offline users</h3><table><tr><th>Username</th><th>Punish</th></tr>`;
					for (const username of results.offline) {
						buf += `<tr><td><username>${Utils.escapeHTML(username)}</username></td>`;
						buf += `<td>${getPunishmentHTML(toID(username), target)}</td></tr>`;
					}
				}
			}
		}
		buf += `</div>`;
	}
	return buf;
}

function saveNames() {
	FS('config/chat-plugins/usersearch.json').writeUpdate(() => JSON.stringify(nameList));
}

export const commands: Chat.ChatCommands = {
	us: 'usersearch',
	uspage: 'usersearch',
	usersearchpage: 'usersearch',
	usersearch(target, room, user, connection, cmd) {
		this.checkCan('lock');
		target = toID(target);
		if (!target) { // just join directly if it's the page cmd, they're likely looking for the full list
			if (cmd.includes('page')) return this.parse(`/j view-usersearch`);
			return this.parse(`/help usersearch`);
		}
		if (target.length < 3) {
			throw new Chat.ErrorMessage(`That's too short of a term to search for.`);
		}
		const showPage = cmd.includes('page');
		if (showPage) {
			this.parse(`/j view-usersearch-${target}`);
			return;
		}
		return this.sendReplyBox(searchUsernames(target));
	},
	usersearchhelp: [
		`/usersearch [pattern]: Looks for all names matching the [pattern]. Requires: % @ &`,
		`Adding "page" to the end of the command, i.e. /usersearchpage OR /uspage will bring up a page.`,
	],
	usnames: 'usersearchnames',
	usersearchnames: {
		'': 'list',
		list() {
			this.parse(`/join view-usersearch`);
		},
		add(target, room, user) {
			this.checkCan('lock');
			const targets = target.split(',').map(toID).filter(Boolean);
			if (!targets.length) {
				return this.errorReply(`Specify at least one term.`);
			}
			for (const [i, arg] of targets.entries()) {
				if (nameList.includes(arg)) {
					targets.splice(i, 1);
					this.errorReply(`Term ${arg} is already on the usersearch term list.`);
					continue;
				}
				if (arg.length < 3) {
					targets.splice(i, 1);
					this.errorReply(`Term ${arg} is too short for the usersearch term list. Must be more than 3 characters.`);
					continue;
				}
			}
			nameList.push(...targets);
			Rooms.get('staff')?.addByUser(user, `${user.name} added ${Chat.count(targets, 'terms')} to the usersearch name list.`);
			this.globalModlog(`USERSEARCH ADD`, null, targets.join(', '));
			if (!room || room.roomid !== 'staff') {
				this.sendReply(`Added ${Chat.count(targets, 'terms')} to the usersearch name list.`);
			}
			saveNames();
		},
		remove(target, room, user) {
			this.checkCan('lock');
			const targets = target.split(',').map(toID).filter(Boolean);
			if (!targets.length) {
				return this.errorReply(`Specify at least one term.`);
			}
			for (const [i, arg] of targets.entries()) {
				const idx = nameList.indexOf(arg);
				if (idx < 0) {
					targets.splice(i, 1);
					this.errorReply(`${arg} is not in the usersearch name list, and has been skipped.`);
					continue;
				}
				nameList.splice(idx, 1);
			}
			Rooms.get('staff')?.addByUser(user, `${user.name} removed ${Chat.count(targets, 'terms')} from the usersearch name list.`);
			this.globalModlog(`USERSEARCH REMOVE`, null, targets.join(', '));
			if (!room || room.roomid !== 'staff') {
				this.sendReply(`You removed ${Chat.count(targets, 'terms')} from the usersearch name list.`);
			}
			saveNames();
		},
	},
};

export const pages: Chat.PageTable = {
	usersearch(query, user) {
		this.checkCan('lock');
		const target = toID(query.shift());
		if (!target) {
			this.title = `[Usersearch Terms]`;
			let buf = `<div class="pad"><strong>Usersearch term list</strong>`;
			buf += `<button style="float:right;" class="button" name="send" value="/uspage"><i class="fa fa-refresh"></i> Refresh</button>`;
			buf += `<hr />`;
			if (!nameList.length) {
				buf += `None found.`;
				return buf;
			}
			const sorted: {[k: string]: number} = {};
			for (const curUser of Users.users.values()) {
				for (const term of nameList) {
					if (curUser.id.includes(term)) {
						if (!sorted[term]) sorted[term] = 0;
						sorted[term]++;
					}
				}
			}
			buf += `<div class="ladder pad"><table>`;
			buf += `<tr><th>Term</th><th>Current matches</th><th></th></tr>`;
			for (const k of Utils.sortBy(Object.keys(sorted), v => -sorted[v])) {
				buf += `<tr>`;
				buf += `<td>${k}</td>`;
				buf += `<td>${sorted[k]}</td>`;
				buf += `<td><button class="button" name="send" value="/uspage ${k}">Search</button></td>`;
				buf += `</tr>`;
			}
			return buf;
		}
		this.title = `[Usersearch] ${target}`;
		return searchUsernames(target, true);
	},
};