|
35 | 35 |
|
36 | 36 | <script> |
37 | 37 | function setExperimental(event) { |
38 | | - var classList = document.getElementsByTagName("iframe")[0].contentDocument.body.classList; |
| 38 | + var frame = document.querySelector("#storybook-preview-iframe"); |
| 39 | + var classList = frame.contentDocument.body.classList; |
39 | 40 | if(event.target.previousElementSibling) { |
40 | 41 | classList.add("experimental"); |
| 42 | + classList.remove("carbon"); |
41 | 43 | event.target.previousElementSibling.classList.remove("experimental-selected"); |
42 | 44 | } else { |
43 | 45 | classList.remove("experimental"); |
| 46 | + classList.add("carbon"); |
44 | 47 | event.target.nextElementSibling.classList.remove("experimental-selected"); |
45 | 48 | } |
46 | 49 | event.target.classList.add("experimental-selected"); |
47 | 50 | }; |
48 | 51 |
|
49 | 52 | var experimentalToggleInterval = setInterval(function() { |
50 | | - if(!document.getElementsByTagName("iframe")[0] || |
51 | | - !document.getElementsByTagName("iframe")[0].contentDocument || |
52 | | - !document.getElementsByTagName("iframe")[0].contentDocument.getElementById("root")) { |
| 53 | + var frame = document.querySelector("#storybook-preview-iframe"); |
| 54 | + if(!frame || |
| 55 | + !frame.contentDocument || |
| 56 | + !frame.contentDocument.getElementById("root")) { |
53 | 57 | return; |
54 | 58 | } |
55 | 59 |
|
56 | | - document.getElementsByTagName("iframe")[0].contentDocument.head.appendChild( |
| 60 | + frame.contentDocument.head.appendChild( |
57 | 61 | document.getElementById("iframestyle") |
58 | 62 | ); |
59 | 63 |
|
60 | 64 | clearInterval(experimentalToggleInterval); |
61 | | - document.getElementsByTagName("iframe")[0].contentDocument.body.insertAdjacentHTML( |
| 65 | + frame.contentDocument.body.insertAdjacentHTML( |
62 | 66 | "beforeend", |
63 | 67 | `<div style="position: fixed; right: 15px; bottom: 15px"> |
64 | 68 | <button |
|
73 | 77 | </button> |
74 | 78 | </div> |
75 | 79 | `); |
76 | | - document.getElementsByTagName("iframe")[0].contentWindow.setExperimental = setExperimental; |
| 80 | + frame.contentWindow.setExperimental = setExperimental; |
| 81 | + |
| 82 | + var classList = frame.contentDocument.body.classList; |
| 83 | + classList.add("carbon"); |
77 | 84 | }, 100); |
78 | 85 | </script> |
0 commit comments