gitea-mirror/src/hooks/useConfigStatus.ts

import { useCallback, useEffect, useState, useRef } from 'react';
import { useAuth } from './useAuth';
import { apiRequest } from '@/lib/utils';
import type { ConfigApiResponse, SourceApiRecord } from '@/types/config';
import {
  DEFAULT_SOURCE_PROVIDER,
  normalizeSourceProviderKind,
  normalizeSourceUrl,
  type SourceProviderKind,
} from '@/lib/source-providers/kinds';

interface ConfigStatus {
  isGitHubConfigured: boolean;
  isGiteaConfigured: boolean;
  isFullyConfigured: boolean;
  isLoading: boolean;
  error: string | null;
  autoMirrorStarred: boolean;
  githubOwner: string;
  sources: SourceApiRecord[];
  /** Any source row exists, including tokenless public-only sources. */
  hasAnySource: boolean;
  /** The configured (primary) source kind and its normalized instance URL. */
  sourceProvider: SourceProviderKind;
  sourceUrl: string;
}

// Cache to prevent duplicate API calls across components
let configCache: { data: ConfigApiResponse | null; timestamp: number; userId: string | null } = {
  data: null,
  timestamp: 0,
  userId: null
};

const CACHE_DURATION = 30000; // 30 seconds cache

// Mounted hooks, so invalidating the cache refreshes what is on screen
// instead of only clearing the store behind it.
const cacheSubscribers = new Set<() => void>();

// One in-flight GET shared by every mounted hook: an invalidation wakes all
// of them at once, and without this each would fire its own request.
let inFlightRequest: { userId: string; promise: Promise<ConfigApiResponse> } | null = null;

export function hasConfiguredSource(configResponse: ConfigApiResponse | null): boolean {
  return Boolean(
    configResponse?.githubConfig?.token ||
    configResponse?.sources?.some(source => source.provider === 'github' && source.authType === 'app')
  );
}

function fetchConfigOnce(userId: string): Promise<ConfigApiResponse> {
  if (inFlightRequest && inFlightRequest.userId === userId) {
    return inFlightRequest.promise;
  }
  const promise = apiRequest<ConfigApiResponse>(`/config?userId=${userId}`, { method: 'GET' });
  const tracked = promise.finally(() => {
    if (inFlightRequest?.promise === tracked) {
      inFlightRequest = null;
    }
  });
  inFlightRequest = { userId, promise: tracked };
  return tracked;
}

/**
 * Hook to check if GitHub and Gitea are properly configured
 * Returns configuration status and prevents unnecessary API calls when not configured
 * Uses caching to prevent duplicate API calls across components
 */
