channel: create channel.pins blueprint

- schemas: add CHAN_UPDATE and CHAN_OVERWRITE

litecord/blueprints/channel/__init__.py

@@ -1,2 +1,3 @@
 from .messages import bp as channel_messages
 from .reactions import bp as channel_reactions
+from .pins import bp as channel_pins

litecord/blueprints/channel/pins.py

@@ -0,0 +1,93 @@
+from quart import Blueprint, current_app as app, request, jsonify
+
+from litecord.auth import token_check
+from litecord.blueprints.checks import channel_check
+from litecord.snowflake import snowflake_datetime
+
+bp = Blueprint('channel_pins', __name__)
+
+
+@bp.route('/<int:channel_id>/pins', methods=['GET'])
+async def get_pins(channel_id):
+    """Get the pins for a channel"""
+    user_id = await token_check()
+    await channel_check(user_id, channel_id)
+
+    ids = await app.db.fetch("""
+    SELECT message_id
+    FROM channel_pins
+    WHERE channel_id = $1
+    ORDER BY message_id ASC
+    """, channel_id)
+
+    ids = [r['message_id'] for r in ids]
+    res = []
+
+    for message_id in ids:
+        message = await app.storage.get_message(message_id)
+        if message is not None:
+            res.append(message)
+
+    return jsonify(res)
+
+
+@bp.route('/<int:channel_id>/pins/<int:message_id>', methods=['PUT'])
+async def add_pin(channel_id, message_id):
+    """Add a pin to a channel"""
+    user_id = await token_check()
+    _ctype, guild_id = await channel_check(user_id, channel_id)
+
+    # TODO: check MANAGE_MESSAGES permission
+
+    await app.db.execute("""
+    INSERT INTO channel_pins (channel_id, message_id)
+    VALUES ($1, $2)
+    """, channel_id, message_id)
+
+    row = await app.db.fetchrow("""
+    SELECT message_id
+    FROM channel_pins
+    WHERE channel_id = $1
+    ORDER BY message_id ASC
+    LIMIT 1
+    """, channel_id)
+
+    timestamp = snowflake_datetime(row['message_id'])
+
+    await app.dispatcher.dispatch_guild(guild_id, 'CHANNEL_PINS_UPDATE', {
+        'channel_id': str(channel_id),
+        'last_pin_timestamp': timestamp.isoformat()
+    })
+
+    return '', 204
+
+
+@bp.route('/<int:channel_id>/pins/<int:message_id>', methods=['DELETE'])
+async def delete_pin(channel_id, message_id):
+    user_id = await token_check()
+    _ctype, guild_id = await channel_check(user_id, channel_id)
+
+    # TODO: check MANAGE_MESSAGES permission
+
+    await app.db.execute("""
+    DELETE FROM channel_pins
+    WHERE channel_id = $1 AND message_id = $2
+    """, channel_id, message_id)
+
+    row = await app.db.fetchrow("""
+    SELECT message_id
+    FROM channel_pins
+    WHERE channel_id = $1
+    ORDER BY message_id ASC
+    LIMIT 1
+    """, channel_id)
+
+    timestamp = snowflake_datetime(row['message_id'])
+
+    await app.dispatcher.dispatch(
+        'channel', channel_id, 'CHANNEL_PINS_UPDATE', {
+            'channel_id': str(channel_id),
+            'last_pin_timestamp': timestamp.isoformat()
+        })
+
+    return '', 204

litecord/blueprints/channels.py

@@ -4,7 +4,6 @@ from quart import Blueprint, request, current_app as app, jsonify
 from logbook import Logger
 
 from ..auth import token_check
-from ..snowflake import snowflake_datetime
 from ..enums import ChannelType, GUILD_CHANS
 from ..errors import ChannelNotFound
 
@@ -214,90 +213,6 @@ async def close_channel(channel_id):
     raise ChannelNotFound()
 
 
