Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
41 changes: 41 additions & 0 deletions apps/web/components/dashboard-view.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -40,6 +40,7 @@ import {
type Profession,
} from "@/hooks/use-personalization"
import { normalizePluginClientId } from "@/lib/plugin-catalog"
import { detectPluginSpace } from "@/lib/plugin-space"

type DocumentsResponse = z.infer<typeof DocumentsWithMemoriesResponseSchema>
type DocumentWithMemories = DocumentsResponse["documents"][0]
Expand Down Expand Up @@ -443,6 +444,23 @@ function hasClaudeCodeContainer(document: DocumentWithMemories): boolean {
)
}

function getPluginClientFromSpace(
document: DocumentWithMemories,
): string | null {
const containerTags =
(document as { containerTags?: string[] }).containerTags ?? []
const memorySpaceTags = (document.memoryEntries ?? [])
.map((entry) => entry.spaceContainerTag)
.filter((tag): tag is string => !!tag)

for (const tag of [...containerTags, ...memorySpaceTags]) {
const plugin = detectPluginSpace(tag)
if (plugin) return normalizePluginClientId(plugin.pluginId)
}

return null
}

function getPluginClientFromDocument(
document: DocumentWithMemories,
): string | null {
Expand All @@ -462,6 +480,9 @@ function getPluginClientFromDocument(

if (hasClaudeCodeContainer(document)) return "claude_code"

const pluginClientFromSpace = getPluginClientFromSpace(document)
if (pluginClientFromSpace) return pluginClientFromSpace

const content = getDocumentText(document)
const title = document.title ?? ""
if (
Expand Down Expand Up @@ -689,6 +710,26 @@ function parseToolUsage(
}

// Attach latest document info to plugin items where available from MCP documents
for (const [pluginId, docInfo] of latestDocPerPlugin) {
const itemKey = `plugin_${pluginId}`
if (toolMap.has(itemKey)) continue

const catalog = PLUGIN_DISPLAY_CATALOG[pluginId]
toolMap.set(itemKey, {
id: itemKey,
name: catalog?.name ?? docInfo.title,
type: "Plugin",
icon: catalog?.icon ?? null,
lastUsedAt: docInfo.at,
hasBeenUsed: true,
connectedAt: null,
lastDocumentTitle: null,
lastDocumentId: null,
lastDocumentPreview: null,
lastDocument: null,
})
}

for (const [, item] of toolMap) {
if (item.type === "Plugin" && !item.lastDocumentTitle) {
const pluginId = item.id.replace(/^plugin_/, "")
Expand Down
20 changes: 15 additions & 5 deletions apps/web/components/document-modal/index.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,7 @@ import type { UseMutationResult } from "@tanstack/react-query"
import { toast } from "sonner"
import { useIsMobile } from "@hooks/use-mobile"
import { parsePluginDocument } from "@/lib/plugin-document"
import { useFullDocumentContent } from "@/hooks/use-full-document"

type DocumentsResponse = z.infer<typeof DocumentsWithMemoriesResponseSchema>
type DocumentWithMemories = DocumentsResponse["documents"][0]
Expand Down Expand Up @@ -198,16 +199,25 @@ export function DocumentModal({
const isMobile = useIsMobile()
const { updateMutation, deleteMutation } = useDocumentMutations({ onClose })

const fullContent = useFullDocumentContent(_document?.id, isOpen)
const effectiveDocument = useMemo(
() =>
_document && fullContent !== null
? { ..._document, content: fullContent }
: _document,
[_document, fullContent],
)

const { initialEditorContent, initialEditorString } = useMemo(() => {
const content = _document?.content as string | null | undefined
const content = effectiveDocument?.content as string | null | undefined
return {
initialEditorContent: content ?? undefined,
initialEditorString: content ?? "",
}
}, [_document?.content])
}, [effectiveDocument?.content])
const pluginDocument = useMemo(
() => parsePluginDocument(_document),
[_document],
() => parsePluginDocument(effectiveDocument),
[effectiveDocument],
)

const [draftContentString, setDraftContentString] =
Expand Down Expand Up @@ -281,7 +291,7 @@ export function DocumentModal({
className="bg-[#14161A] rounded-[14px] overflow-hidden flex min-h-0 flex-1 flex-col shadow-[inset_0_2px_4px_rgba(0,0,0,0.3),inset_0_1px_2px_rgba(0,0,0,0.1)] relative"
>
<DocumentContent
document={_document}
document={effectiveDocument}
textEditorProps={textEditorProps}
pluginDocument={pluginDocument}
/>
Expand Down
16 changes: 16 additions & 0 deletions apps/web/hooks/use-document-mutations.ts
Original file line number Diff line number Diff line change
Expand Up @@ -212,6 +212,8 @@ function restoreQueriesFromSnapshot(
}

const FILE_UPLOAD_CONCURRENCY = 3
const fullDocumentQueryKey = (documentId: string) =>
["document-full", documentId] as const

export type FileUploadEntry = { id: string; file: File }

Expand Down Expand Up @@ -544,6 +546,10 @@ export function useDocumentMutations({
onSuccess: (_data, variables) => {
analytics.documentEdited({ document_id: variables.documentId })
toast.success("Document saved successfully!")
queryClient.setQueryData(
fullDocumentQueryKey(variables.documentId),
variables.content,
)
queryClient.invalidateQueries({ queryKey: ["documents-with-memories"] })
},
onError: (error) => {
Expand Down Expand Up @@ -584,6 +590,10 @@ export function useDocumentMutations({
onSuccess: (_data, variables) => {
analytics.documentDeleted({ document_id: variables.documentId })
toast.success("Document deleted successfully!")
queryClient.removeQueries({
queryKey: fullDocumentQueryKey(variables.documentId),
exact: true,
})
queryClient.invalidateQueries({ queryKey: ["documents-with-memories"] })
onClose?.()
},
Expand Down Expand Up @@ -623,6 +633,12 @@ export function useDocumentMutations({
toast.success(
`${variables.documentIds.length} document${variables.documentIds.length === 1 ? "" : "s"} deleted`,
)
for (const documentId of variables.documentIds) {
queryClient.removeQueries({
queryKey: fullDocumentQueryKey(documentId),
exact: true,
})
}
queryClient.invalidateQueries({ queryKey: ["documents-with-memories"] })
},
})
Expand Down
29 changes: 29 additions & 0 deletions apps/web/hooks/use-full-document.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,29 @@
import { useQuery } from "@tanstack/react-query"
import { $fetch } from "@lib/api"

// List endpoint truncates `content` to ~2000 chars; fetch the full doc by id for the viewer.
export function useFullDocumentContent(
documentId: string | null | undefined,
enabled: boolean,
): string | null {
const isReal = !!documentId && !documentId.startsWith("temp-")

const { data } = useQuery({
queryKey: ["document-full", documentId],
enabled: enabled && isReal,
staleTime: 5 * 60 * 1000,
queryFn: async () => {
const res = await $fetch("@get/documents/:id", {
params: { id: documentId as string },
disableValidation: true,
})
if (res.error) {
throw new Error(res.error?.message || "Failed to fetch document")
}
const doc = res.data as { content?: string | null } | null
return typeof doc?.content === "string" ? doc.content : null
},
})

return data ?? null
}
31 changes: 31 additions & 0 deletions apps/web/lib/plugin-document.ts
Original file line number Diff line number Diff line change
Expand Up @@ -456,6 +456,34 @@ function parseRoleBlockTranscript(
}
}

function parsePluginSpaceNote(
document: DocumentWithMemories,
content: string,
plugin: PluginIdentity | null,
): ParsedPluginDocument | null {
if (!plugin) return null

const summary = typeof document.summary === "string" ? document.summary : ""
const memoryText = firstMemoryEntryText(document)
const preview = takePreview(memoryText || summary || content, 220)

return {
kind: plugin.pluginId === "codex" ? "codex-save" : "plugin-save",
pluginLabel: plugin.label,
pluginIconSrc: plugin.iconSrc ?? undefined,
formatLabel: "Note",
title:
(typeof document.title === "string" && document.title.trim()) ||
`${plugin.label} memory`,
preview,
summary: takePreview(summary || memoryText || content, 220),
artifacts: [],
messages: [],
sections: [],
rawContent: content,
}
}

function withIcon(
parsed: ParsedPluginDocument | null,
): ParsedPluginDocument | null {
Expand Down Expand Up @@ -694,5 +722,8 @@ export function parsePluginDocument(
if (roleBlockSession) return withIcon(roleBlockSession)
}

const pluginSpaceNote = parsePluginSpaceNote(document, content, plugin)
if (pluginSpaceNote) return withIcon(pluginSpaceNote)

return withIcon(parseClaudeCodeByMetadata(document, metadata))
}
Loading