zulip-desktop/app/renderer/js/notification/helpers.ts

import {remote} from 'electron';

import Logger from '../utils/logger-util';

const logger = new Logger({
	file: 'errors.log',
	timestamp: true
});

// Do not change this
export const appId = 'org.zulip.zulip-electron';

export type BotListItem = [string, string];
const botsList: BotListItem[] = [];
let botsListLoaded = false;

// This function load list of bots from the server
// in case botsList isn't already completely loaded when required in parseRely
export async function loadBots(): Promise<void> {
	botsList.length = 0;
	const response = await fetch('/json/users');
	if (response.ok) {
		const {members} = await response.json();
		members.forEach(({is_bot, full_name}: any) => {
			if (is_bot && typeof full_name === 'string') {
				const bot = `@${full_name}`;
				const mention = `@**${bot.replace(/^@/, '')}**`;
				botsList.push([bot, mention]);
			}
		});
		botsListLoaded = true;
	} else {
		logger.log('Load bots request failed: ', await response.text());
		logger.log('Load bots request status: ', response.status);
	}
}

export function checkElements(...elements: any[]): boolean {
	let status = true;
	elements.forEach(element => {
		if (element === null || element === undefined) {
			status = false;
		}
	});
	return status;
}

export function customReply(reply: string): void {
	// Server does not support notification reply yet.
	const buttonSelector = '.messagebox #send_controls button[type=submit]';
	const messageboxSelector = '.selected_message .messagebox .messagebox-border .messagebox-content';
	const textarea: HTMLInputElement = document.querySelector('#compose-textarea');
	const messagebox: HTMLButtonElement = document.querySelector(messageboxSelector);
	const sendButton: HTMLButtonElement = document.querySelector(buttonSelector);

	// Sanity check for old server versions
	const elementsExists = checkElements(textarea, messagebox, sendButton);
	if (!elementsExists) {
		return;
	}

	textarea.value = reply;
	messagebox.click();
	sendButton.click();
}

const currentWindow = remote.getCurrentWindow();
const webContents = remote.getCurrentWebContents();
const webContentsId = webContents.id;

// This function will focus the server that sent
// the notification. Main function implemented in main.js
export function focusCurrentServer(): void {
	currentWindow.webContents.send('focus-webview-with-id', webContentsId);
}

// This function parses the reply from to notification
// making it easier to reply from notification eg
// @username in reply will be converted to @**username**
// #stream in reply will be converted to #**stream**
// bot mentions are not yet supported
export async function parseReply(reply: string): Promise<string> {
	const usersDiv = document.querySelectorAll('#user_presences li');
	const streamHolder = document.querySelectorAll('#stream_filters li');

	type UsersItem = BotListItem;
	type StreamsItem = BotListItem;
	const users: UsersItem[] = [];
	const streams: StreamsItem[] = [];

	usersDiv.forEach(userRow => {
		const anchor = userRow.querySelector('span a');
		if (anchor !== null) {
			const user = `@${anchor.textContent.trim()}`;
			const mention = `@**${user.replace(/^@/, '')}**`;
			users.push([user, mention]);
		}
	});

	streamHolder.forEach(stream => {
		const streamAnchor = stream.querySelector('div a');
		if (streamAnchor !== null) {
			const streamName = `#${streamAnchor.textContent.trim()}`;
			const streamMention = `#**${streamName.replace(/^#/, '')}**`;
			streams.push([streamName, streamMention]);
		}
	});

	users.forEach(([user, mention]) => {
		if (reply.includes(user)) {
			const regex = new RegExp(user, 'g');
			reply = reply.replace(regex, mention);
		}
	});

	streams.forEach(([stream, streamMention]) => {
		const regex = new RegExp(stream, 'g');
		reply = reply.replace(regex, streamMention);
	});

	// If botsList isn't completely loaded yet, make a request for list
	if (!botsListLoaded) {
		await loadBots();
	}

	// Iterate for every bot name and replace in reply
	// @botname with @**botname**
	botsList.forEach(([bot, mention]) => {
		if (reply.includes(bot)) {
			const regex = new RegExp(bot, 'g');
			reply = reply.replace(regex, mention);
		}
	});

	reply = reply.replace(/\\n/, '\n');
	return reply;
}