<?php
require_once( "LanguageZh_cn.php");
require_once( "LanguageZh_tw.php");
require_once( "LanguageZh_sg.php");
require_once( "LanguageZh_hk.php");

/*
   hook to refresh the cache of conversion tables when 
   MediaWiki:zhconversiontable* is updated
*/
function zhOnArticleSaveComplete($article, $user, $text, $summary, $isminor, $iswatch, $section) {
	$titleobj = $article->getTitle();
	if($titleobj->getNamespace() == NS_MEDIAWIKI) { 
		global $wgContLang; // should be an LanguageZh.
		if(get_class($wgContLang) != 'languagezh')	
			return true;

		$title = $titleobj->getDBkey();
		$t = explode('/', $title, 2);
		if( $t[0] == 'Zhconversiontable' ) {
			if(!in_array($t[1], array('zh-cn', 'zh-tw', 'zh-sg', 'zh-hk')))
				return true;
			$wgContLang->reloadTables();			
		}
	}
}

$wgHooks['ArticleSaveComplete'][] = 'zhOnArticleSaveComplete';

/* class that handles both Traditional and Simplified Chinese
   right now it only distinguish zh_cn and zh_tw (actuall, zh_cn and
   non-zh_cn), will add support for zh_sg, zh_hk, etc, later.
*/
class LanguageZh extends LanguageZh_cn {
	
	var $mZhLanguageCode=false;
	var $mTables=false; //the mapping tables
	var $mTablesLoaded = false;
	var $mCacheKey;
	var $mDoTitleConvert = true;
	function LanguageZh() {
		global $wgDBname;
		$this->mCacheKey = $wgDBname . ":zhtables";
	}

	function reloadTables() {
		global $wgMemc;
		$wgMemc->delete($this->mCacheKey);
		$this->mTablesLoaded=false;
		$this->loadTables();
	}

	// load conversion tables either from the cache or the disk
	function loadTables() {
		global $wgMemc;
		if( $this->mTablesLoaded )
			return;
		$this->mTablesLoaded = true;
		$this->mTables = $wgMemc->get( $this->mCacheKey );
		if( empty( $this->mTables ) ) {
			global $wgMessageCache;
			require( "includes/ZhConversion.php" );
			$this->mTables = array();
			$this->mTables['zh-cn'] = $zh2CN;
			$this->mTables['zh-tw'] = $zh2TW;
			$this->mTables['zh-sg'] = $zh2SG;
			$this->mTables['zh-hk'] = $zh2HK;
			if( is_object( $wgMessageCache ) ){
				$cached = $this->parseCachedTable( $wgMessageCache->get( 'zhconversiontable/zh-cn', true, true, true ) );
				$this->mTables['zh-cn'] = array_merge($this->mTables['zh-cn'], $cached);

				$cached = $this->parseCachedTable( $wgMessageCache->get( 'zhconversiontable/zh-tw', true, true, true ) );
				$this->mTables['zh-tw'] = array_merge($this->mTables['zh-tw'], $cached);

				$cached = $this->parseCachedTable( $wgMessageCache->get( 'zhconversiontable/zh-sg', true, true, true ) );
				$this->mTables['zh-sg'] = array_merge($this->mTables['zh-sg'], $cached);

				$cached = $this->parseCachedTable( $wgMessageCache->get( 'zhconversiontable/zh-hk', true, true, true ) );
				$this->mTables['zh-hk'] = array_merge($this->mTables['zh-hk'], $cached);

			}
			$wgMemc->set($this->mCacheKey, $this->mTables, 43200);
		}
	}
	
	/*
		parse the conversion table stored in the cache 

		the table should be in the following format:

			-{
				word => word ;
				word => word ;
				...
			-}
	*/
	function parseCachedTable($txt) {
		/* $txt should be enclosed by -{ and }- */
		$a = explode( '-{', $txt);
		if( count($a) < 2)
			return array();
		array_shift($a);
		$b = explode( '}-', $a[0]);

		$stripped = str_replace(array('*','#'), '', $b[0]);
		$table = explode( ';', $stripped );
		$ret = array();
		foreach( $table as $t ) {
			$m = explode( '=>', $t );
			if( count( $m ) != 2)
				continue;
			$ret[trim($m[0])] = trim($m[1]);
		}
		return $ret;
	}

