import { createContext, useContext, useState, useEffect, ReactNode } from "react";

interface ApiKey {
  id: string;
  name: string;
  key: string;
  createdAt: string;
}

interface ApiKeyContextType {
  apiKeys: ApiKey[];
  activeKeyId: string | null;
  activeKey: string | null;
  addApiKey: (name: string, key: string) => void;
  removeApiKey: (id: string) => void;
  renameApiKey: (id: string, newName: string) => void;
  setActiveKeyId: (id: string) => void;
}

const ApiKeyContext = createContext<ApiKeyContextType | undefined>(undefined);

export function ApiKeyProvider({ children }: { children: ReactNode }) {
  const [apiKeys, setApiKeys] = useState<ApiKey[]>(() => {
    const saved = localStorage.getItem("smartlead_api_keys");
    return saved ? JSON.parse(saved) : [];
  });
  
  const [activeKeyId, setActiveKeyIdState] = useState<string | null>(() => {
    return localStorage.getItem("smartlead_active_key_id");
  });

  useEffect(() => {
    localStorage.setItem("smartlead_api_keys", JSON.stringify(apiKeys));
  }, [apiKeys]);

  useEffect(() => {
    if (activeKeyId) {
      localStorage.setItem("smartlead_active_key_id", activeKeyId);
    } else {
      localStorage.removeItem("smartlead_active_key_id");
    }
  }, [activeKeyId]);

  const addApiKey = (name: string, key: string) => {
    const newKey: ApiKey = {
      id: Math.random().toString(36).substring(7),
      name,
      key,
      createdAt: new Date().toISOString()
    };
    setApiKeys((prev) => [...prev, newKey]);
    if (!activeKeyId) {
      setActiveKeyIdState(newKey.id);
    }
  };

  const removeApiKey = (id: string) => {
    setApiKeys((prev) => prev.filter((k) => k.id !== id));
    if (activeKeyId === id) {
      setActiveKeyIdState(null);
    }
  };

  const renameApiKey = (id: string, newName: string) => {
    setApiKeys((prev) => prev.map((k) => k.id === id ? { ...k, name: newName } : k));
  };

  const activeKey = apiKeys.find((k) => k.id === activeKeyId)?.key || null;

  return (
    <ApiKeyContext.Provider
      value={{
        apiKeys,
        activeKeyId,
        activeKey,
        addApiKey,
        removeApiKey,
        renameApiKey,
        setActiveKeyId: setActiveKeyIdState,
      }}
    >
      {children}
    </ApiKeyContext.Provider>
  );
}

export function useApiKey() {
  const context = useContext(ApiKeyContext);
  if (context === undefined) {
    throw new Error("useApiKey must be used within an ApiKeyProvider");
  }
  return context;
}