mirror of
https://github.com/supermemoryai/supermemory.git
synced 2026-05-17 12:20:04 +00:00
4e607f9fd7
<h3>Implement comprehensive plugin document rendering support including MCP previews and plugin specific content handling.</h3> <br> <br> <img width="1680" height="471" alt="Screenshot 2026-05-12 at 8 24 49 PM" src="https://github.com/user-attachments/assets/f1294bc2-2841-4833-9f01-ac47b8c52c01" /> <br> <br> <img width="1680" height="963" alt="Screenshot 2026-05-12 at 8 28 25 PM" src="https://github.com/user-attachments/assets/9436c7ab-3b9b-4366-86fd-1465407ff0f9" />
636 lines
18 KiB
TypeScript
636 lines
18 KiB
TypeScript
"use client"
|
|
|
|
import { useState, useMemo, useEffect, useCallback } from "react"
|
|
import Image from "next/image"
|
|
import { useQuery } from "@tanstack/react-query"
|
|
import { cn } from "@lib/utils"
|
|
import { $fetch } from "@lib/api"
|
|
import { dmSans125ClassName, dmSansClassName } from "@/lib/fonts"
|
|
import { DEFAULT_PROJECT_ID } from "@lib/constants"
|
|
import { XIcon, Loader2 } from "lucide-react"
|
|
import type { ContainerTagListType } from "@lib/types"
|
|
import { AddSpaceModal } from "./add-space-modal"
|
|
import { SelectSpacesModal } from "./select-spaces-modal"
|
|
import { useProjectMutations } from "@/hooks/use-project-mutations"
|
|
import { useContainerTags } from "@/hooks/use-container-tags"
|
|
import { motion } from "motion/react"
|
|
import * as DialogPrimitive from "@radix-ui/react-dialog"
|
|
import {
|
|
Dialog,
|
|
DialogContent,
|
|
DialogTitle,
|
|
DialogDescription,
|
|
} from "@repo/ui/components/dialog"
|
|
import {
|
|
Select,
|
|
SelectContent,
|
|
SelectItem,
|
|
SelectTrigger,
|
|
SelectValue,
|
|
} from "@repo/ui/components/select"
|
|
import { Button } from "@repo/ui/components/button"
|
|
import { Tooltip, TooltipContent, TooltipTrigger } from "@ui/components/tooltip"
|
|
import { analytics } from "@/lib/analytics"
|
|
import {
|
|
compareSpacesUserFirst,
|
|
spaceSelectorDisplayName,
|
|
} from "@/lib/ingest-auto-space"
|
|
import { detectPluginSpace, pluginInitial } from "@/lib/plugin-space"
|
|
import { usePluginSpaceMeta } from "@/hooks/use-plugin-space-meta"
|
|
|
|
export interface SpaceSelectorProps {
|
|
selectedProjects: string[]
|
|
onValueChange: (containerTags: string[]) => void
|
|
variant?: "default" | "insideOut"
|
|
triggerClassName?: string
|
|
showNewSpace?: boolean
|
|
enableDelete?: boolean
|
|
compact?: boolean
|
|
}
|
|
|
|
const triggerVariants = {
|
|
default:
|
|
"h-10 min-h-10 shrink-0 rounded-full border border-[#161F2C] bg-muted px-3 gap-2 " +
|
|
"hover:bg-white/5 hover:border-[#2261CA33] " +
|
|
"focus-visible:outline-none focus-visible:ring-2 focus-visible:ring-[#2261CA33]/35",
|
|
insideOut:
|
|
"h-10 min-h-10 gap-2 px-3 rounded-full bg-[#0D121A] shadow-inside-out hover:bg-[#121820]",
|
|
}
|
|
|
|
const RECENTS_KEY = "nova:space-selector:recents"
|
|
const RECENTS_MAX = 10
|
|
|
|
function readRecents(): string[] {
|
|
if (typeof window === "undefined") return []
|
|
try {
|
|
const raw = window.localStorage.getItem(RECENTS_KEY)
|
|
if (!raw) return []
|
|
const parsed = JSON.parse(raw)
|
|
return Array.isArray(parsed)
|
|
? parsed.filter((x) => typeof x === "string")
|
|
: []
|
|
} catch {
|
|
return []
|
|
}
|
|
}
|
|
|
|
function writeRecents(tags: string[]) {
|
|
if (typeof window === "undefined") return
|
|
try {
|
|
window.localStorage.setItem(RECENTS_KEY, JSON.stringify(tags))
|
|
} catch {
|
|
// ignore
|
|
}
|
|
}
|
|
|
|
function formatCount(n: number): string {
|
|
if (n < 1000) return String(n)
|
|
if (n < 10_000) return `${(n / 1000).toFixed(1).replace(/\.0$/, "")}k`
|
|
if (n < 1_000_000) return `${Math.floor(n / 1000)}k`
|
|
return `${(n / 1_000_000).toFixed(1).replace(/\.0$/, "")}m`
|
|
}
|
|
|
|
export function SpaceSelector({
|
|
selectedProjects,
|
|
onValueChange,
|
|
variant = "default",
|
|
triggerClassName,
|
|
showNewSpace = true,
|
|
enableDelete = false,
|
|
compact = false,
|
|
}: SpaceSelectorProps) {
|
|
const [showCreateDialog, setShowCreateDialog] = useState(false)
|
|
const [showSelectSpacesModal, setShowSelectSpacesModal] = useState(false)
|
|
const [recents, setRecents] = useState<string[]>([])
|
|
const [deleteDialog, setDeleteDialog] = useState<{
|
|
open: boolean
|
|
project: { id: string; name: string; containerTag: string } | null
|
|
action: "move" | "delete"
|
|
targetProjectId: string
|
|
}>({
|
|
open: false,
|
|
project: null,
|
|
action: "move",
|
|
targetProjectId: "",
|
|
})
|
|
|
|
const { deleteProjectMutation } = useProjectMutations()
|
|
const { allProjects, isLoading } = useContainerTags()
|
|
|
|
useEffect(() => {
|
|
setRecents(readRecents())
|
|
}, [])
|
|
|
|
const activeTag = selectedProjects[0] ?? DEFAULT_PROJECT_ID
|
|
const { data: spaceCountData } = useQuery({
|
|
queryKey: ["space-selector-count", activeTag],
|
|
queryFn: async (): Promise<number> => {
|
|
const response = await $fetch("@post/documents/documents", {
|
|
body: {
|
|
page: 1,
|
|
limit: 1,
|
|
sort: "createdAt",
|
|
order: "desc",
|
|
containerTags: [activeTag],
|
|
},
|
|
disableValidation: true,
|
|
})
|
|
if (response.error) return 0
|
|
const data = response.data as {
|
|
pagination?: { totalItems?: number }
|
|
} | null
|
|
return data?.pagination?.totalItems ?? 0
|
|
},
|
|
staleTime: 30 * 1000,
|
|
enabled: !!activeTag,
|
|
})
|
|
|
|
const pluginTags = useMemo(
|
|
() =>
|
|
allProjects
|
|
.filter(
|
|
(p: ContainerTagListType) => !!detectPluginSpace(p.containerTag),
|
|
)
|
|
.map((p: ContainerTagListType) => p.containerTag),
|
|
[allProjects],
|
|
)
|
|
const pluginMetaMap = usePluginSpaceMeta(pluginTags)
|
|
|
|
const displayInfo = useMemo<{
|
|
name: string
|
|
emoji: string | null
|
|
plugin: ReturnType<typeof detectPluginSpace>
|
|
}>(() => {
|
|
const containerTag = selectedProjects[0] ?? ""
|
|
if (!containerTag || containerTag === DEFAULT_PROJECT_ID) {
|
|
return { name: "My Space", emoji: "📁", plugin: null }
|
|
}
|
|
const found = allProjects.find(
|
|
(p: ContainerTagListType) => p.containerTag === containerTag,
|
|
)
|
|
const plugin = detectPluginSpace(containerTag)
|
|
const projectName = pluginMetaMap.get(containerTag)?.projectName
|
|
const idForLabel = projectName || plugin?.projectId
|
|
return {
|
|
name: plugin
|
|
? idForLabel
|
|
? `${plugin.label} · ${idForLabel}`
|
|
: plugin.label
|
|
: spaceSelectorDisplayName(found, containerTag),
|
|
emoji: found?.emoji || "📁",
|
|
plugin,
|
|
}
|
|
}, [allProjects, selectedProjects, pluginMetaMap])
|
|
|
|
const pushRecent = useCallback((tag: string) => {
|
|
setRecents((prev) => {
|
|
const next = [tag, ...prev.filter((t) => t !== tag)].slice(0, RECENTS_MAX)
|
|
writeRecents(next)
|
|
return next
|
|
})
|
|
}, [])
|
|
|
|
const handleSelectSpacesApply = useCallback(
|
|
(selected: string[]) => {
|
|
const next = selected.slice(0, 1)
|
|
if (next[0]) {
|
|
analytics.spaceSwitched({ space_id: next[0] })
|
|
pushRecent(next[0])
|
|
}
|
|
onValueChange(next)
|
|
setShowSelectSpacesModal(false)
|
|
},
|
|
[onValueChange, pushRecent],
|
|
)
|
|
|
|
const handleNewSpace = useCallback(() => {
|
|
setShowSelectSpacesModal(false)
|
|
setShowCreateDialog(true)
|
|
}, [])
|
|
|
|
const handleDeleteRequest = useCallback(
|
|
(project: { id: string; name: string; containerTag: string }) => {
|
|
setShowSelectSpacesModal(false)
|
|
setDeleteDialog({
|
|
open: true,
|
|
project,
|
|
action: "move",
|
|
targetProjectId: "",
|
|
})
|
|
},
|
|
[],
|
|
)
|
|
|
|
const handleDeleteConfirm = () => {
|
|
if (!deleteDialog.project) return
|
|
deleteProjectMutation.mutate(
|
|
{
|
|
projectId: deleteDialog.project.id,
|
|
action: deleteDialog.action,
|
|
targetProjectId:
|
|
deleteDialog.action === "move"
|
|
? deleteDialog.targetProjectId
|
|
: undefined,
|
|
},
|
|
{
|
|
onSuccess: () => {
|
|
setDeleteDialog({
|
|
open: false,
|
|
project: null,
|
|
action: "move",
|
|
targetProjectId: "",
|
|
})
|
|
},
|
|
},
|
|
)
|
|
}
|
|
|
|
const handleDeleteCancel = () => {
|
|
setDeleteDialog({
|
|
open: false,
|
|
project: null,
|
|
action: "move",
|
|
targetProjectId: "",
|
|
})
|
|
}
|
|
|
|
const availableTargetProjects = useMemo(() => {
|
|
const filtered = allProjects.filter(
|
|
(p: ContainerTagListType) =>
|
|
p.id !== deleteDialog.project?.id &&
|
|
p.containerTag !== deleteDialog.project?.containerTag,
|
|
)
|
|
const defaultProject = allProjects.find(
|
|
(p: ContainerTagListType) => p.containerTag === DEFAULT_PROJECT_ID,
|
|
)
|
|
const isDefaultProjectBeingDeleted =
|
|
deleteDialog.project?.containerTag === DEFAULT_PROJECT_ID
|
|
if (defaultProject && !isDefaultProjectBeingDeleted) {
|
|
const defaultProjectIncluded = filtered.some(
|
|
(p: ContainerTagListType) => p.containerTag === DEFAULT_PROJECT_ID,
|
|
)
|
|
if (!defaultProjectIncluded) return [defaultProject, ...filtered]
|
|
}
|
|
return filtered.sort(compareSpacesUserFirst)
|
|
}, [allProjects, deleteDialog.project])
|
|
|
|
return (
|
|
<>
|
|
<Tooltip>
|
|
<TooltipTrigger asChild>
|
|
<button
|
|
type="button"
|
|
onClick={() => setShowSelectSpacesModal(true)}
|
|
aria-label={
|
|
isLoading
|
|
? "Loading spaces"
|
|
: `Space: ${displayInfo.name}. Open selector.`
|
|
}
|
|
className={cn(
|
|
"flex min-w-0 max-w-full items-center cursor-pointer transition-colors",
|
|
triggerVariants[variant],
|
|
variant === "default" && compact && "h-9 min-h-9 gap-1.5 px-2.5",
|
|
dmSansClassName(),
|
|
triggerClassName,
|
|
)}
|
|
>
|
|
{displayInfo.plugin ? (
|
|
displayInfo.plugin.iconSrc ? (
|
|
<Image
|
|
src={displayInfo.plugin.iconSrc}
|
|
alt=""
|
|
width={16}
|
|
height={16}
|
|
className={cn(
|
|
"shrink-0 rounded-[3px]",
|
|
compact ? "size-3.5" : "size-4",
|
|
)}
|
|
aria-hidden
|
|
/>
|
|
) : (
|
|
<span
|
|
className={cn(
|
|
"shrink-0 flex items-center justify-center rounded-[3px] bg-[#1E232B] text-[#FAFAFA] text-[10px] font-semibold uppercase",
|
|
compact ? "size-3.5" : "size-4",
|
|
)}
|
|
aria-hidden
|
|
>
|
|
{pluginInitial(displayInfo.plugin.label)}
|
|
</span>
|
|
)
|
|
) : (
|
|
<span
|
|
className="shrink-0 text-sm font-bold tracking-[-0.98px]"
|
|
aria-hidden
|
|
>
|
|
{displayInfo.emoji}
|
|
</span>
|
|
)}
|
|
{!compact && (
|
|
<span
|
|
className={cn(
|
|
"min-w-0 truncate text-sm font-medium text-white",
|
|
"max-w-[10rem] md:max-w-[15rem]",
|
|
)}
|
|
title={isLoading ? undefined : displayInfo.name}
|
|
>
|
|
{isLoading ? "…" : displayInfo.name}
|
|
</span>
|
|
)}
|
|
{!compact && spaceCountData !== undefined && spaceCountData > 0 && (
|
|
<span className="shrink-0 text-[11px] text-[#737373] tabular-nums">
|
|
· {formatCount(spaceCountData)}
|
|
</span>
|
|
)}
|
|
{compact && (
|
|
<span className="sr-only">
|
|
{isLoading ? "Loading" : displayInfo.name}
|
|
</span>
|
|
)}
|
|
</button>
|
|
</TooltipTrigger>
|
|
<TooltipContent side="bottom" className={dmSansClassName()}>
|
|
Switch space
|
|
</TooltipContent>
|
|
</Tooltip>
|
|
|
|
<AddSpaceModal
|
|
isOpen={showCreateDialog}
|
|
onClose={() => setShowCreateDialog(false)}
|
|
onCreated={(containerTag) => {
|
|
pushRecent(containerTag)
|
|
onValueChange([containerTag])
|
|
}}
|
|
/>
|
|
|
|
<SelectSpacesModal
|
|
isOpen={showSelectSpacesModal}
|
|
onClose={() => setShowSelectSpacesModal(false)}
|
|
selectedProjects={selectedProjects}
|
|
onApply={handleSelectSpacesApply}
|
|
projects={allProjects}
|
|
recents={recents}
|
|
showNewSpace={showNewSpace}
|
|
onNewSpace={handleNewSpace}
|
|
enableDelete={enableDelete}
|
|
onDeleteRequest={handleDeleteRequest}
|
|
/>
|
|
|
|
<Dialog
|
|
open={deleteDialog.open}
|
|
onOpenChange={(open: boolean) => {
|
|
if (!open) {
|
|
setDeleteDialog({
|
|
open: false,
|
|
project: null,
|
|
action: "move",
|
|
targetProjectId: "",
|
|
})
|
|
}
|
|
}}
|
|
>
|
|
<DialogContent
|
|
className={cn(
|
|
"w-[90%]! max-w-[500px]! border-none bg-[#1B1F24] flex flex-col p-4 gap-4 rounded-[22px]",
|
|
dmSansClassName(),
|
|
)}
|
|
style={{
|
|
boxShadow:
|
|
"0 2.842px 14.211px 0 rgba(0, 0, 0, 0.25), 0.711px 0.711px 0.711px 0 rgba(255, 255, 255, 0.10) inset",
|
|
}}
|
|
showCloseButton={false}
|
|
>
|
|
<div className="flex flex-col gap-4">
|
|
<div className="flex justify-between items-start gap-4">
|
|
<div className="pl-1 space-y-1 flex-1">
|
|
<DialogTitle
|
|
className={cn(
|
|
"font-semibold text-[#fafafa]",
|
|
dmSans125ClassName(),
|
|
)}
|
|
>
|
|
Delete space
|
|
</DialogTitle>
|
|
<DialogDescription className="text-[#737373] font-medium text-[16px] leading-[1.35]">
|
|
What would you like to do with the documents and memories in{" "}
|
|
<span className="text-[#fafafa] font-medium">
|
|
"{deleteDialog.project?.name}"
|
|
</span>
|
|
?
|
|
</DialogDescription>
|
|
</div>
|
|
<DialogPrimitive.Close
|
|
className="bg-[#0D121A] w-7 h-7 flex items-center justify-center focus:ring-ring rounded-full transition-opacity hover:opacity-100 focus:ring-2 focus:ring-offset-2 focus:outline-hidden disabled:pointer-events-none [&_svg]:pointer-events-none [&_svg]:shrink-0 [&_svg:not([class*='size-'])]:size-4 border border-[rgba(115,115,115,0.2)] shrink-0"
|
|
style={{
|
|
boxShadow:
|
|
"inset 1.313px 1.313px 3.938px 0px rgba(0,0,0,0.7)",
|
|
}}
|
|
>
|
|
<XIcon stroke="#737373" />
|
|
<span className="sr-only">Close</span>
|
|
</DialogPrimitive.Close>
|
|
</div>
|
|
|
|
<div className="space-y-3">
|
|
<button
|
|
type="button"
|
|
onClick={() =>
|
|
setDeleteDialog((prev) => ({ ...prev, action: "move" }))
|
|
}
|
|
className={cn(
|
|
"flex items-center gap-3 p-3 rounded-[12px] cursor-pointer transition-colors w-full text-left",
|
|
deleteDialog.action === "move"
|
|
? "bg-[#14161A] shadow-inside-out"
|
|
: "bg-[#14161A]/50 hover:bg-[#14161A]/70",
|
|
)}
|
|
>
|
|
<div
|
|
className={cn(
|
|
"w-4 h-4 rounded-full border-2 flex items-center justify-center shrink-0",
|
|
deleteDialog.action === "move"
|
|
? "border-[#4BA0FA]"
|
|
: "border-[#737373]",
|
|
)}
|
|
>
|
|
{deleteDialog.action === "move" && (
|
|
<div className="w-2 h-2 rounded-full bg-[#4BA0FA]" />
|
|
)}
|
|
</div>
|
|
<span className="text-[#fafafa] text-sm font-medium">
|
|
Move to another space
|
|
</span>
|
|
</button>
|
|
|
|
{deleteDialog.action === "move" && (
|
|
<motion.div
|
|
initial={{ opacity: 0, height: 0 }}
|
|
animate={{ opacity: 1, height: "auto" }}
|
|
exit={{ opacity: 0, height: 0 }}
|
|
className="ml-7"
|
|
>
|
|
<Select
|
|
value={deleteDialog.targetProjectId}
|
|
onValueChange={(val: string) =>
|
|
setDeleteDialog((prev) => ({
|
|
...prev,
|
|
targetProjectId: val,
|
|
}))
|
|
}
|
|
>
|
|
<SelectTrigger
|
|
className={cn(
|
|
"bg-[#14161A] shadow-inside-out rounded-[12px] text-[#fafafa] text-[14px] h-[45px]",
|
|
dmSansClassName(),
|
|
)}
|
|
>
|
|
<SelectValue placeholder="Select target space" />
|
|
</SelectTrigger>
|
|
<SelectContent
|
|
className={cn(
|
|
"bg-[#14161A] border border-[rgba(82,89,102,0.2)] rounded-[12px]",
|
|
dmSansClassName(),
|
|
)}
|
|
>
|
|
{availableTargetProjects.map(
|
|
(p: ContainerTagListType) => {
|
|
const plugin = detectPluginSpace(p.containerTag)
|
|
return (
|
|
<SelectItem
|
|
key={p.id}
|
|
value={p.id}
|
|
className="text-[#fafafa] hover:bg-[#1B1F24] cursor-pointer rounded-md"
|
|
>
|
|
<span className="flex items-center gap-2 min-w-0">
|
|
{plugin ? (
|
|
plugin.iconSrc ? (
|
|
<Image
|
|
src={plugin.iconSrc}
|
|
alt=""
|
|
width={16}
|
|
height={16}
|
|
className="shrink-0 rounded-[3px]"
|
|
aria-hidden
|
|
/>
|
|
) : (
|
|
<span
|
|
className="shrink-0 flex items-center justify-center w-4 h-4 rounded-[3px] bg-[#1E232B] text-[#FAFAFA] text-[10px] font-semibold uppercase"
|
|
aria-hidden
|
|
>
|
|
{pluginInitial(plugin.label)}
|
|
</span>
|
|
)
|
|
) : (
|
|
<span>{p.emoji || "📁"}</span>
|
|
)}
|
|
<span className="truncate">
|
|
{p.containerTag === DEFAULT_PROJECT_ID ? (
|
|
"My Space"
|
|
) : plugin ? (
|
|
<>
|
|
{plugin.label}
|
|
{plugin.projectId && (
|
|
<span className="ml-1.5 text-[11px] text-[#737373]">
|
|
· {plugin.projectId}
|
|
</span>
|
|
)}
|
|
</>
|
|
) : (
|
|
spaceSelectorDisplayName(p, p.containerTag)
|
|
)}
|
|
</span>
|
|
</span>
|
|
</SelectItem>
|
|
)
|
|
},
|
|
)}
|
|
</SelectContent>
|
|
</Select>
|
|
</motion.div>
|
|
)}
|
|
|
|
<button
|
|
type="button"
|
|
onClick={() =>
|
|
setDeleteDialog((prev) => ({ ...prev, action: "delete" }))
|
|
}
|
|
className={cn(
|
|
"flex items-center gap-3 p-3 rounded-[12px] cursor-pointer transition-colors w-full text-left",
|
|
deleteDialog.action === "delete"
|
|
? "bg-[#14161A] shadow-inside-out"
|
|
: "bg-[#14161A]/50 hover:bg-[#14161A]/70",
|
|
)}
|
|
>
|
|
<div
|
|
className={cn(
|
|
"w-4 h-4 rounded-full border-2 flex items-center justify-center shrink-0",
|
|
deleteDialog.action === "delete"
|
|
? "border-red-500"
|
|
: "border-[#737373]",
|
|
)}
|
|
>
|
|
{deleteDialog.action === "delete" && (
|
|
<div className="w-2 h-2 rounded-full bg-red-500" />
|
|
)}
|
|
</div>
|
|
<span className="text-[#fafafa] text-sm font-medium">
|
|
Delete everything permanently
|
|
</span>
|
|
</button>
|
|
|
|
{deleteDialog.action === "delete" && (
|
|
<motion.p
|
|
initial={{ opacity: 0 }}
|
|
animate={{ opacity: 1 }}
|
|
className="text-xs text-red-400 ml-7"
|
|
>
|
|
All documents and memories will be permanently deleted.
|
|
</motion.p>
|
|
)}
|
|
</div>
|
|
|
|
<div className="flex items-center justify-end gap-[22px]">
|
|
<button
|
|
type="button"
|
|
onClick={handleDeleteCancel}
|
|
disabled={deleteProjectMutation.isPending}
|
|
className={cn(
|
|
"text-[#737373] font-medium text-[14px] cursor-pointer transition-colors hover:text-[#999]",
|
|
dmSansClassName(),
|
|
)}
|
|
>
|
|
Cancel
|
|
</button>
|
|
<Button
|
|
variant="insideOut"
|
|
onClick={handleDeleteConfirm}
|
|
disabled={
|
|
deleteProjectMutation.isPending ||
|
|
(deleteDialog.action === "move" &&
|
|
!deleteDialog.targetProjectId)
|
|
}
|
|
className={cn(
|
|
"px-4 py-[10px] rounded-full",
|
|
deleteDialog.action === "delete" &&
|
|
"bg-red-600 hover:bg-red-700 border-red-700",
|
|
)}
|
|
>
|
|
{deleteProjectMutation.isPending ? (
|
|
<>
|
|
<Loader2 className="size-4 animate-spin mr-2" />
|
|
{deleteDialog.action === "move"
|
|
? "Moving..."
|
|
: "Deleting..."}
|
|
</>
|
|
) : deleteDialog.action === "move" ? (
|
|
"Move & Delete"
|
|
) : (
|
|
"Delete Everything"
|
|
)}
|
|
</Button>
|
|
</div>
|
|
</div>
|
|
</DialogContent>
|
|
</Dialog>
|
|
</>
|
|
)
|
|
}
|