PLATFORM-9358 | Ignore Parser processing limit in DPL

Wikia · May 15, 2024 · 7121868 · 7121868
1 parent f022e19
commit 7121868
Showing 1 changed file with 32 additions and 0 deletions.
diff --git a/includes/LST.php b/includes/LST.php
@@ -915,6 +915,7 @@ public static function spaceOrUnderscore( $pattern ) {
 	protected static function callParserPreprocess( Parser $parser, $text, $page, $options ): string {
 		global $wgDplUseRecursivePreprocess;
 		if ( $wgDplUseRecursivePreprocess ) {
+			self::softResetParser( $parser );
 			$parser->setOutputType( OT_PREPROCESS );
 			$text = $parser->recursivePreprocess( $text );
 
@@ -923,4 +924,35 @@ protected static function callParserPreprocess( Parser $parser, $text, $page, $o
 			return $parser->preprocess( $text, $page, $options );
 		}
 	}
+
+	private static function softResetParser( Parser $parser ): void {
+		self::setParserProperties( $parser, [
+			'mStripState' => new \StripState( $parser ),
+			'mIncludeSizes' => [
+				'post-expand' => 0,
+				'arg' => 0,
+			],
+			'mPPNodeCount' => 0,
+			'mHighestExpansionDepth' => 0,
+			'mExpensiveFunctionCount' => 0,
+		] );
+	}
+
+	private static function setParserProperties( Parser $parser, array $properties ): void {
+		static $reflectionCache = [];
+		foreach ( $properties as $property => $value ) {
+			if ( !array_key_exists( $property, $reflectionCache ) ) {
+				try {
+					$reflectionCache[$property] = ( new \ReflectionClass( Parser::class ) )->getProperty( $property );
+				} catch ( \ReflectionException ) {
+					$reflectionCache[$property] = null;
+				}
+			}
+		}
+		foreach ( $properties as $property => $value ) {
+			if ( $reflectionCache[$property] ) {
+				$reflectionCache[$property]->setValue( $parser, $value );
+			}
+		}
+	}
 }