basic enhancer function

Author: i-hardy

dotcom-rendering/src/types/hostedContent.ts

@@ -1,5 +1,29 @@
 import { FEHostedContent } from 'src/frontend/feHostedContent';
 
+enum HostedContentType {
+	Article,
+	Video,
+	Gallery,
+}
+
 export type HostedContent = {
 	frontendData: FEHostedContent;
+	type: HostedContentType;
+};
+
+export const enhanceHostedContentType = (
+	data: FEHostedContent,
+): HostedContent => {
+	let type = HostedContentType.Article;
+
+	if (data.video) {
+		type = HostedContentType.Video;
+	} else if (data.images.length) {
+		type = HostedContentType.Gallery;
+	}
+
+	return {
+		frontendData: data,
+		type,
+	};
 };