From 96d3d11906e801fa3505bac149f46465d6094cff Mon Sep 17 00:00:00 2001
From: Dylan Jeffers <dylan@audius.co>
Date: Fri, 10 Feb 2023 23:46:16 -0800
Subject: [PATCH] [C-2107] Add local saves to offline queue (#2824)

---
 .../sagas/requestDownloadAllFavoritesSaga.ts  | 56 +++++++++++++------
 1 file changed, 40 insertions(+), 16 deletions(-)

diff --git a/packages/mobile/src/store/offline-downloads/sagas/requestDownloadAllFavoritesSaga.ts b/packages/mobile/src/store/offline-downloads/sagas/requestDownloadAllFavoritesSaga.ts
index ce88fd52c6..47fe10ae08 100644
--- a/packages/mobile/src/store/offline-downloads/sagas/requestDownloadAllFavoritesSaga.ts
+++ b/packages/mobile/src/store/offline-downloads/sagas/requestDownloadAllFavoritesSaga.ts
@@ -1,4 +1,8 @@
-import { accountSelectors, getContext } from '@audius/common'
+import {
+  accountSelectors,
+  getContext,
+  savedPageSelectors
+} from '@audius/common'
 import { takeEvery, select, call, put } from 'typed-redux-saga'
 
 import { getAccountCollections } from 'app/screens/favorites-screen/selectors'
@@ -9,6 +13,8 @@ import { addOfflineItems, requestDownloadAllFavorites } from '../slice'
 
 const { getUserId } = accountSelectors
 
+const { getLocalSaves } = savedPageSelectors
+
 export function* requestDownloadAllFavoritesSaga() {
   yield* takeEvery(requestDownloadAllFavorites.type, downloadAllFavorites)
 }
@@ -25,31 +31,49 @@ function* downloadAllFavorites() {
     metadata: { reasons_for_download: [{ is_from_favorites: true }] }
   })
 
+  const trackReasonsForDownload = [
+    { is_from_favorites: true, collection_id: DOWNLOAD_REASON_FAVORITES }
+  ]
+
+  // Add local saves
+  const now = Date.now().toString()
+  const localSaves = yield* select(getLocalSaves)
+  const localSavesToAdd: OfflineItem[] = Object.keys(localSaves)
+    .map((id) => parseInt(id, 10))
+    .map((id) => ({
+      type: 'track',
+      id,
+      metadata: {
+        favorite_created_at: now,
+        reasons_for_download: trackReasonsForDownload
+      }
+    }))
+
+  offlineItemsToAdd.push(...localSavesToAdd)
+
+  // Add favorited tracks from api
   const apiClient = yield* getContext('apiClient')
   const allFavoritedTracks = yield* call([apiClient, apiClient.getFavorites], {
     currentUserId,
     limit: 10000
   })
 
-  if (!allFavoritedTracks) return
+  if (allFavoritedTracks) {
+    for (const favoritedTrack of allFavoritedTracks) {
+      const { save_item_id: trackId, created_at } = favoritedTrack
 
-  for (const favoritedTrack of allFavoritedTracks) {
-    const { save_item_id: trackId, created_at } = favoritedTrack
-    const downloadReason = {
-      is_from_favorites: true,
-      collection_id: DOWNLOAD_REASON_FAVORITES
+      offlineItemsToAdd.push({
+        type: 'track',
+        id: trackId,
+        metadata: {
+          favorite_created_at: created_at,
+          reasons_for_download: trackReasonsForDownload
+        }
+      })
     }
-
-    offlineItemsToAdd.push({
-      type: 'track',
-      id: trackId,
-      metadata: {
-        favorite_created_at: created_at,
-        reasons_for_download: [downloadReason]
-      }
-    })
   }
 
+  // Add favorited collections and their tracks
   const favoritedCollections = yield* select(getAccountCollections)
 
   for (const favoritedCollection of favoritedCollections) {