From 010ff9407b3c9f9b2342ace57df42bf606d37240 Mon Sep 17 00:00:00 2001
From: Oreo <oreothehyena@gmail.com>
Date: Sun, 7 Jul 2024 00:45:08 -0700
Subject: [PATCH] Add e621 flags object to SeachResult site_info

---
 fuzzysearch-api/src/models.rs   |  7 ++++++-
 fuzzysearch-common/src/types.rs | 11 +++++++++++
 2 files changed, 17 insertions(+), 1 deletion(-)

diff --git a/fuzzysearch-api/src/models.rs b/fuzzysearch-api/src/models.rs
index 70ed1ad..9e510ee 100644
--- a/fuzzysearch-api/src/models.rs
+++ b/fuzzysearch-api/src/models.rs
@@ -91,6 +91,7 @@ pub async fn image_query(
             hashes.searched_hash,
             hashes.distance,
             submission.file_sha256 sha256
+            null flags
         FROM hashes
         JOIN submission ON hashes.found_hash = submission.hash_int
         JOIN artist ON submission.artist_id = artist.id
@@ -109,7 +110,8 @@ pub async fn image_query(
             to_timestamp(data->>'created_at', 'YYYY-MM-DD"T"HH24:MI:SS"Z"') posted_at,
             hashes.searched_hash,
             hashes.distance,
-            e621.sha256
+            e621.sha256,
+            e621.data->>'flags' flags
         FROM hashes
         JOIN e621 ON hashes.found_hash = e621.hash
         WHERE e621.hash IN (SELECT hashes.found_hash)
@@ -128,6 +130,7 @@ pub async fn image_query(
             hashes.searched_hash,
             hashes.distance,
             weasyl.sha256
+            null flags
         FROM hashes
         JOIN weasyl ON hashes.found_hash = weasyl.hash
         WHERE weasyl.hash IN (SELECT hashes.found_hash)
@@ -149,6 +152,7 @@ pub async fn image_query(
             hashes.searched_hash,
             hashes.distance,
             null sha256
+            null flags
         FROM hashes
         JOIN tweet_media ON hashes.found_hash = tweet_media.hash
         JOIN tweet ON tweet_media.tweet_id = tweet.id
@@ -164,6 +168,7 @@ pub async fn image_query(
             },
             Some("e621") => SiteInfo::E621 {
                 sources: row.sources,
+                flags: row.flags,
             },
             Some("Twitter") => SiteInfo::Twitter,
             Some("Weasyl") => SiteInfo::Weasyl,
diff --git a/fuzzysearch-common/src/types.rs b/fuzzysearch-common/src/types.rs
index ed967f8..e2bdeac 100644
--- a/fuzzysearch-common/src/types.rs
+++ b/fuzzysearch-common/src/types.rs
@@ -54,6 +54,16 @@ pub struct SearchResult {
     pub searched_hash: Option<i64>,
 }
 
+#[derive(Clone, Debug, Deserialize, Serialize)]
+pub struct E621Flags {
+    pending: bool,
+    flagged: bool,
+    note_locked: bool,
+    status_locked: bool,
+    rating_locked: bool,
+    deleted: bool,
+}
+
 #[derive(Clone, Debug, Deserialize, Serialize)]
 #[serde(tag = "site", content = "site_info")]
 pub enum SiteInfo {
@@ -63,6 +73,7 @@ pub enum SiteInfo {
     #[serde(rename = "e621")]
     E621 {
         sources: Option<Vec<String>>,
+        flags: Option<E621Flags>,
     },
     Twitter,
     Weasyl,