docs: ensure only single demo iframes are selected

Financial-Times · Mar 4, 2024 · 78194f3 · 78194f3
1 parent 2e99d70
commit 78194f3
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/apps/o2-storybook-composition/addons/html/src/Panel.js b/apps/o2-storybook-composition/addons/html/src/Panel.js
@@ -6,6 +6,7 @@ import {PanelContent} from './components/PanelContent';
 import {format as prettierFormat} from 'prettier/standalone';
 import prettierHtml from 'prettier/parser-html';
 
+const demoUrl = '__origami/service/build/v3/demo';
 export const Panel = props => {
 	// https://storybook.js.org/docs/react/addons/addons-api#useaddonstate
 	const [{code}, setState] = useAddonState(ADDON_ID, {
@@ -20,7 +21,10 @@ export const Panel = props => {
 
 			const foundIframe = codeDoc.getElementsByTagName('iframe');
 
-			if (foundIframe.length === 1) {
+			if (
+				foundIframe.length === 1 &&
+				foundIframe.item(0).src.includes(demoUrl)
+			) {
 				console.log(foundIframe.item(0).src);
 				const demoEndpoint = parser.parseFromString(
 					foundIframe.item(0).src,