1 files changed, 143 insertions, 0 deletions
diff --git a/pysite/views/api/bot/user_complete.py b/pysite/views/api/bot/user_complete.py
new file mode 100644
index 00000000..877eee34
--- /dev/null
+++ b/pysite/views/api/bot/user_complete.py
@@ -0,0 +1,143 @@
+import logging
+
+from flask import jsonify, request
+
+from pysite.base_route import APIView
+from pysite.constants import ErrorCodes, ValidationTypes
+from pysite.decorators import api_key, api_params
+from pysite.mixins import DBMixin
+
+
+BANNABLE_STATES = ("preparing", "running")
+
+log = logging.getLogger(__name__)
+
+
+class UserView(APIView, DBMixin):
+    path = "/bot/users/complete"
+    name = "bot.users.complete"
+
+    chunks_table = "member_chunks"
+    infractions_table = "code_jam_infractions"
+    jams_table = "code_jams"
+    oauth_table_name = "oauth_data"
+    participants_table = "code_jam_participants"
+    responses_table = "code_jam_responses"
+    table_name = "users"
+    teams_table = "code_jam_teams"
+
+    @api_key
+    @api_params(validation_type=ValidationTypes.none)
+    def post(self, _):
+        log.debug(f"Size of request: {len(request.data)} bytes")
+
+        documents = self.db.get_all(self.chunks_table)
+        chunks = []
+
+        for doc in documents:
+            log.info(f"Got member chunk with {len(doc['chunk'])} users")
+            chunks.append(doc["chunk"])
+
+            self.db.delete(self.chunks_table, doc["id"], durability="soft")
+        self.db.sync(self.chunks_table)
+
+        log.info(f"Got {len(chunks)} member chunks")
+
+        data = []
+
+        for chunk in chunks:
+            data += chunk
+
+        log.info(f"Got {len(data)} members")
+
+        if not data:
+            return self.error(ErrorCodes.bad_data_format, "No users supplied")
+
+        deletions = 0
+        oauth_deletions = 0
+        profile_deletions = 0
+        response_deletions = 0
+        bans = 0
+
+        user_ids = [user["user_id"] for user in data]
+
+        all_users = self.db.run(self.db.query(self.table_name), coerce=list)
+
+        for user in all_users:
+            if user["user_id"] not in user_ids:
+                self.db.delete(self.table_name, user["user_id"], durability="soft")
+                deletions += 1
+
+        all_oauth_data = self.db.run(self.db.query(self.oauth_table_name), coerce=list)
+
+        for item in all_oauth_data:
+            if item["snowflake"] not in user_ids:
+                user_id = item["snowflake"]
+
+                oauth_deletions += self.db.delete(
+                    self.oauth_table_name, item["id"], durability="soft", return_changes=True
+                ).get("deleted", 0)
+                profile_deletions += self.db.delete(
+                    self.participants_table, user_id, durability="soft", return_changes=True
+                ).get("deleted", 0)
+
+                banned = False
+                responses = self.db.run(
+                    self.db.query(self.responses_table).filter({"snowflake": user_id}),
+                    coerce=list
+                )
+
+                for response in responses:
+                    jam = response["jam"]
+                    jam_obj = self.db.get(self.jams_table, jam)
+
+                    if jam_obj:
+                        if jam_obj["state"] in BANNABLE_STATES:
+                            banned = True
+
+                    self.db.delete(self.responses_table, response["id"], durability="soft")
+                    response_deletions += 1
+
+                teams = self.db.run(
+                    self.db.query(self.teams_table).filter(lambda row: row["members"].contains(user_id)),
+                    coerce=list
+                )
+
+                for team in teams:
+                    team["members"].remove(user_id)
+
+                    self.db.insert(self.teams_table, team, conflict="replace", durability="soft")
+
+                if banned:
+                    self.db.insert(
+                        self.infractions_table, {
+                            "participant": user_id,
+                            "reason": "Automatic ban: Removed jammer profile in the middle of a code jam",
+                            "number": -1,
+                            "decremented_for": []
+                        }, durability="soft"
+                    )
+                    bans += 1
+
+        del user_ids
+
+        changes = self.db.insert(
+            self.table_name, *data,
+            conflict="update",
+            durability="soft"
+        )
+
+        self.db.sync(self.infractions_table)
+        self.db.sync(self.oauth_table_name)
+        self.db.sync(self.participants_table)
+        self.db.sync(self.responses_table)
+        self.db.sync(self.table_name)
+        self.db.sync(self.teams_table)
+
+        changes["deleted"] = deletions
+        changes["deleted_oauth"] = oauth_deletions
+        changes["deleted_jam_profiles"] = profile_deletions
+        changes["deleted_responses"] = response_deletions
+        changes["jam_bans"] = bans
+
+        return jsonify(changes)  # pragma: no cover