@fluidframework/cell 2.0.0-internal.3.0.2 → 2.0.0-internal.3.2.0

package/src/cell.ts

@@ -6,21 +6,21 @@
 import { assert } from "@fluidframework/common-utils";
 import { ISequencedDocumentMessage, MessageType } from "@fluidframework/protocol-definitions";
 import {
-    IChannelAttributes,
-    IFluidDataStoreRuntime,
-    IChannelStorageService,
-    IChannelFactory,
-    Serializable,
+	IChannelAttributes,
+	IFluidDataStoreRuntime,
+	IChannelStorageService,
+	IChannelFactory,
+	Serializable,
 } from "@fluidframework/datastore-definitions";
-import { ISummaryTreeWithStats } from "@fluidframework/runtime-definitions";
+import { AttributionKey, ISummaryTreeWithStats } from "@fluidframework/runtime-definitions";
 import { readAndParse } from "@fluidframework/driver-utils";
-import { createSingleBlobSummary, IFluidSerializer, SharedObject } from "@fluidframework/shared-object-base";
-import { CellFactory } from "./cellFactory";
 import {
-    ISharedCell,
-    ISharedCellEvents,
-    ICellLocalOpMetadata,
-} from "./interfaces";
+	createSingleBlobSummary,
+	IFluidSerializer,
+	SharedObject,
+} from "@fluidframework/shared-object-base";
+import { CellFactory } from "./cellFactory";
+import { ISharedCell, ISharedCellEvents, ICellLocalOpMetadata, ICellOptions } from "./interfaces";
 
 /**
  * Description of a cell delta operation
@@ -28,19 +28,24 @@ import {
 type ICellOperation = ISetCellOperation | IDeleteCellOperation;
 
 interface ISetCellOperation {
-    type: "setCell";
-    value: ICellValue;
+	type: "setCell";
+	value: ICellValue;
 }
 
 interface IDeleteCellOperation {
-    type: "deleteCell";
+	type: "deleteCell";
 }
 
 interface ICellValue {
-    /**
-     * The actual value contained in the `Cell`, which needs to be wrapped to handle `undefined`.
-     */
-    value: unknown;
+	/**
+	 * The actual value contained in the `Cell`, which needs to be wrapped to handle `undefined`.
+	 */
+	value: unknown;
+	/**
+	 * The attribution key contained in the `Cell`.
+	 * @alpha
+	 */
+	attribution?: AttributionKey;
 }
 
 const snapshotFileName = "header";
