content: generateReleaseInfoListenLine: consider album, too

1 files changed, 23 insertions, 12 deletions

diff --git a/src/content/dependencies/generateReleaseInfoListenLine.js b/src/content/dependencies/generateReleaseInfoListenLine.js
index 9b1901c3..8cc2c984 100644
--- a/src/content/dependencies/generateReleaseInfoListenLine.js
+++ b/src/content/dependencies/generateReleaseInfoListenLine.js
@@ -43,25 +43,36 @@ export default {
         .filter(url => url.hostname.endsWith('.bandcamp.com'))
         .map(url => url.hostname);
 
-    let releaseContexts =
-      thing.urls.map(urlString => {
-        const url = new URL(urlString);
+    const getReleaseContext = urlString => {
+      const url = new URL(urlString);
+
+      if (url.hostname === 'homestuck.bandcamp.com') {
+        return 'officialRelease';
+      }
 
-        if (url.hostname === 'homestuck.bandcamp.com') {
-          return 'officialRelease';
-        }
+      if (artistBandcampDomains.includes(url.hostname)) {
+        return 'artistRelease';
+      }
 
-        if (artistBandcampDomains.includes(url.hostname)) {
-          return 'artistRelease';
-        }
+      return null;
+    };
 
-        return null;
-      });
+    let releaseContexts =
+      thing.urls.map(getReleaseContext);
+
+    const albumReleaseContexts =
+      query.album.urls.map(getReleaseContext);
 
     const presentReleaseContexts =
       unique(releaseContexts.filter(Boolean));
 
-    if (presentReleaseContexts.length <= 1) {
+    const presentAlbumReleaseContexts =
+      unique(albumReleaseContexts.filter(Boolean));
+
+    if (
+      presentReleaseContexts.length <= 1 &&
+      presentAlbumReleaseContexts.length <= 1
+    ) {
       releaseContexts =
         thing.urls.map(() => null);
     }