From 31396f33284f543c7fb73cf7cea1c573484cc5fd Mon Sep 17 00:00:00 2001
From: Razmig Sarkissian <razmig@thomasbutton.com>
Date: Tue, 12 Mar 2019 12:27:37 +0100
Subject: [PATCH] clean up fix for tournamentFeatures usecount

---
 .../pokeranalytics/android/model/LiveData.kt  | 20 +++++++++++++++----
 1 file changed, 16 insertions(+), 4 deletions(-)

diff --git a/app/src/main/java/net/pokeranalytics/android/model/LiveData.kt b/app/src/main/java/net/pokeranalytics/android/model/LiveData.kt
index a6979295..bcb1b64f 100644
--- a/app/src/main/java/net/pokeranalytics/android/model/LiveData.kt
+++ b/app/src/main/java/net/pokeranalytics/android/model/LiveData.kt
@@ -40,10 +40,22 @@ enum class LiveData : Localizable {
 	fun setUseCount(realm: Realm, realmResults: RealmResults<*>) {
 		realm.executeTransaction {
 			realmResults.forEach { countableUsage ->
-				(countableUsage as CountableUsage).useCount = it.where<Session>().equalTo(
-					"${relatedEntity.simpleName.decapitalize()}.id",
-					countableUsage.uniqueIdentifier()
-				).count().toInt()
+
+				when (this) {
+					TOURNAMENT_FEATURE -> {
+						(countableUsage as CountableUsage).useCount = it.where<Session>().contains(
+							"tournamentFeatures.id",
+							countableUsage.uniqueIdentifier()
+						).count().toInt()
+					}
+					else -> {
+						(countableUsage as CountableUsage).useCount = it.where<Session>().equalTo(
+							"${relatedEntity.simpleName.decapitalize()}.id",
+							countableUsage.uniqueIdentifier()
+						).count().toInt()
+					}
+				}
+
 			}
 		}
 	}