refactor(media): collection to common template

1 files changed, 26 insertions, 30 deletions

diff --git a/src/lib/AniList/media.ts b/src/lib/AniList/media.ts
index 0dfcecf6..7fd52645 100644
--- a/src/lib/AniList/media.ts
+++ b/src/lib/AniList/media.ts
@@ -63,6 +63,30 @@ export const flattenLists = (lists: { name: string; entries: { media: Media }[]
 	return Array.from(new Set(flattenedList.filter((media) => !ignoredMediaIds.includes(media.id))));
 };
 
+const collectionQueryTemplate = (type: Type, userId: number, includeCompleted: boolean) =>
+	`{
+	MediaListCollection(
+		userId: ${userId},
+		type: ${type === Type.Anime ? 'ANIME' : 'MANGA'}
+		${includeCompleted ? '' : ', status_not_in: [ COMPLETED ]'}) {
+		lists {
+			name entries {
+				media {
+					id status type episodes chapters format
+					title { romaji english native }
+					nextAiringEpisode { episode timeUntilAiring }
+					mediaListEntry {
+						progress progressVolumes status
+						score(format: POINT_100) startedAt { year }
+					}
+					startDate { year }
+					coverImage { extraLarge }
+				}
+			}
+		}
+	}
+}`;
+
 export const mediaListCollection = async (
 	anilistAuthorisation: AniListAuthorisation,
 	userIdentity: UserIdentity,
@@ -115,22 +139,7 @@ export const mediaListCollection = async (
 				Accept: 'application/json'
 			},
 			body: JSON.stringify({
-				query: `{ MediaListCollection(userId: ${userIdentity.id}, type: ${
-					type === Type.Anime ? 'ANIME' : 'MANGA'
-				}${includeCompleted ? '' : ', status_not_in: [ COMPLETED ]'}) {
-            lists { name entries { media {
-              id status type episodes chapters format
-							title { romaji english native }
-              nextAiringEpisode { episode timeUntilAiring }
-              mediaListEntry {
-								progress progressVolumes status
-								score(format: POINT_100) startedAt { year }
-							}
-              startDate { year }
-							coverImage { extraLarge }
-            } } }
-          }
-        }`
+				query: collectionQueryTemplate(type, userIdentity.id, includeCompleted)
 			})
 		})
 	).json();
@@ -159,20 +168,7 @@ export const publicMediaListCollection = async (userId: number, type: Type): Pro
 				Accept: 'application/json'
 			},
 			body: JSON.stringify({
-				query: `{ MediaListCollection(userId: ${userId}, type: ${
-					type === Type.Anime ? 'ANIME' : 'MANGA'
-				}, status_not_in: [ COMPLETED ]) { lists { name entries { media {
-						id
-						status
-						type
-						episodes
-						format
-						title { romaji english native }
-						nextAiringEpisode { episode timeUntilAiring }
-						mediaListEntry { progress status }
-						startDate { year }
-          } } } }
-        }`
+				query: collectionQueryTemplate(type, userId, false)
 			})
 		})
 	).json();