pidgey.js

#!/bin/env node

const Discord = require("discord.js");
const client = new Discord.Client();
const config = require("./config.json");
const strings = require("./strings.json");
const fs = require('fs');

const poifinder = require("./poifind.js");
poifinder.load(config.poifile);

let POIfile = require("./data/poi.json");

// Discord limitation
const MAX_MESSAGE_SIZE = 2000;
// Number of hits to show in guild
const MAX_HITS_CHANNEL = 10;
// Number if hits to show in DM
const MAX_HITS_DM = 25;
// Discord's stupid way of displaying the numbers emojis 0-10
const REACTION_EMOJI = ["\u0030\u20E3", "\u0031\u20E3", "\u0032\u20E3", "\u0033\u20E3", "\u0034\u20E3", "\u0035\u20E3", "\u0036\u20E3", "\u0037\u20E3", "\u0038\u20E3", "\u0039\u20E3", "\ud83d\udd1f"]

let writeLog = function (logfile, message) {
    if (config.logdir == null) return;
    if (!fs.existsSync(config.logdir)) fs.mkdirSync(config.logdir);
    fs.appendFile(config.logdir + "/" + logfile,
		  (new Date()).toISOString() + "\t" + message + "\n",
		  function () { });
}

let createClientMsgSingleMatch = function (singleMatch) {
    let msg = {};
    let coord = singleMatch[2] + "," + singleMatch[3];
    let mapurl = 'https://maps.googleapis.com/maps/api/staticmap?size=512x512&zoom=15&scale=2&key=' + config.google_api_key;
    mapurl = mapurl + '&center=' + coord;
    if (singleMatch['1'] == 'gym')
	mapurl = mapurl + "&markers=color:green%7Clabel:G%7C" + coord;
    else if (singleMatch['1'] == 'portal')
	mapurl = mapurl + "&markers=color:blue%7Clabel:P%7C" + coord;
    else
	mapurl = mapurl + "&markers=color:red%7Clabel:S%7C" + coord;
    
    const embed = new Discord.RichEmbed();
    embed
	.setTitle(singleMatch[0] + " (" + singleMatch[1] + ")")
	.setImage(mapurl)
	.setDescription("[OpenStreetMap](http://www.openstreetmap.org/?mlat=" +
			singleMatch[2] + "&mlon=" +
			singleMatch[3] + "&zoom=15&layers=M)" +
			" / " +
			"[Google Maps](https://www.google.com/maps/dir/?api=1&dir_action=travelmode=walking&navigate&destination=" + coord + ")" +
			" / " +
			"[StreetView](https://www.google.com/maps/?q=&layer=c&cbll=" + coord + ")"
		       );
    msg = { embed };
    let portalLookup = false;
    if (singleMatch[1] == "portal") {
	//console.log("A portal has been looked up");
	portalLookup = true;
    }
    return [msg, portalLookup];
}

let portalLookupMsg = async function (originalMsg, portalReplyMsg, singleMatch) {
    const stopEmoji = originalMsg.guild.emojis.find(x => x.name === 'pokestop');
    const gymEmoji = originalMsg.guild.emojis.find(x => x.name === 'gym');
	portalReplyMsg.react(stopEmoji).then(() => portalReplyMsg.react(gymEmoji));
    
    const filter = (reaction, user) => {
	return [stopEmoji, gymEmoji].includes(reaction.emoji) && user.id === originalMsg.author.id;
    };
    
    portalReplyMsg.awaitReactions(filter, { max: 1, time: 60000, errors: ['time'] })
	.then(collected => {
	    const reaction = collected.first();
	    
	    if (reaction.emoji === stopEmoji) {
		setPOItype("pokestop", singleMatch[2], singleMatch[3]);
		originalMsg.reply('Registrert som pokestop. OBS: vises ved neste omstart av Pidgey');
	    }
	    else if (reaction.emoji === gymEmoji) {
		setPOItype("gym", singleMatch[2], singleMatch[3]);
		originalMsg.reply('Registrert som gym. OBS: vises ved neste omstart av Pidgey');
	    }
	    portalReplyMsg.clearReactions();
	})
	.catch(function(){
	    originalMsg.reply("Ingen reaksjoner registert. Vennligst registrer om dette er en gym eller stop ved å søke på nytt.");
	    portalReplyMsg.clearReactions();
	});
    
}

