automaker/apps/ui/src/components/views/setup-view/hooks/use-cli-status.ts

import { useState, useCallback } from 'react';
import { createLogger } from '@automaker/utils/logger';

interface UseCliStatusOptions {
  cliType: 'claude' | 'codex';
  statusApi: () => Promise<any>;
  setCliStatus: (status: any) => void;
  setAuthStatus: (status: any) => void;
}

const VALID_AUTH_METHODS = {
  claude: [
    'oauth_token_env',
    'oauth_token',
    'api_key',
    'api_key_env',
    'credentials_file',
    'cli_authenticated',
    'none',
  ],
  codex: ['cli_authenticated', 'api_key', 'api_key_env', 'none'],
} as const;

// Create logger outside of the hook to avoid re-creating it on every render
const logger = createLogger('CliStatus');

export function useCliStatus({
  cliType,
  statusApi,
  setCliStatus,
  setAuthStatus,
}: UseCliStatusOptions) {
  const [isChecking, setIsChecking] = useState(false);

  const checkStatus = useCallback(async () => {
    logger.info(`Starting status check for ${cliType}...`);
    setIsChecking(true);
    try {
      const result = await statusApi();
      logger.info(`Raw status result for ${cliType}:`, result);

      if (result.success) {
        const cliStatus = {
          installed: result.status === 'installed',
          path: result.path || null,
          version: result.version || null,
          method: result.method || 'none',
        };
        logger.info(`CLI Status for ${cliType}:`, cliStatus);
        setCliStatus(cliStatus);

        if (result.auth) {
          // Validate method is one of the expected values, default to "none"
          const validMethods = VALID_AUTH_METHODS[cliType] ?? ['none'] as const;
          type AuthMethod = (typeof validMethods)[number];
          const method: AuthMethod = validMethods.includes(result.auth.method as AuthMethod)
            ? (result.auth.method as AuthMethod)
            : 'none';

          if (cliType === 'claude') {
            setAuthStatus({
              authenticated: result.auth.authenticated,
              method,
              hasCredentialsFile: false,
              oauthTokenValid: result.auth.hasStoredOAuthToken || result.auth.hasEnvOAuthToken,
              apiKeyValid: result.auth.hasStoredApiKey || result.auth.hasEnvApiKey,
              hasEnvOAuthToken: result.auth.hasEnvOAuthToken,
              hasEnvApiKey: result.auth.hasEnvApiKey,
            });
          } else {
            setAuthStatus({
              authenticated: result.auth.authenticated,
              method,
              hasAuthFile: result.auth.hasAuthFile ?? false,
              hasApiKey: result.auth.hasApiKey ?? false,
              hasEnvApiKey: result.auth.hasEnvApiKey ?? false,
            });
          }
        }
      }
    } catch (error) {
      logger.error(`Failed to check status for ${cliType}:`, error);
    } finally {
      setIsChecking(false);
    }
  }, [cliType, statusApi, setCliStatus, setAuthStatus]);

  return { isChecking, checkStatus };
}