From 41041fa296be4f719e653e200de81f24af3180e4 Mon Sep 17 00:00:00 2001
From: icarus <eurfelux@gmail.com>
Date: Mon, 13 Oct 2025 14:20:07 +0800
Subject: [PATCH] feat(video): add download button for completed videos

Implement download button UI for completed videos. Currently shows a toast notification as the functionality is not yet implemented.
---
 src/renderer/src/pages/video/VideoPanel.tsx  |  7 ++++++-
 src/renderer/src/pages/video/VideoViewer.tsx | 22 +++++++++++++-------
 2 files changed, 21 insertions(+), 8 deletions(-)

diff --git a/src/renderer/src/pages/video/VideoPanel.tsx b/src/renderer/src/pages/video/VideoPanel.tsx
index a10d3a9b97..a7532121d9 100644
--- a/src/renderer/src/pages/video/VideoPanel.tsx
+++ b/src/renderer/src/pages/video/VideoPanel.tsx
@@ -69,6 +69,10 @@ export const VideoPanel = ({ provider, video, params, updateParams }: VideoPanel
     }
   }, [addOpenAIVideo, couldCreateVideo, params, t, video])
 
+  const handleDownloadVideo = (videoId: string) => {
+    window.toast.info('Not implemented')
+  }
+
   const handleUploadFile = useCallback(() => {
     fileInputRef.current?.click()
   }, [])
@@ -114,7 +118,8 @@ export const VideoPanel = ({ provider, video, params, updateParams }: VideoPanel
     <div className="flex flex-1 flex-col p-2">
       <div className="m-8 flex-1">
         <Skeleton className="h-full w-full rounded-2xl" classNames={{ content: 'h-full w-full' }} isLoaded={true}>
-          <VideoViewer video={video} />
+          {video && <VideoViewer video={video} onDownload={handleDownloadVideo} />}
+          {!video && <VideoViewer video={video} />}
         </Skeleton>
       </div>
       <div className="relative">
diff --git a/src/renderer/src/pages/video/VideoViewer.tsx b/src/renderer/src/pages/video/VideoViewer.tsx
index 611456260c..a63413cd19 100644
--- a/src/renderer/src/pages/video/VideoViewer.tsx
+++ b/src/renderer/src/pages/video/VideoViewer.tsx
@@ -14,11 +14,17 @@ import { CheckCircleIcon, CircleXIcon } from 'lucide-react'
 import { useMemo, useState } from 'react'
 import { useTranslation } from 'react-i18next'
 
-export interface VideoViewerProps {
-  video?: Video
-}
+export type VideoViewerProps =
+  | {
+      video: undefined
+      onDownload?: never
+    }
+  | {
+      video: Video
+      onDownload: (videoId: string) => void
+    }
 
-export const VideoViewer = ({ video }: VideoViewerProps) => {
+export const VideoViewer = ({ video, onDownload }: VideoViewerProps) => {
   const { t } = useTranslation()
   const [loadSuccess, setLoadSuccess] = useState<boolean | undefined>(undefined)
   return (
@@ -27,7 +33,7 @@ export const VideoViewer = ({ video }: VideoViewerProps) => {
         {video === undefined && t('video.undefined')}
         {video && video.status === 'queued' && <QueuedVideo />}
         {video && video.status === 'in_progress' && <InProgressVideo progress={video.progress} />}
-        {video && video.status === 'completed' && <CompletedVideo />}
+        {video && video.status === 'completed' && <CompletedVideo onDownload={() => onDownload(video.id)} />}
         {video && video.status === 'downloading' && <DownloadingVideo progress={video.progress} />}
         {video && video.status === 'downloaded' && loadSuccess !== false && (
           <video
@@ -72,12 +78,14 @@ const InProgressVideo = ({ progress }: { progress: number }) => {
   )
 }
 
-const CompletedVideo = () => {
+const CompletedVideo = ({ onDownload }: { onDownload: () => void }) => {
   const { t } = useTranslation()
+
   return (
-    <div className="flex h-full w-full flex-col items-center justify-center rounded-2xl bg-success-200">
+    <div className="flex h-full w-full flex-col items-center justify-center gap-2 rounded-2xl bg-success-200">
       <CheckCircleIcon size={64} className="text-success" />
       <span className="font-bold text-2xl">{t('video.status.completed')}</span>
+      <Button onPress={onDownload}>{t('common.download')}</Button>
     </div>
   )
 }