RestbaseVirtualRESTService.php

Go to the documentation of this file.

<?php
class RestbaseVirtualRESTService extends VirtualRESTService {
    public function __construct( array $params ) {
        // set up defaults and merge them with the given params
        $mparams = array_merge( [
            'name' => 'restbase',
            'url' => 'http://localhost:7231/',
            'domain' => 'localhost',
            'timeout' => 100,
            'forwardCookies' => false,
            'HTTPProxy' => null,
            'parsoidCompat' => false,
            'fixedUrl' => false,
        ], $params );
        // Ensure that the url parameter has a trailing slash.
        if ( substr( $mparams['url'], -1 ) !== '/' ) {
            $mparams['url'] .= '/';
        }
        // Ensure the correct domain format: strip protocol, port,
        // and trailing slash if present.  This lets us use
        // $wgCanonicalServer as a default value, which is very convenient.
        $mparams['domain'] = preg_replace(
            '/^(https?:\/\/)?([^\/:]+?)(:\d+)?\/?$/',
            '$2',
            $mparams['domain']
        );
        parent::__construct( $mparams );
    }
 
    public function onRequests( array $reqs, Closure $idGenFunc ) {
        if ( $this->params['parsoidCompat'] ) {
            return $this->onParsoidRequests( $reqs, $idGenFunc );
        }
 
        $result = [];
        foreach ( $reqs as $key => $req ) {
            if ( $this->params['fixedUrl'] ) {
                $version = explode( '/', $req['url'] )[1];
                $req['url'] =
                    str_replace( '#version#', $version, $this->params['url'] ) .
                    preg_replace( '#^local/v./#', '', $req['url'] );
            } else {
                // replace /local/ with the current domain
                $req['url'] = preg_replace( '#^local/#', $this->params['domain'] . '/', $req['url'] );
                // and prefix it with the service URL
                $req['url'] = $this->params['url'] . $req['url'];
            }
 
            // set the appropriate proxy, timeout and headers
            if ( $this->params['HTTPProxy'] ) {
                $req['proxy'] = $this->params['HTTPProxy'];
            }
            if ( $this->params['timeout'] != null ) {
                $req['reqTimeout'] = $this->params['timeout'];
            }
            if ( $this->params['forwardCookies'] ) {
                $req['headers']['Cookie'] = $this->params['forwardCookies'];
            }
            $result[$key] = $req;
        }
 
        return $result;
    }
 
    public function onParsoidRequests( array $reqs, Closure $idGeneratorFunc ) {
        $result = [];
        foreach ( $reqs as $key => $req ) {
            $version = explode( '/', $req['url'] )[1];
            if ( $version === 'v3' ) {
                $result[$key] = $this->onParsoid3Request( $req, $idGeneratorFunc );
            } elseif ( $version === 'v1' ) {
                $result[$key] = $this->onParsoid1Request( $req, $idGeneratorFunc );
            } else {
                throw new Exception( "Only v1 and v3 are supported." );
            }
        }
 
        return $result;
    }
 
    public function onParsoid1Request( array $req, Closure $idGeneratorFunc ) {
        $parts = explode( '/', $req['url'] );
        list(
            $targetWiki, // 'local'
            $version, // 'v1'
            $reqType // 'page' or 'transform'
        ) = $parts;
        if ( $targetWiki !== 'local' ) {
            throw new Exception( "Only 'local' target wiki is currently supported" );
        } elseif ( $version !== 'v1' ) {
            throw new Exception( "Version mismatch: should not happen." );
        } elseif ( $reqType !== 'page' && $reqType !== 'transform' ) {
            throw new Exception( "Request type must be either 'page' or 'transform'" );
        }
        $req['url'] = $this->params['url'] . $this->params['domain'] . '/v1/' . $reqType . '/';
        if ( $reqType === 'page' ) {
            $title = $parts[3];
            if ( $parts[4] !== 'html' ) {
                throw new Exception( "Only 'html' output format is currently supported" );
            }
            $req['url'] .= 'html/' . $title;
            if ( isset( $parts[5] ) ) {
                $req['url'] .= '/' . $parts[5];
            } elseif ( isset( $req['query']['oldid'] ) && $req['query']['oldid'] ) {
                $req['url'] .= '/' . $req['query']['oldid'];
                unset( $req['query']['oldid'] );
            }
        } elseif ( $reqType === 'transform' ) {
            // from / to transform
            $req['url'] .= $parts[3] . '/to/' . $parts[5];
            // the title
            if ( isset( $parts[6] ) ) {
                $req['url'] .= '/' . $parts[6];
            }
            // revision id
            if ( isset( $parts[7] ) ) {
                $req['url'] .= '/' . $parts[7];
            } elseif ( isset( $req['body']['oldid'] ) && $req['body']['oldid'] ) {
                $req['url'] .= '/' . $req['body']['oldid'];
                unset( $req['body']['oldid'] );
            }
            if ( $parts[4] !== 'to' ) {
                throw new Exception( "Part index 4 is not 'to'" );
            }
            if ( $parts[3] === 'html' && $parts[5] === 'wikitext' ) {
                if ( !isset( $req['body']['html'] ) ) {
                    throw new Exception( "You must set an 'html' body key for this request" );
                }
            } elseif ( $parts[3] == 'wikitext' && $parts[5] == 'html' ) {
                if ( !isset( $req['body']['wikitext'] ) ) {
                    throw new Exception( "You must set a 'wikitext' body key for this request" );
                }
                if ( isset( $req['body']['body'] ) ) {
                    $req['body']['body_only'] = $req['body']['body'];
                    unset( $req['body']['body'] );
                }
            } else {
                throw new Exception( "Transformation unsupported" );
            }
        }
        // set the appropriate proxy, timeout and headers
        if ( $this->params['HTTPProxy'] ) {
            $req['proxy'] = $this->params['HTTPProxy'];
        }
        if ( $this->params['timeout'] != null ) {
            $req['reqTimeout'] = $this->params['timeout'];
        }
        if ( $this->params['forwardCookies'] ) {
            $req['headers']['Cookie'] = $this->params['forwardCookies'];
        }
 
        return $req;
    }
 
    public function onParsoid3Request( array $req, Closure $idGeneratorFunc ) {
        $parts = explode( '/', $req['url'] );
        list(
            $targetWiki, // 'local'
            $version, // 'v3'
            $action, // 'transform' or 'page'
            $format, // 'html' or 'wikitext'
            // $title, // optional
            // $revision, // optional
        ) = $parts;
        if ( $targetWiki !== 'local' ) {
            throw new Exception( "Only 'local' target wiki is currently supported" );
        } elseif ( $version !== 'v3' ) {
            throw new Exception( "Version mismatch: should not happen." );
        }
        // replace /local/ with the current domain, change v3 to v1,
        $req['url'] = preg_replace( '#^local/v3/#', $this->params['domain'] . '/v1/', $req['url'] );
        // and prefix it with the service URL
        $req['url'] = $this->params['url'] . $req['url'];
        // set the appropriate proxy, timeout and headers
        if ( $this->params['HTTPProxy'] ) {
            $req['proxy'] = $this->params['HTTPProxy'];
        }
        if ( $this->params['timeout'] != null ) {
            $req['reqTimeout'] = $this->params['timeout'];
        }
        if ( $this->params['forwardCookies'] ) {
            $req['headers']['Cookie'] = $this->params['forwardCookies'];
        }
 
        return $req;
    }
 
}