From 63ab8ea0648b57fcb5af2827e6817998e6108483 Mon Sep 17 00:00:00 2001
From: Luna <git@l4.pm>
Date: Sun, 1 Sep 2019 17:31:24 -0300
Subject: [PATCH] add messages.flags column

---
 litecord/storage.py                                  | 5 ++++-
 manage/cmd/migration/scripts/3_add_message_flags.sql | 2 ++
 2 files changed, 6 insertions(+), 1 deletion(-)
 create mode 100644 manage/cmd/migration/scripts/3_add_message_flags.sql

diff --git a/litecord/storage.py b/litecord/storage.py
index 58c7bd6..31d7da2 100644
--- a/litecord/storage.py
+++ b/litecord/storage.py
@@ -876,7 +876,7 @@ class Storage:
         row = await self.fetchrow_with_json("""
         SELECT id::text, channel_id::text, author_id, content,
             created_at AS timestamp, edited_at AS edited_timestamp,
-            tts, mention_everyone, nonce, message_type, embeds
+            tts, mention_everyone, nonce, message_type, embeds, flags
         FROM messages
         WHERE id = $1
         """, message_id)
@@ -961,6 +961,9 @@ class Storage:
         if guild_id:
             res['guild_id'] = str(guild_id)
 
+        if res['flags'] is None:
+            res.pop('flags')
+
         return res
 
     async def get_invite(self, invite_code: str) -> Optional[Dict]:
diff --git a/manage/cmd/migration/scripts/3_add_message_flags.sql b/manage/cmd/migration/scripts/3_add_message_flags.sql
new file mode 100644
index 0000000..319ccde
--- /dev/null
+++ b/manage/cmd/migration/scripts/3_add_message_flags.sql
@@ -0,0 +1,2 @@
+ALTER TABLE messages
+    ADD COLUMN flags bigint DEFAULT NULL;