Improvements for sidebar

This commit is contained in:
Utkarsh-Patel-13 2025-08-02 21:20:36 -07:00
parent d98dfd40b5
commit 8bc369cd94
10 changed files with 560 additions and 373 deletions

View file

@ -6,7 +6,13 @@ from sqlalchemy.ext.asyncio import AsyncSession
from sqlalchemy.future import select from sqlalchemy.future import select
from app.db import Chat, SearchSpace, User, get_async_session from app.db import Chat, SearchSpace, User, get_async_session
from app.schemas import AISDKChatRequest, ChatCreate, ChatRead, ChatUpdate from app.schemas import (
AISDKChatRequest,
ChatCreate,
ChatRead,
ChatReadWithoutMessages,
ChatUpdate,
)
from app.tasks.stream_connector_search_results import stream_connector_search_results from app.tasks.stream_connector_search_results import stream_connector_search_results
from app.users import current_active_user from app.users import current_active_user
from app.utils.check_ownership import check_ownership from app.utils.check_ownership import check_ownership
@ -112,7 +118,7 @@ async def create_chat(
) from None ) from None
@router.get("/chats/", response_model=list[ChatRead]) @router.get("/chats/", response_model=list[ChatReadWithoutMessages])
async def read_chats( async def read_chats(
skip: int = 0, skip: int = 0,
limit: int = 100, limit: int = 100,
@ -121,14 +127,26 @@ async def read_chats(
user: User = Depends(current_active_user), user: User = Depends(current_active_user),
): ):
try: try:
query = select(Chat).join(SearchSpace).filter(SearchSpace.user_id == user.id) # Select specific fields excluding messages
query = (
select(
Chat.id,
Chat.type,
Chat.title,
Chat.initial_connectors,
Chat.search_space_id,
Chat.created_at,
)
.join(SearchSpace)
.filter(SearchSpace.user_id == user.id)
)
# Filter by search_space_id if provided # Filter by search_space_id if provided
if search_space_id is not None: if search_space_id is not None:
query = query.filter(Chat.search_space_id == search_space_id) query = query.filter(Chat.search_space_id == search_space_id)
result = await session.execute(query.offset(skip).limit(limit)) result = await session.execute(query.offset(skip).limit(limit))
return result.scalars().all() return result.all()
except OperationalError: except OperationalError:
raise HTTPException( raise HTTPException(
status_code=503, detail="Database operation failed. Please try again later." status_code=503, detail="Database operation failed. Please try again later."

View file

@ -1,5 +1,12 @@
from .base import IDModel, TimestampModel from .base import IDModel, TimestampModel
from .chats import AISDKChatRequest, ChatBase, ChatCreate, ChatRead, ChatUpdate from .chats import (
AISDKChatRequest,
ChatBase,
ChatCreate,
ChatRead,
ChatReadWithoutMessages,
ChatUpdate,
)
from .chunks import ChunkBase, ChunkCreate, ChunkRead, ChunkUpdate from .chunks import ChunkBase, ChunkCreate, ChunkRead, ChunkUpdate
from .documents import ( from .documents import (
DocumentBase, DocumentBase,
@ -37,6 +44,7 @@ __all__ = [
"ChatBase", "ChatBase",
"ChatCreate", "ChatCreate",
"ChatRead", "ChatRead",
"ChatReadWithoutMessages",
"ChatUpdate", "ChatUpdate",
"ChunkBase", "ChunkBase",
"ChunkCreate", "ChunkCreate",

View file

@ -15,6 +15,12 @@ class ChatBase(BaseModel):
search_space_id: int search_space_id: int
class ChatBaseWithoutMessages(BaseModel):
type: ChatType
title: str
search_space_id: int
class ClientAttachment(BaseModel): class ClientAttachment(BaseModel):
name: str name: str
content_type: str content_type: str
@ -50,3 +56,7 @@ class ChatUpdate(ChatBase):
class ChatRead(ChatBase, IDModel, TimestampModel): class ChatRead(ChatBase, IDModel, TimestampModel):
model_config = ConfigDict(from_attributes=True) model_config = ConfigDict(from_attributes=True)
class ChatReadWithoutMessages(ChatBaseWithoutMessages, IDModel, TimestampModel):
model_config = ConfigDict(from_attributes=True)

View file

@ -19,14 +19,7 @@ import { useEffect, useState } from "react";
import { toast } from "sonner"; import { toast } from "sonner";
import { Badge } from "@/components/ui/badge"; import { Badge } from "@/components/ui/badge";
import { Button } from "@/components/ui/button"; import { Button } from "@/components/ui/button";
import { import { Card, CardDescription, CardFooter, CardHeader, CardTitle } from "@/components/ui/card";
Card,
CardContent,
CardDescription,
CardFooter,
CardHeader,
CardTitle,
} from "@/components/ui/card";
import { import {
Dialog, Dialog,
DialogContent, DialogContent,
@ -61,24 +54,16 @@ import {
SelectTrigger, SelectTrigger,
SelectValue, SelectValue,
} from "@/components/ui/select"; } from "@/components/ui/select";
import { cn } from "@/lib/utils";
interface Chat { interface Chat {
created_at: string; created_at: string;
id: number; id: number;
type: string; type: string;
title: string; title: string;
messages: ChatMessage[];
search_space_id: number; search_space_id: number;
} }
interface ChatMessage {
id: string;
createdAt: string;
role: string;
content: string;
parts?: any;
}
interface ChatsPageClientProps { interface ChatsPageClientProps {
searchSpaceId: string; searchSpaceId: string;
} }
@ -580,12 +565,12 @@ export default function ChatsPageClient({ searchSpaceId }: ChatsPageClientProps)
animate="animate" animate="animate"
exit="exit" exit="exit"
transition={{ duration: 0.2, delay: index * 0.05 }} transition={{ duration: 0.2, delay: index * 0.05 }}
className={`overflow-hidden hover:shadow-md transition-shadow className={cn(
${ "overflow-hidden hover:shadow-md transition-shadow",
selectionMode && selectedChats.includes(chat.id) selectionMode && selectedChats.includes(chat.id)
? "ring-2 ring-primary ring-offset-2" ? "ring-2 ring-primary ring-offset-2"
: "" : ""
}`} )}
onClick={(e) => { onClick={(e) => {
if (!selectionMode) return; if (!selectionMode) return;
// Ignore clicks coming from interactive elements // Ignore clicks coming from interactive elements
@ -672,24 +657,21 @@ export default function ChatsPageClient({ searchSpaceId }: ChatsPageClientProps)
)} )}
</div> </div>
</CardHeader> </CardHeader>
<CardContent>
<div className="text-sm text-muted-foreground line-clamp-3"> <CardFooter className="flex items-center justify-between gap-2 w-full">
{chat.messages && chat.messages.length > 0
? typeof chat.messages[0] === "string"
? chat.messages[0]
: chat.messages[0]?.content || "No message content"
: "No messages in this chat."}
</div>
</CardContent>
<CardFooter className="flex justify-between pt-2">
<div className="flex items-center text-xs text-muted-foreground">
<MessageCircleMore className="mr-1 h-3.5 w-3.5" />
<span>{chat.messages?.length || 0} messages</span>
</div>
<Badge variant="secondary" className="text-xs"> <Badge variant="secondary" className="text-xs">
<Tag className="mr-1 h-3 w-3" /> <Tag className="mr-1 h-3 w-3" />
{chat.type || "Unknown"} {chat.type || "Unknown"}
</Badge> </Badge>
<Button
size="sm"
onClick={() =>
router.push(`/dashboard/${chat.search_space_id}/researcher/${chat.id}`)
}
>
<MessageCircleMore className="h-4 w-4" />
<span>View Chat</span>
</Button>
</CardFooter> </CardFooter>
</MotionCard> </MotionCard>
))} ))}

