feat: Allow disabling episode count

Author: lostb1t

Makefile

@@ -29,7 +29,7 @@ docker-run:
 
 
 run:
-	REPLEX_TMDB_API_KEY=0d73e0cb91f39e670b0efa6913afbd58 REPLEX_ENABLE_CONSOLE=0 REPLEX_CACHE_TTL=0 REPLEX_HOST=https://46-4-30-217.01b0839de64b49138531cab1bf32f7c2.plex.direct:42405 RUST_LOG="info,replex=info" cargo watch -x run
+	REPLEX_DISABLE_LEAF_COUNT=0 REPLEX_DISABLE_USER_STATE=1 REPLEX_TMDB_API_KEY=0d73e0cb91f39e670b0efa6913afbd58 REPLEX_ENABLE_CONSOLE=0 REPLEX_CACHE_TTL=0 REPLEX_HOST=https://46-4-30-217.01b0839de64b49138531cab1bf32f7c2.plex.direct:42405 RUST_LOG="info,replex=info" cargo watch -x run
 
 # run:
 # 	REPLEX_ENABLE_CONSOLE=0 REPLEX_CACHE_TTL=0 REPLEX_HOST=https://46-4-30-217.01b0839de64b49138531cab1bf32f7c2.plex.direct:42405 RUST_LOG="info" cargo run

README.md

@@ -10,7 +10,8 @@ Remix your plex recommendations.
 - Remove watched items from recommendations.
 - Choose between styles, shelf (default) or hero.
 - Auto load tmdb artwork for hero styles.
-- Disable user state: unwatched and episode count banners are removed from the artwork.
+- Disable user state: remove unwatched labels from artwork.
+- Disable leaf count: remove episode count from artwork.
 - Works on every client/app not only plex web!
 - Plays nice with PMM (and without).
 
@@ -72,7 +73,8 @@ Settings are set via [environment variables](https://kinsta.com/knowledgebase/wh
 | REPLEX_HOST              |        	| Plex target host to proxy                                             	  |
 | REPLEX_INCLUDE_WATCHED   | false    | If set to false, hide watched items.                                      |
 | REPLEX_CACHE_TTL         | 300    	| Time to live for caches in seconds. Set to 0 to disable            	      |
-| REPLEX_DISABLE_USER_STATE| false    | Disable user state, unwatched and episode count banners are removed from the artwork  |
+| REPLEX_DISABLE_USER_STATE| false    | Unwatched and episode count banners will be removec from artwork. *does currently not work on IOS/AppleTV |
+| REPLEX_DISABLE_LEAF_COUNT| false    | remove episode count label from show artwork                              |
 | REPLEX_TMDB_API_KEY      |     	    | Enables tmdb artwork for hero hubs instead of plex background artwork     |
 | REPLEX_SSL_ENABLE        | false    | Enable automatic SSL generation. http will be disabled                    |
 |                          |          | (stored in /data/acme/letsencrypt so make sure to mount a volume)         |

src/config.rs

@@ -35,6 +35,11 @@ pub struct Config {
         deserialize_with = "figment::util::bool_from_str_or_int"
     )]
     pub disable_user_state: bool,
+    #[serde(
+        default = "default_as_false",
+        deserialize_with = "figment::util::bool_from_str_or_int"
+    )]
+    pub disable_leaf_count: bool,
 }
 
 fn default_cache_ttl() -> u64 {

src/main.rs

@@ -72,8 +72,7 @@ async fn main() {
     //         dbg!("we are being runned");
     //     }
     // });
-    let version =
-        env::var("CARGO_PKG_VERSION").unwrap_or("unknown".to_string());
+    let version = env!("CARGO_PKG_VERSION");
     tracing::info!("Replex version {}", version);
 
     let router = route();

src/transform.rs

@@ -691,15 +691,25 @@ impl Transform for UserStateTransform {
         options: PlexParams,
     ) {
         let config: Config = Config::figment().extract().unwrap();
-        if !config.disable_user_state {
+        if !config.disable_user_state && !config.disable_leaf_count {
             return
         }
         if item.is_hub() {
             for child in item.children_mut() {
-                child.user_state = Some(false);
+                if config.disable_user_state {
+                    child.user_state = Some(false);
+                }
+                if config.disable_leaf_count {
+                    child.leaf_count = None;
+                }
             }
         } else {
-            item.user_state = Some(false);
+            if config.disable_user_state {
+                item.user_state = Some(false);
+            }
+            if config.disable_leaf_count {
+                item.leaf_count = None;
+            }
         }
     }
 }