fix: content status for PieceInstances can be under the id of the PieceInstance or the wrapped Piece SOFIE-2993

Author: Julusian

meteor/client/ui/SegmentTimeline/withMediaObjectStatus.tsx

@@ -70,6 +70,35 @@ export function withMediaObjectStatus<IProps extends AnyPiece, IState>(): (
 				}
 			}
 
+			private getStatusDocForPiece(
+				piece: BucketAdLibUi | IAdLibListItem | AdLibPieceUi | PieceUi | BucketAdLibActionUi
+			) {
+				const pieceUnwrapped = WithMediaObjectStatusHOCComponent.unwrapPieceInstance(piece)
+
+				// Bucket items use a different collection
+				if (RundownUtils.isBucketAdLibItem(piece)) {
+					return UIBucketContentStatuses.findOne({
+						bucketId: piece.bucketId,
+						docId: pieceUnwrapped._id,
+					})
+				}
+
+				// PieceInstance's might have a dedicated status
+				if (RundownUtils.isPieceInstance(piece)) {
+					const status = UIPieceContentStatuses.findOne({
+						// Future: It would be good for this to be stricter.
+						pieceId: piece.instance._id,
+					})
+					if (status) return status
+				}
+
+				// Fallback to using the one from the source piece
+				return UIPieceContentStatuses.findOne({
+					// Future: It would be good for this to be stricter.
+					pieceId: pieceUnwrapped._id,
+				})
+			}
+
 			updateDataTracker() {
 				if (this.destroyed) return
 
@@ -80,19 +109,8 @@ export function withMediaObjectStatus<IProps extends AnyPiece, IState>(): (
 
 					// Check item status
 					if (piece && (piece.sourceLayer || layer) && studio) {
-						const pieceUnwrapped = WithMediaObjectStatusHOCComponent.unwrapPieceInstance(piece)
-						const statusDoc = RundownUtils.isBucketAdLibItem(piece)
-							? UIBucketContentStatuses.findOne({
-									bucketId: piece.bucketId,
-									docId: pieceUnwrapped._id,
-							  })
-							: UIPieceContentStatuses.findOne({
-									// Future: It would be good for this to be stricter.
-									pieceId: pieceUnwrapped._id,
-							  })
-
 						// Extract the status or populate some default values
-						const statusObj = statusDoc?.status ?? DEFAULT_STATUS
+						const statusObj = this.getStatusDocForPiece(piece)?.status ?? DEFAULT_STATUS
 
 						if (RundownUtils.isAdLibPieceOrAdLibListItem(piece)) {
 							if (!overrides.piece || !_.isEqual(statusObj, (overrides.piece as AdLibPieceUi).contentStatus)) {