import React, { useEffect, useState } from 'react';
import { Switch } from '~/components/ui/Switch';
import { useSettings } from '~/lib/hooks/useSettings';
import { LOCAL_PROVIDERS, URL_CONFIGURABLE_PROVIDERS } from '~/lib/stores/settings';
import type { IProviderConfig } from '~/types/model';
import { logStore } from '~/lib/stores/logs';

// Import a default fallback icon
import DefaultIcon from '/icons/Default.svg';
const ADVANCED_PROVIDERS = ['Ollama', 'OpenAILike', 'LMStudio'];
import { providerBaseUrlEnvKeys } from '~/utils/constants';


export default function ProvidersTab() {
  const { providers, updateProviderSettings, isLocalModel } = useSettings();
  const [filteredProviders, setFilteredProviders] = useState<IProviderConfig[]>([]);
  const [advancedProviders, setAdvancedProviders] = useState<IProviderConfig[]>([]);
  const [regularProviders, setRegularProviders] = useState<IProviderConfig[]>([]);

  const [searchTerm, setSearchTerm] = useState('');

  useEffect(() => {
    let newFilteredProviders: IProviderConfig[] = Object.entries(providers).map(([key, value]) => ({
      ...value,
      name: key,
    }));

    if (searchTerm && searchTerm.length > 0) {
      newFilteredProviders = newFilteredProviders.filter((provider) =>
        provider.name.toLowerCase().includes(searchTerm.toLowerCase()),
      );
    }

    if (!isLocalModel) {
      newFilteredProviders = newFilteredProviders.filter((provider) => !LOCAL_PROVIDERS.includes(provider.name));
    }

    // Split providers into regular and advanced
    const regular = newFilteredProviders.filter(
      (provider) => !ADVANCED_PROVIDERS.includes(provider.name)
    );
    const advanced = newFilteredProviders.filter(
      (provider) => ADVANCED_PROVIDERS.includes(provider.name)
    );

    // Sort advanced providers in specific order - OpenAILike at the end
    const advancedOrder = ['Ollama', 'LMStudio', 'OpenAILike'];
    advanced.sort((a, b) => {
      return advancedOrder.indexOf(a.name) - advancedOrder.indexOf(b.name);
    });

    // Sort regular providers alphabetically
    regular.sort((a, b) => a.name.localeCompare(b.name));

    setRegularProviders(regular);
    setAdvancedProviders(advanced);
    setFilteredProviders(newFilteredProviders);
  }, [providers, searchTerm, isLocalModel]);

  const ProviderCard = ({ provider }: { provider: IProviderConfig }) => (
    <div className="flex flex-col provider-item hover:bg-bolt-elements-bg-depth-3 p-4 rounded-lg border border-bolt-elements-borderColor">
      <div className="flex items-center justify-between mb-2">
        <div className="flex items-center gap-2">
          <img
            src={`/icons/${provider.name}.svg`}
            onError={(e) => {
              e.currentTarget.src = DefaultIcon;
            }}
            alt={`${provider.name} icon`}
            className="w-6 h-6 dark:invert"
          />
          <span className="text-bolt-elements-textPrimary">{provider.name}</span>
        </div>
        <Switch
          className="ml-auto"
          checked={provider.settings.enabled}
          onCheckedChange={(enabled) => {
            updateProviderSettings(provider.name, { ...provider.settings, enabled });
            if (enabled) {
              logStore.logProvider(`Provider ${provider.name} enabled`, { provider: provider.name });
            } else {
              logStore.logProvider(`Provider ${provider.name} disabled`, { provider: provider.name });
            }
          }}
        />
      </div>
      {URL_CONFIGURABLE_PROVIDERS.includes(provider.name) && provider.settings.enabled && (
        <div className="mt-2">
          <label className="block text-sm text-bolt-elements-textSecondary mb-1">Base URL:</label>
          <input
            type="text"
            value={provider.settings.baseUrl || ''}
            onChange={(e) => {
              const newBaseUrl = e.target.value;
              updateProviderSettings(provider.name, { ...provider.settings, baseUrl: newBaseUrl });
              logStore.logProvider(`Base URL updated for ${provider.name}`, {
                provider: provider.name,
                baseUrl: newBaseUrl,
              });
            }}
            placeholder={`Enter ${provider.name} base URL`}
            className="w-full bg-white dark:bg-bolt-elements-background-depth-4 relative px-2 py-1.5 rounded-md focus:outline-none placeholder-bolt-elements-textTertiary text-bolt-elements-textPrimary dark:text-bolt-elements-textPrimary border border-bolt-elements-borderColor"
          />
        </div>
      )}
    </div>
  );

  return (
    <div className="p-4">
      <div className="flex mb-4">
        <input
          type="text"
          placeholder="Search providers..."
          value={searchTerm}
          onChange={(e) => setSearchTerm(e.target.value)}
          className="w-full bg-white dark:bg-bolt-elements-background-depth-4 relative px-2 py-1.5 rounded-md focus:outline-none placeholder-bolt-elements-textTertiary text-bolt-elements-textPrimary dark:text-bolt-elements-textPrimary border border-bolt-elements-borderColor"
        />
      </div>

      {filteredProviders.map((provider) => {
        const envBaseUrlKey = providerBaseUrlEnvKeys[provider.name].baseUrlKey;
        const envBaseUrl = envBaseUrlKey ? import.meta.env[envBaseUrlKey] : undefined;

        return (
          <div
            key={provider.name}
            className="flex flex-col mb-2 provider-item hover:bg-bolt-elements-bg-depth-3 p-4 rounded-lg border border-bolt-elements-borderColor "
          >
            <div className="flex items-center justify-between mb-2">
              <div className="flex items-center gap-2">
                <img
                  src={`/icons/${provider.name}.svg`} // Attempt to load the specific icon
                  onError={(e) => {
                    // Fallback to default icon on error
                    e.currentTarget.src = DefaultIcon;
                  }}
                  alt={`${provider.name} icon`}
                  className="w-6 h-6 dark:invert"
                />
                <span className="text-bolt-elements-textPrimary">{provider.name}</span>
              </div>
              <Switch
                className="ml-auto"
                checked={provider.settings.enabled}
                onCheckedChange={(enabled) => {
                  updateProviderSettings(provider.name, { ...provider.settings, enabled });

                  if (enabled) {
                    logStore.logProvider(`Provider ${provider.name} enabled`, { provider: provider.name });
                  } else {
                    logStore.logProvider(`Provider ${provider.name} disabled`, { provider: provider.name });
                  }
                }}
              />
            </div>
            {/* Base URL input for configurable providers */}
            {URL_CONFIGURABLE_PROVIDERS.includes(provider.name) && provider.settings.enabled && (
              <div className="mt-2">
                {envBaseUrl && (
                  <label className="block text-xs text-bolt-elements-textSecondary text-green-300 mb-2">
                    Set On (.env) : {envBaseUrl}
                  </label>
                )}
                <label className="block text-sm text-bolt-elements-textSecondary mb-2">
                  {envBaseUrl ? 'Override Base Url' : 'Base URL '}:{' '}
                </label>
                <input
                  type="text"
                  value={provider.settings.baseUrl || ''}
                  onChange={(e) => {
                    let newBaseUrl: string | undefined = e.target.value;

                    if (newBaseUrl && newBaseUrl.trim().length === 0) {
                      newBaseUrl = undefined;
                    }

                    updateProviderSettings(provider.name, { ...provider.settings, baseUrl: newBaseUrl });
                    logStore.logProvider(`Base URL updated for ${provider.name}`, {
                      provider: provider.name,
                      baseUrl: newBaseUrl,
                    });
                  }}
                  placeholder={`Enter ${provider.name} base URL`}
                  className="w-full bg-white dark:bg-bolt-elements-background-depth-4 relative px-2 py-1.5 rounded-md focus:outline-none placeholder-bolt-elements-textTertiary text-bolt-elements-textPrimary dark:text-bolt-elements-textPrimary border border-bolt-elements-borderColor"
                />
              </div>
            )}
          </div>
        );
      })}
    </div>
  );
}
