MadelineProto/TL.php

<?php

set_include_path(get_include_path().PATH_SEPARATOR.dirname(__FILE__).DIRECTORY_SEPARATOR.'libpy2php');
require_once 'libpy2php.php';
$__author__ = 'agrigoryev';
require_once 'os.php';
class TlConstructor
{
    public function __construct($json_dict)
    {
        $this->id = (int) $json_dict['id'];
        $this->type = $json_dict['type'];
        $this->predicate = $json_dict['predicate'];
        $this->params = [];
        foreach ($json_dict['params'] as $param) {
            if (($param['type'] == 'Vector<long>')) {
                $param['type'] = 'Vector t';
                $param['subtype'] = 'long';
            } elseif (($param['type'] == 'vector<%Message>')) {
                $param['type'] = 'vector';
                $param['subtype'] = 'message';
            } elseif (($param['type'] == 'vector<future_salt>')) {
                $param['type'] = 'vector';
                $param['subtype'] = 'future_salt';
            } else {
                $param['subtype'] = null;
            }
            $this->params[] = $param;
        }
    }
}
class TlMethod
{
    public function __construct($json_dict)
    {
        $this->id = (int) $json_dict['id'];
        $this->type = $json_dict['type'];
        $this->method = $json_dict['method'];
        $this->params = $json_dict['params'];
    }
}
class TLObject extends ArrayObject
{
    public function __construct($tl_elem)
    {
        parent::__construct();
        $this->name = $tl_elem->predicate;
    }
}
class TL
{
    public function __construct($filename)
    {
        $TL_dict = json_decode(file_get_contents($filename), true);
        $this->constructors = $TL_dict['constructors'];
        $this->constructor_id = [];
        $this->constructor_type = [];
        foreach ($this->constructors as $elem) {
            $z = new TlConstructor($elem);
            $this->constructor_id[$z->id] = $z;
            $this->constructor_type[$z->predicate] = $z;
        }
        $this->methods = $TL_dict['methods'];
        $this->method_id = [];
        $this->method_name = [];
        foreach ($this->methods as $elem) {
            $z = new TlMethod($elem);
            $this->method_id[$z->id] = $z;
            $this->method_name[$z->method] = $z;
        }
        $this->struct = new \danog\PHP\Struct();
    }

    public function serialize_obj($type_, $kwargs)
    {
        $bytes_io = fopen('php://memory', 'rw+b');
        if (isset($this->constructor_type[$type_])) {
            $tl_constructor = $this->constructor_type[$type_];
        } else {
            throw new Exception(sprintf('Could not extract type: %s', $type_));
        }
        fwrite($bytes_io, $this->struct->pack('<i', $tl_constructor->id));
        foreach ($tl_constructor->params as $arg) {
            $this->serialize_param($bytes_io, $arg['type'], $kwargs[$arg['name']]);
        }

        return fread_all($bytes_io);
    }

    public function serialize_method($type_, $kwargs)
    {
        $bytes_io = fopen('php://memory', 'rw+b');
        if (isset($this->method_name[$type_])) {
            $tl_method = $this->method_name[$type_];
        } else {
            throw new Exception(sprintf('Could not extract type: %s', $type_));
        }
        fwrite($bytes_io, $this->struct->pack('<i', $tl_method->id));
        foreach ($tl_method->params as $arg) {
            $this->serialize_param($bytes_io, $arg['type'], $kwargs[$arg['name']]);
        }

        return fread_all($bytes_io);
    }

    public function serialize_param($bytes_io, $type_, $value)
    {
        if (($type_ == 'int')) {
            assert(is_numeric($value));
            assert(strlen(decbin($value)) <= 32);
            fwrite($bytes_io, $this->struct->pack('<i', $value));
        } elseif (($type_ == 'long')) {
            assert(is_numeric($value));
            fwrite($bytes_io, $this->struct->pack('<q', $value));
        } elseif (in_array($type_, ['int128', 'int256'])) {
            assert(is_string($value));
            fwrite($bytes_io, $value);
        } elseif ($type_ == 'string' || $type_ == 'bytes') {
            $l = len($value);
            if (($l < 254)) {
                fwrite($bytes_io, $this->struct->pack('<b', $l));
                fwrite($bytes_io, $value);
                fwrite($bytes_io, pack('@'.((-$l - 1) % 4)));
            } else {
                fwrite($bytes_io, string2bin('\xfe'));
                fwrite($bytes_io, substr($this->struct->pack('<i', $l), 0, 3));
                fwrite($bytes_io, $value);
                fwrite($bytes_io, pack('@'.(-$l % 4)));
            }
        }
    }

    /**
     * :type bytes_io: io.BytesIO object.
     */
    public function deserialize($bytes_io, $type_ = null, $subtype = null)
    {
        assert(get_resource_type($bytes_io) == 'file' || get_resource_type($bytes_io) == 'stream');
        if (($type_ == 'int')) {
            $x = $this->struct->unpack('<i', fread($bytes_io, 4)) [0];
        } elseif (($type_ == '#')) {
            $x = $this->struct->unpack('<I', fread($bytes_io, 4)) [0];
        } elseif (($type_ == 'long')) {
            $x = $this->struct->unpack('<q', fread($bytes_io, 8)) [0];
        } elseif (($type_ == 'double')) {
            $x = $this->struct->unpack('<d', fread($bytes_io, 8)) [0];
        } elseif (($type_ == 'int128')) {
            $x = fread($bytes_io, 16);
        } elseif (($type_ == 'int256')) {
            $x = fread($bytes_io, 32);
        } elseif (($type_ == 'string') || ($type_ == 'bytes')) {
            $l = $this->struct->unpack('<B', fread($bytes_io, 1)) [0];
            assert($l <= 254);
            if (($l == 254)) {
                $long_len = $this->struct->unpack('<I', fread($bytes_io, 3).string2bin('\x00')) [0];
                $x = fread($bytes_io, $long_len);
                fread($bytes_io, (-$long_len % 4));
            } else {
                $x = fread($bytes_io, $l);
                fread($bytes_io, (($l + 1) % 4));
            }
            assert(is_string($x));
        } elseif (($type_ == 'vector')) {
            assert($subtype != null);
            $count = $this->struct->unpack('<l', fread($bytes_io, 4)) [0];
            $x = [];
            foreach (pyjslib_range($count) as $i) {
                $x[] = $this->deserialize($bytes_io, $subtype);
            }
        } else {
            if (isset($this->constructor_type[$type_])) {
                $tl_elem = $this->constructor_type[$type_];
            } else {
                $i = $this->struct->unpack('<i', fread($bytes_io, 4)) [0];
                if (isset($this->constructor_id[$i])) {
                    $tl_elem = $this->constructor_id[$i];
                } else {
                    throw new Exception(sprintf('Could not extract type: %s', $type_));
                }
            }

            $base_boxed_types = ['Vector t', 'Int', 'Long', 'Double', 'String', 'Int128', 'Int256'];
            if (in_array($tl_elem->type, $base_boxed_types)) {
                $x = $this->deserialize($bytes_io, $tl_elem->predicate, $subtype);
            } else {
                $x = new TLObject($tl_elem);
                foreach ($tl_elem->params as $arg) {
                    $x[$arg['name']] = $this->deserialize($bytes_io, $arg['type'], $arg['subtype']);
                }
            }
        }

        return $x;
    }
}