From ad9b3d21663ae51fe04cd59e835bb9a77ffaa51f Mon Sep 17 00:00:00 2001
From: Ske <voltasalt@gmail.com>
Date: Sun, 15 Jul 2018 21:09:55 +0200
Subject: [PATCH] Add data export command

---
 bot/pluralkit/commands.py | 48 ++++++++++++++++++++++++++++++++++++++-
 1 file changed, 47 insertions(+), 1 deletion(-)

diff --git a/bot/pluralkit/commands.py b/bot/pluralkit/commands.py
index 1657af26..947d586f 100644
--- a/bot/pluralkit/commands.py
+++ b/bot/pluralkit/commands.py
@@ -1,5 +1,7 @@
 from datetime import datetime
+import io
 import itertools
+import json
 import os
 import re
 from urllib.parse import urlparse
@@ -627,4 +629,48 @@ async def invite_link(conn, message, args):
 
     url = oauth_url(client_id, permissions)
     logger.debug("Sending invite URL: {}".format(url))
-    return True, url
\ No newline at end of file
+    return True, url
+
+@command(cmd="export", description="Exports system data to a machine-readable format.")
+async def export(conn, message, args):
+    system = await db.get_system_by_account(conn, message.author.id)
+
+    if system is None:
+        return False, "No system is registered to this account."
+
+    members = await db.get_all_members(conn, system["id"])
+    accounts = await db.get_linked_accounts(conn, system["id"])
+    switches = await get_front_history(conn, system["id"], 999999)
+
+    data = {
+        "name": system["name"],
+        "hid": system["hid"],
+        "description": system["description"],
+        "tag": system["tag"],
+        "avatar_url": system["avatar_url"],
+        "created": system["created"].isoformat(),
+        "members": [
+            {
+                "name": member["name"],
+                "hid": member["hid"],
+                "color": member["color"],
+                "avatar_url": member["avatar_url"],
+                "birthday": member["birthday"].isoformat() if member["birthday"] else None,
+                "pronouns": member["pronouns"],
+                "description": member["description"],
+                "prefix": member["prefix"],
+                "suffix": member["suffix"],
+                "created": member["created"].isoformat()
+            } for member in members
+        ],
+        "accounts": [str(uid) for uid in accounts],
+        "switches": [
+            {
+                "timestamp": timestamp.isoformat(),
+                "members": [member["hid"] for member in members]
+            } for timestamp, members in switches
+        ]
+    }
+
+    f = io.BytesIO(json.dumps(data).encode("utf-8"))
+    await client.send_file(message.channel, f, filename="system.json")