From dd1f02a13b8a96eba1a5ea5757050a7ee4751a25 Mon Sep 17 00:00:00 2001
From: Fredrik Burmester <fredrik.burmester@gmail.com>
Date: Tue, 1 Oct 2024 14:16:02 +0200
Subject: [PATCH] fix: manual download button

---
 components/downloads/ActiveDownloads.tsx | 92 +++++++++++++++---------
 providers/DownloadProvider.tsx           | 54 ++++++++++----
 2 files changed, 99 insertions(+), 47 deletions(-)

diff --git a/components/downloads/ActiveDownloads.tsx b/components/downloads/ActiveDownloads.tsx
index 9d2d3e50..15b1557c 100644
--- a/components/downloads/ActiveDownloads.tsx
+++ b/components/downloads/ActiveDownloads.tsx
@@ -18,11 +18,12 @@ import {
   ViewProps,
 } from "react-native";
 import { toast } from "sonner-native";
+import { Button } from "../Button";
 
 interface Props extends ViewProps {}
 
 export const ActiveDownloads: React.FC<Props> = ({ ...props }) => {
-  const { processes } = useDownload();
+  const { processes, startDownload } = useDownload();
   if (processes?.length === 0)
     return (
       <View {...props} className="bg-neutral-900 p-4 rounded-2xl">
@@ -48,6 +49,7 @@ interface DownloadCardProps extends TouchableOpacityProps {
 }
 
 const DownloadCard = ({ process, ...props }: DownloadCardProps) => {
+  const { processes, startDownload } = useDownload();
   const router = useRouter();
   const { removeProcess, setProcesses } = useDownload();
   const [settings] = useSettings();
@@ -99,43 +101,67 @@ const DownloadCard = ({ process, ...props }: DownloadCardProps) => {
       className="relative bg-neutral-900 border border-neutral-800 rounded-2xl overflow-hidden"
       {...props}
     >
-      <View
-        className={`
+      {process.status === "optimizing" && (
+        <View
+          className={`
         bg-purple-600 h-1 absolute bottom-0 left-0
         `}
-        style={{
-          width: process.progress ? `${Math.max(5, process.progress)}%` : "5%",
-        }}
-      ></View>
-      <View className="p-4 flex flex-row items-center justify-between w-full">
-        <View className="shrink">
-          <Text className="font-semibold shrink">{process.item.Name}</Text>
-          <Text className="text-xs opacity-50">{process.item.Type}</Text>
-          <View className="flex flex-row items-center space-x-2 mt-1 text-purple-600">
-            <Text className="text-xs">{process.progress.toFixed(0)}%</Text>
-            {process.speed && (
-              <Text className="text-xs">{process.speed?.toFixed(2)}x</Text>
-            )}
-            {eta(process) && (
-              <View>
+          style={{
+            width: process.progress
+              ? `${Math.max(5, process.progress)}%`
+              : "5%",
+          }}
+        ></View>
+      )}
+      <View className="p-4 flex flex-col w-full">
+        <View className="flex flex-row items-center justify-between w-full">
+          <View className="shrink">
+            <Text className="text-xs opacity-50">{process.item.Type}</Text>
+            <Text className="font-semibold shrink">{process.item.Name}</Text>
+            <Text className="text-xs opacity-50">
+              {process.item.ProductionYear}
+            </Text>
+            <View className="flex flex-row items-center space-x-2 mt-2 text-purple-600">
+              {process.progress === 0 ? (
+                <ActivityIndicator size={"small"} color={"white"} />
+              ) : (
+                <Text className="text-xs">{process.progress.toFixed(0)}%</Text>
+              )}
+              {process.speed && (
+                <Text className="text-xs">{process.speed?.toFixed(2)}x</Text>
+              )}
+              {eta(process) && (
                 <Text className="text-xs">ETA {eta(process)}</Text>
-              </View>
+              )}
+            </View>
+
+            <View className="flex flex-row items-center space-x-2 mt-1 text-purple-600">
+              <Text className="text-xs capitalize">{process.status}</Text>
+            </View>
+          </View>
+          <TouchableOpacity
+            disabled={cancelJobMutation.isPending}
+            onPress={() => cancelJobMutation.mutate(process.id)}
+          >
+            {cancelJobMutation.isPending ? (
+              <ActivityIndicator size="small" color="white" />
+            ) : (
+              <Ionicons name="close" size={24} color="red" />
             )}
-          </View>
-          <View className="flex flex-row items-center space-x-2 mt-1 text-purple-600">
-            <Text className="text-xs capitalize">{process.status}</Text>
-          </View>
+          </TouchableOpacity>
         </View>
-        <TouchableOpacity
-          disabled={cancelJobMutation.isPending}
-          onPress={() => cancelJobMutation.mutate(process.id)}
-        >
-          {cancelJobMutation.isPending ? (
-            <ActivityIndicator size="small" color="white" />
-          ) : (
-            <Ionicons name="close" size={24} color="red" />
-          )}
-        </TouchableOpacity>
+        {process.status === "completed" && (
+          <View className="flex flex-row mt-4 space-x-4">
+            <Button
+              onPress={() => {
+                startDownload(process);
+              }}
+              className="w-full"
+            >
+              Download now
+            </Button>
+          </View>
+        )}
       </View>
     </TouchableOpacity>
   );
diff --git a/providers/DownloadProvider.tsx b/providers/DownloadProvider.tsx
index 568faae1..0246cab5 100644
--- a/providers/DownloadProvider.tsx
+++ b/providers/DownloadProvider.tsx
@@ -129,22 +129,21 @@ function useDownloadProvider() {
 
   useEffect(() => {
     const checkIfShouldStartDownload = async () => {
-      if (!processes) return;
-      for (let i = 0; i < processes.length; i++) {
-        const job = processes[i];
+      const tasks = await checkForExistingDownloads();
+      // for (let i = 0; i < processes.length; i++) {
+      //   const job = processes[i];
 
-        if (job.status === "completed") {
-          // Check if the download is already in progress
-          const tasks = await checkForExistingDownloads();
-          if (tasks.find((task) => task.id === job.id)) continue;
-          await startDownload(job);
-          continue;
-        }
-      }
+      //   if (job.status === "completed") {
+      //     // Check if the download is already in progress
+      //     if (tasks.find((task) => task.id === job.id)) continue;
+      //     await startDownload(job);
+      //     continue;
+      //   }
+      // }
     };
 
     checkIfShouldStartDownload();
-  }, [processes]);
+  }, []);
 
   /********************
    * Background task
@@ -208,6 +207,19 @@ function useDownloadProvider() {
     async (process: JobStatus) => {
       if (!process?.item.Id || !authHeader) throw new Error("No item id");
 
+      setProcesses((prev) =>
+        prev.map((p) =>
+          p.id === process.id
+            ? {
+                ...p,
+                speed: undefined,
+                status: "downloading",
+                progress: 0,
+              }
+            : p
+        )
+      );
+
       setConfig({
         isLogsEnabled: true,
         progressInterval: 500,
@@ -256,9 +268,11 @@ function useDownloadProvider() {
         })
         .done(async () => {
           await saveDownloadedItemInfo(process.item);
-          removeProcess(process.id);
-          completeHandler(process.id);
           toast.success(`Download completed for ${process.item.Name}`);
+          setTimeout(() => {
+            completeHandler(process.id);
+            removeProcess(process.id);
+          }, 1000);
         })
         .error(async (error) => {
           removeProcess(process.id);
@@ -267,9 +281,20 @@ function useDownloadProvider() {
           if (error.errorCode === 1000) {
             errorMsg = "No space left";
           }
+          if (error.errorCode === 404) {
+            errorMsg = "File not found on server";
+          }
           toast.error(`Download failed for ${process.item.Name} - ${errorMsg}`);
           writeToLog("ERROR", `Download failed for ${process.item.Name}`, {
             error,
+            processDetails: {
+              id: process.id,
+              itemName: process.item.Name,
+              itemId: process.item.Id,
+            },
+          });
+          console.error("Error details:", {
+            errorCode: error.errorCode,
           });
         });
     },
@@ -463,6 +488,7 @@ function useDownloadProvider() {
     saveDownloadedItemInfo,
     removeProcess,
     setProcesses,
+    startDownload,
   };
 }