@rivetkit/cloudflare-workers/src/util.ts

Cloudflare Workers adapter for RivetKit actors

// Constants for key handling
export const EMPTY_KEY = "(none)";
export const KEY_SEPARATOR = ",";

/**
 * Serializes an array of key strings into a single string for use with idFromName
 *
 * @param name The actor name
 * @param key Array of key strings to serialize
 * @returns A single string containing the serialized name and key
 */
export function serializeNameAndKey(name: string, key: string[]): string {
	// Escape colons in the name
	const escapedName = name.replace(/:/g, "\\:");

	// For empty keys, just return the name and a marker
	if (key.length === 0) {
		return `${escapedName}:${EMPTY_KEY}`;
	}

	// Serialize the key array
	const serializedKey = serializeKey(key);

	// Combine name and serialized key
	return `${escapedName}:${serializedKey}`;
}

/**
 * Serializes an array of key strings into a single string
 *
 * @param key Array of key strings to serialize
 * @returns A single string containing the serialized key
 */
export function serializeKey(key: string[]): string {
	// Use a special marker for empty key arrays
	if (key.length === 0) {
		return EMPTY_KEY;
	}

	// Escape each key part to handle the separator and the empty key marker
	const escapedParts = key.map((part) => {
		// First check if it matches our empty key marker
		if (part === EMPTY_KEY) {
			return `\\${EMPTY_KEY}`;
		}

		// Escape backslashes first, then commas
		let escaped = part.replace(/\\/g, "\\\\");
		escaped = escaped.replace(/,/g, "\\,");
		return escaped;
	});

	return escapedParts.join(KEY_SEPARATOR);
}

/**
 * Deserializes a key string back into an array of key strings
 *
 * @param keyString The serialized key string
 * @returns Array of key strings
 */
export function deserializeKey(keyString: string): string[] {
	// Handle empty values
	if (!keyString) {
		return [];
	}

	// Check for special empty key marker
	if (keyString === EMPTY_KEY) {
		return [];
	}

	// Split by unescaped commas and unescape the escaped characters
	const parts: string[] = [];
	let currentPart = "";
	let escaping = false;

	for (let i = 0; i < keyString.length; i++) {
		const char = keyString[i];

		if (escaping) {
			// This is an escaped character, add it directly
			currentPart += char;
			escaping = false;
		} else if (char === "\\") {
			// Start of an escape sequence
			escaping = true;
		} else if (char === KEY_SEPARATOR) {
			// This is a separator
			parts.push(currentPart);
			currentPart = "";
		} else {
			// Regular character
			currentPart += char;
		}
	}

	// Add the last part if it exists
	if (currentPart || parts.length > 0) {
		parts.push(currentPart);
	}

	return parts;
}