	/* 
		get preferred language variants.
	*/
	function getPreferredVariant() {
		global $wgUser, $wgRequest;
		
		if($this->mZhLanguageCode)
			return $this->mZhLanguageCode;

		// see if the preference is set in the request
		$zhreq = $wgRequest->getText( 'variant' );
		if( in_array( $zhreq, $this->getVariants() ) ) {
			$this->mZhLanguageCode = $zhreq;
			return $zhreq;
		}

		// get language variant preference from logged in users 
		if($wgUser->getID()!=0) {
			$this->mZhLanguageCode = $wgUser->getOption('variant');
		}

		if( !$this->mZhLanguageCode ) {
			// see if some zh- variant is set in the http header,
			$this->mZhLanguageCode="zh";
			$header = str_replace( '_', '-', strtolower($_SERVER["HTTP_ACCEPT_LANGUAGE"]));
			$zh = strstr($header, 'zh-');
			if($zh) {
				$this->mZhLanguageCode = substr($zh,0,5);
			}
		}
		return $this->mZhLanguageCode;
	}
	
	# this should give much better diff info
	function segmentForDiff( $text ) {
		return preg_replace(
			"/([\\xc0-\\xff][\\x80-\\xbf]*)/e",
			"' ' .\"$1\"", $text);
	}

	function unsegmentForDiff( $text ) {
		return preg_replace(
			"/ ([\\xc0-\\xff][\\x80-\\xbf]*)/e",
			"\"$1\"", $text);
	}

	function autoConvert($text, $toVariant=false) {
		$fname="LanguageZh::autoConvert";
		wfProfileIn( $fname );

		if(!$this->mTablesLoaded)
			$this->loadTables();

		if(!$toVariant) 
			$toVariant = $this->getPreferredVariant();
		$ret = '';
		switch( $toVariant ) {
			case 'zh-cn': $ret = strtr($text, $this->mTables['zh-cn']);break;
			case 'zh-tw': $ret = strtr($text, $this->mTables['zh-tw']);break;
			case 'zh-sg': $ret = strtr(strtr($text, $this->mTables['zh-cn']), $this->mTables['zh-sg']);break;
			case 'zh-hk': $ret = strtr(strtr($text, $this->mTables['zh-tw']), $this->mTables['zh-hk']);break;
			default: $ret = $text;
		}
		wfProfileOut( $fname );
		return $ret;
	}
    
	function autoConvertToAllVariants($text) {
		$fname="LanguageZh::autoConvertToAllVariants";
		wfProfileIn( $fname );
		if( !$this->mTablesLoaded )
			$this->loadTables();

		$ret = array();
		$ret['zh-cn'] = strtr($text, $this->mTables['zh-cn']);
		$ret['zh-tw'] = strtr($text, $this->mTables['zh-tw']);
		$ret['zh-sg'] = strtr(strtr($text, $this->mTables['zh-cn']), $this->mTables['zh-sg']);
		$ret['zh-hk'] = strtr(strtr($text, $this->mTables['zh-tw']), $this->mTables['zh-hk']);
		wfProfileOut( $fname );
		return $ret;
	}
    
