management.py

import operator

import PIL.Image
import PIL.ImageDraw
import PIL.ImageFont

import config

def join(cmd):
	guild_id, role_id = _ids(cmd)
	if config.bot.roles is None or guild_id != config.bot.roles['server']:
		return
	roles = cmd.bot.guilds[guild_id].roles
	if role_id is None or cmd.args not in _allowed_role_names(roles):
		cmd.reply('no joinable role named %s' % cmd.args)
	else:
		cmd.bot.post('/guilds/%s/members/%s/roles/%s' % (guild_id, cmd.sender['id'], role_id), None,
				method='PUT')
		cmd.reply('put <@!%s> in %s' % (cmd.sender['id'], cmd.args))

def leave(cmd):
	guild_id, role_id = _ids(cmd)
	if config.bot.roles is None or guild_id != config.bot.roles['server']:
		return
	roles = cmd.bot.guilds[guild_id].roles
	if role_id is None or cmd.args not in _allowed_role_names(roles):
		cmd.reply('no joinable role named %s' % cmd.args)
	else:
		cmd.bot.post('/guilds/%s/members/%s/roles/%s' % (guild_id, cmd.sender['id'], role_id), None,
				method='DELETE')
		cmd.reply('removed <@!%s> from %s' % (cmd.sender['id'], cmd.args))

def list_roles(cmd):
	bot = cmd.bot
	guild_id = bot.channels[cmd.channel_id]
	if config.bot.roles is None or guild_id != config.bot.roles['server']:
		return

	roles = list(_allowed_roles(bot.guilds[guild_id].roles))
	roles.sort(key=operator.itemgetter('position'), reverse=True)

	img = PIL.Image.new('RGB', (200, 18 * len(roles) + 2), '#202225')
	font = PIL.ImageFont.truetype('whitney500.ttf', 16)
	draw = PIL.ImageDraw.Draw(img)
	offset = 0
	for role in roles:
		color = role['color']
		if color == 0:
			fill = (153, 170, 181)
		else:
			r = color >> 16
			g = (color >> 8) & 0xff
			b = color & 0xff
			fill = (r, g, b)
		draw.text((0, offset), role['name'], fill=fill, font=font)
		offset += 18
	with open(config.bot.roles['image_path'], 'wb') as f:
		img.save(f, 'png')

	embed = {'image': {'url': config.bot.roles['public_url']}}
	cmd.reply('', embed)

def _ids(cmd):
	bot = cmd.bot
	guild_id = bot.channels[cmd.channel_id]
	roles = bot.guilds[guild_id].roles
	try:
		role_id = roles[cmd.args]['id']
		return guild_id, role_id
	except KeyError:
		return guild_id, None

def _allowed_roles(roles):
	sbot_position = roles['sbot']['position']
	for role in roles.values():
		# exclude roles higher than ours, @everyone (position 0), and bots
		if 0 < role['position'] < sbot_position and role['name'] != 'bots':
			yield role

def _allowed_role_names(roles):
	for role in _allowed_roles(roles):
		yield role['name']