Stirling-PDF/frontend/src/core/hooks/useGoogleDrivePicker.ts

/**
 * React hook for Google Drive file picker
 */

import { useState, useCallback, useEffect, useMemo } from 'react';
import { useAppConfig } from '@app/contexts/AppConfigContext';
import {
  getGoogleDrivePickerService,
  isGoogleDriveConfigured,
  getGoogleDriveConfig,
  extractGoogleDriveBackendConfig,
} from '@app/services/googleDrivePickerService';

interface UseGoogleDrivePickerOptions {
  multiple?: boolean;
  mimeTypes?: string;
}

interface UseGoogleDrivePickerReturn {
  isEnabled: boolean;
  isLoading: boolean;
  error: string | null;
  openPicker: (options?: UseGoogleDrivePickerOptions) => Promise<File[]>;
}

/**
 * Hook to use Google Drive file picker
 */
export function useGoogleDrivePicker(): UseGoogleDrivePickerReturn {
  const { config } = useAppConfig();
  const [isEnabled, setIsEnabled] = useState(false);
  const [isLoading, setIsLoading] = useState(false);
  const [error, setError] = useState<string | null>(null);
  const [isInitialized, setIsInitialized] = useState(false);

  // Memoize backend config to only track Google Drive specific properties
  const googleDriveBackendConfig = useMemo(
    () => extractGoogleDriveBackendConfig(config),
    [config?.googleDriveEnabled, config?.googleDriveClientId, config?.googleDriveApiKey, config?.googleDriveAppId]
  );

  // Check if Google Drive is configured and reset initialization if disabled
  useEffect(() => {
    const configured = isGoogleDriveConfigured(googleDriveBackendConfig);
    setIsEnabled(configured);
    // Reset initialization state if Google Drive becomes disabled
    if (!configured) {
      setIsInitialized(false);
    }
  }, [googleDriveBackendConfig]);

  /**
   * Initialize the Google Drive service (lazy initialization)
   */
  const initializeService = useCallback(async () => {
    if (isInitialized) return;

    const googleDriveConfig = getGoogleDriveConfig(googleDriveBackendConfig);
    if (!googleDriveConfig) {
      throw new Error('Google Drive is not configured');
    }

    const service = getGoogleDrivePickerService();
    await service.initialize(googleDriveConfig);
    setIsInitialized(true);
  }, [isInitialized, googleDriveBackendConfig]);

  /**
   * Open the Google Drive picker
   */
  const openPicker = useCallback(
    async (options: UseGoogleDrivePickerOptions = {}): Promise<File[]> => {
      if (!isEnabled) {
        setError('Google Drive is not configured');
        return [];
      }

      try {
        setIsLoading(true);
        setError(null);

        // Initialize service if needed
        await initializeService();

        // Open picker
        const service = getGoogleDrivePickerService();
        const files = await service.openPicker({
          multiple: options.multiple ?? true,
          mimeTypes: options.mimeTypes,
        });

        return files;
      } catch (err) {
        const errorMessage = err instanceof Error ? err.message : 'Failed to open Google Drive picker';
        setError(errorMessage);
        console.error('Google Drive picker error:', err);
        return [];
      } finally {
        setIsLoading(false);
      }
    },
    [isEnabled, initializeService]
  );

  return {
    isEnabled,
    isLoading,
    error,
    openPicker,
  };
}