From 80f945555ccff5666609013afc4d1a84f1e9d3cf Mon Sep 17 00:00:00 2001
From: Claudio Ortolina <cloud8421@gmail.com>
Date: Wed, 13 Nov 2024 11:53:45 +0000
Subject: [PATCH] Add status for each track in the scrobble activity

---
 lib/music_library/collection.ex               | 10 ++++++
 lib/music_library/wishlist.ex                 | 10 ++++++
 .../live/stats_live/index.ex                  | 10 ++++++
 .../live/stats_live/index.html.heex           | 35 ++++++++++++++++++-
 priv/gettext/default.pot                      | 20 ++++++++---
 5 files changed, 79 insertions(+), 6 deletions(-)

diff --git a/lib/music_library/collection.ex b/lib/music_library/collection.ex
index 55f200c1..0a294fa0 100644
--- a/lib/music_library/collection.ex
+++ b/lib/music_library/collection.ex
@@ -46,6 +46,16 @@ defmodule MusicLibrary.Collection do
     Repo.one!(q)
   end
 
+  def collected_release_ids(release_ids) do
+    q =
+      from r in fragment("records, json_each(records.release_ids)"),
+        where: r.value in ^release_ids,
+        where: fragment("records.purchased_at IS NOT NULL"),
+        select: r.value
+
+    Repo.all(q)
+  end
+
   defp base_search do
     from r in Record,
       where: not is_nil(r.purchased_at)
diff --git a/lib/music_library/wishlist.ex b/lib/music_library/wishlist.ex
index 5bca2e45..b1fb9669 100644
--- a/lib/music_library/wishlist.ex
+++ b/lib/music_library/wishlist.ex
@@ -20,6 +20,16 @@ defmodule MusicLibrary.Wishlist do
     Repo.aggregate(base_search(), :count)
   end
 
+  def wishlisted_release_ids(release_ids) do
+    q =
+      from r in fragment("records, json_each(records.release_ids)"),
+        where: r.value in ^release_ids,
+        where: fragment("records.purchased_at IS NULL"),
+        select: r.value
+
+    Repo.all(q)
+  end
+
   defp base_search do
     from r in Record,
       where: is_nil(r.purchased_at)
diff --git a/lib/music_library_web/live/stats_live/index.ex b/lib/music_library_web/live/stats_live/index.ex
index 13835280..065c8326 100644
--- a/lib/music_library_web/live/stats_live/index.ex
+++ b/lib/music_library_web/live/stats_live/index.ex
@@ -20,6 +20,14 @@ defmodule MusicLibraryWeb.StatsLive.Index do
 
     recent_tracks = LastFm.Feed.all()
 
+    release_ids =
+      recent_tracks
+      |> Enum.map(fn t -> t.album.musicbrainz_id end)
+      |> Enum.uniq()
+
+    collected_release_ids = Collection.collected_release_ids(release_ids)
+    wishlisted_release_ids = Wishlist.wishlisted_release_ids(release_ids)
+
     if connected?(socket) do
       LastFm.Feed.subscribe()
     end
@@ -37,6 +45,8 @@ defmodule MusicLibraryWeb.StatsLive.Index do
        collection_count: collection_count,
        wishlist_count: wishlist_count,
        latest_record: latest_record,
+       collected_release_ids: collected_release_ids,
+       wishlisted_release_ids: wishlisted_release_ids,
        nav_section: :stats
      )}
   end
diff --git a/lib/music_library_web/live/stats_live/index.html.heex b/lib/music_library_web/live/stats_live/index.html.heex
index 608def8a..baaa89d1 100644
--- a/lib/music_library_web/live/stats_live/index.html.heex
+++ b/lib/music_library_web/live/stats_live/index.html.heex
@@ -100,7 +100,40 @@
             </div>
           </div>
 
