Added WishlistOnly filter

#23
Citrinate · Feb 25, 2024 · 3819e5f · 3819e5f
1 parent 39eb291
commit 3819e5f
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 0 deletions.
diff --git a/FreePackages/Data/FilterConfig.cs b/FreePackages/Data/FilterConfig.cs
@@ -55,6 +55,9 @@ internal sealed class FilterConfig {
 		[JsonProperty(Required = Required.Default)]
 		internal HashSet<string> Systems = new();
 
+		[JsonProperty(Required = Required.Default)]
+		internal bool WishlistOnly = false;
+
 		[JsonConstructor]
 		internal FilterConfig() { }
 	}

diff --git a/FreePackages/Data/UserData.cs b/FreePackages/Data/UserData.cs
@@ -21,6 +21,12 @@ internal sealed class UserData {
 		[JsonProperty(PropertyName = "rgExcludedContentDescriptorIDs", Required = Required.Always)]
 		internal HashSet<uint> ExcludedContentDescriptorIDs = new();
 
+		[JsonProperty(PropertyName = "rgWishlist", Required = Required.Always)]
+		internal HashSet<uint> WishlistedApps = new();
+
+		[JsonProperty(PropertyName = "rgFollowedApps", Required = Required.Always)]
+		internal HashSet<uint> FollowedApps = new();
+
 		[JsonExtensionData]
 		internal Dictionary<string, JToken> AdditionalData = new();
 

diff --git a/FreePackages/Handlers/PackageFilter.cs b/FreePackages/Handlers/PackageFilter.cs
@@ -83,6 +83,10 @@ internal bool IsRedeemableApp(FilterableApp app, HashSet<uint>? includedAppIDs =
 		}
 
 		internal bool IsAppWantedByFilter(FilterableApp app, FilterConfig filter) {
+			if (UserData == null) {
+				throw new InvalidOperationException(nameof(UserData));
+			}
+
 			if (filter.Types.Count > 0 && app.Type != EAppType.Beta && !app.HasType(filter.Types)) {
 				// Don't require user to specify they want playtests (Beta), this is already implied by the PlaytestMode filter
 				// App isn't a wanted type
@@ -115,6 +119,11 @@ internal bool IsAppWantedByFilter(FilterableApp app, FilterConfig filter) {
 				return false;
 			}
 
+			if (filter.WishlistOnly && !UserData.WishlistedApps.Contains(app.ID) && !UserData.FollowedApps.Contains(app.ID)) {
+				// Unwated to due not being wishlisted or followed on the Steam storefront
+				return false;
+			}
+
 			return true;
 		}