2 files changed, 112 insertions, 19 deletions

diff --git a/extensions/SyntaxHighlight_GeSHi/maintenance/updateLanguageList.php b/extensions/SyntaxHighlight_GeSHi/maintenance/updateCSS.php
index 6e9feb33..9299cd74 100644
--- a/extensions/SyntaxHighlight_GeSHi/maintenance/updateLanguageList.php
+++ b/extensions/SyntaxHighlight_GeSHi/maintenance/updateCSS.php
@@ -1,6 +1,6 @@
 <?php
 /**
- * Script to update list of supported languages.
+ * Script to update Pygments CSS
  *
  * This program is free software; you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
@@ -22,33 +22,49 @@
  * @ingroup Maintenance
  */
 
+use Symfony\Component\Process\ProcessBuilder;
+
 $IP = getenv( 'MW_INSTALL_PATH' ) ?: __DIR__ . '/../../..';
 
 require_once "$IP/maintenance/Maintenance.php";
-require_once __DIR__ . "/../geshi/geshi.php";
 
-class UpdateLanguageList extends Maintenance {
+class UpdateCSS extends Maintenance {
+
 	public function __construct() {
 		parent::__construct();
-		$this->addDescription( 'Update list of languages supported by SyntaxHighlight_GeSHi' );
+		$this->addDescription( 'Generate CSS code for SyntaxHighlight_GeSHi' );
 	}
 
 	public function execute() {
-		global $IP;
-
-		$geshi = new GeSHi;
-		$header = '// Generated by ' . basename( __FILE__ ) . ' on ' . date( DATE_RFC2822 ) . "\n";
-		$langs = array_values( array_filter( $geshi->get_supported_languages( false ), 'ctype_alnum' ) );
-		sort( $langs );
-		$replace = array( '[' => "array(\n\t", ']' => "\n);\n", '",' => "\",\n\t" );
-		$code = "<?php\n" .
-			$header .
-			"global \$wgGeSHiSupportedLanguages;\n" .
-			'$wgGeSHiSupportedLanguages = ' . strtr( json_encode( $langs ), $replace );
-		file_put_contents( __DIR__ . '/../SyntaxHighlight_GeSHi.langs.php', $code );
-		$this->output( "Updated language list written to SyntaxHighlight_GeSHi.langs.php\n" );
+		global $wgPygmentizePath;
+
+		$target = __DIR__ . '/../modules/pygments.generated.css';
+		$css = "/* Stylesheet generated by updateCSS.php */\n";
+
+		$builder = new ProcessBuilder();
+		$builder->setPrefix( $wgPygmentizePath );
+
+		$process = $builder
+			->add( '-f' )->add( 'html' )
+			->add( '-S' )->add( 'default' )
+			->add( '-a' )->add( '.' . SyntaxHighlight_GeSHi::HIGHLIGHT_CSS_CLASS )
+			->getProcess();
+
+		$process->run();
+
+		if ( !$process->isSuccessful() ) {
+			throw new \RuntimeException( $process->getErrorOutput() );
+		}
+
+		$css .= $process->getOutput();
+
+		if ( file_put_contents( $target, $css ) === false ) {
+			$this->output( "Failed to write to {$target}\n" );
+		} else {
+			$this->output( 'CSS written to ' . realpath( $target ) . "\n" );
+		}
 	}
 }
 
-$maintClass = "UpdateLanguageList";
-require_once( RUN_MAINTENANCE_IF_MAIN );
+$maintClass = "UpdateCSS";
+require_once RUN_MAINTENANCE_IF_MAIN;
diff --git a/extensions/SyntaxHighlight_GeSHi/maintenance/updateLexerList.php b/extensions/SyntaxHighlight_GeSHi/maintenance/updateLexerList.php
new file mode 100644
index 00000000..b5a7fc5a
--- /dev/null
+++ b/extensions/SyntaxHighlight_GeSHi/maintenance/updateLexerList.php
@@ -0,0 +1,77 @@
+<?php
+/**
+ * Script to update list of supported lexers.
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program; if not, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
+ * http://www.gnu.org/copyleft/gpl.html
+ *
+ * @file
+ * @author Ori Livneh <ori@wikimedia.org>
+ * @ingroup Maintenance
+ */
+
+use Symfony\Component\Process\ProcessBuilder;
+
+$IP = getenv( 'MW_INSTALL_PATH' ) ?: __DIR__ . '/../../..';
+
+require_once "$IP/maintenance/Maintenance.php";
+
+class UpdateLanguageList extends Maintenance {
+	public function __construct() {
+		parent::__construct();
+		$this->addDescription( 'Update list of lexers supported by SyntaxHighlight_GeSHi' );
+	}
+
+	public function execute() {
+		global $wgPygmentizePath;
+
+		function lang_filter( $val ) {
+			return preg_match( '/^[a-zA-Z0-9\-_]+$/', $val );
+		}
+
+		$header = '// Generated by ' . basename( __FILE__ ) . "\n\n";
+
+		$lexers = array();
+
+		$builder = new ProcessBuilder();
+		$builder->setPrefix( $wgPygmentizePath );
+
+		$process = $builder->add( '-L' )->add( 'lexer' )->getProcess();
+		$process->run();
+
+		if ( !$process->isSuccessful() ) {
+			throw new \RuntimeException( $process->getErrorOutput() );
+		}
+
+		$output = $process->getOutput();
+		foreach ( explode( "\n", $output ) as $line ) {
+			if ( substr( $line, 0, 1 ) === '*' ) {
+				$newLexers = explode( ', ', trim( $line, "* :\n" ) );
+				$lexers = array_merge( $lexers, $newLexers );
+			}
+		}
+		sort( $lexers );
+
+		$code = "<?php\n" .  $header .  'return ' . var_export( $lexers, true ) . ";\n";
+		$code = preg_replace( '/(\d+ \=\>| (?=\())/i', '', $code );
+		$code = preg_replace( "/^ +/m", "\t", $code );
+
+		file_put_contents( __DIR__ . '/../SyntaxHighlight_GeSHi.lexers.php', $code );
+		$this->output( "Updated language list written to SyntaxHighlight_GeSHi.lexers.php\n" );
+	}
+}
+
+$maintClass = "UpdateLanguageList";
+require_once ( RUN_MAINTENANCE_IF_MAIN );