#106 changes for the code

santiagoparradev · Feb 21, 2024 · e0aa76f · e0aa76f
1 parent cf61521
commit e0aa76f
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 12 deletions.
diff --git a/force-app/main/default/classes/entities/TrackResponse.cls b/force-app/main/default/classes/entities/TrackResponse.cls
@@ -2,8 +2,19 @@ public class TrackResponse {
 
     public List<ItemTrack> items;
     public Integer total;
-    
+
     public static TrackResponse parse(String jsonBody) {
         return (TrackResponse) JSON.deserialize(jsonBody, TrackResponse.class);
     }
+
+    public List<SimplefieldTracksObject> getTracks() { 
+
+        List<SimplefieldTracksObject> result = new List<SimplefieldTracksObject>();
+
+        for (ItemTrack item : this.items) {
+            result.add(item.track);
+        }
+
+        return result;
+    }
 }
diff --git a/force-app/main/default/classes/service/SyncTrack.cls b/force-app/main/default/classes/service/SyncTrack.cls
@@ -3,7 +3,7 @@ public class SyncTrack {
     @TestVisible
     private static Database.SaveResult errorEventResult;
     public static void syncTrack(List<ClientCredential__c> credentials) {
-        
+
         try {
             Map<String, Track__c> recordsToUpsertTracksMap = new Map<String, Track__c>();
 
@@ -13,14 +13,14 @@ public class SyncTrack {
                     cred.ClientId__c,
                     cred.ClientSecret__c
                 );
-                
+
                 if (token.getStatusCode() != SpotifyApiHttpClient.SUCCESS_STATUS_CODE)
                 continue;
 
                 List<SimplefieldTracksObject> tracks = getTracksList(
-                    TokenResponse.parse(token.getBody()).access_token, 
+                    TokenResponse.parse(token.getBody()).access_token,
                     cred.SpotifyUserId__c);
-                    
+
                 for (SimplefieldTracksObject track : tracks) {
 
                     if (!recordsToUpsertTracksMap.containsKey(track.id)) {
@@ -36,6 +36,7 @@ public class SyncTrack {
                             Name = track.name,
                             Popularity__c = track.popularity
                         );
+
                         recordsToUpsertTracksMap.put(track.id, trackRecord);
                     }
                 }
@@ -64,10 +65,10 @@ public class SyncTrack {
             HttpResponse playlistsRes = SpotifyApiHttpClient.getPlaylist(
                 token,
                 spotifyUserId,
-                cont 
+                cont
             );
 
-            if (playlistsRes.getStatusCode() != SpotifyApiHttpClient.SUCCESS_STATUS_CODE) 
+            if (playlistsRes.getStatusCode() != SpotifyApiHttpClient.SUCCESS_STATUS_CODE)
             continue;
 
             playlists = PlaylistResponse.parse(playlistsRes.getBody());
@@ -76,15 +77,13 @@ public class SyncTrack {
 
                 HttpResponse trackRes = SpotifyApiHttpClient.getTrack(token, playlist.id);
 
-                if (trackRes.getStatusCode() != SpotifyApiHttpClient.SUCCESS_STATUS_CODE) 
+                if (trackRes.getStatusCode() != SpotifyApiHttpClient.SUCCESS_STATUS_CODE)
                 continue;
 
                 TrackResponse tracks = TrackResponse.parse(trackRes.getBody());
-
-                for (ItemTrack item : tracks.items) {
-                    result.add(item.track);
-                }
+                result.addAll(tracks.getTracks());
             }
+
             cont += playlists.items.size();
 
         } while (cont < playlists.total);