Quick start

import { createHmac } from 'node:crypto';
import { build } from 'esbuild';

process.loadEnvFile();

const { ENVIRONMENT_ID, WEBSOCKET_URL, API_SECRET } = process.env;
const BUNDLE_VERSION = 'coauthor-quick-start';

// 1. Build the editor as one file that sets the `CKEditorCS` global.
const { outputFiles } = await build( {
	stdin: {
		contents: `
			import { ClassicEditor, CloudServices, Essentials, Paragraph, Heading, Bold, Italic, List, Table } from 'ckeditor5';
			import {
				Comments, TrackChanges, DocumentCompare,
				RealTimeCollaborativeEditing, RealTimeCollaborativeComments, RealTimeCollaborativeTrackChanges
			} from 'ckeditor5-premium-features';

			class CKEditorCS extends ClassicEditor {}

			CKEditorCS.builtinPlugins = [
				CloudServices, Essentials, Paragraph, Heading, Bold, Italic, List, Table,
				Comments, TrackChanges, DocumentCompare,
				RealTimeCollaborativeEditing, RealTimeCollaborativeComments, RealTimeCollaborativeTrackChanges
			];

			globalThis.CKEditorCS = CKEditorCS;
		`,
		resolveDir: import.meta.dirname
	},
	bundle: true,
	format: 'iife',
	minify: true,
	write: false
} );

// 2. Upload it with the version that `editor.html` names. The API secret signs the request.
const url = new URL( `https://${ new URL( WEBSOCKET_URL ).host }/api/v5/${ ENVIRONMENT_ID }/editors` );
const body = JSON.stringify( {
	bundle: outputFiles[ 0 ].text,
	config: { cloudServices: { bundleVersion: BUNDLE_VERSION } }
} );
const timestamp = Date.now();
const signature = createHmac( 'sha256', API_SECRET ).update( `POST${ url.pathname }${ timestamp }${ body }` ).digest( 'hex' );

const response = await fetch( url, {
	method: 'POST',
	headers: { 'Content-Type': 'application/json', 'X-CS-Timestamp': String( timestamp ), 'X-CS-Signature': signature },
	body
} );

if ( response.status === 201 ) {
	console.log( `Uploaded the editor bundle ${ BUNDLE_VERSION }.` );
} else if ( response.status === 409 ) {
	console.log( `The editor bundle ${ BUNDLE_VERSION } is already uploaded.` );
} else {
	throw new Error( `The upload answered ${ response.status }: ${ await response.text() }` );
}

import { SignJWT } from 'jose';

process.loadEnvFile();

const { ENVIRONMENT_ID, WEBSOCKET_URL, ACCESS_KEY } = process.env;
const COAUTHOR_URL = `https://${ new URL( WEBSOCKET_URL ).host }/toolkit/v1/environments/${ ENVIRONMENT_ID }`;

const token = await new SignJWT( {
	user: { id: 'quick-start-user', name: 'Quick start user' },
	auth: {
		toolkit: {
			permissions: [ 'sessions:open', 'sessions:read', 'sessions:tools', 'sessions:apply', 'sessions:close' ],
			documents: { pricing: {} },
			mode: 'suggest'
		}
	}
} )
	.setProtectedHeader( { alg: 'HS256' } )
	.setAudience( ENVIRONMENT_ID )
	.setSubject( 'quick-start-user' )
	.setIssuedAt()
	.setExpirationTime( '1h' )
	.sign( new TextEncoder().encode( ACCESS_KEY ) );

console.log( JSON.stringify( { type: 'http', url: `${ COAUTHOR_URL }/mcp`, headers: { Authorization: `Bearer ${ token }` } } ) );

import { anthropic } from '@ai-sdk/anthropic';
import { createMCPClient } from '@ai-sdk/mcp';
import { generateText, isStepCount } from 'ai';
import { SignJWT } from 'jose';

process.loadEnvFile();

const { ENVIRONMENT_ID, WEBSOCKET_URL, ACCESS_KEY } = process.env;
const DOCUMENT_ID = 'pricing';

// Coauthor runs on the host of your WebSocket URL.
const COAUTHOR_URL = `https://${ new URL( WEBSOCKET_URL ).host }/toolkit/v1/environments/${ ENVIRONMENT_ID }`;

// The backend token lets this script open, apply, and close sessions on `pricing`.
const backendToken = await new SignJWT( {
	auth: {
		toolkit: {
			permissions: [ 'sessions:open', 'sessions:read', 'sessions:apply', 'sessions:close' ],
			documents: { [ DOCUMENT_ID ]: {} },
			mode: 'suggest'
		}
	}
} )
	.setProtectedHeader( { alg: 'HS256' } )
	.setAudience( ENVIRONMENT_ID )
	.setSubject( 'quick-start-backend' )
	.setIssuedAt()
	.setExpirationTime( '1h' )
	.sign( new TextEncoder().encode( ACCESS_KEY ) );

async function send( path, body ) {
	const response = await fetch( `${ COAUTHOR_URL }${ path }`, {
		method: 'POST',
		headers: { Authorization: `Bearer ${ backendToken }`, 'Content-Type': 'application/json' },
		body: body === undefined ? null : JSON.stringify( body )
	} );

	if ( !response.ok ) {
		throw new Error( `${ path } answered ${ response.status }: ${ await response.text() }` );
	}

	return response.status === 204 ? null : response.json();
}

// 1. Open a session. Quick start agent becomes the author of the suggestions.
const session = await send( '/sessions', {
	target: { type: 'live', documentId: DOCUMENT_ID },
	actor: { id: 'quick-start-agent', name: 'Quick start agent' },
	mode: 'suggest'
} );

try {
	// 2. Connect with the session token, and let the model call the tools.
	const coauthor = await createMCPClient( {
		transport: { type: 'http', url: `${ COAUTHOR_URL }/mcp`, headers: { Authorization: `Bearer ${ session.token }` } }
	} );
	const { text } = await generateText( {
		model: anthropic( 'claude-sonnet-5-5' ),
		tools: await coauthor.tools(),
		instructions: coauthor.instructions,
		prompt: process.argv[ 2 ],
		stopWhen: isStepCount( 30 )
	} );

	await coauthor.close();
	console.log( text );

	// 3. Apply the edits. They reach the document as suggestions.
	await send( `/sessions/${ session.id }/apply` );
} finally {
	// 4. Close the session.
	await send( `/sessions/${ session.id }/close` );
}