"use client"; import { invoke } from "@tauri-apps/api/core"; import { listen } from "@tauri-apps/api/event"; import { getCurrent } from "@tauri-apps/plugin-deep-link"; import { useCallback, useEffect, useMemo, useState } from "react"; import { CamoufoxConfigDialog } from "@/components/camoufox-config-dialog"; import { CreateProfileDialog } from "@/components/create-profile-dialog"; import { DeleteConfirmationDialog } from "@/components/delete-confirmation-dialog"; import { GroupAssignmentDialog } from "@/components/group-assignment-dialog"; import { GroupBadges } from "@/components/group-badges"; import { GroupManagementDialog } from "@/components/group-management-dialog"; import HomeHeader from "@/components/home-header"; import { ImportProfileDialog } from "@/components/import-profile-dialog"; import { PermissionDialog } from "@/components/permission-dialog"; import { ProfilesDataTable } from "@/components/profile-data-table"; import { ProfileSelectorDialog } from "@/components/profile-selector-dialog"; import { ProxyManagementDialog } from "@/components/proxy-management-dialog"; import { SettingsDialog } from "@/components/settings-dialog"; import { useAppUpdateNotifications } from "@/hooks/use-app-update-notifications"; import { useGroupEvents } from "@/hooks/use-group-events"; import type { PermissionType } from "@/hooks/use-permissions"; import { usePermissions } from "@/hooks/use-permissions"; import { useProfileEvents } from "@/hooks/use-profile-events"; import { useProxyEvents } from "@/hooks/use-proxy-events"; import { useUpdateNotifications } from "@/hooks/use-update-notifications"; import { useVersionUpdater } from "@/hooks/use-version-updater"; import { showErrorToast, showToast } from "@/lib/toast-utils"; import type { BrowserProfile, CamoufoxConfig } from "@/types"; type BrowserTypeString = | "mullvad-browser" | "firefox" | "firefox-developer" | "chromium" | "brave" | "zen" | "tor-browser" | "camoufox"; interface PendingUrl { id: string; url: string; } export default function Home() { // Mount global version update listener/toasts useVersionUpdater(); const [isInitializing, setIsInitializing] = useState(true); // Use the new profile events hook for centralized profile management const { profiles, runningProfiles, isLoading: profilesLoading, error: profilesError, } = useProfileEvents(); const { groups: groupsData, isLoading: groupsLoading, error: groupsError, } = useGroupEvents(); const { isLoading: proxiesLoading, error: proxiesError } = useProxyEvents(); const [createProfileDialogOpen, setCreateProfileDialogOpen] = useState(false); const [settingsDialogOpen, setSettingsDialogOpen] = useState(false); const [importProfileDialogOpen, setImportProfileDialogOpen] = useState(false); const [proxyManagementDialogOpen, setProxyManagementDialogOpen] = useState(false); const [camoufoxConfigDialogOpen, setCamoufoxConfigDialogOpen] = useState(false); const [groupManagementDialogOpen, setGroupManagementDialogOpen] = useState(false); const [groupAssignmentDialogOpen, setGroupAssignmentDialogOpen] = useState(false); const [selectedGroupId, setSelectedGroupId] = useState("default"); const [selectedProfilesForGroup, setSelectedProfilesForGroup] = useState< string[] >([]); const [selectedProfiles, setSelectedProfiles] = useState([]); const [searchQuery, setSearchQuery] = useState(""); const [pendingUrls, setPendingUrls] = useState([]); const [currentProfileForCamoufoxConfig, setCurrentProfileForCamoufoxConfig] = useState(null); const [hasCheckedStartupPrompt, setHasCheckedStartupPrompt] = useState(false); const [permissionDialogOpen, setPermissionDialogOpen] = useState(false); const [currentPermissionType, setCurrentPermissionType] = useState("microphone"); const [showBulkDeleteConfirmation, setShowBulkDeleteConfirmation] = useState(false); const [isBulkDeleting, setIsBulkDeleting] = useState(false); const { isMicrophoneAccessGranted, isCameraAccessGranted, isInitialized } = usePermissions(); const handleSelectGroup = useCallback((groupId: string) => { setSelectedGroupId(groupId); setSelectedProfiles([]); }, []); // Check for missing binaries and offer to download them const checkMissingBinaries = useCallback(async () => { try { const missingBinaries = await invoke<[string, string, string][]>( "check_missing_binaries", ); // Also check for missing GeoIP database const missingGeoIP = await invoke( "check_missing_geoip_database", ); if (missingBinaries.length > 0 || missingGeoIP) { if (missingBinaries.length > 0) { console.log("Found missing binaries:", missingBinaries); } if (missingGeoIP) { console.log("Found missing GeoIP database for Camoufox"); } // Group missing binaries by browser type to avoid concurrent downloads const browserMap = new Map(); for (const [profileName, browser, version] of missingBinaries) { if (!browserMap.has(browser)) { browserMap.set(browser, []); } const versions = browserMap.get(browser); if (versions) { versions.push(`${version} (for ${profileName})`); } } // Show a toast notification about missing binaries and auto-download them let missingList = Array.from(browserMap.entries()) .map(([browser, versions]) => `${browser}: ${versions.join(", ")}`) .join(", "); if (missingGeoIP) { if (missingList) { missingList += ", GeoIP database for Camoufox"; } else { missingList = "GeoIP database for Camoufox"; } } console.log(`Downloading missing components: ${missingList}`); try { // Download missing binaries and GeoIP database sequentially to prevent conflicts const downloaded = await invoke( "ensure_all_binaries_exist", ); if (downloaded.length > 0) { console.log( "Successfully downloaded missing components:", downloaded, ); } } catch (downloadError) { console.error( "Failed to download missing components:", downloadError, ); } } } catch (err: unknown) { console.error("Failed to check missing components:", err); } }, []); const [processingUrls, setProcessingUrls] = useState>(new Set()); const handleUrlOpen = useCallback( async (url: string) => { // Prevent duplicate processing of the same URL if (processingUrls.has(url)) { console.log("URL already being processed:", url); return; } setProcessingUrls((prev) => new Set(prev).add(url)); try { console.log("URL received for opening:", url); // Always show profile selector for manual selection - never auto-open // Replace any existing pending URL with the new one setPendingUrls([{ id: Date.now().toString(), url }]); } finally { // Remove URL from processing set after a short delay to prevent rapid duplicates setTimeout(() => { setProcessingUrls((prev) => { const next = new Set(prev); next.delete(url); return next; }); }, 1000); } }, [processingUrls], ); // Auto-update functionality - use the existing hook for compatibility const updateNotifications = useUpdateNotifications(); const { checkForUpdates, isUpdating } = updateNotifications; useAppUpdateNotifications(); // Check for startup URLs but only process them once const [hasCheckedStartupUrl, setHasCheckedStartupUrl] = useState(false); const checkCurrentUrl = useCallback(async () => { if (hasCheckedStartupUrl) return; try { const currentUrl = await getCurrent(); if (currentUrl && currentUrl.length > 0) { console.log("Startup URL detected:", currentUrl[0]); void handleUrlOpen(currentUrl[0]); } } catch (error) { console.error("Failed to check current URL:", error); } finally { setHasCheckedStartupUrl(true); } }, [handleUrlOpen, hasCheckedStartupUrl]); const checkStartupPrompt = useCallback(async () => { // Only check once during app startup to prevent reopening after dismissing notifications if (hasCheckedStartupPrompt) return; try { const shouldShow = await invoke( "should_show_settings_on_startup", ); if (shouldShow) { setSettingsDialogOpen(true); } } catch (error) { console.error("Failed to check startup prompt:", error); } finally { setHasCheckedStartupPrompt(true); } }, [hasCheckedStartupPrompt]); // Warm up nodecar at startup and block UI until complete useEffect(() => { let cancelled = false; (async () => { try { await invoke("warm_up_nodecar"); } catch (err) { if (!cancelled) { // Don't set error here since useProfileEvents handles profile errors console.error("Initialization failed:", err); } } finally { if (!cancelled) setIsInitializing(false); } })(); return () => { cancelled = true; }; }, []); // Handle profile errors from useProfileEvents hook useEffect(() => { if (profilesError) { showErrorToast(profilesError); } }, [profilesError]); // Handle group errors from useGroupEvents hook useEffect(() => { if (groupsError) { showErrorToast(groupsError); } }, [groupsError]); // Handle proxy errors from useProxyEvents hook useEffect(() => { if (proxiesError) { showErrorToast(proxiesError); } }, [proxiesError]); const checkAllPermissions = useCallback(async () => { try { // Wait for permissions to be initialized before checking if (!isInitialized) { return; } // Check if any permissions are not granted - prioritize missing permissions if (!isMicrophoneAccessGranted) { setCurrentPermissionType("microphone"); setPermissionDialogOpen(true); } else if (!isCameraAccessGranted) { setCurrentPermissionType("camera"); setPermissionDialogOpen(true); } } catch (error) { console.error("Failed to check permissions:", error); } }, [isMicrophoneAccessGranted, isCameraAccessGranted, isInitialized]); const checkNextPermission = useCallback(() => { try { if (!isMicrophoneAccessGranted) { setCurrentPermissionType("microphone"); setPermissionDialogOpen(true); } else if (!isCameraAccessGranted) { setCurrentPermissionType("camera"); setPermissionDialogOpen(true); } else { setPermissionDialogOpen(false); } } catch (error) { console.error("Failed to check next permission:", error); } }, [isMicrophoneAccessGranted, isCameraAccessGranted]); const listenForUrlEvents = useCallback(async () => { try { // Listen for URL open events from the deep link handler (when app is already running) await listen("url-open-request", (event) => { console.log("Received URL open request:", event.payload); void handleUrlOpen(event.payload); }); // Listen for show profile selector events await listen("show-profile-selector", (event) => { console.log("Received show profile selector request:", event.payload); void handleUrlOpen(event.payload); }); // Listen for show create profile dialog events await listen("show-create-profile-dialog", (event) => { console.log( "Received show create profile dialog request:", event.payload, ); showErrorToast( "No profiles available. Please create a profile first before opening URLs.", ); setCreateProfileDialogOpen(true); }); // Listen for custom logo click events const handleLogoUrlEvent = (event: CustomEvent) => { console.log("Received logo URL event:", event.detail); void handleUrlOpen(event.detail); }; window.addEventListener( "url-open-request", handleLogoUrlEvent as EventListener, ); // Return cleanup function return () => { window.removeEventListener( "url-open-request", handleLogoUrlEvent as EventListener, ); }; } catch (error) { console.error("Failed to setup URL listener:", error); } }, [handleUrlOpen]); const handleConfigureCamoufox = useCallback((profile: BrowserProfile) => { setCurrentProfileForCamoufoxConfig(profile); setCamoufoxConfigDialogOpen(true); }, []); const handleSaveCamoufoxConfig = useCallback( async (profile: BrowserProfile, config: CamoufoxConfig) => { try { await invoke("update_camoufox_config", { profileId: profile.id, config, }); // No need to manually reload - useProfileEvents will handle the update setCamoufoxConfigDialogOpen(false); } catch (err: unknown) { console.error("Failed to update camoufox config:", err); showErrorToast( `Failed to update camoufox config: ${JSON.stringify(err)}`, ); throw err; } }, [], ); const handleCreateProfile = useCallback( async (profileData: { name: string; browserStr: BrowserTypeString; version: string; releaseType: string; proxyId?: string; camoufoxConfig?: CamoufoxConfig; groupId?: string; }) => { try { await invoke("create_browser_profile_new", { name: profileData.name, browserStr: profileData.browserStr, version: profileData.version, releaseType: profileData.releaseType, proxyId: profileData.proxyId, camoufoxConfig: profileData.camoufoxConfig, groupId: profileData.groupId || (selectedGroupId !== "default" ? selectedGroupId : undefined), }); // No need to manually reload - useProfileEvents will handle the update } catch (error) { showErrorToast( `Failed to create profile: ${ error instanceof Error ? error.message : String(error) }`, ); throw error; } }, [selectedGroupId], ); const launchProfile = useCallback(async (profile: BrowserProfile) => { console.log("Starting launch for profile:", profile.name); try { const result = await invoke("launch_browser_profile", { profile, }); console.log("Successfully launched profile:", result.name); } catch (err: unknown) { console.error("Failed to launch browser:", err); const errorMessage = err instanceof Error ? err.message : String(err); showErrorToast(`Failed to launch browser: ${errorMessage}`); // Re-throw the error so the table component can handle loading state cleanup throw err; } }, []); const handleDeleteProfile = useCallback(async (profile: BrowserProfile) => { console.log("Attempting to delete profile:", profile.name); try { // First check if the browser is running for this profile const isRunning = await invoke("check_browser_status", { profile, }); if (isRunning) { showErrorToast( "Cannot delete profile while browser is running. Please stop the browser first.", ); return; } // Attempt to delete the profile await invoke("delete_profile", { profileId: profile.id }); console.log("Profile deletion command completed successfully"); // No need to manually reload - useProfileEvents will handle the update console.log("Profile deleted successfully"); } catch (err: unknown) { console.error("Failed to delete profile:", err); const errorMessage = err instanceof Error ? err.message : String(err); showErrorToast(`Failed to delete profile: ${errorMessage}`); } }, []); const handleRenameProfile = useCallback( async (profileId: string, newName: string) => { try { await invoke("rename_profile", { profileId, newName }); // No need to manually reload - useProfileEvents will handle the update } catch (err: unknown) { console.error("Failed to rename profile:", err); showErrorToast(`Failed to rename profile: ${JSON.stringify(err)}`); throw err; } }, [], ); const handleKillProfile = useCallback(async (profile: BrowserProfile) => { console.log("Starting kill for profile:", profile.name); try { await invoke("kill_browser_profile", { profile }); console.log("Successfully killed profile:", profile.name); // No need to manually reload - useProfileEvents will handle the update } catch (err: unknown) { console.error("Failed to kill browser:", err); const errorMessage = err instanceof Error ? err.message : String(err); showErrorToast(`Failed to kill browser: ${errorMessage}`); // Re-throw the error so the table component can handle loading state cleanup throw err; } }, []); const handleDeleteSelectedProfiles = useCallback( async (profileIds: string[]) => { try { await invoke("delete_selected_profiles", { profileIds }); // No need to manually reload - useProfileEvents will handle the update } catch (err: unknown) { console.error("Failed to delete selected profiles:", err); showErrorToast( `Failed to delete selected profiles: ${JSON.stringify(err)}`, ); } }, [], ); const handleAssignProfilesToGroup = useCallback((profileIds: string[]) => { setSelectedProfilesForGroup(profileIds); setGroupAssignmentDialogOpen(true); }, []); const handleBulkDelete = useCallback(() => { if (selectedProfiles.length === 0) return; setShowBulkDeleteConfirmation(true); }, [selectedProfiles]); const confirmBulkDelete = useCallback(async () => { if (selectedProfiles.length === 0) return; setIsBulkDeleting(true); try { await invoke("delete_selected_profiles", { profileIds: selectedProfiles, }); // No need to manually reload - useProfileEvents will handle the update setSelectedProfiles([]); setShowBulkDeleteConfirmation(false); } catch (error) { console.error("Failed to delete selected profiles:", error); showErrorToast( `Failed to delete selected profiles: ${JSON.stringify(error)}`, ); } finally { setIsBulkDeleting(false); } }, [selectedProfiles]); const handleBulkGroupAssignment = useCallback(() => { if (selectedProfiles.length === 0) return; handleAssignProfilesToGroup(selectedProfiles); setSelectedProfiles([]); }, [selectedProfiles, handleAssignProfilesToGroup]); const handleGroupAssignmentComplete = useCallback(async () => { // No need to manually reload - useProfileEvents will handle the update setGroupAssignmentDialogOpen(false); setSelectedProfilesForGroup([]); }, []); const handleGroupManagementComplete = useCallback(async () => { // No need to manually reload - useProfileEvents will handle the update }, []); useEffect(() => { // Check for startup default browser prompt void checkStartupPrompt(); // Listen for URL open events and get cleanup function const setupListeners = async () => { const cleanup = await listenForUrlEvents(); return cleanup; }; let cleanup: (() => void) | undefined; setupListeners().then((cleanupFn) => { cleanup = cleanupFn; }); // Check for startup URLs (when app was launched as default browser) void checkCurrentUrl(); // Set up periodic update checks (every 30 minutes) const updateInterval = setInterval( () => { void checkForUpdates(); }, 30 * 60 * 1000, ); // Check for missing binaries after initial profile load if (!profilesLoading && profiles.length > 0) { void checkMissingBinaries(); } return () => { clearInterval(updateInterval); if (cleanup) { cleanup(); } }; }, [ checkForUpdates, checkStartupPrompt, listenForUrlEvents, checkCurrentUrl, checkMissingBinaries, profilesLoading, profiles.length, ]); // Show deprecation warning for unsupported profiles (with names) useEffect(() => { if (profiles.length === 0) return; const deprecatedProfiles = profiles.filter( (p) => ["tor-browser", "mullvad-browser"].includes(p.browser) || (p.release_type === "nightly" && p.browser !== "firefox-developer"), ); if (deprecatedProfiles.length > 0) { const deprecatedNames = deprecatedProfiles.map((p) => p.name).join(", "); // Use a stable id to avoid duplicate toasts on re-renders showToast({ id: "deprecated-profiles-warning", type: "error", title: "Some profiles will be deprecated soon", description: `The following profiles will be deprecated soon: ${deprecatedNames}. Tor Browser, Mullvad Browser, and nightly profiles (except Firefox Developers Edition) will be removed in upcoming versions. Please check GitHub for migration instructions.`, duration: 15000, action: { label: "Learn more", onClick: () => { const event = new CustomEvent("url-open-request", { detail: "https://github.com/zhom/donutbrowser/discussions/66", }); window.dispatchEvent(event); }, }, }); } }, [profiles]); // Check permissions when they are initialized useEffect(() => { if (isInitialized) { void checkAllPermissions(); } }, [isInitialized, checkAllPermissions]); // Filter data by selected group and search query const filteredProfiles = useMemo(() => { let filtered = profiles; // Filter by group if (!selectedGroupId || selectedGroupId === "default") { filtered = profiles.filter((profile) => !profile.group_id); } else { filtered = profiles.filter( (profile) => profile.group_id === selectedGroupId, ); } // Filter by search query if (searchQuery.trim()) { const query = searchQuery.toLowerCase().trim(); filtered = filtered.filter((profile) => { // Search in profile name if (profile.name.toLowerCase().includes(query)) return true; // Search in browser name if (profile.browser.toLowerCase().includes(query)) return true; // Search in tags if (profile.tags?.some((tag) => tag.toLowerCase().includes(query))) return true; return false; }); } return filtered; }, [profiles, selectedGroupId, searchQuery]); // Update loading states const isLoading = profilesLoading || groupsLoading || proxiesLoading; return (
{isInitializing && (
Initializing
Please don't close the app
)} { setCreateProfileDialogOpen(false); }} onCreateProfile={handleCreateProfile} selectedGroupId={selectedGroupId} /> { setSettingsDialogOpen(false); }} /> { setImportProfileDialogOpen(false); }} /> { setProxyManagementDialogOpen(false); }} /> {pendingUrls.map((pendingUrl) => ( { setPendingUrls((prev) => prev.filter((u) => u.id !== pendingUrl.id), ); }} url={pendingUrl.url} isUpdating={isUpdating} runningProfiles={runningProfiles} /> ))} { setPermissionDialogOpen(false); }} permissionType={currentPermissionType} onPermissionGranted={checkNextPermission} /> { setCamoufoxConfigDialogOpen(false); }} profile={currentProfileForCamoufoxConfig} onSave={handleSaveCamoufoxConfig} /> { setGroupManagementDialogOpen(false); }} onGroupManagementComplete={handleGroupManagementComplete} /> { setGroupAssignmentDialogOpen(false); }} selectedProfiles={selectedProfilesForGroup} onAssignmentComplete={handleGroupAssignmentComplete} profiles={profiles} /> setShowBulkDeleteConfirmation(false)} onConfirm={confirmBulkDelete} title="Delete Selected Profiles" description={`This action cannot be undone. This will permanently delete ${selectedProfiles.length} profile${selectedProfiles.length !== 1 ? "s" : ""} and all associated data.`} confirmButtonText={`Delete ${selectedProfiles.length} Profile${selectedProfiles.length !== 1 ? "s" : ""}`} isLoading={isBulkDeleting} profileIds={selectedProfiles} profiles={profiles.map((p) => ({ id: p.id, name: p.name }))} />
); }