<?php
require_once dirname(__DIR__, 3) . '/maintenance/Maintenance.php';
require_once dirname(__DIR__) . '/includes/Hanzi2Pinyin.php';
require_once dirname(__DIR__) . '/includes/Utils.php';

use LatinizeUrl\Hanzi2Pinyin;
use LatinizeUrl\Utils;

/**
 * Maintenance script to normalize double-byte Latin UTF-8 characters.
 *
 * @ingroup Maintenance
 */
class UpdateLatinizeUrl extends Maintenance {

	public function __construct() {
		parent::__construct();
		$this->addDescription( '更新所有页面的拼音url' );
		$this->addArg( 'pairfile', '生成新旧网址对', false );
		$this->addOption( 'force', '强制更新所有页面的url（除了自定义url的页面）' );
	}

	public function getDbType() {
		return Maintenance::DB_ADMIN;
    }

	public function execute() {
		global $wgLatinizeUrlConfig;
		
		$force = $this->hasOption( 'force' );
		$outputFileName = $this->getArg( 0 );
		$outputFile = false;
		if($outputFileName){
			$outputFile = fopen($outputFileName, 'w');
		}

		$dbw = $this->getDB( DB_MASTER );
		if ( $dbw->getType() !== 'mysql' ) {
			$this->fatalError( "This change is only needed on MySQL, quitting.\n" );
		}
        
		$convertor = new Hanzi2Pinyin($wgLatinizeUrlConfig);

		$res = $this->findRows( $dbw );
		foreach($res as $one){
			$title = $one->page_title;
			$isCustom = boolval(intval($dbw->selectField('url_slug', 'is_custom', ['title' => $title], __METHOD__)));
            if(!$force && !$isCustom && Utils::titleSlugExists($title)) continue;

            $pinyin = $convertor->parse($title);
            $slug = $convertor->pinyin2String($pinyin);
			echo $title . ' -> ' . $slug . PHP_EOL;
			if($outputFile){
				$pair = [$this->getFullUrl($title), $this->getFullUrl($slug)];
				fwrite($outputFile, implode(' ', $pair) . "\r\n");
			}

            Utils::replaceTitleSlugMap($title, $slug, $pinyin);
		}
		
		fclose($outputFile);

		$this->output( "Done\n" );
	}

	public function searchIndexUpdateCallback( $dbw, $row ) {
		return $this->updateSearchIndexForPage( $dbw, $row->si_page );
	}

	private function getFullUrl($pageName){
		global $wgServer, $wgArticlePath, $wgUsePathInfo;
		$pageName = implode("/", array_map("urlencode", explode("/", $pageName)));
		if($wgUsePathInfo){
			return $wgServer . str_replace('$1', $pageName, $wgArticlePath);
		} else {
			return $wgServer . '/index.php?title=' . $pageName;
		}
	}

	private function findRows( $dbi ) {
		return $dbi->select('page', ['page_title'], [
            'page_namespace' => NS_MAIN,
        ], __METHOD__, []);
	}
}

$maintClass = UpdateLatinizeUrl::class;
require_once RUN_MAINTENANCE_IF_MAIN;