YamlFFS.php

<?php
 
use MediaWiki\Extension\Translate\MessageGroupConfiguration\MetaYamlSchemaExtender;
use MediaWiki\Extension\Translate\MessageLoading\MessageCollection;
use MediaWiki\Extension\Translate\MessageProcessing\ArrayFlattener;
use MediaWiki\Extension\Translate\Utilities\Utilities;
 
class YamlFFS extends SimpleFFS implements MetaYamlSchemaExtender {
    private $flattener;
 
    public function __construct( FileBasedMessageGroup $group ) {
        parent::__construct( $group );
        $this->flattener = $this->getFlattener();
    }
 
    public function getFileExtensions() {
        return [ '.yaml', '.yml' ];
    }
 
    public function readFromVariable( $data ) {
        // Authors first.
        $matches = [];
        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()->mangleArray( $messages );
        foreach ( $messages as &$value ) {
            $value = rtrim( $value, "\n" );
        }
 
        return [
            'AUTHORS' => $authors,
            'MESSAGES' => $messages,
        ];
    }
 
    protected function writeReal( MessageCollection $collection ) {
        $output = $this->doHeader( $collection );
        $output .= $this->doAuthors( $collection );
 
        $mangler = $this->group->getMangler();
 
        $messages = [];
 
        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 = [ $code => $messages ];
        }
 
        $output .= TranslateYaml::dump( $messages );
 
        return $output;
    }
 
    protected function doHeader( MessageCollection $collection ) {
        global $wgSitename;
        global $wgTranslateYamlLibrary;
 
        $code = $collection->code;
        $name = Utilities::getLanguageName( $code );
        $native = Utilities::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 getFlattener() {
        $nestingSeparator = $this->extra['nestingSeparator'] ?? '.';
        $parseCLDRPlurals = $this->extra['parseCLDRPlurals'] ?? false;
 
        // Instantiate helper class for flattening and unflattening nested arrays
        return new ArrayFlattener( $nestingSeparator, $parseCLDRPlurals );
    }
 
    protected function flatten( $messages ) {
        return $this->flattener->flatten( $messages );
    }
 
    protected function unflatten( $messages ) {
        return $this->flattener->unflatten( $messages );
    }
 
    public function isContentEqual( $a, $b ) {
        return $this->flattener->compareContent( $a, $b );
    }
 
    public static function getExtraSchema(): array {
        $schema = [
            'root' => [
                '_type' => 'array',
                '_children' => [
                    'FILES' => [
                        '_type' => 'array',
                        '_children' => [
                            'codeAsRoot' => [
                                '_type' => 'boolean',
                            ],
                            'nestingSeparator' => [
                                '_type' => 'text',
                            ],
                            'parseCLDRPlurals' => [
                                '_type' => 'boolean',
                            ]
                        ]
                    ]
                ]
            ]
        ];
 
        return $schema;
    }
}