-          <div class="relative flex-none">
+          <span
+            :if={track.album.musicbrainz_id in @collected_release_ids}
+            class={[
+              "inline-flex items-center rounded-md",
+              "px-2 py-1 text-xs font-medium",
+              "ring-1 ring-inset",
+              "bg-green-50 dark:bg-green-500/10",
+              "text-green-700 dark:text-green-400",
+              "ring-green-600/20 dark:ring-green-500/20"
+            ]}
+          >
+            <%= gettext("Collected") %>
+          </span>
+          <span
+            :if={track.album.musicbrainz_id in @wishlisted_release_ids}
+            class={[
+              "inline-flex items-center rounded-md",
+              "px-2 py-1 text-xs font-medium",
+              "ring-1 ring-inset",
+              "bg-yellow-50 dark:bg-yellow-400/10",
+              "text-yellow-800 dark:text-yellow-500",
+              "ring-yellow-600/20 dark:ring-yellow-400/20"
+            ]}
+          >
+            <%= gettext("Wishlisted") %>
+          </span>
+
+          <div
+            :if={
+              track.album.musicbrainz_id not in @collected_release_ids and
+                track.album.musicbrainz_id not in @wishlisted_release_ids
+            }
+            class="relative flex-none"
+          >
             <button
               type="button"
               class="text-zinc-500 hover:text-zinc-900 dark:text-zinc-400 dark:hover:text-zinc-300"
diff --git a/priv/gettext/default.pot b/priv/gettext/default.pot
index a89b14cc..268b80f1 100644
--- a/priv/gettext/default.pot
+++ b/priv/gettext/default.pot
@@ -82,8 +82,8 @@ msgid "Edit"
 msgstr ""
 
 #: lib/music_library_web/live/record_live/index.ex:121
-#: lib/music_library_web/live/stats_live/index.ex:60
-#: lib/music_library_web/live/stats_live/index.ex:66
+#: lib/music_library_web/live/stats_live/index.ex:70
+#: lib/music_library_web/live/stats_live/index.ex:76
 #: lib/music_library_web/live/wishlist_live/index.ex:117
 #: lib/music_library_web/live/wishlist_live/index.ex:124
 #, elixir-autogen, elixir-format
@@ -177,7 +177,7 @@ msgid "No results"
 msgstr ""
 
 #: lib/music_library_web/live/record_live/index.html.heex:89
-#: lib/music_library_web/live/stats_live/index.html.heex:112
+#: lib/music_library_web/live/stats_live/index.html.heex:145
 #: lib/music_library_web/live/wishlist_live/index.html.heex:89
 #, elixir-autogen, elixir-format
 msgid "Open options"
@@ -210,7 +210,7 @@ msgid "Purchased on"
 msgstr ""
 
 #: lib/music_library_web/live/record_live/index.ex:112
-#: lib/music_library_web/live/stats_live/index.ex:52
+#: lib/music_library_web/live/stats_live/index.ex:62
 #: lib/music_library_web/live/wishlist_live/index.ex:109
 #, elixir-autogen, elixir-format
 msgid "Record imported successfully"
@@ -273,7 +273,7 @@ msgid "Something went wrong!"
 msgstr ""
 
 #: lib/music_library_web/components/layouts/app.html.heex:7
-#: lib/music_library_web/live/stats_live/index.ex:34
+#: lib/music_library_web/live/stats_live/index.ex:42
 #, elixir-autogen, elixir-format
 msgid "Stats"
 msgstr ""
@@ -371,3 +371,13 @@ msgstr ""
 #, elixir-autogen, elixir-format
 msgid "Choose which format to import"
 msgstr ""
+
+#: lib/music_library_web/live/stats_live/index.html.heex:114
+#, elixir-autogen, elixir-format
+msgid "Collected"
+msgstr ""
+
+#: lib/music_library_web/live/stats_live/index.html.heex:127
+#, elixir-autogen, elixir-format
+msgid "Wishlisted"
+msgstr ""