From ce77e926a518f7f657e3c7ea4e9ba0c6b7d4d567 Mon Sep 17 00:00:00 2001
From: Paul Makles <me@insrt.uk>
Date: Fri, 3 Mar 2023 20:14:30 +0000
Subject: [PATCH] fix(safety): implement fetch_snapshot fully

---
 Cargo.lock                                       | 6 +++---
 crates/bonfire/Cargo.toml                        | 2 +-
 crates/delta/Cargo.toml                          | 2 +-
 crates/delta/src/routes/safety/fetch_snapshot.rs | 9 +++++++--
 crates/quark/Cargo.toml                          | 2 +-
 5 files changed, 13 insertions(+), 8 deletions(-)

diff --git a/Cargo.lock b/Cargo.lock
index 96da304c..24f86ce5 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -2792,7 +2792,7 @@ dependencies = [
 
 [[package]]
 name = "revolt-bonfire"
-version = "0.5.12"
+version = "0.5.13"
 dependencies = [
  "async-std",
  "async-tungstenite",
@@ -2808,7 +2808,7 @@ dependencies = [
 
 [[package]]
 name = "revolt-delta"
-version = "0.5.12"
+version = "0.5.13"
 dependencies = [
  "async-channel",
  "async-std",
@@ -2848,7 +2848,7 @@ dependencies = [
 
 [[package]]
 name = "revolt-quark"
-version = "0.5.12"
+version = "0.5.13"
 dependencies = [
  "async-lock",
  "async-recursion",
diff --git a/crates/bonfire/Cargo.toml b/crates/bonfire/Cargo.toml
index 2a870f86..4ed4b6bd 100644
--- a/crates/bonfire/Cargo.toml
+++ b/crates/bonfire/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "revolt-bonfire"
-version = "0.5.12"
+version = "0.5.13"
 license = "AGPL-3.0-or-later"
 edition = "2021"
 
diff --git a/crates/delta/Cargo.toml b/crates/delta/Cargo.toml
index 05e9a17c..d6a3377c 100644
--- a/crates/delta/Cargo.toml
+++ b/crates/delta/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "revolt-delta"
-version = "0.5.12"
+version = "0.5.13"
 license = "AGPL-3.0-or-later"
 authors = ["Paul Makles <paulmakles@gmail.com>"]
 edition = "2018"
diff --git a/crates/delta/src/routes/safety/fetch_snapshot.rs b/crates/delta/src/routes/safety/fetch_snapshot.rs
index 81b8bef2..7d32bcf2 100644
--- a/crates/delta/src/routes/safety/fetch_snapshot.rs
+++ b/crates/delta/src/routes/safety/fetch_snapshot.rs
@@ -44,8 +44,13 @@ pub async fn fetch_snapshot(
             user_ids.insert(&message.author);
             channel_ids.insert(&message.channel);
         }
-        _ => {
-            todo!()
+        SnapshotContent::User(user) => {
+            user_ids.insert(&user.id);
+        }
+        SnapshotContent::Server(server) => {
+            for channel in &server.channels {
+                channel_ids.insert(channel);
+            }
         }
     }
 
diff --git a/crates/quark/Cargo.toml b/crates/quark/Cargo.toml
index e7279117..2d3e37d9 100644
--- a/crates/quark/Cargo.toml
+++ b/crates/quark/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "revolt-quark"
-version = "0.5.12"
+version = "0.5.13"
 license = "AGPL-3.0-or-later"
 edition = "2021"