TPParse.php

Go to the documentation of this file.

<?php
class TPParse {
    protected $title = null;

    public $sections = array();
    public $template = null;
    protected $dbSections = null;

    public function __construct( Title $title ) {
        $this->title = $title;
    }

    public function countSections() {
        return count( $this->sections );
    }

    public function getTemplate() {
        return $this->template;
    }

    public function getTemplatePretty() {
        $text = $this->template;
        $sections = $this->getSectionsForSave();
        foreach ( $sections as $ph => $s ) {
            $text = str_replace( $ph, "<!--T:{$s->id}-->", $text );
        }

        return $text;
    }

    public function getSectionsForSave( $highest = 0 ) {
        $this->loadFromDatabase();

        $sections = $this->sections;
        foreach ( array_keys( $this->dbSections ) as $key ) {
            $highest = max( $highest, intval( $key ) );
        }

        foreach ( $sections as $_ ) {
            $highest = max( $highest, intval( $_->id ) );
        }

        foreach ( $sections as $s ) {
            $s->type = 'old';

            if ( $s->id === -1 ) {
                $s->type = 'new';
                $s->id = ++$highest;
            } else {
                if ( isset( $this->dbSections[$s->id] ) ) {
                    $storedText = $this->dbSections[$s->id]->text;
                    if ( $s->text !== $storedText ) {
                        $s->type = 'changed';
                        $s->oldText = $storedText;
                    }
                }
            }
        }

        return $sections;
    }

    public function getDeletedSections() {
        $sections = $this->getSectionsForSave();
        $deleted = $this->dbSections;

        foreach ( $sections as $s ) {
            if ( isset( $deleted[$s->id] ) ) {
                unset( $deleted[$s->id] );
            }
        }

        return $deleted;
    }

    protected function loadFromDatabase() {
        if ( $this->dbSections !== null ) {
            return;
        }

        $this->dbSections = array();

        $db = wfGetDB( DB_SLAVE );
        $tables = 'translate_sections';
        $vars = array( 'trs_key', 'trs_text' );
        $conds = array( 'trs_page' => $this->title->getArticleID() );

        $res = $db->select( $tables, $vars, $conds, __METHOD__ );
        foreach ( $res as $r ) {
            $section = new TPSection;
            $section->id = $r->trs_key;
            $section->text = $r->trs_text;
            $section->type = 'db';
            $this->dbSections[$r->trs_key] = $section;
        }
    }

    public function getSourcePageText() {
        $text = $this->template;

        foreach ( $this->sections as $ph => $s ) {
            $text = str_replace( $ph, $s->getMarkedText(), $text );
        }

        return $text;
    }

    public function getTranslationPageText( $collection ) {
        $text = $this->template; // The source

        // For finding the messages
        $prefix = $this->title->getPrefixedDBKey() . '/';

        if ( $collection instanceof MessageCollection ) {
            $collection->loadTranslations( DB_MASTER );
            $collection->filter( 'translated', false );
        }

        foreach ( $this->sections as $ph => $s ) {
            $sectiontext = null;

            if ( isset( $collection[$prefix . $s->id] ) ) {
                $msg = $collection[$prefix . $s->id];
                $sectiontext = $msg->translation();
            }

            // Use the original text if no translation is available.

            // For the source language, this will actually be the source, which
            // contains variable declarations (tvar) instead of variables ($1).
            // The getTextForTrans will convert declarations to normal variables
            // for us so that the variable substitutions below will also work
            // for the source language.
            if ( $sectiontext === null || $sectiontext === $s->getText() ) {
                $sectiontext = $s->getTextForTrans();
            }

            // Substitute variables into section text and substitute text into document
            $sectiontext = strtr( $sectiontext, $s->getVariables() );
            $text = str_replace( $ph, $sectiontext, $text );
        }

        $nph = array();
        $text = TranslatablePage::armourNowiki( $nph, $text );

        // Remove translation markup from the template to produce final text
        $cb = array( __CLASS__, 'replaceTagCb' );
        $text = preg_replace_callback( '~(<translate>)(.*)(</translate>)~sU', $cb, $text );
        $text = TranslatablePage::unArmourNowiki( $nph, $text );

        return $text;
    }

    protected static function replaceTagCb( $matches ) {
        return preg_replace( '~^\n|\n\z~', '', $matches[2] );
    }
}