Put dom getters in dom-elements.js.

Author: sebadoom

src/website/dom-elements.js

@@ -6,6 +6,9 @@ export const menuScrollableLinks =
 
 export const sectionElements = document.getElementsByTagName('section');
 
+export const extensionSection = document.querySelector('.update-site');
+export const ebookSection = document.querySelector('.jtw-ebook-banner');
+
 export const extensionButton = document.getElementById('extension-button');
 export const extensionButtonText = extensionButton ?
   extensionButton.querySelector('.button-text') :

src/website/index.js

@@ -14,6 +14,8 @@ import {
   publicKeyTextArea, 
   codeElements, 
   debuggerSection,
+  extensionSection,
+  ebookSection
 } from './dom-elements.js';
 
 /* For initialization, look at the end of this file */
@@ -57,9 +59,7 @@ function loadToken() {
 }
 
 function pickEbookOrExtensionBanner() {
-  var extensionSection = document.getElementsByClassName("update-site")[0];
-  var ebookSection = document.getElementsByClassName("jtw-ebook-banner")[0];
-  if (Math.random() >= .5) {
+  if(Math.random() >= 0.5) {
     extensionSection.style.display = 'block';
   } else {
     ebookSection.style.display = 'block';