Fix endless page loading with WPML.

src/php/class-main.php

@@ -167,6 +167,8 @@ public function init_hooks() {
 		}
 
 		if ( class_exists( SitePress::class ) ) {
+			$this->wpml_locale = $this->get_wpml_locale();
+
 			// We cannot use locale filter here
 			// as WPML reverts locale at PHP_INT_MAX in \WPML\ST\MO\Hooks\LanguageSwitch::filterLocale.
 			add_filter( 'ctl_locale', [ $this, 'wpml_locale_filter' ], - PHP_INT_MAX );
@@ -609,16 +611,19 @@ public function wpml_locale_filter( $locale ) {
 			return $this->wpml_locale;
 		}
 
+		return $locale;
+	}
+
+	/**
+	 * Get wpml locale.
+	 *
+	 * @return string|null
+	 */
+	private function get_wpml_locale() {
 		$language_code = wpml_get_current_language();
 		$languages     = apply_filters( 'wpml_active_languages', null );
 
-		if ( isset( $languages[ $language_code ] ) ) {
-			$this->wpml_locale = $languages[ $language_code ]['default_locale'];
-
-			return $this->wpml_locale;
-		}
-
-		return $locale;
+		return isset( $languages[ $language_code ] ) ? $languages[ $language_code ]['default_locale'] : null;
 	}
 
 	/**