-@bp.route('/<int:channel_id>/pins', methods=['GET'])
-async def get_pins(channel_id):
-    user_id = await token_check()
-    await channel_check(user_id, channel_id)
-
-    ids = await app.db.fetch("""
-    SELECT message_id
-    FROM channel_pins
-    WHERE channel_id = $1
-    ORDER BY message_id ASC
-    """, channel_id)
-
-    ids = [r['message_id'] for r in ids]
-    res = []
-
-    for message_id in ids:
-        message = await app.storage.get_message(message_id)
-        if message is not None:
-            res.append(message)
-
-    return jsonify(res)
-
-
-@bp.route('/<int:channel_id>/pins/<int:message_id>', methods=['PUT'])
-async def add_pin(channel_id, message_id):
-    user_id = await token_check()
-    _ctype, guild_id = await channel_check(user_id, channel_id)
-
-    # TODO: check MANAGE_MESSAGES permission
-
-    await app.db.execute("""
-    INSERT INTO channel_pins (channel_id, message_id)
-    VALUES ($1, $2)
-    """, channel_id, message_id)
-
-    row = await app.db.fetchrow("""
-    SELECT message_id
-    FROM channel_pins
-    WHERE channel_id = $1
-    ORDER BY message_id ASC
-    LIMIT 1
-    """, channel_id)
-
-    timestamp = snowflake_datetime(row['message_id'])
-
-    await app.dispatcher.dispatch_guild(guild_id, 'CHANNEL_PINS_UPDATE', {
-        'channel_id': str(channel_id),
-        'last_pin_timestamp': timestamp.isoformat()
-    })
-
-    return '', 204
-
-
-@bp.route('/<int:channel_id>/pins/<int:message_id>', methods=['DELETE'])
-async def delete_pin(channel_id, message_id):
-    user_id = await token_check()
-    _ctype, guild_id = await channel_check(user_id, channel_id)
-
-    # TODO: check MANAGE_MESSAGES permission
-
-    await app.db.execute("""
-    DELETE FROM channel_pins
-    WHERE channel_id = $1 AND message_id = $2
-    """, channel_id, message_id)
-
-    row = await app.db.fetchrow("""
-    SELECT message_id
-    FROM channel_pins
-    WHERE channel_id = $1
-    ORDER BY message_id ASC
-    LIMIT 1
-    """, channel_id)
-
-    timestamp = snowflake_datetime(row['message_id'])
-
-    await app.dispatcher.dispatch(
-        'channel', channel_id, 'CHANNEL_PINS_UPDATE', {
-            'channel_id': str(channel_id),
-            'last_pin_timestamp': timestamp.isoformat()
-        })
-
-    return '', 204
-
-
 @bp.route('/<int:channel_id>/typing', methods=['POST'])
 async def trigger_typing(channel_id):
     user_id = await token_check()

litecord/schemas.py

@@ -251,6 +251,57 @@ GUILD_UPDATE = {
 }
 
 
+CHAN_OVERWRITE = {
+    'type': 'dict',
+    'schema': {
+        'id': {'coerce': int},
+        'type': {'type': 'string', 'allowed': ['role', 'member']},
+        'allow': {'coerce': Permissions},
+        'deny': {'coerce': Permissions}
+    }
+}
+
+
+CHAN_UPDATE = {
+    'name': {
+        'type': 'string', 'minlength': 2,
+        'maxlength': 100, 'required': False},
+
+    'position': {'coerce': int, 'required': False},
+
+    'topic': {
+        'type': 'string', 'minlength': 0,
+        'maxlength': 1024, 'required': False},
+
+    'nsfw': {'type': 'boolean', 'required': False},
+    'rate_limit_per_user': {
+        'coerce': int, 'min': 0,
+        'max': 120, 'required': False},
+
+    'bitrate': {
+        'coerce': int, 'min': 8000,
+
+        # NOTE: 'max' is 96000 for non-vip guilds
+        'max': 128000, 'required': False},
+
+    'user_limit': {
+        # user_limit being 0 means infinite.
+        'coerce': int, 'min': 0,
+        'max': 99, 'required': False
+    },
+
+    'permission_overwrites': {
+        'type': 'list',
+        'schema': CHAN_OVERWRITE,
+        'required': False
+    },
+
+    'parent_id': {'coerce': int, 'required': False, 'nullable': True}
+
+
+}
+
+
 ROLE_CREATE = {
     'name': {'type': 'string', 'default': 'new role'},
     'permissions': {'coerce': Permissions, 'nullable': True},

run.py

@@ -25,7 +25,7 @@ from litecord.blueprints.guild import (
 )
 
 from litecord.blueprints.channel import (
-    channel_messages, channel_reactions
+    channel_messages, channel_reactions, channel_pins
 )
 
 from litecord.ratelimits.handler import ratelimit_handler
@@ -79,6 +79,7 @@ bps = {
     channels: '/channels',
     channel_messages: '/channels',
     channel_reactions: '/channels',
+    channel_pins: '/channels',
 
     webhooks: None,
     science: None,