Method webSocketMessage

src/durable-objects/user-notes.ts:914–972

(ws: WebSocket, message: string | ArrayBuffer)

        }

        // --- Tree (full graph for visualization) ---
        if (request.method === "GET" && path.match(/^\/notes\/[^/]+\/tree$/)) {
            const noteId = path.split("/")[2];
            this.ensureDefaultBranch(noteId);
            const branches = this.sql.exec(
                "SELECT id, name, head_version_id, is_default, created_at FROM note_branches WHERE note_id = ?", noteId
            ).toArray();
            const current = this.getCurrentBranch(noteId);
            const versions = this.sql.exec(
                `SELECT id, parent_id, branch_id, title, is_checkpoint, created_by, kind, summary, created_at FROM note_versions
                 WHERE note_id = ? ORDER BY created_at DESC LIMIT 1000`, noteId
            ).toArray();
            const note = this.getNote(noteId);
            return Response.json({
                branches: branches.map((b: any) => ({ ...b, is_current: b.id === current.id ? 1 : 0 })),
                versions,
                sync_mode: note?.sync_mode || "cloud",
            });
        }

        // Note history — git-style version list
        if (request.method === "GET" && path.match(/^\/notes\/[^/]+\/history$/)) {
            const noteId = path.split("/")[2];
            const rows = this.sql.exec(
                `SELECT id, note_id, title, is_checkpoint, branch_id, created_by, kind, summary, created_at FROM note_versions
                 WHERE note_id = ? ORDER BY created_at DESC LIMIT 1000`, noteId
            ).toArray();
            return Response.json(rows);
        }
        // Reconstruct a specific version's full content
        if (request.method === "GET" && path.match(/^\/notes\/[^/]+\/history\/[^/]+$/)) {
            const parts = path.split("/");
            const noteId = parts[2];
            const versionId = parts[4];
            const row = this.sql.exec(
                "SELECT id, note_id, title, is_checkpoint, created_by, kind, summary, created_at FROM note_versions WHERE id = ? AND note_id = ?",
                versionId, noteId
            ).toArray()[0];
            if (!row) return new Response("Version not found", { status: 404 });
            const content = await this.reconstructVersion(versionId, noteId);
            return Response.json({ ...row, content });
        }
        // Restore to a specific version
        if (request.method === "POST" && path.match(/^\/notes\/[^/]+\/history\/restore$/)) {
            const noteId = path.split("/")[2];
            const { version_id } = (await request.json()) as { version_id: string };
            const row = this.sql.exec(
                "SELECT id FROM note_versions WHERE id = ? AND note_id = ?", version_id, noteId
            ).toArray()[0];
            if (!row) return new Response("Version not found", { status: 404 });

            const restoredContent = await this.reconstructVersion(version_id, noteId);

            // Version the current state before restoring (so restore itself is reversible)
            const current = this.getNote(noteId);
            if (current?.content) {
                await this.createVersion(noteId, (current.title as string) || "Untitled", current.content as string, "auto-backup", "backup", "Auto-backup before restore");
            }