YamlFFS.php

Go to the documentation of this file.

<?php

class YamlFFS extends SimpleFFS {
    public function getFileExtensions() {
        return array( '.yaml', '.yml' );
    }

    public function readFromVariable( $data ) {
        // Authors first.
        $matches = array();
        preg_match_all( '/^#\s*Author:\s*(.*)$/m', $data, $matches );
        $authors = $matches[1];

        // Then messages.
        $messages = TranslateYaml::loadString( $data );

        // Some groups have messages under language code
        if ( isset( $this->extra['codeAsRoot'] ) ) {
            $messages = array_shift( $messages );
        }

        $messages = $this->flatten( $messages );
        $messages = $this->group->getMangler()->mangle( $messages );
        foreach ( $messages as &$value ) {
            $value = rtrim( $value, "\n" );
        }

        return array(
            'AUTHORS' => $authors,
            'MESSAGES' => $messages,
        );
    }

    protected function writeReal( MessageCollection $collection ) {
        $output = $this->doHeader( $collection );
        $output .= $this->doAuthors( $collection );

        $mangler = $this->group->getMangler();

        $messages = array();
        foreach ( $collection as $key => $m ) {
            $key = $mangler->unmangle( $key );
            $value = $m->translation();
            $value = str_replace( TRANSLATE_FUZZY, '', $value );

            if ( $value === '' ) {
                continue;
            }

            $messages[$key] = $value;
        }

        if ( !count( $messages ) ) {
            return false;
        }

        $messages = $this->unflatten( $messages );

        // Some groups have messages under language code.
        if ( isset( $this->extra['codeAsRoot'] ) ) {
            $code = $this->group->mapCode( $collection->code );
            $messages = array( $code => $messages );
        }

        $output .= TranslateYaml::dump( $messages );

        return $output;
    }

    protected function doHeader( MessageCollection $collection ) {
        global $wgSitename;
        global $wgTranslateYamlLibrary;

        $code = $collection->code;
        $name = TranslateUtils::getLanguageName( $code );
        $native = TranslateUtils::getLanguageName( $code, $code );
        $output = "# Messages for $name ($native)\n";
        $output .= "# Exported from $wgSitename\n";

        if ( isset( $wgTranslateYamlLibrary ) ) {
            $output .= "# Export driver: $wgTranslateYamlLibrary\n";
        }

        return $output;
    }

    protected function doAuthors( MessageCollection $collection ) {
        $output = '';
        $authors = $collection->getAuthors();
        $authors = $this->filterAuthors( $authors, $collection->code );

        foreach ( $authors as $author ) {
            $output .= "# Author: $author\n";
        }

        return $output;
    }

    protected function flatten( $messages ) {
        $flat = true;

        foreach ( $messages as $v ) {
            if ( !is_array( $v ) ) {
                continue;
            }

            $flat = false;
            break;
        }

        if ( $flat ) {
            return $messages;
        }

        $array = array();
        foreach ( $messages as $key => $value ) {
            if ( !is_array( $value ) ) {
                $array[$key] = $value;
            } else {
                $plural = $this->flattenPlural( $value );
                if ( $plural ) {
                    $array[$key] = $plural;
                } else {
                    $newArray = array();
                    foreach ( $value as $newKey => $newValue ) {
                        $newArray["$key.$newKey"] = $newValue;
                    }
                    $array += $this->flatten( $newArray );
                }
            }

            unset( $messages[$key] );
        }

        return $array;
    }

    protected function unflatten( $messages ) {
        $array = array();
        foreach ( $messages as $key => $value ) {
            $plurals = $this->unflattenPlural( $key, $value );

            if ( $plurals === false ) {
                continue;
            }

            foreach ( $plurals as $keyPlural => $valuePlural ) {
                $path = explode( '.', $keyPlural );
                if ( count( $path ) == 1 ) {
                    $array[$keyPlural] = $valuePlural;
                    continue;
                }

                $pointer = &$array;
                do {
                    $level = array_shift( $path );
                    if ( !isset( $pointer[$level] ) ) {
                        $pointer[$level] = array();
                    }

                    $tmpPointer = &$pointer[$level];
                    unset( $pointer );
                    $pointer = &$tmpPointer;
                    unset( $tmpPointer );

                    if ( count( $path ) === 1 ) {
                        $lastKey = array_shift( $path );
                        $pointer[$lastKey] = $valuePlural;
                    }
                } while ( count( $path ) );
            }
        }

        return $array;
    }

    public function flattenPlural( $value ) {
        return false;
    }

    public function unflattenPlural( $key, $value ) {
        return array( $key => $value );
    }
}