From abc597de9572cf03972f9f6aff0c74e5f230c720 Mon Sep 17 00:00:00 2001
From: trisua <me@trisua.com>
Date: Sat, 17 May 2025 20:02:55 -0400
Subject: [PATCH] add: infinite post drafts for supporters

---
 .../src/public/html/communities/create_post.html  |  3 ++-
 crates/app/src/public/html/profile/settings.html  |  1 +
 crates/core/src/database/drafts.rs                | 15 +++++++++++++++
 crates/core/src/database/stacks.rs                |  5 +++--
 4 files changed, 21 insertions(+), 3 deletions(-)

diff --git a/crates/app/src/public/html/communities/create_post.html b/crates/app/src/public/html/communities/create_post.html
index a09da7b..92e8db3 100644
--- a/crates/app/src/public/html/communities/create_post.html
+++ b/crates/app/src/public/html/communities/create_post.html
@@ -333,7 +333,8 @@
         </div>
 
         <div class="card flex flex-col gap-2">
-            {% for draft in drafts %}
+            {{ components::supporter_ad(body="Become a supporter to save
+            infinite post drafts!") }} {% for draft in drafts %}
             <div class="card-nest">
                 <div class="card small flex flex-col gap-2">
                     <span class="no_p_margin"
diff --git a/crates/app/src/public/html/profile/settings.html b/crates/app/src/public/html/profile/settings.html
index d24444d..47a34ad 100644
--- a/crates/app/src/public/html/profile/settings.html
+++ b/crates/app/src/public/html/profile/settings.html
@@ -537,6 +537,7 @@
                                 <li>Ability to upload and use gif emojis</li>
                                 <li>Create infinite stack timelines</li>
                                 <li>Ability to upload images to posts</li>
+                                <li>Save infinite post drafts</li>
                             </ul>
 
                             <a
diff --git a/crates/core/src/database/drafts.rs b/crates/core/src/database/drafts.rs
index 239d035..d50d157 100644
--- a/crates/core/src/database/drafts.rs
+++ b/crates/core/src/database/drafts.rs
@@ -83,6 +83,8 @@ impl DataManager {
         Ok(res.unwrap())
     }
 
+    const MAXIMUM_FREE_DRAFTS: usize = 10;
+
     /// Create a new post draft in the database.
     ///
     /// # Arguments
@@ -95,6 +97,19 @@ impl DataManager {
             return Err(Error::DataTooLong("content".to_string()));
         }
 
+        // check number of stacks
+        let owner = self.get_user_by_id(data.owner).await?;
+
+        if !owner.permissions.check(FinePermission::SUPPORTER) {
+            let stacks = self.get_stacks_by_owner(data.owner).await?;
+
+            if stacks.len() >= Self::MAXIMUM_FREE_DRAFTS {
+                return Err(Error::MiscError(
+                    "You already have the maximum number of drafts you can have".to_string(),
+                ));
+            }
+        }
+
         // ...
         let conn = match self.connect().await {
             Ok(c) => c,
diff --git a/crates/core/src/database/stacks.rs b/crates/core/src/database/stacks.rs
index 6eac313..c214f38 100644
--- a/crates/core/src/database/stacks.rs
+++ b/crates/core/src/database/stacks.rs
@@ -117,6 +117,8 @@ impl DataManager {
         Ok(res.unwrap())
     }
 
+    const MAXIMUM_FREE_STACKS: usize = 5;
+
     /// Create a new stack in the database.
     ///
     /// # Arguments
@@ -134,9 +136,8 @@ impl DataManager {
 
         if !owner.permissions.check(FinePermission::SUPPORTER) {
             let stacks = self.get_stacks_by_owner(data.owner).await?;
-            let maximum_count = 5;
 
-            if stacks.len() >= maximum_count {
+            if stacks.len() >= Self::MAXIMUM_FREE_STACKS {
                 return Err(Error::MiscError(
                     "You already have the maximum number of stacks you can have".to_string(),
                 ));