LCStoreStaticArray.php

Go to the documentation of this file.

<?php
use Wikimedia\StaticArrayWriter;
 
class LCStoreStaticArray implements LCStore {
    private $currentLang = null;
 
    private $data = [];
 
    private $fname = null;
 
    private $directory;
 
    public function __construct( $conf = [] ) {
        $this->directory = $conf['directory'];
    }
 
    public function startWrite( $code ) {
        if ( !file_exists( $this->directory ) && !wfMkdirParents( $this->directory, null, __METHOD__ ) ) {
            throw new MWException( "Unable to create the localisation store " .
                "directory \"{$this->directory}\"" );
        }
 
        $this->currentLang = $code;
        $this->fname = $this->directory . '/' . $code . '.l10n.php';
        $this->data[$code] = [];
        if ( file_exists( $this->fname ) ) {
            $this->data[$code] = require $this->fname;
        }
    }
 
    public function set( $key, $value ) {
        $this->data[$this->currentLang][$key] = self::encode( $value );
    }
 
    public static function encode( $value ) {
        if ( is_array( $value ) ) {
            // [a]rray
            return [ 'a', array_map( 'LCStoreStaticArray::encode', $value ) ];
        }
        if ( is_object( $value ) ) {
            // [s]erialized
            return [ 's', serialize( $value ) ];
        }
        if ( is_scalar( $value ) || $value === null ) {
            // Scalar value, written directly without array
            return $value;
        }
 
        throw new RuntimeException( 'Cannot encode ' . var_export( $value, true ) );
    }
 
    public static function decode( $encoded ) {
        if ( !is_array( $encoded ) ) {
            // Scalar values are written directly without array
            return $encoded;
        }
 
        list( $type, $data ) = $encoded;
 
        switch ( $type ) {
            case 'a':
                return array_map( 'LCStoreStaticArray::decode', $data );
            case 's':
                return unserialize( $data );
            case 'v':
                // Support: MediaWiki 1.32 and earlier
                // Backward compatibility with older file format
                return $data;
            default:
                throw new RuntimeException(
                    'Unable to decode ' . var_export( $encoded, true ) );
        }
    }
 
    public function finishWrite() {
        $writer = new StaticArrayWriter();
        $out = $writer->create(
            $this->data[$this->currentLang],
            'Generated by LCStoreStaticArray.php -- do not edit!'
        );
        file_put_contents( $this->fname, $out );
        // Release the data to manage the memory in rebuildLocalisationCache
        unset( $this->data[$this->currentLang] );
        $this->currentLang = null;
        $this->fname = null;
    }
 
    public function get( $code, $key ) {
        if ( !array_key_exists( $code, $this->data ) ) {
            $fname = $this->directory . '/' . $code . '.l10n.php';
            if ( !file_exists( $fname ) ) {
                return null;
            }
            $this->data[$code] = require $fname;
        }
        $data = $this->data[$code];
        if ( array_key_exists( $key, $data ) ) {
            return self::decode( $data[$key] );
        }
        return null;
    }
}