View file

@ -1,7 +1,7 @@
"use client"; "use client";
import { Trash2 } from "lucide-react"; import { AlertCircle, Trash2 } from "lucide-react";
import { useEffect, useState } from "react"; import { useCallback, useEffect, useMemo, useState } from "react";
import { AppSidebar } from "@/components/sidebar/app-sidebar"; import { AppSidebar } from "@/components/sidebar/app-sidebar";
import { Button } from "@/components/ui/button"; import { Button } from "@/components/ui/button";
import { import {
@ -12,7 +12,7 @@ import {
DialogHeader, DialogHeader,
DialogTitle, DialogTitle,
} from "@/components/ui/dialog"; } from "@/components/ui/dialog";
import { apiClient } from "@/lib/api"; // Import the API client import { apiClient } from "@/lib/api";
interface Chat { interface Chat {
created_at: string; created_at: string;
@ -50,6 +50,26 @@ interface AppSidebarProviderProps {
}[]; }[];
} }
// Loading skeleton component
const LoadingSkeleton = () => (
<div className="space-y-2 p-2">
{Array.from({ length: 3 }).map((_, i) => (
<div key={i} className="h-8 bg-muted animate-pulse rounded-md" />
))}
</div>
);
// Error state component
const ErrorState = ({ error, onRetry }: { error: string; onRetry: () => void }) => (
<div className="p-4 text-center">
<AlertCircle className="h-8 w-8 text-destructive mx-auto mb-2" />
<p className="text-sm text-muted-foreground mb-2">{error}</p>
<Button variant="outline" size="sm" onClick={onRetry}>
Retry
</Button>
</div>
);
export function AppSidebarProvider({ export function AppSidebarProvider({
searchSpaceId, searchSpaceId,
navSecondary, navSecondary,
@ -80,15 +100,12 @@ export function AppSidebarProvider({
setIsClient(true); setIsClient(true);
}, []); }, []);
// Fetch recent chats // Memoized fetch function for chats
useEffect(() => { const fetchRecentChats = useCallback(async () => {
const fetchRecentChats = async () => {
try { try {
// Only run on client-side // Only run on client-side
if (typeof window === "undefined") return; if (typeof window === "undefined") return;
try {
// Use the API client instead of direct fetch - filter by current search space ID
const chats: Chat[] = await apiClient.get<Chat[]>( const chats: Chat[] = await apiClient.get<Chat[]>(
`api/v1/chats/?limit=5&skip=0&search_space_id=${searchSpaceId}` `api/v1/chats/?limit=5&skip=0&search_space_id=${searchSpaceId}`
); );
@ -97,22 +114,15 @@ export function AppSidebarProvider({
const sortedChats = chats.sort( const sortedChats = chats.sort(
(a, b) => new Date(b.created_at).getTime() - new Date(a.created_at).getTime() (a, b) => new Date(b.created_at).getTime() - new Date(a.created_at).getTime()
); );
// console.log("sortedChats", sortedChats);
// Transform API response to the format expected by AppSidebar // Transform API response to the format expected by AppSidebar
const formattedChats = sortedChats.map((chat) => ({ const formattedChats = sortedChats.map((chat) => ({
name: chat.title || `Chat ${chat.id}`, // Fallback if title is empty name: chat.title || `Chat ${chat.id}`,
url: `/dashboard/${chat.search_space_id}/researcher/${chat.id}`, url: `/dashboard/${chat.search_space_id}/researcher/${chat.id}`,
icon: "MessageCircleMore", icon: "MessageCircleMore",
id: chat.id, id: chat.id,
search_space_id: chat.search_space_id, search_space_id: chat.search_space_id,
actions: [ actions: [
{
name: "View Details",
icon: "ExternalLink",
onClick: () => {
window.location.href = `/dashboard/${chat.search_space_id}/researcher/${chat.id}`;
},
},
{ {
name: "Delete", name: "Delete",
icon: "Trash2", icon: "Trash2",
@ -129,56 +139,18 @@ export function AppSidebarProvider({
} catch (error) { } catch (error) {
console.error("Error fetching chats:", error); console.error("Error fetching chats:", error);
setChatError(error instanceof Error ? error.message : "Unknown error occurred"); setChatError(error instanceof Error ? error.message : "Unknown error occurred");
// Provide empty array to ensure UI still renders
setRecentChats([]); setRecentChats([]);
} finally { } finally {
setIsLoadingChats(false); setIsLoadingChats(false);
} }
} catch (error) {
console.error("Error in fetchRecentChats:", error);
setIsLoadingChats(false);
}
};
fetchRecentChats();
// Set up a refresh interval (every 5 minutes)
const intervalId = setInterval(fetchRecentChats, 5 * 60 * 1000);
// Clean up interval on component unmount
return () => clearInterval(intervalId);
}, [searchSpaceId]); }, [searchSpaceId]);
// Handle delete chat // Memoized fetch function for search space
const handleDeleteChat = async () => { const fetchSearchSpace = useCallback(async () => {
if (!chatToDelete) return;
try {
setIsDeleting(true);
// Use the API client instead of direct fetch
await apiClient.delete(`api/v1/chats/${chatToDelete.id}`);
// Close dialog and refresh chats
setRecentChats(recentChats.filter((chat) => chat.id !== chatToDelete.id));
} catch (error) {
console.error("Error deleting chat:", error);
} finally {
setIsDeleting(false);
setShowDeleteDialog(false);
setChatToDelete(null);
}
};
// Fetch search space details
useEffect(() => {
const fetchSearchSpace = async () => {
try { try {
// Only run on client-side // Only run on client-side
if (typeof window === "undefined") return; if (typeof window === "undefined") return;
try {
// Use the API client instead of direct fetch
const data: SearchSpace = await apiClient.get<SearchSpace>( const data: SearchSpace = await apiClient.get<SearchSpace>(
`api/v1/searchspaces/${searchSpaceId}` `api/v1/searchspaces/${searchSpaceId}`
); );
@ -190,38 +162,101 @@ export function AppSidebarProvider({
} finally { } finally {
setIsLoadingSearchSpace(false); setIsLoadingSearchSpace(false);
} }
} catch (error) {
console.error("Error in fetchSearchSpace:", error);
setIsLoadingSearchSpace(false);
}
};
fetchSearchSpace();
}, [searchSpaceId]); }, [searchSpaceId]);
// Create a fallback chat if there's an error or no chats // Retry function
const fallbackChats = const retryFetch = useCallback(() => {
chatError || (!isLoadingChats && recentChats.length === 0) setChatError(null);
? [ setSearchSpaceError(null);
setIsLoadingChats(true);
setIsLoadingSearchSpace(true);
fetchRecentChats();
fetchSearchSpace();
}, [fetchRecentChats, fetchSearchSpace]);
// Fetch recent chats
useEffect(() => {
fetchRecentChats();
// Set up a refresh interval (every 5 minutes)
const intervalId = setInterval(fetchRecentChats, 5 * 60 * 1000);
// Clean up interval on component unmount
return () => clearInterval(intervalId);
}, [fetchRecentChats]);
// Fetch search space details
useEffect(() => {
fetchSearchSpace();
}, [fetchSearchSpace]);
// Handle delete chat with better error handling
const handleDeleteChat = useCallback(async () => {
if (!chatToDelete) return;
try {
setIsDeleting(true);
await apiClient.delete(`api/v1/chats/${chatToDelete.id}`);
// Update local state
setRecentChats((prev) => prev.filter((chat) => chat.id !== chatToDelete.id));
} catch (error) {
console.error("Error deleting chat:", error);
// You could show a toast notification here
} finally {
setIsDeleting(false);
setShowDeleteDialog(false);
setChatToDelete(null);
}
}, [chatToDelete]);
// Memoized fallback chats
const fallbackChats = useMemo(() => {
if (chatError) {
return [
{ {
name: chatError ? "Error loading chats" : "No recent chats", name: "Error loading chats",
url: "#", url: "#",
icon: chatError ? "AlertCircle" : "MessageCircleMore", icon: "AlertCircle",
id: 0,
search_space_id: Number(searchSpaceId),
actions: [
{
name: "Retry",
icon: "RefreshCw",
onClick: retryFetch,
},
],
},
];
}
if (!isLoadingChats && recentChats.length === 0) {
return [
{
name: "No recent chats",
url: "#",
icon: "MessageCircleMore",
id: 0, id: 0,
search_space_id: Number(searchSpaceId), search_space_id: Number(searchSpaceId),
actions: [], actions: [],
}, },
] ];
: []; }
return [];
}, [chatError, isLoadingChats, recentChats.length, searchSpaceId, retryFetch]);
// Use fallback chats if there's an error or no chats // Use fallback chats if there's an error or no chats
const displayChats = recentChats.length > 0 ? recentChats : fallbackChats; const displayChats = recentChats.length > 0 ? recentChats : fallbackChats;
// Update the first item in navSecondary to show the search space name // Memoized updated navSecondary
const updatedNavSecondary = [...navSecondary]; const updatedNavSecondary = useMemo(() => {
if (updatedNavSecondary.length > 0 && isClient) { const updated = [...navSecondary];
updatedNavSecondary[0] = { if (updated.length > 0 && isClient) {
...updatedNavSecondary[0], updated[0] = {
...updated[0],
title: title:
searchSpace?.name || searchSpace?.name ||
(isLoadingSearchSpace (isLoadingSearchSpace
@ -231,17 +266,19 @@ export function AppSidebarProvider({
: "Unknown Search Space"), : "Unknown Search Space"),
}; };
} }
return updated;
}, [navSecondary, isClient, searchSpace?.name, isLoadingSearchSpace, searchSpaceError]);
// Show loading state if not client-side
if (!isClient) {
return <AppSidebar navSecondary={navSecondary} navMain={navMain} RecentChats={[]} />;
}
return ( return (
<> <>
<AppSidebar <AppSidebar navSecondary={updatedNavSecondary} navMain={navMain} RecentChats={displayChats} />
navSecondary={updatedNavSecondary}
navMain={navMain}
RecentChats={isClient ? displayChats : []}
/>
{/* Delete Confirmation Dialog - Only render on client */} {/* Delete Confirmation Dialog */}
{isClient && (
<Dialog open={showDeleteDialog} onOpenChange={setShowDeleteDialog}> <Dialog open={showDeleteDialog} onOpenChange={setShowDeleteDialog}>
<DialogContent className="sm:max-w-md"> <DialogContent className="sm:max-w-md">
<DialogHeader> <DialogHeader>
@ -284,7 +321,6 @@ export function AppSidebarProvider({
</DialogFooter> </DialogFooter>
</DialogContent> </DialogContent>
</Dialog> </Dialog>
)}
</> </>
); );
} }

View file

@ -17,15 +17,17 @@ import {
Trash2, Trash2,
Undo2, Undo2,
} from "lucide-react"; } from "lucide-react";
import { useMemo } from "react"; import { memo, useMemo } from "react";
import { Logo } from "@/components/Logo"; import { Logo } from "@/components/Logo";
import { NavMain } from "@/components/sidebar/nav-main"; import { NavMain } from "@/components/sidebar/nav-main";
import { NavProjects } from "@/components/sidebar/nav-projects"; import { NavProjects } from "@/components/sidebar/nav-projects";
import { NavSecondary } from "@/components/sidebar/nav-secondary"; import { NavSecondary } from "@/components/sidebar/nav-secondary";
import { NavUser } from "@/components/sidebar/nav-user";
import { import {
Sidebar, Sidebar,
SidebarContent, SidebarContent,
SidebarFooter,
SidebarHeader, SidebarHeader,
SidebarMenu, SidebarMenu,
SidebarMenuButton, SidebarMenuButton,
@ -64,7 +66,6 @@ const defaultData = {
isActive: true, isActive: true,
items: [], items: [],
}, },
{ {
title: "Documents", title: "Documents",
url: "#", url: "#",
@ -154,12 +155,12 @@ interface AppSidebarProps extends React.ComponentProps<typeof Sidebar> {
navSecondary?: { navSecondary?: {
title: string; title: string;
url: string; url: string;
icon: string; // Changed to string (icon name) icon: string;
}[]; }[];
RecentChats?: { RecentChats?: {
name: string; name: string;
url: string; url: string;
icon: string; // Changed to string (icon name) icon: string;
id?: number; id?: number;
search_space_id?: number; search_space_id?: number;
actions?: { actions?: {
@ -168,19 +169,26 @@ interface AppSidebarProps extends React.ComponentProps<typeof Sidebar> {
onClick: () => void; onClick: () => void;
}[]; }[];
}[]; }[];
user?: {
name: string;
email: string;
avatar: string;
};
} }
export function AppSidebar({ // Memoized AppSidebar component for better performance
export const AppSidebar = memo(function AppSidebar({
navMain = defaultData.navMain, navMain = defaultData.navMain,
navSecondary = defaultData.navSecondary, navSecondary = defaultData.navSecondary,
RecentChats = defaultData.RecentChats, RecentChats = defaultData.RecentChats,
user = defaultData.user,
...props ...props
}: AppSidebarProps) { }: AppSidebarProps) {
// Process navMain to resolve icon names to components // Process navMain to resolve icon names to components
const processedNavMain = useMemo(() => { const processedNavMain = useMemo(() => {
return navMain.map((item) => ({ return navMain.map((item) => ({
...item, ...item,
icon: iconMap[item.icon] || SquareTerminal, // Fallback to SquareTerminal if icon not found icon: iconMap[item.icon] || SquareTerminal,
})); }));
}, [navMain]); }, [navMain]);
@ -188,7 +196,7 @@ export function AppSidebar({
const processedNavSecondary = useMemo(() => { const processedNavSecondary = useMemo(() => {
return navSecondary.map((item) => ({ return navSecondary.map((item) => ({
...item, ...item,
icon: iconMap[item.icon] || Undo2, // Fallback to Undo2 if icon not found icon: iconMap[item.icon] || Undo2,
})); }));
}, [navSecondary]); }, [navSecondary]);
@ -197,17 +205,17 @@ export function AppSidebar({
return ( return (
RecentChats?.map((item) => ({ RecentChats?.map((item) => ({
...item, ...item,
icon: iconMap[item.icon] || MessageCircleMore, // Fallback to MessageCircleMore if icon not found icon: iconMap[item.icon] || MessageCircleMore,
})) || [] })) || []
); );
}, [RecentChats]); }, [RecentChats]);
return ( return (
<Sidebar variant="inset" {...props}> <Sidebar variant="inset" collapsible="icon" aria-label="Main navigation" {...props}>
<SidebarHeader> <SidebarHeader>
<SidebarMenu> <SidebarMenu>
<SidebarMenuItem> <SidebarMenuItem>
<SidebarMenuButton size="lg" asChild> <SidebarMenuButton size="lg" asChild aria-label="Go to home page">
<div> <div>
<div className="bg-sidebar-primary text-sidebar-primary-foreground flex aspect-square size-8 items-center justify-center rounded-lg"> <div className="bg-sidebar-primary text-sidebar-primary-foreground flex aspect-square size-8 items-center justify-center rounded-lg">
<Logo className="rounded-lg" /> <Logo className="rounded-lg" />
@ -221,11 +229,22 @@ export function AppSidebar({
</SidebarMenuItem> </SidebarMenuItem>
</SidebarMenu> </SidebarMenu>
</SidebarHeader> </SidebarHeader>
<SidebarContent>
<SidebarContent className="space-y-6">
<NavMain items={processedNavMain} /> <NavMain items={processedNavMain} />
{processedRecentChats.length > 0 && <NavProjects chats={processedRecentChats} />}
<NavSecondary items={processedNavSecondary} className="mt-auto" /> {processedRecentChats.length > 0 && (
<div className="space-y-2">
<NavProjects chats={processedRecentChats} />
</div>
)}
</SidebarContent> </SidebarContent>
<SidebarFooter>
<NavSecondary items={processedNavSecondary} className="mt-auto" />
{/* User Profile Section */}
<NavUser user={user} />
</SidebarFooter>
</Sidebar> </Sidebar>
); );
} });

View file

@ -1,6 +1,7 @@
"use client"; "use client";
import { ChevronRight, type LucideIcon } from "lucide-react"; import { ChevronRight, type LucideIcon } from "lucide-react";
import { useMemo } from "react";
import { Collapsible, CollapsibleContent, CollapsibleTrigger } from "@/components/ui/collapsible"; import { Collapsible, CollapsibleContent, CollapsibleTrigger } from "@/components/ui/collapsible";
import { import {
@ -15,10 +16,7 @@ import {
SidebarMenuSubItem, SidebarMenuSubItem,
} from "@/components/ui/sidebar"; } from "@/components/ui/sidebar";
export function NavMain({ interface NavItem {
items,
}: {
items: {
title: string; title: string;
url: string; url: string;
icon: LucideIcon; icon: LucideIcon;
@ -27,34 +25,47 @@ export function NavMain({
title: string; title: string;
url: string; url: string;
}[]; }[];
}[]; }
}) {
export function NavMain({ items }: { items: NavItem[] }) {
// Memoize items to prevent unnecessary re-renders
const memoizedItems = useMemo(() => items, [items]);
return ( return (
<SidebarGroup> <SidebarGroup>
<SidebarGroupLabel>Platform</SidebarGroupLabel> <SidebarGroupLabel>Platform</SidebarGroupLabel>
<SidebarMenu> <SidebarMenu>
{items.map((item, index) => ( {memoizedItems.map((item, index) => (
<Collapsible key={`${item.title}-${index}`} asChild defaultOpen={item.isActive}> <Collapsible key={`${item.title}-${index}`} asChild defaultOpen={item.isActive}>
<SidebarMenuItem> <SidebarMenuItem>
<SidebarMenuButton asChild tooltip={item.title}> <SidebarMenuButton
asChild
tooltip={item.title}
isActive={item.isActive}
aria-label={`${item.title}${item.items?.length ? " with submenu" : ""}`}
>
<a href={item.url}> <a href={item.url}>
<item.icon /> <item.icon />
<span>{item.title}</span> <span>{item.title}</span>
</a> </a>
</SidebarMenuButton> </SidebarMenuButton>
{item.items?.length ? ( {item.items?.length ? (
<> <>
<CollapsibleTrigger asChild> <CollapsibleTrigger asChild>
<SidebarMenuAction className="data-[state=open]:rotate-90"> <SidebarMenuAction
className="data-[state=open]:rotate-90 transition-transform duration-200"
aria-label={`Toggle ${item.title} submenu`}
>
<ChevronRight /> <ChevronRight />
<span className="sr-only">Toggle</span> <span className="sr-only">Toggle submenu</span>
</SidebarMenuAction> </SidebarMenuAction>
</CollapsibleTrigger> </CollapsibleTrigger>
<CollapsibleContent> <CollapsibleContent className="data-[state=open]:animate-in data-[state=closed]:animate-out data-[state=closed]:fade-out-0 data-[state=open]:fade-in-0 data-[state=closed]:slide-out-to-top-2 data-[state=open]:slide-in-from-top-2 duration-200">
<SidebarMenuSub> <SidebarMenuSub>
{item.items?.map((subItem, subIndex) => ( {item.items?.map((subItem, subIndex) => (
<SidebarMenuSubItem key={`${subItem.title}-${subIndex}`}> <SidebarMenuSubItem key={`${subItem.title}-${subIndex}`}>
<SidebarMenuSubButton asChild> <SidebarMenuSubButton asChild aria-label={subItem.title}>
<a href={subItem.url}> <a href={subItem.url}>
<span>{subItem.title}</span> <span>{subItem.title}</span>
</a> </a>

View file

@ -1,17 +1,27 @@
"use client"; "use client";
import { ExternalLink, Folder, type LucideIcon, MoreHorizontal, Share, Trash2 } from "lucide-react"; import {
ExternalLink,
Folder,
type LucideIcon,
MoreHorizontal,
RefreshCw,
Search,
Share,
Trash2,
} from "lucide-react";
import { useRouter } from "next/navigation"; import { useRouter } from "next/navigation";
import { useCallback, useMemo, useState } from "react";
import { import {
DropdownMenu, DropdownMenu,
DropdownMenuContent, DropdownMenuContent,
DropdownMenuItem, DropdownMenuItem,
DropdownMenuSeparator,
DropdownMenuTrigger, DropdownMenuTrigger,
} from "@/components/ui/dropdown-menu"; } from "@/components/ui/dropdown-menu";
import { import {
SidebarGroup, SidebarGroup,
SidebarGroupLabel, SidebarGroupLabel,
SidebarInput,
SidebarMenu, SidebarMenu,
SidebarMenuAction, SidebarMenuAction,
SidebarMenuButton, SidebarMenuButton,
@ -26,6 +36,8 @@ const actionIconMap: Record<string, LucideIcon> = {
Share, Share,
Trash2, Trash2,
MoreHorizontal, MoreHorizontal,
Search,
RefreshCw,
}; };
interface ChatAction { interface ChatAction {
@ -34,33 +46,57 @@ interface ChatAction {
onClick: () => void; onClick: () => void;
} }
export function NavProjects({ interface ChatItem {
chats,
}: {
chats: {
name: string; name: string;
url: string; url: string;
icon: LucideIcon; icon: LucideIcon;
id?: number; id?: number;
search_space_id?: number; search_space_id?: number;
actions?: ChatAction[]; actions?: ChatAction[];
}[]; }
}) {
export function NavProjects({ chats }: { chats: ChatItem[] }) {
const { isMobile } = useSidebar(); const { isMobile } = useSidebar();
const router = useRouter(); const router = useRouter();
const [searchQuery, setSearchQuery] = useState("");
const [isDeleting, setIsDeleting] = useState<number | null>(null);
const searchSpaceId = chats[0]?.search_space_id || ""; const searchSpaceId = chats[0]?.search_space_id || "";
// Memoized filtered chats
const filteredChats = useMemo(() => {
if (!searchQuery.trim()) return chats;
return chats.filter((chat) => chat.name.toLowerCase().includes(searchQuery.toLowerCase()));
}, [chats, searchQuery]);
// Handle chat deletion with loading state
const handleDeleteChat = useCallback(async (chatId: number, deleteAction: () => void) => {
setIsDeleting(chatId);
try {
await deleteAction();
} finally {
setIsDeleting(null);
}
}, []);
// Enhanced chat item component
const ChatItemComponent = useCallback(
({ chat }: { chat: ChatItem }) => {
const isDeletingChat = isDeleting === chat.id;
return ( return (
<SidebarGroup className="group-data-[collapsible=icon]:hidden"> <SidebarMenuItem key={chat.id ? `chat-${chat.id}` : `chat-${chat.name}`}>
<SidebarGroupLabel>Recent Chats</SidebarGroupLabel> <SidebarMenuButton
<SidebarMenu> onClick={() => router.push(chat.url)}
{chats.map((item, index) => ( disabled={isDeletingChat}
<SidebarMenuItem key={item.id ? `chat-${item.id}` : `chat-${item.name}-${index}`}> className={isDeletingChat ? "opacity-50" : ""}
<SidebarMenuButton> >
<item.icon /> <chat.icon />
<span>{item.name}</span> <span className={isDeletingChat ? "opacity-50" : ""}>{chat.name}</span>
</SidebarMenuButton> </SidebarMenuButton>
{chat.actions && chat.actions.length > 0 && (
<DropdownMenu> <DropdownMenu>
<DropdownMenuTrigger asChild> <DropdownMenuTrigger asChild>
<SidebarMenuAction showOnHover> <SidebarMenuAction showOnHover>
@ -73,44 +109,79 @@ export function NavProjects({
side={isMobile ? "bottom" : "right"} side={isMobile ? "bottom" : "right"}
align={isMobile ? "end" : "start"} align={isMobile ? "end" : "start"}
> >
{item.actions ? ( {chat.actions.map((action, actionIndex) => {
// Use the actions provided by the item
item.actions.map((action, actionIndex) => {
const ActionIcon = actionIconMap[action.icon] || Folder; const ActionIcon = actionIconMap[action.icon] || Folder;
const isDeleteAction = action.name.toLowerCase().includes("delete");
return ( return (
<DropdownMenuItem <DropdownMenuItem
key={`${action.name}-${actionIndex}`} key={`${action.name}-${actionIndex}`}
onClick={action.onClick} onClick={() => {
if (isDeleteAction) {
handleDeleteChat(chat.id || 0, action.onClick);
} else {
action.onClick();
}
}}
disabled={isDeletingChat}
className={isDeleteAction ? "text-destructive" : ""}
> >
<ActionIcon className="text-muted-foreground" /> <ActionIcon className="text-muted-foreground" />
<span>{action.name}</span> <span>{isDeletingChat && isDeleteAction ? "Deleting..." : action.name}</span>
</DropdownMenuItem> </DropdownMenuItem>
); );
}) })}
) : (
// Default actions if none provided
<>
<DropdownMenuItem>
<Folder className="text-muted-foreground" />
<span>View Chat</span>
</DropdownMenuItem>
<DropdownMenuSeparator />
<DropdownMenuItem>
<Trash2 className="text-muted-foreground" />
<span>Delete Chat</span>
</DropdownMenuItem>
</>
)}
</DropdownMenuContent> </DropdownMenuContent>
</DropdownMenu> </DropdownMenu>
)}
</SidebarMenuItem> </SidebarMenuItem>
))} );
},
[isDeleting, router, isMobile, handleDeleteChat]
);
// Show search input if there are chats
const showSearch = chats.length > 0;
return (
<SidebarGroup className="group-data-[collapsible=icon]:hidden">
<SidebarGroupLabel>Recent Chats</SidebarGroupLabel>
{/* Search Input */}
{showSearch && (
<div className="px-2 pb-2">
<SidebarInput
placeholder="Search chats..."
value={searchQuery}
onChange={(e) => setSearchQuery(e.target.value)}
className="h-8"
/>
</div>
)}
<SidebarMenu>
{/* Chat Items */}
{filteredChats.length > 0 ? (
filteredChats.map((chat) => <ChatItemComponent key={chat.id || chat.name} chat={chat} />)
) : (
/* No results state */
<SidebarMenuItem>
<SidebarMenuButton disabled className="text-muted-foreground">
<Search className="h-4 w-4" />
<span>{searchQuery ? "No chats found" : "No recent chats"}</span>
</SidebarMenuButton>
</SidebarMenuItem>
)}
{/* View All Chats */}
{chats.length > 0 && (
<SidebarMenuItem> <SidebarMenuItem>
<SidebarMenuButton onClick={() => router.push(`/dashboard/${searchSpaceId}/chats`)}> <SidebarMenuButton onClick={() => router.push(`/dashboard/${searchSpaceId}/chats`)}>
<MoreHorizontal /> <MoreHorizontal />
<span>View All Chats</span> <span>View All Chats</span>
</SidebarMenuButton> </SidebarMenuButton>
</SidebarMenuItem> </SidebarMenuItem>
)}
</SidebarMenu> </SidebarMenu>
</SidebarGroup> </SidebarGroup>
); );

View file

@ -2,6 +2,7 @@
import type { LucideIcon } from "lucide-react"; import type { LucideIcon } from "lucide-react";
import type * as React from "react"; import type * as React from "react";
import { useMemo } from "react";
import { import {
SidebarGroup, SidebarGroup,
@ -11,23 +12,28 @@ import {
SidebarMenuItem, SidebarMenuItem,
} from "@/components/ui/sidebar"; } from "@/components/ui/sidebar";
interface NavSecondaryItem {
title: string;
url: string;
icon: LucideIcon;
}
export function NavSecondary({ export function NavSecondary({
items, items,
...props ...props
}: { }: {
items: { items: NavSecondaryItem[];
title: string;
url: string;
icon: LucideIcon;
}[];
} & React.ComponentPropsWithoutRef<typeof SidebarGroup>) { } & React.ComponentPropsWithoutRef<typeof SidebarGroup>) {
// Memoize items to prevent unnecessary re-renders
const memoizedItems = useMemo(() => items, [items]);
return ( return (
<SidebarGroup {...props}> <SidebarGroup {...props}>
<SidebarGroupLabel>SearchSpace</SidebarGroupLabel> <SidebarGroupLabel>SearchSpace</SidebarGroupLabel>
<SidebarMenu> <SidebarMenu>
{items.map((item, index) => ( {memoizedItems.map((item, index) => (
<SidebarMenuItem key={`${item.title}-${index}`}> <SidebarMenuItem key={`${item.title}-${index}`}>
<SidebarMenuButton asChild size="sm"> <SidebarMenuButton asChild size="sm" aria-label={item.title}>
<a href={item.url}> <a href={item.url}>
<item.icon /> <item.icon />
<span>{item.title}</span> <span>{item.title}</span>

View file

@ -1,7 +1,8 @@
"use client"; "use client";
import { BadgeCheck, ChevronsUpDown, LogOut, Settings } from "lucide-react"; import { BadgeCheck, ChevronsUpDown, LogOut, Settings, User } from "lucide-react";
import { useParams, useRouter } from "next/navigation"; import { useParams, useRouter } from "next/navigation";
import { memo, useCallback } from "react";
import { Avatar, AvatarFallback, AvatarImage } from "@/components/ui/avatar"; import { Avatar, AvatarFallback, AvatarImage } from "@/components/ui/avatar";
import { import {
DropdownMenu, DropdownMenu,
@ -13,32 +14,43 @@ import {
DropdownMenuTrigger, DropdownMenuTrigger,
} from "@/components/ui/dropdown-menu"; } from "@/components/ui/dropdown-menu";
import { import {
SidebarGroup,
SidebarMenu, SidebarMenu,
SidebarMenuButton, SidebarMenuButton,
SidebarMenuItem, SidebarMenuItem,
useSidebar, useSidebar,
} from "@/components/ui/sidebar"; } from "@/components/ui/sidebar";
export function NavUser({ interface UserData {
user,
}: {
user: {
name: string; name: string;
email: string; email: string;
avatar: string; avatar: string;
}; }
}) {
// Memoized NavUser component for better performance
export const NavUser = memo(function NavUser({ user }: { user: UserData }) {
const { isMobile } = useSidebar(); const { isMobile } = useSidebar();
const router = useRouter(); const router = useRouter();
const { search_space_id } = useParams(); const { search_space_id } = useParams();
const handleLogout = () => { // Memoized logout handler
const handleLogout = useCallback(() => {
if (typeof window !== "undefined") { if (typeof window !== "undefined") {
localStorage.removeItem("surfsense_bearer_token"); localStorage.removeItem("surfsense_bearer_token");
router.push("/"); router.push("/");
} }
}; }, [router]);
// Get user initials for avatar fallback
const userInitials = user.name
.split(" ")
.map((n) => n[0])
.join("")
.toUpperCase()
.slice(0, 2);
return ( return (
<SidebarGroup className="mt-auto">
<SidebarMenu> <SidebarMenu>
<SidebarMenuItem> <SidebarMenuItem>
<DropdownMenu> <DropdownMenu>
@ -46,14 +58,17 @@ export function NavUser({
<SidebarMenuButton <SidebarMenuButton
size="lg" size="lg"
className="data-[state=open]:bg-sidebar-accent data-[state=open]:text-sidebar-accent-foreground" className="data-[state=open]:bg-sidebar-accent data-[state=open]:text-sidebar-accent-foreground"
aria-label="User menu"
> >
<Avatar className="h-8 w-8 rounded-lg"> <Avatar className="h-8 w-8 rounded-lg">
<AvatarImage src={user.avatar} alt={user.name} /> <AvatarImage src={user.avatar} alt={user.name} />
<AvatarFallback className="rounded-lg">CN</AvatarFallback> <AvatarFallback className="rounded-lg">
{userInitials || <User className="h-4 w-4" />}
</AvatarFallback>
</Avatar> </Avatar>
<div className="grid flex-1 text-left text-sm leading-tight"> <div className="grid flex-1 text-left text-sm leading-tight">
<span className="truncate font-medium">{user.name}</span> <span className="truncate font-medium">{user.name}</span>
<span className="truncate text-xs">{user.email}</span> <span className="truncate text-xs text-muted-foreground">{user.email}</span>
</div> </div>
<ChevronsUpDown className="ml-auto size-4" /> <ChevronsUpDown className="ml-auto size-4" />
</SidebarMenuButton> </SidebarMenuButton>
@ -68,11 +83,13 @@ export function NavUser({
<div className="flex items-center gap-2 px-1 py-1.5 text-left text-sm"> <div className="flex items-center gap-2 px-1 py-1.5 text-left text-sm">
<Avatar className="h-8 w-8 rounded-lg"> <Avatar className="h-8 w-8 rounded-lg">
<AvatarImage src={user.avatar} alt={user.name} /> <AvatarImage src={user.avatar} alt={user.name} />
<AvatarFallback className="rounded-lg">CN</AvatarFallback> <AvatarFallback className="rounded-lg">
{userInitials || <User className="h-4 w-4" />}
</AvatarFallback>
</Avatar> </Avatar>
<div className="grid flex-1 text-left text-sm leading-tight"> <div className="grid flex-1 text-left text-sm leading-tight">
<span className="truncate font-medium">{user.name}</span> <span className="truncate font-medium">{user.name}</span>
<span className="truncate text-xs">{user.email}</span> <span className="truncate text-xs text-muted-foreground">{user.email}</span>
</div> </div>
</div> </div>
</DropdownMenuLabel> </DropdownMenuLabel>
@ -80,23 +97,32 @@ export function NavUser({
<DropdownMenuGroup> <DropdownMenuGroup>
<DropdownMenuItem <DropdownMenuItem
onClick={() => router.push(`/dashboard/${search_space_id}/api-key`)} onClick={() => router.push(`/dashboard/${search_space_id}/api-key`)}
aria-label="Manage API key"
> >
<BadgeCheck /> <BadgeCheck className="h-4 w-4" />
API Key API Key
</DropdownMenuItem> </DropdownMenuItem>
</DropdownMenuGroup> </DropdownMenuGroup>
<DropdownMenuSeparator /> <DropdownMenuSeparator />
<DropdownMenuItem onClick={() => router.push(`/settings`)}> <DropdownMenuItem
<Settings /> onClick={() => router.push(`/settings`)}
aria-label="Go to settings"
>
<Settings className="h-4 w-4" />
Settings Settings
</DropdownMenuItem> </DropdownMenuItem>
<DropdownMenuItem onClick={handleLogout}> <DropdownMenuItem
<LogOut /> onClick={handleLogout}
Log out aria-label="Sign out"
className="text-destructive focus:text-destructive"
>
<LogOut className="h-4 w-4" />
Sign out
</DropdownMenuItem> </DropdownMenuItem>
</DropdownMenuContent> </DropdownMenuContent>
</DropdownMenu> </DropdownMenu>
</SidebarMenuItem> </SidebarMenuItem>
</SidebarMenu> </SidebarMenu>
</SidebarGroup>
); );
} });