export function useConfigStatus(): ConfigStatus {
  const { user } = useAuth();
  const [configStatus, setConfigStatus] = useState<ConfigStatus>({
    isGitHubConfigured: false,
    isGiteaConfigured: false,
    isFullyConfigured: false,
    isLoading: true,
    error: null,
    autoMirrorStarred: false,
    githubOwner: '',
    sources: [],
    hasAnySource: false,
    sourceProvider: DEFAULT_SOURCE_PROVIDER,
    sourceUrl: normalizeSourceUrl(undefined, DEFAULT_SOURCE_PROVIDER),
  });

  // Track if this hook has already checked config to prevent multiple calls
  const hasCheckedRef = useRef(false);

  const checkConfiguration = useCallback(async () => {
    if (!user?.id) {
      setConfigStatus({
        isGitHubConfigured: false,
        isGiteaConfigured: false,
        isFullyConfigured: false,
        isLoading: false,
        error: 'No user found',
        autoMirrorStarred: false,
        githubOwner: '',
        sources: [],
        hasAnySource: false,
        sourceProvider: DEFAULT_SOURCE_PROVIDER,
    sourceUrl: normalizeSourceUrl(undefined, DEFAULT_SOURCE_PROVIDER),
      });
      return;
    }

    // Check cache first
    const now = Date.now();
    const isCacheValid = configCache.data &&
                        configCache.userId === user.id &&
                        (now - configCache.timestamp) < CACHE_DURATION;

    if (isCacheValid && hasCheckedRef.current) {
      const configResponse = configCache.data!;

      // App private keys are redacted from the config response, so connected
      // App sources count alongside a visible PAT for dashboard access.
      const isGitHubConfigured = hasConfiguredSource(configResponse);

      const isGiteaConfigured = !!(
        configResponse?.giteaConfig?.url &&
        configResponse?.giteaConfig?.token
      );

      const isFullyConfigured = isGitHubConfigured && isGiteaConfigured;

      const sources = configResponse?.sources ?? [];

      setConfigStatus({
        isGitHubConfigured,
        isGiteaConfigured,
        isFullyConfigured,
        isLoading: false,
        error: null,
        autoMirrorStarred: configResponse?.advancedOptions?.autoMirrorStarred ?? false,
        githubOwner: configResponse?.githubConfig?.username ?? '',
        sources,
        hasAnySource: sources.length > 0,
        sourceProvider: normalizeSourceProviderKind(configResponse?.githubConfig?.provider),
        sourceUrl: normalizeSourceUrl(
          configResponse?.githubConfig?.url,
          normalizeSourceProviderKind(configResponse?.githubConfig?.provider)
        ),
      });
      return;
    }

    try {
      // Only show loading if we haven't checked before or cache is invalid
      if (!hasCheckedRef.current) {
        setConfigStatus(prev => ({ ...prev, isLoading: true, error: null }));
      }

      const configResponse = await fetchConfigOnce(user.id);

      // Update cache
      configCache = {
        data: configResponse,
        timestamp: now,
        userId: user.id
      };

      // Match the cached path: a redacted App source is still configured.
      const isGitHubConfigured = hasConfiguredSource(configResponse);

      const isGiteaConfigured = !!(
        configResponse?.giteaConfig?.url &&
        configResponse?.giteaConfig?.token
      );

      const isFullyConfigured = isGitHubConfigured && isGiteaConfigured;

      const sources = configResponse?.sources ?? [];

      setConfigStatus({
        isGitHubConfigured,
        isGiteaConfigured,
        isFullyConfigured,
        isLoading: false,
        error: null,
        autoMirrorStarred: configResponse?.advancedOptions?.autoMirrorStarred ?? false,
        githubOwner: configResponse?.githubConfig?.username ?? '',
        sources,
        hasAnySource: sources.length > 0,
        sourceProvider: normalizeSourceProviderKind(configResponse?.githubConfig?.provider),
        sourceUrl: normalizeSourceUrl(
          configResponse?.githubConfig?.url,
          normalizeSourceProviderKind(configResponse?.githubConfig?.provider)
        ),
      });

      hasCheckedRef.current = true;
    } catch (error) {
      setConfigStatus({
        isGitHubConfigured: false,
        isGiteaConfigured: false,
        isFullyConfigured: false,
        isLoading: false,
        error: error instanceof Error ? error.message : 'Failed to check configuration',
        autoMirrorStarred: false,
        githubOwner: '',
        sources: [],
        hasAnySource: false,
        sourceProvider: DEFAULT_SOURCE_PROVIDER,
    sourceUrl: normalizeSourceUrl(undefined, DEFAULT_SOURCE_PROVIDER),
      });
      hasCheckedRef.current = true;
    }
  }, [user?.id]);

  useEffect(() => {
    checkConfiguration();
  }, [checkConfiguration]);

  // Re-read after invalidateConfigCache(): saving the configuration or
  // adding a public organization (which creates its tokenless source row)
  // has to reach the components already on screen, or source-dependent UI
  // like the organization card's "Public only" badge stays stale until a
  // reload.
  useEffect(() => {
    const refresh = () => {
      void checkConfiguration();
    };
    cacheSubscribers.add(refresh);
    return () => {
      cacheSubscribers.delete(refresh);
    };
  }, [checkConfiguration]);

  return configStatus;
}

// Export function to invalidate cache when config is updated. Mounted hooks
// re-fetch (through the shared in-flight request, so this costs one call).
export function invalidateConfigCache() {
  configCache = { data: null, timestamp: 0, userId: null };
  inFlightRequest = null;
  for (const refresh of [...cacheSubscribers]) {
    refresh();
  }
}

// Export function to get cached config data for other hooks
export function getCachedConfig(): ConfigApiResponse | null {
  const now = Date.now();
  const isCacheValid = configCache.data && (now - configCache.timestamp) < CACHE_DURATION;
  return isCacheValid ? configCache.data : null;
}