From 36229b9846ea69ec7dc7cd13b89cfe86d0821a53 Mon Sep 17 00:00:00 2001
From: Dragon Fire <danielbodendahl@gmail.com>
Date: Fri, 27 Mar 2020 12:22:43 -0400
Subject: [PATCH] Add Dragon Fire to company

---
 commands/search/company.js | 24 ++++++++++++++++++------
 package.json               |  2 +-
 2 files changed, 19 insertions(+), 7 deletions(-)

diff --git a/commands/search/company.js b/commands/search/company.js
index fb32611f..79f7b92c 100644
--- a/commands/search/company.js
+++ b/commands/search/company.js
@@ -2,6 +2,7 @@ const Command = require('../../structures/Command');
 const { MessageEmbed } = require('discord.js');
 const request = require('node-superfetch');
 const { CLEARBIT_KEY } = process.env;
+const dragonFireAliases = ['dragonfire535', 'dragon fire'];
 
 module.exports = class CompanyCommand extends Command {
 	constructor(client) {
@@ -31,12 +32,8 @@ module.exports = class CompanyCommand extends Command {
 
 	async run(msg, { query }) {
 		try {
-			const { body } = await request
-				.get(`https://autocomplete.clearbit.com/v1/companies/suggest`)
-				.query({ query })
-				.set({ Authorization: `Bearer ${CLEARBIT_KEY}` });
-			if (!body.length) return msg.say('Could not find any results.');
-			const data = body[0];
+			const data = await this.fetchCompany(query);
+			if (!data) return msg.say('Could not find any results.');
 			const embed = new MessageEmbed()
 				.setTitle(data.name)
 				.setImage(data.logo)
@@ -48,4 +45,19 @@ module.exports = class CompanyCommand extends Command {
 			return msg.reply(`Oh no, an error occurred: \`${err.message}\`. Try again later!`);
 		}
 	}
+
+	async fetchCompany(query) {
+		if (dragonFireAliases.includes(query.toLowerCase())) {
+			return {
+				name: 'Dragon Fire',
+				logo: 'https://i.imgur.com/G5BP0kB.png'
+			};
+		}
+		const { body } = await request
+			.get(`https://autocomplete.clearbit.com/v1/companies/suggest`)
+			.query({ query })
+			.set({ Authorization: `Bearer ${CLEARBIT_KEY}` });
+		if (!body.length) return null;
+		return body[0];
+	}
 };
diff --git a/package.json b/package.json
index 9d8e2077..ce755f9c 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
 	"name": "xiao",
-	"version": "112.18.0",
+	"version": "112.18.1",
 	"description": "Your personal server companion.",
 	"main": "Xiao.js",
 	"scripts": {