client.on("message", async message => {
    
    // Ignore all bots.
    if (message.author.bot) return;
    
    // Ignore messages not starting with ! 
    if (message.content.indexOf(config.prefix) !== 0) return;
    
    // Get command and argument if given. 
    const args = message.content.slice(config.prefix.length).trim().split(/ +/g);
    const command = args.shift().toLowerCase();
    const isDirectMessage = (message.guild === null);
    
    let singleMatch;
    
    if (command === config.command) {
	
	let scope = "";
	let clientMessage;
	let matches = null, query, showHelp;
	
	// Shop up to 10 matches in chats, 250 hits in dm's
	let maxHits = isDirectMessage ? MAX_HITS_DM : MAX_HITS_CHANNEL;
	
	//If portal lookup is true, add reaction message to edit poi.json
	var portalLookup = false;
	
	// If no arguments, or single argument pokestop|gym, show help text
	if (!args[0])
	    showHelp = true;
	if (!showHelp && poifinder.isPoiType(args[0].toLowerCase())) {
	    scope = args.shift().toLowerCase();
	}
	if (!args[0])
	    showHelp = true;
	
	if (!showHelp) {
	    // If numeric query, return poi by number. This works
	    // becase all numbers which are part of poi names are
	    // are exclude.
	    query = args.join(" ");
	    if (/^[0-9]+$/.test(query)) {
		matches = poifinder.getByNumber(query);
	    }
	    if (!matches || !matches.length)
		matches = poifinder.find(query, scope);
	}
	
	if (showHelp) {
	    const embed = new Discord.RichEmbed();
	    embed.setTitle(config.description)
		.setDescription(config.prefix + config.command + strings[config.language]["searchstring"]);
	    clientMessage = { embed };
	    
	} else if (!matches || matches.length == 0) {
	    
	    clientMessage = strings[config.language]["nomatches"].replace('{term}', query);
	    
	} else if(singleMatch = poifinder.singleMatch(matches, query, scope)) {
	    
	    [clientMessage, portalLookup] = createClientMsgSingleMatch(singleMatch)
	    
	} else if (matches.length <= maxHits) {
	    
	    clientMessage = strings[config.language]["selectmap"];
	    clientMessage += "\n";
	    clientMessage += poifinder.listResults(matches).join("\n");
	    
	    // Too long messages will be rejected by server
	    if (clientMessage.length > MAX_MESSAGE_SIZE) {
		clientMessage = strings[config.language]["toomany"].replace('{num}', matches.length);
		clientMessage += strings[config.language]["refinequery"];
	    }
	    
	} else {
	    clientMessage = strings[config.language]["toomany"].replace('{num}', matches.length);
	    
	    if (!isDirectMessage && matches.length < MAX_HITS_DM)
		clientMessage += strings[config.language]["senddm"];
	    else
		clientMessage += strings[config.language]["refinequery"];
	}
	
	//Send message to channel
	await message.channel.send(clientMessage)
	    .then(async function (msg) {
		let currentMap=false;
		if (portalLookup == true && config.portalmenu &&  !isDirectMessage) { 
		    // Emoji does not exist on pm, it only exist in guild / server
		    portalLookupMsg(message, msg, singleMatch);
		}
		else if (matches.length > 1 && matches.length <= maxHits) {
		    let i;
		    let maxReactions = Math.min(MAX_HITS_CHANNEL, matches.length);
		    
		    const filter = (reaction, user) => {
			return REACTION_EMOJI.includes(reaction.emoji.name) && user.id === message.author.id;
		    };
                    
		    let collector = msg.createReactionCollector(filter, { time: 60000 });
		    collector.on('collect', (reaction, collector) => {                                                                    
			let index = REACTION_EMOJI.findIndex(x => x === reaction.emoji.name);
                        singleMatch = matches[index-1];
                        [reply, portalLookup] = createClientMsgSingleMatch(singleMatch);
			if(currentMap) {
			    currentMap.delete();
			    currentMap=false;
			}
                        
                        message.channel.send(reply).then(async function (replymsg) {
			    currentMap = replymsg;
                            if (portalLookup == true && config.portalmenu  && !isDirectMessage) {
                                // Emoji does not exist on pm, it only exist in guild / server)
				portalLookupMsg(message, replymsg);
                            }
                        })
                    })
		    collector.on('end',() => {
			// Avoid msg.clearReactions() since pidgey does not reqiure the necessary permimssions.
                        msg.reactions.filter(u => u.me).forEach(function(r) { 
                            r.remove(); 
                        })
                    });
		    
		  for (i = 1; i <= maxReactions; i++) { // Await makes the emoji appear in order
		      await msg.react(REACTION_EMOJI[i]); 
		  }
		}
		writeLog("searches.log", " " + message.channel.name + ": [" + matches.length + "] " + message);
	    }).catch(function () {
		writeLog("error.log", "ERROR:  " + message.channel.name + ":" + message);
	    });
    }
});

function setPOItype(type, lat, lng) {
    for (var i = POIfile.length - 1; i >= 0; i--) {
	if (POIfile[i][2] === lat && POIfile[i][3] === lng) {
	    POIfile[i][1] = type;
	    fs.writeFileSync('./data/poi.json', JSON.stringify(POIfile, null, 4));
	    return;
	}
    }
}

client.on("ready", () => {
    client.user.setActivity(config.prefix + config.command);
});

client.on('error', (error) => {
    writeLog("error.log", error);
});

client.login(config.token);