# © lambda#0987 <lambda@lambda.dance>
# SPDX-License-Identifier: AGPL-3.0-or-later

import functools
from discord.ext.commands import BadArgument

_emote_type_predicates = {
	'all': lambda _: True,
	'static': lambda e: not e.animated,
	'animated': lambda e: e.animated}

# this is kind of a hack to ensure that the last argument is always converted, even if the default is used.
def emote_type_filter_default(command):
	old_callback = command.callback

	@functools.wraps(old_callback)
	async def callback(self, ctx, *args):
		image_type = args[-1]
		try:
			image_type = _emote_type_predicates[image_type]
		except KeyError:
			raise BadArgument(f'Invalid emote type. Specify one of "all", "static", or "animated".')
		return await old_callback(self, ctx, *args[:-1], image_type)

	command.callback = callback
	return command