@@ -50,269 +55,318 @@ const snapshotFileName = "header";
  */
 // TODO: use `unknown` instead (breaking change).
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
-export class SharedCell<T = any> extends SharedObject<ISharedCellEvents<T>> implements ISharedCell<T> {
-    /**
-     * Create a new `SharedCell`.
-     *
-     * @param runtime - The data store runtime to which the `SharedCell` belongs.
-     * @param id - Unique identifier for the `SharedCell`.
-     *
-     * @returns The newly create `SharedCell`. Note that it will not yet be attached.
-     */
-    public static create(runtime: IFluidDataStoreRuntime, id?: string): SharedCell {
-        return runtime.createChannel(id, CellFactory.Type) as SharedCell;
-    }
-
-    /**
-     * Gets the factory for the `SharedCell` to register with the data store.
-     *
-     * @returns A factory that creates and loads `SharedCell`s.
-     */
-    public static getFactory(): IChannelFactory {
-        return new CellFactory();
-    }
-
-    /**
-     * The data held by this cell.
-     */
-    private data: Serializable<T> | undefined;
-
-    /**
-     * This is used to assign a unique id to outgoing messages. It is used to track messages until
-     * they are ack'd.
-     */
-    private messageId: number = -1;
-
-    /**
-     * This keeps track of the messageId of messages that have been ack'd. It is updated every time
-     * we a message is ack'd with it's messageId.
-     */
-    private messageIdObserved: number = -1;
-
-    private readonly pendingMessageIds: number[] = [];
-
-    /**
-     * Constructs a new `SharedCell`.
-     * If the object is non-local an id and service interfaces will be provided.
-     *
-     * @param runtime - The data store runtime to which the `SharedCell` belongs.
-     * @param id - Unique identifier for the `SharedCell`.
-     */
-    public constructor(id: string, runtime: IFluidDataStoreRuntime, attributes: IChannelAttributes) {
-        super(id, runtime, attributes, "fluid_cell_");
-    }
-
-    /**
-     * {@inheritDoc ISharedCell.get}
-     */
-    public get(): Serializable<T> | undefined {
-        return this.data;
-    }
-
-    /**
-     * {@inheritDoc ISharedCell.set}
-     */
-    public set(value: Serializable<T>): void {
-        // Serialize the value if required.
-        const operationValue: ICellValue = {
-            value: this.serializer.encode(value, this.handle),
-        };
-
-        // Set the value locally.
-        const previousValue = this.setCore(value);
-
-        // If we are not attached, don't submit the op.
-        if (!this.isAttached()) {
-            return;
-        }
-
-        const op: ISetCellOperation = {
-            type: "setCell",
-            value: operationValue,
-        };
-        this.submitCellMessage(op, previousValue);
-    }
-
-    /**
-     * {@inheritDoc ISharedCell.delete}
-     */
-    public delete(): void {
-        // Delete the value locally.
-        const previousValue = this.deleteCore();
-
-        // If we are not attached, don't submit the op.
-        if (!this.isAttached()) {
-            return;
-        }
-
-        const op: IDeleteCellOperation = {
-            type: "deleteCell",
-        };
-        this.submitCellMessage(op, previousValue);
-    }
-
-    /**
-     * {@inheritDoc ISharedCell.empty}
-     */
-    public empty(): boolean {
-        return this.data === undefined;
-    }
-
-    /**
-     * Creates a summary for the Cell.
-     *
-     * @returns The summary of the current state of the Cell.
-     */
-    protected summarizeCore(serializer: IFluidSerializer): ISummaryTreeWithStats {
-        const content: ICellValue = { value: this.data };
-        return createSingleBlobSummary(snapshotFileName, serializer.stringify(content, this.handle));
-    }
-
-    /**
-     * {@inheritDoc @fluidframework/shared-object-base#SharedObject.loadCore}
-     */
-    protected async loadCore(storage: IChannelStorageService): Promise<void> {
-        const content = await readAndParse<ICellValue>(storage, snapshotFileName);
-
-        this.data = this.decode(content);
-    }
-
-    /**
-     * Initialize a local instance of cell.
-     */
-    protected initializeLocalCore(): void {
-        this.data = undefined;
-    }
-
-    /**
-     * Call back on disconnect.
-     */
-    protected onDisconnect(): void { }
-
-    /**
-     * Apply inner op.
-     *
-     * @param content - ICellOperation content
-     */
-    private applyInnerOp(content: ICellOperation): Serializable<T> | undefined {
-        switch (content.type) {
-            case "setCell":
-                return this.setCore(this.decode(content.value));
-
-            case "deleteCell":
-                return this.deleteCore();
-
-            default:
-                throw new Error("Unknown operation");
-        }
-    }
-
-    /**
-     * Process a cell operation (op).
-     *
-     * @param message - The message to prepare.
-     * @param local - Whether or not the message was sent by the local client.
-     * @param localOpMetadata - For local client messages, this is the metadata that was submitted with the message.
-     * For messages from a remote client, this will be `undefined`.
-     */
-    protected processCore(message: ISequencedDocumentMessage, local: boolean, localOpMetadata: unknown): void {
-        const cellOpMetadata = localOpMetadata as ICellLocalOpMetadata;
-        if (this.messageId !== this.messageIdObserved) {
-            // We are waiting for an ACK on our change to this cell - we will ignore all messages until we get it.
-            if (local) {
-                const messageIdReceived = cellOpMetadata.pendingMessageId;
-                assert(messageIdReceived !== undefined && messageIdReceived <= this.messageId,
-                    0x00c /* "messageId is incorrect from from the local client's ACK" */);
-                assert(this.pendingMessageIds !== undefined &&
-                    this.pendingMessageIds[0] === cellOpMetadata.pendingMessageId,
-                    0x471 /* Unexpected pending message received */);
-                this.pendingMessageIds.shift();
-                // We got an ACK. Update messageIdObserved.
-                this.messageIdObserved = cellOpMetadata.pendingMessageId;
-            }
-            return;
-        }
-
-        if (message.type === MessageType.Operation && !local) {
-            const op = message.contents as ICellOperation;
-            this.applyInnerOp(op);
-        }
-    }
-
-    private setCore(value: Serializable<T>): Serializable<T> | undefined {
-        const previousLocalValue = this.get();
-        this.data = value;
-        this.emit("valueChanged", value);
-        return previousLocalValue;
-    }
-
-    private deleteCore(): Serializable<T> | undefined {
-        const previousLocalValue = this.get();
-        this.data = undefined;
-        this.emit("delete");
-        return previousLocalValue;
-    }
-
-    private decode(cellValue: ICellValue): Serializable<T> {
-        const value = cellValue.value;
-        return this.serializer.decode(value) as Serializable<T> ;
-    }
-
-    private createLocalOpMetadata(op: ICellOperation, previousValue?: Serializable<T>): ICellLocalOpMetadata {
-        const pendingMessageId = ++this.messageId;
-        this.pendingMessageIds.push(pendingMessageId);
-        const localMetadata: ICellLocalOpMetadata = {
-            pendingMessageId, previousValue,
-        };
-        return localMetadata;
-    }
-
-    /**
-     * {@inheritDoc @fluidframework/shared-object-base#SharedObjectCore.applyStashedOp}
-     *
-     * @internal
-     */
-    protected applyStashedOp(content: unknown): unknown {
-        const cellContent = content as ICellOperation;
-        const previousValue = this.applyInnerOp(cellContent);
-        return this.createLocalOpMetadata(cellContent, previousValue);
-    }
-
-    /**
-     * Rollback a local op.
-     *
-     * @param content - The operation to rollback.
-     * @param localOpMetadata - The local metadata associated with the op.
-     */
-    // TODO: use `unknown` instead (breaking change).
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any, @typescript-eslint/explicit-module-boundary-types
-    protected rollback(content: any, localOpMetadata: unknown): void {
-        const cellOpMetadata = localOpMetadata as ICellLocalOpMetadata;
-        // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
-        if (content.type === "setCell" || content.type === "deleteCell") {
-            if (cellOpMetadata.previousValue === undefined) {
-                this.deleteCore();
-            } else {
-                this.setCore(cellOpMetadata.previousValue as Serializable<T>);
-            }
-
-            const lastPendingMessageId = this.pendingMessageIds.pop();
-            if (lastPendingMessageId !== cellOpMetadata.pendingMessageId) {
-                throw new Error("Rollback op does not match last pending");
-            }
-        } else {
-            throw new Error("Unsupported op for rollback");
-        }
-    }
-
-     /**
-     * Submit a cell message to remote clients.
-     *
-     * @param op - The cell message.
-     * @param previousValue - The value of the cell before this op.
-     */
-    private submitCellMessage(op: ICellOperation, previousValue?: Serializable<T>): void {
-        const localMetadata = this.createLocalOpMetadata(op, previousValue);
-        this.submitLocalMessage(op, localMetadata);
-    }
+export class SharedCell<T = any>
+	extends SharedObject<ISharedCellEvents<T>>
+	implements ISharedCell<T>
+{
+	/**
+	 * Create a new `SharedCell`.
+	 *
+	 * @param runtime - The data store runtime to which the `SharedCell` belongs.
+	 * @param id - Unique identifier for the `SharedCell`.
+	 *
+	 * @returns The newly create `SharedCell`. Note that it will not yet be attached.
+	 */
+	public static create(runtime: IFluidDataStoreRuntime, id?: string): SharedCell {
+		return runtime.createChannel(id, CellFactory.Type) as SharedCell;
+	}
+
+	/**
+	 * Gets the factory for the `SharedCell` to register with the data store.
+	 *
+	 * @returns A factory that creates and loads `SharedCell`s.
+	 */
+	public static getFactory(): IChannelFactory {
+		return new CellFactory();
+	}
+
+	/**
+	 * The data held by this cell.
+	 */
+	private data: Serializable<T> | undefined;
+
+	/**
+	 * This is used to assign a unique id to outgoing messages. It is used to track messages until
+	 * they are ack'd.
+	 */
+	private messageId: number = -1;
+
+	/**
+	 * This keeps track of the messageId of messages that have been ack'd. It is updated every time
+	 * we a message is ack'd with it's messageId.
+	 */
+	private messageIdObserved: number = -1;
+
+	private readonly pendingMessageIds: number[] = [];
+
+	private attribution: AttributionKey | undefined;
+
+	private readonly options: ICellOptions | undefined;
+
+	/**
+	 * Constructs a new `SharedCell`.
+	 * If the object is non-local an id and service interfaces will be provided.
+	 *
+	 * @alpha
+	 *
+	 * @param runtime - The data store runtime to which the `SharedCell` belongs.
+	 * @param id - Unique identifier for the `SharedCell`.
+	 */
+	// eslint-disable-next-line @typescript-eslint/explicit-member-accessibility
+	constructor(id: string, runtime: IFluidDataStoreRuntime, attributes: IChannelAttributes) {
+		super(id, runtime, attributes, "fluid_cell_");
+
+		this.options = runtime.options as ICellOptions;
+	}
+
+	/**
+	 * {@inheritDoc ISharedCell.get}
+	 */
+	public get(): Serializable<T> | undefined {
+		return this.data;
+	}
+
+	/**
+	 * {@inheritDoc ISharedCell.set}
+	 */
+	public set(value: Serializable<T>): void {
+		// Serialize the value if required.
+		const operationValue: ICellValue = {
+			value: this.serializer.encode(value, this.handle),
+		};
+
+		// Set the value locally.
+		const previousValue = this.setCore(value);
+
+		// If we are not attached, don't submit the op.
+		if (!this.isAttached()) {
+			return;
+		}
+
+		const op: ISetCellOperation = {
+			type: "setCell",
+			value: operationValue,
+		};
+		this.submitCellMessage(op, previousValue);
+	}
+
+	/**
+	 * {@inheritDoc ISharedCell.delete}
+	 */
+	public delete(): void {
+		// Delete the value locally.
+		const previousValue = this.deleteCore();
+
+		// If we are not attached, don't submit the op.
+		if (!this.isAttached()) {
+			return;
+		}
+
+		const op: IDeleteCellOperation = {
+			type: "deleteCell",
+		};
+		this.submitCellMessage(op, previousValue);
+	}
+
+	/**
+	 * {@inheritDoc ISharedCell.empty}
+	 */
+	public empty(): boolean {
+		return this.data === undefined;
+	}
+
+	/**
+	 * {@inheritDoc ISharedCell.getAttribution}
+	 * @alpha
+	 */
+	public getAttribution(): AttributionKey | undefined {
+		return this.attribution;
+	}
+
+	/**
+	 * Set the attribution through the SequencedDocumentMessage
+	 */
+	private setAttribution(message: ISequencedDocumentMessage): void {
+		if (this.options?.attribution?.track ?? false) {
+			this.attribution = { type: "op", seq: message.sequenceNumber };
+		}
+	}
+
+	/**
+	 * Creates a summary for the Cell.
+	 *
+	 * @returns The summary of the current state of the Cell.
+	 */
+	protected summarizeCore(serializer: IFluidSerializer): ISummaryTreeWithStats {
+		const content: ICellValue = { value: this.data, attribution: this.attribution };
+		return createSingleBlobSummary(
+			snapshotFileName,
+			serializer.stringify(content, this.handle),
+		);
+	}
+
+	/**
+	 * {@inheritDoc @fluidframework/shared-object-base#SharedObject.loadCore}
+	 */
+	protected async loadCore(storage: IChannelStorageService): Promise<void> {
+		const content = await readAndParse<ICellValue>(storage, snapshotFileName);
+
+		this.data = this.decode(content);
+		this.attribution = content.attribution;
+	}
+
+	/**
+	 * Initialize a local instance of cell.
+	 */
+	protected initializeLocalCore(): void {
+		this.data = undefined;
+	}
+
+	/**
+	 * Call back on disconnect.
+	 */
+	protected onDisconnect(): void {}
+
+	/**
+	 * Apply inner op.
+	 *
+	 * @param content - ICellOperation content
+	 */
+	private applyInnerOp(content: ICellOperation): Serializable<T> | undefined {
+		switch (content.type) {
+			case "setCell":
+				return this.setCore(this.decode(content.value));
+
+			case "deleteCell":
+				return this.deleteCore();
+
+			default:
+				throw new Error("Unknown operation");
+		}
+	}
+
+	/**
+	 * Process a cell operation (op).
+	 *
+	 * @param message - The message to prepare.
+	 * @param local - Whether or not the message was sent by the local client.
+	 * @param localOpMetadata - For local client messages, this is the metadata that was submitted with the message.
+	 * For messages from a remote client, this will be `undefined`.
+	 */
+	protected processCore(
+		message: ISequencedDocumentMessage,
+		local: boolean,
+		localOpMetadata: unknown,
+	): void {
+		const cellOpMetadata = localOpMetadata as ICellLocalOpMetadata;
+		if (this.messageId !== this.messageIdObserved) {
+			// We are waiting for an ACK on our change to this cell - we will ignore all messages until we get it.
+			if (local) {
+				const messageIdReceived = cellOpMetadata.pendingMessageId;
+				assert(
+					messageIdReceived !== undefined && messageIdReceived <= this.messageId,
+					0x00c /* "messageId is incorrect from from the local client's ACK" */,
+				);
+				assert(
+					this.pendingMessageIds !== undefined &&
+						this.pendingMessageIds[0] === cellOpMetadata.pendingMessageId,
+					0x471 /* Unexpected pending message received */,
+				);
+				this.pendingMessageIds.shift();
+				// We got an ACK. Update messageIdObserved.
+				this.messageIdObserved = cellOpMetadata.pendingMessageId;
+				// update the attributor
+				this.setAttribution(message);
+			}
+			return;
+		}
+
+		if (message.type === MessageType.Operation && !local) {
+			const op = message.contents as ICellOperation;
+			// update the attributor
+			this.setAttribution(message);
+			this.applyInnerOp(op);
+		}
+	}
+
+	private setCore(value: Serializable<T>): Serializable<T> | undefined {
+		const previousLocalValue = this.get();
+		this.data = value;
+		this.emit("valueChanged", value);
+		return previousLocalValue;
+	}
+
+	private deleteCore(): Serializable<T> | undefined {
+		const previousLocalValue = this.get();
+		this.data = undefined;
+		this.emit("delete");
+		return previousLocalValue;
+	}
+
+	private decode(cellValue: ICellValue): Serializable<T> {
+		const value = cellValue.value;
+		return this.serializer.decode(value) as Serializable<T>;
+	}
+
+	private createLocalOpMetadata(
+		op: ICellOperation,
+		previousValue?: Serializable<T>,
+	): ICellLocalOpMetadata {
+		const pendingMessageId = ++this.messageId;
+		this.pendingMessageIds.push(pendingMessageId);
+		const localMetadata: ICellLocalOpMetadata = {
+			pendingMessageId,
+			previousValue,
+		};
+		return localMetadata;
+	}
+
+	/**
+	 * {@inheritDoc @fluidframework/shared-object-base#SharedObjectCore.applyStashedOp}
+	 *
+	 * @internal
+	 */
+	protected applyStashedOp(content: unknown): unknown {
+		const cellContent = content as ICellOperation;
+		const previousValue = this.applyInnerOp(cellContent);
+		return this.createLocalOpMetadata(cellContent, previousValue);
+	}
+
+	/**
+	 * Rollback a local op.
+	 *
+	 * @param content - The operation to rollback.
+	 * @param localOpMetadata - The local metadata associated with the op.
+	 */
+	// TODO: use `unknown` instead (breaking change).
+	// eslint-disable-next-line @typescript-eslint/no-explicit-any, @typescript-eslint/explicit-module-boundary-types
+	protected rollback(content: any, localOpMetadata: unknown): void {
+		const cellOpMetadata = localOpMetadata as ICellLocalOpMetadata;
+		// eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
+		if (content.type === "setCell" || content.type === "deleteCell") {
+			if (cellOpMetadata.previousValue === undefined) {
+				this.deleteCore();
+			} else {
+				this.setCore(cellOpMetadata.previousValue as Serializable<T>);
+			}
+
+			const lastPendingMessageId = this.pendingMessageIds.pop();
+			if (lastPendingMessageId !== cellOpMetadata.pendingMessageId) {
+				throw new Error("Rollback op does not match last pending");
+			}
+		} else {
+			throw new Error("Unsupported op for rollback");
+		}
+	}
+
+	/**
+	 * Submit a cell message to remote clients.
+	 *
+	 * @param op - The cell message.
+	 * @param previousValue - The value of the cell before this op.
+	 */
+	private submitCellMessage(op: ICellOperation, previousValue?: Serializable<T>): void {
+		const localMetadata = this.createLocalOpMetadata(op, previousValue);
+		this.submitLocalMessage(op, localMetadata);
+	}
 }