SpotiFLAC/frontend/src/hooks/useDownloadProgress.ts

import { useState, useEffect, useRef } from "react";
import { GetDownloadProgress } from "../../wailsjs/go/main/App";

export interface DownloadProgressInfo {
  is_downloading: boolean;
  mb_downloaded: number;
  speed_mbps: number;
}

export function useDownloadProgress() {
  const [progress, setProgress] = useState<DownloadProgressInfo>({
    is_downloading: false,
    mb_downloaded: 0,
    speed_mbps: 0,
  });
  const intervalRef = useRef<number | null>(null);

  useEffect(() => {
    // Poll progress every 200ms for smooth updates
    const pollProgress = async () => {
      try {
        const progressInfo = await GetDownloadProgress();
        setProgress(progressInfo);
      } catch (error) {
        console.error("Failed to get download progress:", error);
      }
    };

    // Start polling
    intervalRef.current = window.setInterval(pollProgress, 200);

    // Initial fetch
    pollProgress();

    // Cleanup
    return () => {
      if (intervalRef.current) {
        clearInterval(intervalRef.current);
      }
    };
  }, []);

  return progress;
}