main.js: Wait until headers are on the page before acting

Don't set ids for the page headers until those headers are available on
the page.

1 files changed, 28 insertions, 5 deletions

diff --git a/main.js b/main.js
index b7bc92b..24d09ac 100644
--- a/main.js
+++ b/main.js
@@ -1,16 +1,39 @@
 (function() {
 	var jira_person_header_class_name = '.ghx-heading';
+	var person_headers;
 	
 	// Removes spaces from a string
 	function slugify (str) {
 		return str.replace(/ /g, '');
 	}
 	
-	Array.prototype.forEach.call(
-		document.querySelectorAll(jira_person_header_class_name),
-		function(element) {
-			element.setAttribute('id', slugify(element.firstChild.textContent));
-			console.log(element);
+	function do_when (when, action) {
+		var timeout;
+		
+		window.clearTimeout(timeout);
+		
+		if (when()) {
+			action();
+		}
+		else {
+			timeout = window.setTimeout(function() {
+				do_when(when, action)
+			}, 500);
+		}
+	}
+	
+	do_when(
+		function() {
+			person_headers = document.querySelectorAll(jira_person_header_class_name);
+			return person_headers.length;
+		},
+		function () {
+			Array.prototype.forEach.call(
+				person_headers,
+				function(element) {
+					element.setAttribute('id', slugify(element.firstChild.textContent));
+				}
+			);
 		}
 	);
 })();