From 3f91edc5b0c886247bcb207403bc17eb3a962efa Mon Sep 17 00:00:00 2001
From: Adrian Castro <22133246+castdrian@users.noreply.github.com>
Date: Thu, 28 Mar 2024 01:13:01 +0100
Subject: [PATCH] chore: cleanup

---
 apps/expo/src/hooks/DownloadManagerContext.tsx | 12 +++++-------
 1 file changed, 5 insertions(+), 7 deletions(-)

diff --git a/apps/expo/src/hooks/DownloadManagerContext.tsx b/apps/expo/src/hooks/DownloadManagerContext.tsx
index c6684dd..f1d23b9 100644
--- a/apps/expo/src/hooks/DownloadManagerContext.tsx
+++ b/apps/expo/src/hooks/DownloadManagerContext.tsx
@@ -291,15 +291,15 @@ export const DownloadManagerProvider: React.FC<{ children: ReactNode }> = ({
       localSegmentPaths.push(segmentFile);
 
       try {
-        await downloadSegment(segment, segmentFile, headers, downloadId, () => {
-          segmentsDownloaded++;
-          updateProgress();
-        });
+        await downloadSegment(downloadId, segment, segmentFile, headers);
 
         if (getCancellationFlag(downloadId)) {
           await cleanupDownload(segmentDir, downloadId);
           return;
         }
+
+        segmentsDownloaded++;
+        updateProgress();
       } catch (e) {
         console.error(e);
         if (getCancellationFlag(downloadId)) {
@@ -326,11 +326,10 @@ export const DownloadManagerProvider: React.FC<{ children: ReactNode }> = ({
   };
 
   const downloadSegment = async (
+	downloadId: string,
     segmentUrl: string,
     segmentFile: string,
     headers: Record<string, string>,
-    downloadId: string,
-    onSegmentDownloadComplete: () => void,
   ) => {
     return new Promise<void>((resolve, reject) => {
       const task = download({
@@ -342,7 +341,6 @@ export const DownloadManagerProvider: React.FC<{ children: ReactNode }> = ({
 
       task
         .done(() => {
-          onSegmentDownloadComplete();
           resolve();
         })
         .error((error) => {