galactic-bot/structure/client/Resolver.js

503 lines
19 KiB
JavaScript
Raw Normal View History

2020-06-04 19:59:09 +02:00
const timestring = require('timestring');
2020-04-08 18:08:46 +02:00
class Resolver {
constructor(client) {
this.client = client;
}
/**
*
*
* @param {string} [str=''] Keyword to search for
* @param {string} type Type of the component (ex. command)
* @param {boolean} [exact=true] Exact matching
* @returns
* @memberof Resolver
*/
components(str = '', type, exact = true) {
2020-04-08 18:08:46 +02:00
const string = str.toLowerCase();
const components = this.client.registry.components
.filter((c) => c.type === type)
.filter(exact ? filterExact(string) : filterInexact(string)) //eslint-disable-line no-use-before-define
2020-04-08 18:08:46 +02:00
.array();
return components || [];
}
2020-05-24 23:39:37 +02:00
timeAgo(diff, extraMin = false, extraHours = false, extraDays = false) {
diff = parseInt(diff);
if (isNaN(diff)) return 'that ain\'t it chief (not a number)';
2020-06-09 17:58:09 +02:00
const years = Math.floor(diff / 60 / 60 / 24 / 365),
months = Math.floor(diff / 60 / 60 / 24 / 30.4),
weeks = extraDays ? Math.floor(diff / 60 / 60 / 24 / 7) : (diff / 60 / 60 / 24 / 7).toFixed(),
days = extraHours ? Math.floor(diff / 60 / 60 / 24) : (diff / 60 / 60 / 24).toFixed(),
hours = extraMin ? Math.floor(diff / 60 / 60) : (diff / 60 / 60).toFixed(),
2020-05-24 23:39:37 +02:00
minutes = (diff / 60).toFixed();
if (days > 365) {
2020-06-09 17:58:09 +02:00
return `${years > 0 ? years : 1} year${years > 1 ? 's' : ''}${months % 12 > 0 ? ` ${months % 12} month${months % 12 > 1 ? 's' : ''}` : ''}`;
2020-05-24 23:39:37 +02:00
} else if (weeks > 4) {
2020-06-09 17:58:09 +02:00
return `${months} month${months % 12 > 1 ? 's' : ''}${days % 30 > 0 ? ` ${days % 30} day${days % 30 > 1 ? 's' : ''}` : ''}`;
2020-05-24 23:39:37 +02:00
} else if (days >= 7) {
2020-06-09 17:58:09 +02:00
return `${weeks} week${weeks > 1 ? 's' : ''}${extraDays && days % 7 > 0 ? ` ${days % 7} day${days % 7 > 1 ? 's' : ''}` : ''}`;
2020-05-24 23:39:37 +02:00
} else if (hours >= 24) {
2020-06-09 17:58:09 +02:00
return `${days} day${days > 1 ? 's' : ''}${extraHours && hours % 24 > 0 ? ` ${hours % 24} hour${hours % 24 > 1 ? 's' : ''}` : ''}`;
2020-05-24 23:39:37 +02:00
} else if (minutes >= 60) {
2020-06-09 17:58:09 +02:00
return `${hours} hour${hours > 1 ? 's' : ''}${extraMin && minutes % 60 > 0 ? ` ${minutes % 60} minute${minutes % 60 > 1 ? 's' : ''}` : ''}`;
2020-05-24 23:39:37 +02:00
} else if (diff >= 60) {
return `${minutes} minute${minutes > 1 ? 's' : ''}`;
}
return diff.toFixed() + ` second${diff.toFixed() !== 1 ? 's' : ''}`;
}
resolveBoolean(input) {
input = input.toLowerCase();
const truthy = [ 'on', 'true', 'yes', 'enable', 'y', 't' ];
const falsey = [ 'off', 'false', 'no', 'disable', 'n', 'f' ];
if(truthy.includes(input)) {
return true;
} else if (falsey.includes(input)) {
return false;
}
return null;
}
2020-05-24 00:10:01 +02:00
/**
* Resolves methods used primarily for settings, also deals with appending the arguments into existing lists
*
* @param {Array<String>} args The incoming arguments with the first element being the method ex. ['add','ban','kick']
* @param {Array<String>} valid An array of items to compare to, if an argument doesn't exist in this array it'll be skipped over
* @param {Array<String>} [existing=[]] Existing values in the array, valid elements will be appended to this
2020-06-10 16:48:26 +02:00
* @param {Function} resolver One of the resolver functions used to resolve the passed values into objects (should always be one of the mass resolvers due to the nature of this method, might break otherwise) NOTE: REMEMBER TO BIND 'this' ARG!
* @param {Guild} guild The guild for the resolver to use when resolving
* @param {Boolean} caseSensitive Whether or not the arguments are case sensitive
2020-05-24 00:10:01 +02:00
* @returns {Object}
* @memberof Resolver
*/
async resolveMethod(args, valid, existing = [], resolver, guild, caseSensitive = false) {
2020-05-24 00:10:01 +02:00
const methods = {
2020-05-24 23:39:37 +02:00
list: ['view', 'list', '?', 'show'],
2020-05-24 00:10:01 +02:00
add: ['add', '+'],
2020-06-10 16:48:26 +02:00
set: ['set', '='],
2020-05-24 23:39:37 +02:00
remove: ['remove', 'delete', '-'],
2020-06-20 22:26:02 +02:00
reset: ['clear', 'reset', 'default'],
2020-06-09 17:58:09 +02:00
off: ['off', 'disable', 'false', 'no', 'n', 'f'],
on: ['on', 'enable', 'true', 'yes', 'y', 't']
2020-05-24 00:10:01 +02:00
};
if (!args.length) return false;
// eslint-disable-next-line prefer-const
let [method, ...rest] = args;
method = method.toLowerCase();
2020-05-24 23:39:37 +02:00
let resolved = [];
2020-06-09 17:58:09 +02:00
if (methods.reset.includes(method)) return { method: 'reset' };
2020-06-10 16:48:26 +02:00
if (methods.off.includes(method)) return { method: 'off' };
if (methods.on.includes(method)) return { method: 'on' };
2020-05-24 00:10:01 +02:00
if (!rest.length) {
2020-05-24 23:39:37 +02:00
if (methods.add.includes(method)) return { method: 'add' };
if (methods.remove.includes(method)) return { method: 'remove' };
2020-06-10 16:48:26 +02:00
if (methods.set.includes(method)) return { method: 'set' };
2020-05-24 00:10:01 +02:00
}
if (methods.list.includes(method)) {
2020-05-24 23:39:37 +02:00
2020-06-10 16:48:26 +02:00
if (resolver) resolved = await resolver(existing, false, guild);
2020-05-24 23:39:37 +02:00
return { method: 'list', resolved };
2020-05-24 00:10:01 +02:00
} else if (methods.add.includes(method)) {
2020-05-24 23:39:37 +02:00
2020-05-24 00:10:01 +02:00
const added = [];
for (let elem of rest) {
2020-05-24 23:39:37 +02:00
if (resolver) {
2020-06-10 16:48:26 +02:00
const _resolved = await resolver(elem, false, guild);
2020-05-24 23:39:37 +02:00
if (_resolved) {
if(!resolved.includes(_resolved[0])) resolved.push(_resolved[0]);
elem = _resolved[0].id;
}
}
if(!caseSensitive) elem = elem.toLowerCase();
2020-05-24 00:10:01 +02:00
if (existing.includes(elem) || valid && !valid.includes(elem)) continue;
2020-05-24 23:39:37 +02:00
2020-05-24 00:10:01 +02:00
added.push(elem);
existing.push(elem);
2020-05-24 23:39:37 +02:00
2020-05-24 00:10:01 +02:00
}
2020-05-24 23:39:37 +02:00
return { rest, method: 'add', changed: added, result: existing, resolved };
2020-05-24 00:10:01 +02:00
} else if (methods.remove.includes(method)) {
2020-05-24 23:39:37 +02:00
2020-05-24 00:10:01 +02:00
const removed = [];
for (let elem of rest) {
2020-05-24 23:39:37 +02:00
if (resolver) {
2020-06-10 16:48:26 +02:00
const _resolved = await resolver(elem, false, guild);
2020-05-24 23:39:37 +02:00
if (_resolved) {
if (!resolved.includes(_resolved[0])) resolved.push(_resolved[0]);
elem = _resolved[0].id;
}
}
if (!caseSensitive) elem = elem.toLowerCase();
2020-05-24 00:10:01 +02:00
if (!existing.includes(elem) || removed.includes(elem) || valid && !valid.includes(elem)) continue;
removed.push(elem);
existing.splice(existing.indexOf(elem), 1);
2020-05-24 23:39:37 +02:00
2020-05-24 00:10:01 +02:00
}
2020-05-24 23:39:37 +02:00
return { rest, method: 'remove', changed: removed, result: existing, resolved };
2020-06-10 16:48:26 +02:00
} else if (methods.set.includes(method)) {
const set = [];
for (let elem of rest) {
if (resolver) {
const _resolved = await resolver(elem, false, guild);
if (_resolved) {
if (!resolved.includes(_resolved[0])) resolved.push(_resolved[0]);
elem = _resolved[0].id;
}
}
if (!caseSensitive) elem = elem.toLowerCase();
if (valid && !valid.includes(elem)) continue;
set.push(elem);
2020-06-10 16:48:26 +02:00
}
return { rest, method: 'set', changed: set, result: set, resolved };
2020-05-24 00:10:01 +02:00
}
return false;
}
2020-04-08 18:08:46 +02:00
2020-04-11 10:06:39 +02:00
/**
* Resolve several user resolveables
*
* @param {array<string>} [resolveables=[]] an array of user resolveables (name, id, tag)
* @param {boolean} [strict=false] whether or not to attempt resolving by partial usernames
* @returns {array || boolean} Array of resolved users or false if none were resolved
* @memberof Resolver
*/
2020-04-09 11:18:14 +02:00
async resolveUsers(resolveables = [], strict = false) {
2020-04-08 18:08:46 +02:00
if(typeof resolveables === 'string') resolveables = [ resolveables ];
if(resolveables.length === 0) return false;
const { users } = this.client;
const resolved = [];
2020-04-08 18:08:46 +02:00
for(const resolveable of resolveables) {
2020-04-08 18:08:46 +02:00
if((/<@!?([0-9]{17,21})>/u).test(resolveable)) {
2020-04-08 18:08:46 +02:00
const [, id] = resolveable.match(/<@!?([0-9]{17,21})>/u);
const user = await users.fetch(id).catch((err) => {
if(err.code === 10013) return false;
// this.client.logger.warn(err); return false;
});
2020-04-08 18:08:46 +02:00
if(user) resolved.push(user);
} else if((/(id:)?([0-9]{17,21})/u).test(resolveable)) {
2020-04-08 18:08:46 +02:00
const [,, id] = resolveable.match(/(id:)?([0-9]{17,21})/u);
const user = await users.fetch(id).catch((err) => {
if(err.code === 10013) return false;
// this.client.logger.warn(err); return false;
});
2020-04-08 18:08:46 +02:00
if(user) resolved.push(user);
} else if((/^@?([\S\s]{1,32})#([0-9]{4})/u).test(resolveable)) {
2020-04-08 18:08:46 +02:00
const m = resolveable.match(/^@?([\S\s]{1,32})#([0-9]{4})/u);
const username = m[1].toLowerCase();
const discrim = m[2].toLowerCase();
2020-06-10 16:48:26 +02:00
const user = users.cache.sort((a, b) => a.name.length - b.name.length).filter((u) => u.username.toLowerCase() === username && u.discriminator === discrim).first();
2020-04-08 18:08:46 +02:00
if(user) resolved.push(user);
2020-04-09 11:18:14 +02:00
} else if(!strict) {
const name = resolveable.toLowerCase();
2020-06-10 16:48:26 +02:00
const user = users.cache.sort((a, b) => a.name.length - b.name.length).filter((u) => u.username.toLowerCase().includes(name)).first();
2020-04-09 11:18:14 +02:00
if(user) resolved.push(user);
2020-04-08 18:08:46 +02:00
}
}
return resolved.length ? resolved : false;
2020-04-08 18:08:46 +02:00
}
async resolveUser(resolveable, strict) {
2020-05-22 22:13:47 +02:00
if (!resolveable) return false;
2020-05-24 23:39:37 +02:00
if (resolveable instanceof Array) throw new Error('Resolveable cannot be of type Array, use resolveUsers for resolving arrays of users');
const result = await this.resolveUsers([ resolveable ], strict);
2020-05-17 13:04:58 +02:00
return result ? result[0] : false;
2020-05-22 22:13:47 +02:00
}
2020-04-11 10:06:39 +02:00
/**
* Resolve multiple member resolveables
*
* @param {array<string>} [resolveables=[]] an array of member resolveables (name, nickname, tag, id)
* @param {boolean} [strict=false] whether or not to attempt resolving by partial matches
* @param {Guild} guild the guild in which to look for members
* @returns {array<GuildMember> || boolean} an array of resolved members or false if none were resolved
* @memberof Resolver
*/
async resolveMembers(resolveables = [], strict = false, guild = null) {
2020-04-08 18:08:46 +02:00
if(typeof resolveables === 'string') resolveables = [ resolveables ];
if(resolveables.length === 0) return false;
if(!guild) return false;
const { members } = guild;
const resolved = [];
2020-04-08 18:08:46 +02:00
for(const resolveable of resolveables) {
2020-04-08 18:08:46 +02:00
if((/<@!?([0-9]{17,21})>/u).test(resolveable)) {
2020-04-08 18:08:46 +02:00
const [, id] = resolveable.match(/<@!?([0-9]{17,21})>/u);
const member = await members.fetch(id).catch((err) => {
if(err.code === 10007) return false;
// this.client.logger.warn(err); return false;
});
2020-04-08 18:08:46 +02:00
if(member) resolved.push(member);
} else if((/(id:)?([0-9]{17,21})/u).test(resolveable)) {
2020-04-09 11:18:14 +02:00
const [,, id] = resolveable.match(/(id:)?([0-9]{17,21})/u);
const member = await members.fetch(id).catch((err) => {
if(err.code === 10007) return false;
// this.client.logger.warn(err); return false;
});
2020-04-09 11:18:14 +02:00
if(member) resolved.push(member);
} else if((/^@?([\S\s]{1,32})#([0-9]{4})/u).test(resolveable)) {
2020-04-09 11:18:14 +02:00
const m = resolveable.match(/^@?([\S\s]{1,32})#([0-9]{4})/u);
const username = m[1].toLowerCase();
const discrim = m[2].toLowerCase();
const member = members.cache.filter((m) => m.user.username.toLowerCase() === username && m.user.discriminator === discrim).first();
2020-04-09 11:18:14 +02:00
if(member) resolved.push(member);
} else if((/^@?([\S\s]{1,32})/u).test(resolveable) && guild && !strict) {
2020-04-09 11:18:14 +02:00
const nickname = resolveable.match(/^@?([\S\s]{1,32})/u)[0].toLowerCase();
2020-06-10 16:48:26 +02:00
const member = members.cache.sort((a, b) => a.name.length - b.name.length).filter((m) => m && m.user &&
((!m.nickname ? false : m.nickname.toLowerCase() === nickname) ||
2020-04-09 11:18:14 +02:00
(!m.nickname ? false : m.nickname.toLowerCase().includes(nickname)) ||
m.user.username.toLowerCase().includes(nickname) ||
m.user.username.toLowerCase() === nickname)).first();
2020-04-09 11:18:14 +02:00
if(member) resolved.push(member);
}
2020-04-08 18:08:46 +02:00
}
2020-04-11 10:06:39 +02:00
return resolved.length > 0 ? resolved : false;
}
async resolveMember(resolveable, strict, guild) {
2020-05-22 22:13:47 +02:00
if (!resolveable) return false;
2020-05-24 23:39:37 +02:00
if (resolveable instanceof Array) throw new Error('Resolveable cannot be of type Array, use resolveMembers for resolving arrays of members');
const result = await this.resolveMembers([ resolveable ], strict, guild);
2020-05-17 13:04:58 +02:00
return result ? result[0] : false;
}
2020-04-11 10:06:39 +02:00
/**
2020-04-11 10:11:22 +02:00
* Resolve multiple channels
2020-04-11 10:06:39 +02:00
*
* @param {array<string>} [resolveables=[]] an array of channel resolveables (name, id)
* @param {guild} guild the guild in which to look for channels
* @param {boolean} [strict=false] whether or not partial names are resolved
2020-04-11 10:11:22 +02:00
* @returns {array<GuildChannel> || false} an array of guild channels or false if none were resolved
2020-04-11 10:06:39 +02:00
* @memberof Resolver
*/
async resolveChannels(resolveables = [], strict = false, guild = null, filter = () => true) {
2020-04-11 10:06:39 +02:00
if(typeof resolveables === 'string') resolveables = [ resolveables ];
if(resolveables.length === 0) return false;
if(!guild) return false;
const CM = guild.channels;
const resolved = [];
2020-04-11 10:06:39 +02:00
for(const resolveable of resolveables) {
2020-04-11 10:06:39 +02:00
const channel = CM.resolve(resolveable);
if(channel && filter(channel)) {
2020-04-11 10:06:39 +02:00
resolved.push(channel);
continue;
}
const name = /^#?([a-z0-9\-_0]*)/iu;
const id = /^<?#?([0-9]*)>?/iu;
2020-04-11 10:06:39 +02:00
if (id.test(resolveable)) {
const match = resolveable.match(id);
const [, ch] = match;
2020-07-20 00:42:21 +02:00
const channel = await this.client.channels.fetch(ch).catch((e) => {}); //eslint-disable-line no-empty, no-empty-function, no-unused-vars
if (channel && filter(channel)) resolved.push(channel);
} else if (name.test(resolveable)) {
2020-04-11 10:06:39 +02:00
const match = resolveable.match(name);
const ch = match[1].toLowerCase();
2020-06-10 16:48:26 +02:00
const channel = CM.cache.sort((a, b) => a.name.length - b.name.length).filter(filter).filter((c) => {
if (!strict) return c.name.toLowerCase().includes(ch);
2020-04-11 10:06:39 +02:00
return c.name.toLowerCase() === ch;
2020-06-04 19:59:09 +02:00
}).first();
2020-04-11 10:06:39 +02:00
if(channel) resolved.push(channel);
}
2020-04-11 10:06:39 +02:00
}
return resolved.length > 0 ? resolved : false;
}
2020-05-17 13:04:58 +02:00
async resolveChannel(resolveable, strict, guild, filter) {
2020-05-17 13:04:58 +02:00
2020-05-22 22:13:47 +02:00
if (!resolveable) return false;
2020-05-24 23:39:37 +02:00
if (resolveable instanceof Array) throw new Error('Resolveable cannot be of type Array, use resolveChannels for resolving arrays of channels');
const result = await this.resolveChannels([resolveable], strict, guild, filter);
2020-05-17 13:04:58 +02:00
return result ? result[0] : false;
}
2020-04-11 10:06:39 +02:00
2020-04-11 10:11:22 +02:00
/**
* Resolve multiple roles
*
* @param {array<string>} [resolveables=[]] an array of roles resolveables (name, id)
* @param {Guild} guild the guild in which to look for roles
* @param {boolean} [strict=false] whether or not partial names are resolved
* @returns {array<GuildRole> || false} an array of roles or false if none were resolved
* @memberof Resolver
*/
async resolveRoles(resolveables = [], strict = false, guild = null) {
2020-04-11 10:06:39 +02:00
if(typeof resolveables === 'string') resolveables = [ resolveables ];
if(resolveables.length === 0) return false;
if(!guild) return false;
const { roles } = guild;
const resolved = [];
2020-04-11 10:06:39 +02:00
for(const resolveable of resolveables) {
2020-04-11 10:06:39 +02:00
const id = /^(<@&)?([0-9]{16,22})>?/iu;
2020-04-11 10:06:39 +02:00
if(id.test(resolveable)) {
const match = resolveable.match(id);
const [,, rId] = match;
2020-04-11 10:06:39 +02:00
const role = await roles.fetch(rId).catch(this.client.logger.error);
2020-04-11 10:06:39 +02:00
if(role) resolved.push(role);
} else {
2020-06-10 16:48:26 +02:00
const role = roles.cache.sort((a, b) => a.name.length - b.name.length).filter((r) => {
2020-04-11 10:06:39 +02:00
if(!strict) return r.name.toLowerCase().includes(resolveable.toLowerCase());
return r.name.toLowerCase() === resolveable.toLowerCase();
}).first();
2020-04-11 10:06:39 +02:00
if(role) resolved.push(role);
}
}
return resolved.length > 0 ? resolved : false;
2020-04-09 11:18:14 +02:00
2020-04-08 18:08:46 +02:00
}
2020-05-17 13:04:58 +02:00
async resolveRole(resolveable, strict, guild) {
2020-05-22 22:13:47 +02:00
if (!resolveable) return false;
2020-05-24 23:39:37 +02:00
if (resolveable instanceof Array) throw new Error('Resolveable cannot be of type Array, use resolveRoles for resolving arrays of roles');
const result = await this.resolveRoles([resolveable], strict, guild);
2020-05-17 13:04:58 +02:00
return result ? result[0] : false;
}
2020-06-04 19:59:09 +02:00
resolveTime(string) {
let time = null;
try {
time = timestring(string);
} catch(err) {
return null;
}
return time;
}
async infinite(args = [], resolvers = [], strict, guild) {
let parsed = [], //eslint-disable-line prefer-const
parameters = [];
if(resolvers.length === 0) return { parsed, parameters: args };
for(let i = 0; i < args.length; i++) {
const arg = args[i];
let resolved = null;
for(const resolver of resolvers) {
if(resolved) break;
resolved = resolver(arg, strict, guild);
if(resolved instanceof Promise) resolved = await resolved;
}
if(resolved) {
const ids = parsed.map((p) => p.id);
if(!ids.includes(resolved.id)) parsed.push(resolved);
continue;
} else {
parameters = args.splice(i);
break;
}
}
return { parsed, parameters };
2020-05-17 13:04:58 +02:00
}
2020-04-08 18:08:46 +02:00
}
2020-04-09 16:30:52 +02:00
module.exports = Resolver;
const filterExact = (search) => (comp) => comp.id.toLowerCase() === search ||
2020-04-08 18:08:46 +02:00
comp.resolveable.toLowerCase() === search ||
comp.aliases && (comp.aliases.some((ali) => `${comp.type}:${ali}`.toLowerCase() === search) ||
comp.aliases.some((ali) => ali.toLowerCase() === search));
2020-04-08 18:08:46 +02:00
const filterInexact = (search) => (comp) => comp.id.toLowerCase().includes(search) ||
2020-04-08 18:08:46 +02:00
comp.resolveable.toLowerCase().includes(search) ||
comp.aliases && (comp.aliases.some((ali) => `${comp.type}:${ali}`.toLowerCase().includes(search)) ||
comp.aliases.some((ali) => ali.toLowerCase().includes(search)));