	# convert text to different variants of a language. the automatic
	# conversion is done in autoConvert(). here we parse the text 
	# marked with -{}-, which specifies special conversions of the 
	# text that can not be accomplished in autoConvert()
	#
	# syntax of the markup:
	# -{code1:text1;code2:text2;...}-  or
	# -{text}- in which case no conversion should take place for text
	function convert( $text , $isTitle=false) {
		global $wgDisableLangConversion;
		if($wgDisableLangConversion)
			return $text; 

		$mw =& MagicWord::get( MAG_NOTITLECONVERT );
		if( $mw->matchAndRemove( $text ) )
			$this->mDoTitleConvert = false;

		// no conversion if redirecting
		$mw =& MagicWord::get( MAG_REDIRECT );
		if( $mw->matchStart( $text ))
			return $text;

		if( $isTitle ) {
			if( !$this->mDoTitleConvert )
				return $text;

			global $wgRequest;
			$isredir = $wgRequest->getText( 'redirect', 'yes' );
			$action = $wgRequest->getText( 'action' );
			if ( $isredir == 'no' || $action == 'edit' ) {
				return $text;
			}
			else {
				return $this->autoConvert($text);
			}
		}

		$plang = $this->getPreferredVariant();
		$fallback = $this->getVariantFallback($plang);

		$tarray = explode("-{", $text);
		$tfirst = array_shift($tarray);
		$text = $this->autoConvert($tfirst);
		foreach($tarray as $txt) {
			$marked = explode("}-", $txt);
			
			$choice = explode(";", $marked{0});
			if(!array_key_exists(1, $choice)) {
				/* a single choice */
				$text .= $choice{0};
			} else {
				$choice1=false;
				$choice2=false;
				foreach($choice as $c) {
					$v = explode(":", $c);
					if(!array_key_exists(1, $v)) {
						//syntax error in the markup, give up
						break;			
					}
					$code = trim($v{0});
					$content = trim($v{1});
					if($code == $plang) {
						$choice1 = $content;
						break;
					}
					if($code == $fallback)
						$choice2 = $content;
				}
				if ( $choice1 )
					$text .= $choice1;
				elseif ( $choice2 )
					$text .= $choice2;
				else
					$text .= $marked{0};
			}
			if(array_key_exists(1, $marked))
				$text .= $this->autoConvert($marked{1});
		}
		
		return $text;
	}


	function getVariants() {
		return array("zh", "zh-cn", "zh-tw", "zh-sg", "zh-hk");
	}

	function getVariantFallback($v) {
		switch ($v) {
		case 'zh': return 'zh-cn'; break;
		case 'zh-cn': return 'zh-sg'; break;
		case 'zh-sg': return 'zh-cn'; break;
		case 'zh-tw': return 'zh-hk'; break;
		case 'zh-hk': return 'zh-tw'; break;
		}
		return false;
	}

	// word segmentation
	function stripForSearch( $string ) {
		$fname="LanguageZh::stripForSearch";
		wfProfileIn( $fname );

		// eventually this should be a word segmentation
		// for now just treat each character as a word
		$t = preg_replace(
				"/([\\xc0-\\xff][\\x80-\\xbf]*)/e",
				"' ' .\"$1\"", $string);

        //always convert to zh-cn before indexing. it should be
		//better to use zh-cn for search, since conversion from 
		//Traditional to Simplified is less ambiguous than the
		//other way around

		$t = $this->autoConvert($t, 'zh-cn');
		$t = LanguageUtf8::stripForSearch( $t );
		wfProfileOut( $fname );
		return $t;

	}

	function convertForSearchResult( $termsArray ) {
		$terms = implode( '|', $termsArray );
		$terms = implode( '|', $this->autoConvertToAllVariants( $terms ) );
		$ret = array_unique( explode('|', $terms) );
		return $ret;
	}

	function findVariantLink( &$link, &$nt ) {
		static $count=0; //used to limit this operation
		static $cache=array();
		global $wgDisableLangConversion;
		$pref = $this->getPreferredVariant();
		if( $wgDisableLangConversion || $pref == 'zh' || $count > 50)
			return;
		$count++;
		$variants = $this->autoConvertToAllVariants($link);
		if($variants == false) //give up
			return;
		foreach( $variants as $v ) {
			if(isset($cache[$v]))
				continue;
			$cache[$v] = 1;
			$varnt = Title::newFromText( $v );
			if( $varnt && $varnt->getArticleID() > 0 ) {
				$nt = $varnt;
				$link = $v;
				break;
			}
		}
	}

	function getExtraHashOptions() {
		global $wgUser;
		$variant = $this->getPreferredVariant();
		return '!' . $variant ;
	}
}
?>