From 3dedb89c918c11cc7d1f63101b1c987f70994c87 Mon Sep 17 00:00:00 2001
From: Daniel Odendahl Jr <danielbodendahl@gmail.com>
Date: Wed, 4 Apr 2018 00:57:39 +0000
Subject: [PATCH] Allow specifying Allowed Guilds

---
 Xiao.js      | 27 +++++++++++++++++++++++++--
 package.json |  2 +-
 2 files changed, 26 insertions(+), 3 deletions(-)

diff --git a/Xiao.js b/Xiao.js
index e581d5fa..3fae4a39 100644
--- a/Xiao.js
+++ b/Xiao.js
@@ -1,4 +1,4 @@
-const { XIAO_TOKEN, OWNERS, XIAO_PREFIX, INVITE } = process.env;
+const { XIAO_TOKEN, OWNERS, XIAO_PREFIX, INVITE, XIAO_ALLOWED_GUILDS } = process.env;
 const path = require('path');
 const XiaoClient = require('./structures/Client');
 const client = new XiaoClient({
@@ -11,6 +11,7 @@ const client = new XiaoClient({
 });
 const SequelizeProvider = require('./providers/Sequelize');
 const activities = require('./assets/json/activity');
+const allowedGuilds = XIAO_ALLOWED_GUILDS ? XIAO_ALLOWED_GUILDS.split(',') : null;
 
 client.registry
 	.registerDefaultTypes()
@@ -49,12 +50,23 @@ client.dispatcher.addInhibitor(msg => {
 	return 'Blacklisted.';
 });
 
-client.on('ready', () => {
+client.on('ready', async () => {
 	console.log(`[READY] Logged in as ${client.user.tag}! (${client.user.id})`);
 	client.setInterval(() => {
 		const activity = activities[Math.floor(Math.random() * activities.length)];
 		client.user.setActivity(activity.text, { type: activity.type });
 	}, 60000);
+	if (allowedGuilds) {
+		for (const guild of client.guilds.values()) {
+			if (allowedGuilds.includes(guild.id)) return;
+			try {
+				await guild.leave();
+				console.log(`[GUILD] Left non-allowed guild ${guild.name}. (${guild.id})`);
+			} catch (err) {
+				console.error(`[GUILD] Failed to leave non-allowed guild ${guild.name}. (${guild.id})`, err);
+			}
+		}
+	}
 });
 
 client.on('disconnect', event => {
@@ -62,6 +74,17 @@ client.on('disconnect', event => {
 	process.exit(0);
 });
 
+client.on('guildCreate', async guild => {
+	if (!allowedGuilds) return;
+	if (allowedGuilds.includes(guild.id)) return;
+	try {
+		await guild.leave();
+		console.log(`[GUILD] Left non-allowed guild ${guild.name}. (${guild.id})`);
+	} catch (err) {
+		console.error(`[GUILD] Failed to leave non-allowed guild ${guild.name}. (${guild.id})`, err);
+	}
+});
+
 client.on('commandRun', command => console.log(`[COMMAND] Ran command ${command.groupID}:${command.memberName}.`));
 
 client.on('error', err => console.error('[ERROR]', err));
diff --git a/package.json b/package.json
index f04a286d..d07379c8 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
 	"name": "xiao",
-	"version": "71.0.0",
+	"version": "71.0.1",
 	"description": "Your personal server companion.",
 	"main": "Xiao.js",
 	"scripts": {