Update security/class-private-sites.php

Co-authored-by: Volodymyr Kolesnykov <[email protected]>
Automattic · Sep 28, 2023 · 1d64f9a · 1d64f9a
1 parent 3297c96
commit 1d64f9a
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/security/class-private-sites.php b/security/class-private-sites.php
@@ -95,14 +95,21 @@ public function disable_blog_public_ui() {
 		wp_register_script( 'vip-disable-blog-public-option-ui', false, array(), '0.1', true );
 		wp_enqueue_script( 'vip-disable-blog-public-option-ui' );
 		$js_code       = <<<JS
-		document.addEventListener("DOMContentLoaded", function() {
+		function onContentLoaded(callback) {
+			if (document.readyState !== 'loading') {
+				callback();
+			} else {
+				document.addEventListener('DOMContentLoaded', callback);
+			}
+		}
+
+		onContentLoaded(function() {
 			function updateProperty(selector, property, value) {
 				const element = document.querySelector(selector);
 				if (element) {
 					element[property] = value;
 				}
 			}
-
 			var checkbox = 'tr.option-site-visibility input#blog_public[type="checkbox"]';
 			if (document.querySelector(checkbox)) {
 				updateProperty(checkbox, 'disabled', true);