Files
tkmind_go/ui/desktop/src/App.tsx
T
callebtc dbbee1cdbf add encrypted Nostr session sharing (#8922)
Signed-off-by: callebtc <93376500+callebtc@users.noreply.github.com>
Signed-off-by: Douwe Osinga <douwe@squareup.com>
Signed-off-by: Michael Neale <michael.neale@gmail.com>
Co-authored-by: Douwe Osinga <douwe@squareup.com>
Co-authored-by: Michael Neale <michael.neale@gmail.com>
2026-05-13 03:06:44 +00:00

728 lines
24 KiB
TypeScript

import { useEffect, useState, useRef } from 'react';
import { IpcRendererEvent } from 'electron';
import {
HashRouter,
Routes,
Route,
useNavigate,
useLocation,
useSearchParams,
} from 'react-router-dom';
import { openSharedSessionFromDeepLink, importNostrSessionFromDeepLink } from './sessionLinks';
import { type SharedSessionDetails } from './sharedSessions';
import { ErrorUI } from './components/ErrorBoundary';
import { ExtensionInstallModal } from './components/ExtensionInstallModal';
import { toast, ToastContainer } from 'react-toastify';
import AnnouncementModal from './components/AnnouncementModal';
import TelemetryConsentPrompt from './components/TelemetryConsentPrompt';
import OnboardingGuard from './components/onboarding/OnboardingGuard';
import { createSession } from './sessions';
import { ChatType } from './types/chat';
import Hub from './components/Hub';
import { UserInput } from './types/message';
interface PairRouteState {
resumeSessionId?: string;
initialMessage?: UserInput;
}
import SettingsView, { SettingsViewOptions } from './components/settings/SettingsView';
import SessionsView from './components/sessions/SessionsView';
import SharedSessionView from './components/sessions/SharedSessionView';
import SchedulesView from './components/schedule/SchedulesView';
import ProviderSettings from './components/settings/providers/ProviderSettingsPage';
import { AppLayout } from './components/Layout/AppLayout';
import { ChatProvider, DEFAULT_CHAT_TITLE } from './contexts/ChatContext';
import LauncherView from './components/LauncherView';
import 'react-toastify/dist/ReactToastify.css';
import { useConfig } from './components/ConfigContext';
import { ModelAndProviderProvider } from './components/ModelAndProviderContext';
import { ThemeProvider } from './contexts/ThemeContext';
import { FeaturesProvider } from './contexts/FeaturesContext';
import PermissionSettingsView from './components/settings/permission/PermissionSetting';
import ExtensionsView, { ExtensionsViewOptions } from './components/extensions/ExtensionsView';
import RecipesView from './components/recipes/RecipesView';
import SkillsView from './components/skills/SkillsView';
import AppsView from './components/apps/AppsView';
import StandaloneAppView from './components/apps/StandaloneAppView';
import { View, ViewOptions } from './utils/navigationUtils';
import { useNavigation } from './hooks/useNavigation';
import { errorMessage } from './utils/conversionUtils';
import { getInitialWorkingDir } from './utils/workingDir';
import { usePageViewTracking } from './hooks/useAnalytics';
import { trackErrorWithContext } from './utils/analytics';
import { AppEvents } from './constants/events';
import { registerPlatformEventHandlers } from './utils/platform_events';
function PageViewTracker() {
usePageViewTracking();
return null;
}
// Route Components
const HubRouteWrapper = () => {
const setView = useNavigation();
return <Hub setView={setView} />;
};
export function resolveSessionInitialMessage(
session: { recipe?: { prompt?: string | null } | null },
initialMessage?: UserInput
): UserInput | undefined {
return (
initialMessage ??
(session.recipe?.prompt ? { msg: session.recipe.prompt, images: [] } : undefined)
);
}
const PairRouteWrapper = ({
activeSessions,
}: {
activeSessions: Array<{
sessionId: string;
initialMessage?: UserInput;
}>;
setActiveSessions: (sessions: Array<{ sessionId: string; initialMessage?: UserInput }>) => void;
}) => {
const { extensionsList } = useConfig();
const location = useLocation();
const routeState =
(location.state as PairRouteState) || (window.history.state as PairRouteState) || {};
const [searchParams, setSearchParams] = useSearchParams();
const [isCreatingSession, setIsCreatingSession] = useState(false);
const resumeSessionId = searchParams.get('resumeSessionId') ?? undefined;
const recipeDeeplinkFromConfig = window.appConfig?.get('recipeDeeplink') as string | undefined;
const recipeIdFromConfig = window.appConfig?.get('recipeId') as string | undefined;
const initialMessage = routeState.initialMessage;
// Create session if we have an initialMessage, recipeDeeplink, or recipeId but no sessionId
useEffect(() => {
if (
(initialMessage || recipeDeeplinkFromConfig || recipeIdFromConfig) &&
!resumeSessionId &&
!isCreatingSession
) {
setIsCreatingSession(true);
(async () => {
try {
const newSession = await createSession(getInitialWorkingDir(), {
recipeDeeplink: recipeDeeplinkFromConfig,
recipeId: recipeIdFromConfig,
allExtensions: extensionsList,
});
const sessionInitialMessage = resolveSessionInitialMessage(newSession, initialMessage);
window.dispatchEvent(
new CustomEvent(AppEvents.ADD_ACTIVE_SESSION, {
detail: {
sessionId: newSession.id,
initialMessage: sessionInitialMessage,
},
})
);
setSearchParams((prev) => {
prev.set('resumeSessionId', newSession.id);
return prev;
});
} catch (error) {
console.error('Failed to create session:', error);
trackErrorWithContext(error, {
component: 'PairRouteWrapper',
action: 'create_session',
recoverable: true,
});
} finally {
setIsCreatingSession(false);
}
})();
}
// Note: isCreatingSession is intentionally NOT in the dependency array
// It's only used as a guard to prevent concurrent session creation
// eslint-disable-next-line react-hooks/exhaustive-deps
}, [
initialMessage,
recipeDeeplinkFromConfig,
recipeIdFromConfig,
resumeSessionId,
setSearchParams,
extensionsList,
]);
// Add resumed session to active sessions if not already there
useEffect(() => {
if (resumeSessionId && !activeSessions.some((s) => s.sessionId === resumeSessionId)) {
window.dispatchEvent(
new CustomEvent(AppEvents.ADD_ACTIVE_SESSION, {
detail: {
sessionId: resumeSessionId,
initialMessage: initialMessage,
},
})
);
}
}, [resumeSessionId, activeSessions, initialMessage]);
return null;
};
const SettingsRoute = ({ activeSessionId }: { activeSessionId?: string }) => {
const location = useLocation();
const navigate = useNavigate();
const [searchParams] = useSearchParams();
const setView = useNavigation();
// Get viewOptions from location.state, history.state, or URL search params
const viewOptions =
(location.state as SettingsViewOptions) || (window.history.state as SettingsViewOptions) || {};
// If section is provided via URL search params, add it to viewOptions
const sectionFromUrl = searchParams.get('section');
if (sectionFromUrl) {
viewOptions.section = sectionFromUrl;
}
return (
<SettingsView
onClose={() => navigate('/')}
setView={setView}
viewOptions={{ ...viewOptions, sessionId: activeSessionId }}
/>
);
};
const SessionsRoute = () => {
return <SessionsView />;
};
const SchedulesRoute = () => {
const navigate = useNavigate();
return <SchedulesView onClose={() => navigate('/')} />;
};
const RecipesRoute = () => {
return <RecipesView />;
};
const SkillsRoute = () => {
return <SkillsView />;
};
const PermissionRoute = () => {
const location = useLocation();
const navigate = useNavigate();
const parentView = location.state?.parentView as View;
const parentViewOptions = location.state?.parentViewOptions as ViewOptions;
return (
<PermissionSettingsView
onClose={() => {
// Navigate back to parent view with options
switch (parentView) {
case 'chat':
navigate('/');
break;
case 'pair':
navigate('/pair');
break;
case 'settings':
navigate('/settings', { state: parentViewOptions });
break;
case 'sessions':
navigate('/sessions');
break;
case 'schedules':
navigate('/schedules');
break;
case 'recipes':
navigate('/recipes');
break;
case 'skills':
navigate('/skills');
break;
default:
navigate('/');
}
}}
/>
);
};
const ConfigureProvidersRoute = () => {
const navigate = useNavigate();
return (
<div className="w-screen h-screen bg-background-primary">
<ProviderSettings
onClose={() => navigate('/settings', { state: { section: 'models' } })}
isOnboarding={false}
/>
</div>
);
};
// Wrapper component for SharedSessionRoute to access parent state
const SharedSessionRouteWrapper = ({
isLoadingSharedSession,
setIsLoadingSharedSession,
sharedSessionError,
}: {
isLoadingSharedSession: boolean;
setIsLoadingSharedSession: (loading: boolean) => void;
sharedSessionError: string | null;
}) => {
const location = useLocation();
const setView = useNavigation();
const historyState = window.history.state;
const sessionDetails = (location.state?.sessionDetails ||
historyState?.sessionDetails) as SharedSessionDetails | null;
const error = location.state?.error || historyState?.error || sharedSessionError;
const shareToken = location.state?.shareToken || historyState?.shareToken;
const baseUrl = location.state?.baseUrl || historyState?.baseUrl;
return (
<SharedSessionView
session={sessionDetails}
isLoading={isLoadingSharedSession}
error={error}
onRetry={async () => {
if (shareToken && baseUrl) {
setIsLoadingSharedSession(true);
try {
await openSharedSessionFromDeepLink(`goose://sessions/${shareToken}`, setView, baseUrl);
} catch (error) {
console.error('Failed to retry loading shared session:', error);
} finally {
setIsLoadingSharedSession(false);
}
}
}}
/>
);
};
const ExtensionsRoute = () => {
const navigate = useNavigate();
const location = useLocation();
// Get viewOptions from location.state or history.state (for deep link extensions)
const viewOptions =
(location.state as ExtensionsViewOptions) ||
(window.history.state as ExtensionsViewOptions) ||
{};
return (
<ExtensionsView
onClose={() => navigate(-1)}
setView={(view, options) => {
switch (view) {
case 'chat':
navigate('/');
break;
case 'pair':
navigate('/pair', { state: options });
break;
case 'settings':
navigate('/settings', { state: options });
break;
default:
navigate('/');
}
}}
viewOptions={viewOptions}
/>
);
};
export function AppInner() {
const [fatalError, setFatalError] = useState<string | null>(null);
const [isLoadingSharedSession, setIsLoadingSharedSession] = useState(false);
const [sharedSessionError, setSharedSessionError] = useState<string | null>(null);
const navigate = useNavigate();
const setView = useNavigation();
const [chat, setChat] = useState<ChatType>({
sessionId: '',
name: DEFAULT_CHAT_TITLE,
messages: [],
recipe: null,
});
const MAX_ACTIVE_SESSIONS = 10;
const [activeSessions, setActiveSessions] = useState<
Array<{ sessionId: string; initialMessage?: UserInput }>
>([]);
useEffect(() => {
const handleAddActiveSession = (event: Event) => {
const { sessionId, initialMessage } = (
event as CustomEvent<{
sessionId: string;
initialMessage?: UserInput;
}>
).detail;
setActiveSessions((prev) => {
const existingIndex = prev.findIndex((s) => s.sessionId === sessionId);
if (existingIndex !== -1) {
// Session exists - move to end of LRU list (most recently used)
const existing = prev[existingIndex];
return [...prev.slice(0, existingIndex), ...prev.slice(existingIndex + 1), existing];
}
// New session - add to end with LRU eviction if needed
const newSession = { sessionId, initialMessage };
const updated = [...prev, newSession];
if (updated.length > MAX_ACTIVE_SESSIONS) {
return updated.slice(updated.length - MAX_ACTIVE_SESSIONS);
}
return updated;
});
};
const handleClearInitialMessage = (event: Event) => {
const { sessionId } = (event as CustomEvent<{ sessionId: string }>).detail;
setActiveSessions((prev) => {
return prev.map((session) => {
if (session.sessionId === sessionId) {
return { ...session, initialMessage: undefined };
}
return session;
});
});
};
window.addEventListener(AppEvents.ADD_ACTIVE_SESSION, handleAddActiveSession);
window.addEventListener(AppEvents.CLEAR_INITIAL_MESSAGE, handleClearInitialMessage);
return () => {
window.removeEventListener(AppEvents.ADD_ACTIVE_SESSION, handleAddActiveSession);
window.removeEventListener(AppEvents.CLEAR_INITIAL_MESSAGE, handleClearInitialMessage);
};
}, []);
const { addExtension } = useConfig();
useEffect(() => {
try {
window.electron.reactReady();
} catch (error) {
console.error('Error sending reactReady:', error);
setFatalError(`React ready notification failed: ${errorMessage(error, 'Unknown error')}`);
}
}, []);
useEffect(() => {
const handleOpenSharedSession = async (_event: IpcRendererEvent, ...args: unknown[]) => {
const link = args[0] as string;
window.electron.logInfo(`Opening shared session from deep link ${link}`);
setIsLoadingSharedSession(true);
setSharedSessionError(null);
try {
if (link.startsWith('goose://sessions/nostr')) {
await importNostrSessionFromDeepLink(link);
navigate('/sessions');
return;
}
await openSharedSessionFromDeepLink(link, (_view: View, options?: ViewOptions) => {
navigate('/shared-session', { state: options });
});
} catch (error) {
console.error('Unexpected error opening shared session:', error);
trackErrorWithContext(error, {
component: 'AppInner',
action: 'open_shared_session',
recoverable: true,
});
if (link.startsWith('goose://sessions/nostr')) {
toast.error(`Failed to import Nostr session: ${errorMessage(error, 'Unknown error')}`);
navigate('/sessions');
} else {
const shareToken = link.replace('goose://sessions/', '');
const options = {
sessionDetails: null,
error: errorMessage(error, 'Unknown error'),
shareToken,
};
navigate('/shared-session', { state: options });
}
} finally {
setIsLoadingSharedSession(false);
}
};
window.electron.on('open-shared-session', handleOpenSharedSession);
return () => {
window.electron.off('open-shared-session', handleOpenSharedSession);
};
}, [navigate]);
useEffect(() => {
const handleKeyDown = (event: KeyboardEvent) => {
const isMac = window.electron.platform === 'darwin';
if ((isMac ? event.metaKey : event.ctrlKey) && event.key === 'n') {
event.preventDefault();
try {
window.electron.createChatWindow({ dir: getInitialWorkingDir() });
} catch (error) {
console.error('Error creating new window:', error);
}
}
};
window.addEventListener('keydown', handleKeyDown);
return () => {
window.removeEventListener('keydown', handleKeyDown);
};
}, []);
// Show a toast if mesh is the configured provider but isn't running.
useEffect(() => {
const handler = () => {
toast.warn('Inference Mesh is set as your provider but isn\'t running. Open Settings → Mesh to start it. Keep goose running to stay connected.', {
autoClose: false,
toastId: 'mesh-not-running',
});
};
window.electron.on('mesh-not-running', handler);
return () => { window.electron.off('mesh-not-running', handler); };
}, []);
// Prevent default drag and drop behavior globally to avoid opening files in new windows
// but allow our React components to handle drops in designated areas
useEffect(() => {
const preventDefaults = (e: globalThis.DragEvent) => {
// Only prevent default if we're not over a designated drop zone
const target = e.target as HTMLElement;
const isOverDropZone = target.closest('[data-drop-zone="true"]') !== null;
if (!isOverDropZone) {
e.preventDefault();
e.stopPropagation();
}
};
const handleDragOver = (e: globalThis.DragEvent) => {
// Always prevent default for dragover to allow dropping
e.preventDefault();
e.stopPropagation();
};
const handleDrop = (e: globalThis.DragEvent) => {
// Only prevent default if we're not over a designated drop zone
const target = e.target as HTMLElement;
const isOverDropZone = target.closest('[data-drop-zone="true"]') !== null;
if (!isOverDropZone) {
e.preventDefault();
e.stopPropagation();
}
};
// Add event listeners to document to catch drag events
document.addEventListener('dragenter', preventDefaults, false);
document.addEventListener('dragleave', preventDefaults, false);
document.addEventListener('dragover', handleDragOver, false);
document.addEventListener('drop', handleDrop, false);
return () => {
document.removeEventListener('dragenter', preventDefaults, false);
document.removeEventListener('dragleave', preventDefaults, false);
document.removeEventListener('dragover', handleDragOver, false);
document.removeEventListener('drop', handleDrop, false);
};
}, []);
useEffect(() => {
const handleFatalError = (_event: IpcRendererEvent, ...args: unknown[]) => {
const errorMessage = args[0] as string;
console.error('Encountered a fatal error:', errorMessage);
setFatalError(errorMessage);
};
window.electron.on('fatal-error', handleFatalError);
return () => {
window.electron.off('fatal-error', handleFatalError);
};
}, []);
useEffect(() => {
const handleSetView = (_event: IpcRendererEvent, ...args: unknown[]) => {
const newView = args[0] as View;
const section = args[1] as string | undefined;
if (section && newView === 'settings') {
navigate(`/settings?section=${section}`);
} else {
navigate(`/${newView}`);
}
};
window.electron.on('set-view', handleSetView);
return () => window.electron.off('set-view', handleSetView);
}, [navigate]);
useEffect(() => {
const handleNewChat = (_event: IpcRendererEvent, ..._args: unknown[]) => {
window.dispatchEvent(new CustomEvent(AppEvents.TRIGGER_NEW_CHAT));
};
window.electron.on('new-chat', handleNewChat);
return () => window.electron.off('new-chat', handleNewChat);
}, []);
useEffect(() => {
const handleFocusInput = (_event: IpcRendererEvent, ..._args: unknown[]) => {
const inputField = document.querySelector('input[type="text"], textarea') as HTMLInputElement;
if (inputField) {
inputField.focus();
}
};
window.electron.on('focus-input', handleFocusInput);
return () => {
window.electron.off('focus-input', handleFocusInput);
};
}, []);
// Handle initial message from launcher
const isProcessingRef = useRef(false);
useEffect(() => {
const handleSetInitialMessage = async (_event: IpcRendererEvent, ...args: unknown[]) => {
const initialMessage = args[0] as string;
if (initialMessage && !isProcessingRef.current) {
isProcessingRef.current = true;
navigate('/pair', {
state: {
initialMessage: { msg: initialMessage, images: [] },
},
});
setTimeout(() => {
isProcessingRef.current = false;
}, 1000);
}
};
window.electron.on('set-initial-message', handleSetInitialMessage);
return () => {
window.electron.off('set-initial-message', handleSetInitialMessage);
};
}, [navigate]);
// Register platform event handlers for app lifecycle management
useEffect(() => {
return registerPlatformEventHandlers();
}, []);
if (fatalError) {
return <ErrorUI error={errorMessage(fatalError)} />;
}
return (
<>
<PageViewTracker />
<ToastContainer
aria-label="Toast notifications"
toastClassName={() =>
`relative min-h-16 mb-4 p-2 rounded-lg
flex justify-between overflow-hidden cursor-pointer
text-text-inverse bg-background-inverse
`
}
style={{ width: '450px' }}
className="mt-6"
position="top-right"
autoClose={3000}
closeOnClick
pauseOnHover
/>
<ExtensionInstallModal addExtension={addExtension} setView={setView} />
<div className="relative w-screen h-screen overflow-hidden bg-background-secondary flex flex-col">
<div className="titlebar-drag-region" />
<div style={{ position: 'relative', width: '100%', height: '100%' }}>
<Routes>
<Route path="launcher" element={<LauncherView />} />
<Route path="configure-providers" element={<ConfigureProvidersRoute />} />
<Route path="standalone-app" element={<StandaloneAppView />} />
<Route
path="/"
element={
<OnboardingGuard>
<ChatProvider chat={chat} setChat={setChat} contextKey="hub">
<AppLayout activeSessions={activeSessions} />
</ChatProvider>
</OnboardingGuard>
}
>
<Route index element={<HubRouteWrapper />} />
<Route
path="pair"
element={
<PairRouteWrapper
activeSessions={activeSessions}
setActiveSessions={setActiveSessions}
/>
}
/>
<Route
path="settings"
element={
<SettingsRoute
activeSessionId={activeSessions[activeSessions.length - 1]?.sessionId}
/>
}
/>
<Route
path="extensions"
element={
<ChatProvider chat={chat} setChat={setChat} contextKey="extensions">
<ExtensionsRoute />
</ChatProvider>
}
/>
<Route path="apps" element={<AppsView />} />
<Route path="sessions" element={<SessionsRoute />} />
<Route path="schedules" element={<SchedulesRoute />} />
<Route path="recipes" element={<RecipesRoute />} />
<Route path="skills" element={<SkillsRoute />} />
<Route
path="shared-session"
element={
<SharedSessionRouteWrapper
isLoadingSharedSession={isLoadingSharedSession}
setIsLoadingSharedSession={setIsLoadingSharedSession}
sharedSessionError={sharedSessionError}
/>
}
/>
<Route path="permission" element={<PermissionRoute />} />
</Route>
</Routes>
</div>
</div>
</>
);
}
export default function App() {
return (
<ThemeProvider>
<FeaturesProvider>
<ModelAndProviderProvider>
<HashRouter>
<AppInner />
</HashRouter>
<AnnouncementModal />
<TelemetryConsentPrompt />
</ModelAndProviderProvider>
</FeaturesProvider>
</ThemeProvider>
);
}