DynamoDNS/client/src/context/organization-context.tsx

import { createContext, useContext, useState, useEffect, ReactNode } from "react";
import { useQuery, useMutation, UseMutationResult } from "@tanstack/react-query";
import { Organization, InsertOrganization } from "@shared/schema";
import { useAuth } from "@/hooks/use-auth";
import { apiRequest, queryClient } from "@/lib/queryClient";
import { useToast } from "@/hooks/use-toast";

type OrganizationContextType = {
  organizations: Organization[];
  currentOrganization: Organization | null;
  setCurrentOrganization: (organization: Organization) => void;
  isLoading: boolean;
  createOrganizationMutation: UseMutationResult<Organization, Error, InsertOrganization>;
};

const OrganizationContext = createContext<OrganizationContextType | undefined>(undefined);

export function OrganizationProvider({ children }: { children: ReactNode }) {
  const { user } = useAuth();
  const { toast } = useToast();
  const [currentOrganization, setCurrentOrganization] = useState<Organization | null>(null);

  const { data: organizations = [], isLoading } = useQuery<Organization[]>({
    queryKey: ["/api/organizations"],
    // Only fetch if user is logged in
    enabled: !!user,
  });
  
  const createOrganizationMutation = useMutation({
    mutationFn: async (orgData: InsertOrganization) => {
      const res = await apiRequest("POST", "/api/organizations", orgData);
      return await res.json();
    },
    onSuccess: (newOrg: Organization) => {
      queryClient.invalidateQueries({ queryKey: ["/api/organizations"] });
      toast({
        title: "Organization created",
        description: `Organization '${newOrg.name}' has been created successfully.`,
      });
      setCurrentOrganization(newOrg);
    },
    onError: (error: Error) => {
      toast({
        title: "Failed to create organization",
        description: error.message,
        variant: "destructive",
      });
    },
  });

  // Set default organization when organizations are loaded
  useEffect(() => {
    if (organizations.length > 0 && !currentOrganization) {
      // If user has an organization ID, try to find it in the list
      if (user?.organizationId) {
        const userOrg = organizations.find(org => org.id === user.organizationId);
        if (userOrg) {
          setCurrentOrganization(userOrg);
          return;
        }
      }
      
      // Otherwise, use the first organization
      setCurrentOrganization(organizations[0]);
    }
  }, [organizations, currentOrganization, user]);

  return (
    <OrganizationContext.Provider
      value={{ 
        organizations, 
        currentOrganization, 
        setCurrentOrganization, 
        isLoading,
        createOrganizationMutation 
      }}
    >
      {children}
    </OrganizationContext.Provider>
  );
}

export function useOrganization() {
  const context = useContext(OrganizationContext);
  
  if (context === undefined) {
    throw new Error("useOrganization must be used within an OrganizationProvider");
  }
  
  return context;
}