forked from engineering/cloudservices
113 lines
6.6 KiB
TypeScript
113 lines
6.6 KiB
TypeScript
import moment from 'moment';
|
|
import { Message, GuildTextableChannel, Member, Role } from 'eris';
|
|
import { Client, Command, RichEmbed } from '../class';
|
|
import { dataConversion } from '../functions';
|
|
import { AccountInterface } from '../models';
|
|
|
|
export default class Whois extends Command {
|
|
constructor(client: Client) {
|
|
super(client);
|
|
this.name = 'whois';
|
|
this.description = 'Gets information about an account.';
|
|
this.usage = `${this.client.config.prefix}whois <username | user ID>`;
|
|
this.aliases = ['account', 'user'];
|
|
this.enabled = true;
|
|
}
|
|
|
|
public fullRoles = ['662163685439045632', '701454780828221450'];
|
|
|
|
public IP_REGEX = /((^\s*((([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])\.){3}([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5]))\s*$)|(^\s*((([0-9A-Fa-f]{1,4}:){7}([0-9A-Fa-f]{1,4}|:))|(([0-9A-Fa-f]{1,4}:){6}(:[0-9A-Fa-f]{1,4}|((25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)(\.(25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)){3})|:))|(([0-9A-Fa-f]{1,4}:){5}(((:[0-9A-Fa-f]{1,4}){1,2})|:((25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)(\.(25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)){3})|:))|(([0-9A-Fa-f]{1,4}:){4}(((:[0-9A-Fa-f]{1,4}){1,3})|((:[0-9A-Fa-f]{1,4})?:((25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)(\.(25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)){3}))|:))|(([0-9A-Fa-f]{1,4}:){3}(((:[0-9A-Fa-f]{1,4}){1,4})|((:[0-9A-Fa-f]{1,4}){0,2}:((25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)(\.(25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)){3}))|:))|(([0-9A-Fa-f]{1,4}:){2}(((:[0-9A-Fa-f]{1,4}){1,5})|((:[0-9A-Fa-f]{1,4}){0,3}:((25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)(\.(25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)){3}))|:))|(([0-9A-Fa-f]{1,4}:){1}(((:[0-9A-Fa-f]{1,4}){1,6})|((:[0-9A-Fa-f]{1,4}){0,4}:((25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)(\.(25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)){3}))|:))|(:(((:[0-9A-Fa-f]{1,4}){1,7})|((:[0-9A-Fa-f]{1,4}){0,5}:((25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)(\.(25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)){3}))|:)))(%.+)?\s*$))/g
|
|
|
|
public async run(message: Message<GuildTextableChannel>, args: string[]) {
|
|
try {
|
|
let full = false;
|
|
let account: AccountInterface;
|
|
if (args[1] === '--full' && this.fullRoles.some((r) => message.member.roles.includes(r))) full = true;
|
|
|
|
const user = args[0] || message.author.id;
|
|
if (full) account = await this.client.db.Account.findOne({ $or: [{ username: user }, { userID: user }, { emailAddress: user }, { supportKey: user.toUpperCase() }] });
|
|
else account = await this.client.db.Account.findOne({ $or: [{ username: user }, { userID: user }] });
|
|
if (!account) return this.error(message.channel, 'Account not found.');
|
|
|
|
const thumbnail = this.client.users.get(account.userID)?.avatarURL || message.channel.guild.iconURL;
|
|
|
|
const embed = new RichEmbed();
|
|
embed.setTitle('Account Information');
|
|
embed.setThumbnail(thumbnail);
|
|
if (full) await this.full(account, embed, message.member);
|
|
else await this.default(account, embed);
|
|
|
|
let details = '';
|
|
let role: Role;
|
|
if (account.locked) details += '__This account is currently locked.__\n';
|
|
switch (true) {
|
|
case account.permissions.director:
|
|
details += 'This account belongs to a Director.\n';
|
|
role = message.member.guild.roles.get('662163685439045632');
|
|
break;
|
|
case account.permissions.technician:
|
|
details += 'This account belongs to a Technician.\n';
|
|
role = message.member.guild.roles.get('701454780828221450');
|
|
break;
|
|
case account.permissions.staff:
|
|
details += 'This account belongs to a Staff member.\n';
|
|
role = message.member.guild.roles.get('453689940140883988');
|
|
break;
|
|
default:
|
|
role = message.member.guild.roles.get(message.member.guild.id);
|
|
break;
|
|
}
|
|
if (account.root) details += '**This account has root/administrative privileges.**\n';
|
|
embed.setColor(role.color || 0x36393f);
|
|
if (details) embed.addField('Additional Details', details, true);
|
|
|
|
embed.setTimestamp();
|
|
return message.channel.createMessage({ embed });
|
|
} catch (error) {
|
|
return this.client.util.handleError(error, message, this);
|
|
}
|
|
}
|
|
|
|
public async full(account: AccountInterface, embed: RichEmbed, member: Member) {
|
|
const [cpuUsage, data, fingerInformation, chage, memory] = await Promise.all([
|
|
this.client.util.exec(`top -b -n 1 -u ${account.username} | awk 'NR>7 { sum += $9; } END { print sum; }'`),
|
|
this.client.redis.get(`storage-${account.username}`),
|
|
this.client.util.exec(`finger ${account.username}`),
|
|
this.client.util.exec(`chage -l ${account.username}`),
|
|
this.client.util.exec(`memory ${account.username}`),
|
|
]);
|
|
const finger = !member.roles.includes('662163685439045632') ? fingerInformation.replace(this.IP_REGEX, '[MASKED IP ADDRRESS]') : fingerInformation;
|
|
|
|
embed.setDescription(`${finger}\n${chage}`);
|
|
embed.addField('Username', `${account.username} | <@${account.userID}>`, true);
|
|
embed.addField('ID', account.userID, true);
|
|
embed.addField('Email Address', account.emailAddress, true);
|
|
embed.addField('Tier', String(account.tier), true);
|
|
embed.addField('Support Key', account.supportKey, true);
|
|
embed.addField('Created By', `<@${this.client.users.get(account.createdBy).id}>`, true);
|
|
embed.addField('Created At', moment(account.createdAt).format('dddd, MMMM Do YYYY, h:mm:ss A'), true);
|
|
embed.addField('CPU Usage', `${cpuUsage.split('\n')[0] || '0'}%`, true);
|
|
embed.addField('Memory', dataConversion(Number(memory) * 1000), true);
|
|
embed.addField('Storage', data ? dataConversion(Number(data)) : 'N/A', true);
|
|
}
|
|
|
|
public async default(account: AccountInterface, embed: RichEmbed) {
|
|
const [cpuUsage, data, memory] = await Promise.all([
|
|
this.client.util.exec(`top -b -n 1 -u ${account.username} | awk 'NR>7 { sum += $9; } END { print sum; }'`),
|
|
this.client.redis.get(`storage-${account.username}`),
|
|
this.client.util.exec(`memory ${account.username}`),
|
|
]);
|
|
|
|
embed.addField('Username', `${account.username} | <@${account.userID}>`, true);
|
|
embed.addField('ID', account.userID, true);
|
|
embed.addField('Tier', String(account.tier), true);
|
|
embed.addField('Created By', `<@${this.client.users.get(account.createdBy).id}>`, true);
|
|
embed.addField('Created At', moment(account.createdAt).format('dddd, MMMM Do YYYY, h:mm:ss A'), true);
|
|
embed.addField('CPU Usage', `${cpuUsage.split('\n')[0] || '0'}%`, true);
|
|
embed.addField('Memory', dataConversion(Number(memory) * 1000), true);
|
|
embed.addField('Storage', data ? dataConversion(Number(data)) : 'N/A', true);
|
|
}
|
|
}
|
|
|
|
// Whois user only includes username, id, tier, created by, created at, cpu usage, memory, storage and additional details
|