velvox
/
litecord
mirror of https://gitlab.com/litecord/litecord.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

webhooks: add WEBHOOKS_UPDATE event dispatch

This commit is contained in:
Luna 2019-03-18 18:00:27 -03:00
parent 8c43bdcee7
commit 3866b1f2bc
1 changed files with 27 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
litecord/blueprints/webhooks.py
View File

@ -129,6 +129,14 @@ async def webhook_token_check(webhook_id: int, webhook_token: str):
return row['guild_id'], row['channel_id'] return row['guild_id'], row['channel_id']
async def _dispatch_webhook_update(guild_id: int, channel_id):
await app.dispatcher.dispatch('guild', guild_id, 'WEBHOOKS_UPDATE', {
'guild_id': str(guild_id),
'channel_id': str(channel_id)
})
@bp.route('/channels/<int:channel_id>/webhooks', methods=['POST']) @bp.route('/channels/<int:channel_id>/webhooks', methods=['POST'])
async def create_webhook(channel_id: int): async def create_webhook(channel_id: int):
"""Create a webhook given a channel.""" """Create a webhook given a channel."""
@ -161,6 +169,7 @@ async def create_webhook(channel_id: int):
j['name'], webhook_icon.icon_hash, token j['name'], webhook_icon.icon_hash, token
) )
await _dispatch_webhook_update(guild_id, channel_id)
return jsonify(await get_webhook(webhook_id)) return jsonify(await get_webhook(webhook_id))
@ -235,13 +244,20 @@ async def modify_webhook(webhook_id: int):
raise ChannelNotFound('cant assign webhook to channel') raise ChannelNotFound('cant assign webhook to channel')
await _update_webhook(webhook_id, j) await _update_webhook(webhook_id, j)
return jsonify(await get_webhook(webhook_id))
webhook = await get_webhook(webhook_id)
# we don't need to cast channel_id to int since that isn't
# used in the dispatcher call
await _dispatch_webhook_update(guild_id, webhook['channel_id'])
return jsonify(webhook)
@bp.route('/webhooks/<int:webhook_id>/<webhook_token>', methods=['PATCH']) @bp.route('/webhooks/<int:webhook_id>/<webhook_token>', methods=['PATCH'])
async def modify_webhook_tokened(webhook_id, webhook_token): async def modify_webhook_tokened(webhook_id, webhook_token):
"""Modify a webhook, using its token.""" """Modify a webhook, using its token."""
await webhook_token_check(webhook_id, webhook_token) guild_id, channel_id = await webhook_token_check(
webhook_id, webhook_token)
# forcefully pop() the channel id out of the schema # forcefully pop() the channel id out of the schema
# instead of making another, for simplicity's sake # instead of making another, for simplicity's sake
@ -249,11 +265,14 @@ async def modify_webhook_tokened(webhook_id, webhook_token):
WEBHOOK_UPDATE.pop('channel_id')) WEBHOOK_UPDATE.pop('channel_id'))
await _update_webhook(webhook_id, j) await _update_webhook(webhook_id, j)
await _dispatch_webhook_update(guild_id, channel_id)
return jsonify(await get_webhook(webhook_id, secure=False)) return jsonify(await get_webhook(webhook_id, secure=False))
async def delete_webhook(webhook_id: int): async def delete_webhook(webhook_id: int):
"""Delete a webhook.""" """Delete a webhook."""
webhook = await get_webhook(webhook_id)
res = await app.db.execute(""" res = await app.db.execute("""
DELETE FROM webhooks DELETE FROM webhooks
WHERE id = $1 WHERE id = $1
@ -262,6 +281,12 @@ async def delete_webhook(webhook_id: int):
if res.lower() == 'delete 0': if res.lower() == 'delete 0':
raise WebhookNotFound() raise WebhookNotFound()
# only casting the guild id since that's whats used
# on the dispatcher call.
await _dispatch_webhook_update(
int(webhook['guild_id']), webhook['channel_id']
)
@bp.route('/webhooks/<int:webhook_id>', methods=['DELETE']) @bp.route('/webhooks/<int:webhook_id>', methods=['DELETE'])
async def del_webhook(webhook_id): async def del_webhook(webhook_id):