add: custom schema changes

Add:

- `torrents.deleted_at`
- `users.is_donor`
- `users.is_lifetime`

Co-authored-by: HDVinnie <hdinnovations@protonmail.com>

Author: Roardom

src/announce.rs

@@ -358,8 +358,9 @@ pub async fn announce(
             _ => (),
         }
 
-        // Make sure user isn't leeching more torrents than their group allows
+        // Make sure user isn't leeching more torrents than their group allows unless they are a lifetime user
         let has_hit_download_slot_limit = queries.left > 0
+            && !user.is_lifetime
             && matches!(group.download_slots, Some(slots) if slots <= user.num_leeching);
 
         // Change of upload/download compared to previous announce
@@ -662,6 +663,7 @@ pub async fn announce(
             .featured_torrents
             .read()
             .contains(&FeaturedTorrent { torrent_id })
+        || user.is_donor
     {
         0
     } else {
@@ -724,7 +726,7 @@ pub async fn announce(
         queries.downloaded,
         queries.left == 0,
         queries.event != Event::Stopped,
-        group.is_immune,
+        group.is_immune || user.is_donor,
         completed_at,
     );
 

src/scheduler/torrent_update.rs

@@ -155,6 +155,7 @@ impl Queue {
             .await
             .map(|result| result.rows_affected());
 
+        /*
         if rows_affected_res.is_ok() {
             // When torrent records are upserted after they have been deleted
             // since the last flush interval, the record gets unintentionally
@@ -166,6 +167,7 @@ impl Queue {
             .execute(db)
             .await;
         }
+        */
 
         rows_affected_res
     }

src/tracker/torrent.rs

@@ -76,6 +76,8 @@ impl Map {
                     0 as `is_deleted: bool`
                 FROM
                     torrents
+                WHERE
+                    torrents.deleted_at IS NULL
             "#
         )
         .fetch_all(db)

src/tracker/torrent/infohash2id.rs

@@ -36,6 +36,8 @@ impl Map {
                     torrents.info_hash as `info_hash: InfoHash`
                 FROM
                     torrents
+                WHERE
+                    torrents.deleted_at IS NULL
             "#
         )
         .fetch_all(db)

src/tracker/user.rs

@@ -36,7 +36,9 @@ impl Map {
                     users.passkey as `passkey: Passkey`,
                     users.can_download as `can_download: bool`,
                     CAST(COALESCE(SUM(peers.seeder = 1 AND peers.active = 1 AND peers.visible = 1), 0) AS UNSIGNED) as `num_seeding: u32`,
-                    CAST(COALESCE(SUM(peers.seeder = 0 AND peers.active = 1 AND peers.visible = 1), 0) AS UNSIGNED) as `num_leeching: u32`
+                    CAST(COALESCE(SUM(peers.seeder = 0 AND peers.active = 1 AND peers.visible = 1), 0) AS UNSIGNED) as `num_leeching: u32`,
+                    users.is_donor as `is_donor: bool`,
+                    users.is_lifetime as `is_lifetime: bool`
                 FROM
                     users
                 LEFT JOIN
@@ -78,6 +80,8 @@ impl Map {
                     can_download: user.can_download,
                     num_seeding: user.num_seeding,
                     num_leeching: user.num_leeching,
+                    is_donor: user.is_donor,
+                    is_lifetime: user.is_lifetime,
                 },
             );
 
@@ -140,6 +144,8 @@ pub struct User {
     pub can_download: bool,
     pub num_seeding: u32,
     pub num_leeching: u32,
+    pub is_donor: bool,
+    pub is_lifetime: bool,
 }
 
 #[derive(Clone, Deserialize, Hash)]
@@ -150,6 +156,8 @@ pub struct APIInsertUser {
     pub can_download: bool,
     pub num_seeding: u32,
     pub num_leeching: u32,
+    pub is_donor: bool,
+    pub is_lifetime: bool,
 }
 
 #[derive(Clone, Deserialize, Hash)]