WebResponse.php

Go to the documentation of this file.

<?php
class WebResponse {

    protected static $setCookies = [];

    public function header( $string, $replace = true, $http_response_code = null ) {
        header( $string, $replace, $http_response_code );
    }

    public function getHeader( $key ) {
        foreach ( headers_list() as $header ) {
            list( $name, $val ) = explode( ':', $header, 2 );
            if ( !strcasecmp( $name, $key ) ) {
                return trim( $val );
            }
        }
        return null;
    }

    public function statusHeader( $code ) {
        HttpStatus::header( $code );
    }

    public function headersSent() {
        return headers_sent();
    }

    public function setCookie( $name, $value, $expire = 0, $options = [] ) {
        global $wgCookiePath, $wgCookiePrefix, $wgCookieDomain;
        global $wgCookieSecure, $wgCookieExpiration, $wgCookieHttpOnly;

        if ( !is_array( $options ) ) {
            // Backwards compatibility
            $options = [ 'prefix' => $options ];
            if ( func_num_args() >= 5 ) {
                $options['domain'] = func_get_arg( 4 );
            }
            if ( func_num_args() >= 6 ) {
                $options['secure'] = func_get_arg( 5 );
            }
        }
        $options = array_filter( $options, function ( $a ) {
            return $a !== null;
        } ) + [
            'prefix' => $wgCookiePrefix,
            'domain' => $wgCookieDomain,
            'path' => $wgCookiePath,
            'secure' => $wgCookieSecure,
            'httpOnly' => $wgCookieHttpOnly,
            'raw' => false,
        ];

        if ( $expire === null ) {
            $expire = 0; // Session cookie
        } elseif ( $expire == 0 && $wgCookieExpiration != 0 ) {
            $expire = time() + $wgCookieExpiration;
        }

        $func = $options['raw'] ? 'setrawcookie' : 'setcookie';

        if ( Hooks::run( 'WebResponseSetCookie', [ &$name, &$value, &$expire, &$options ] ) ) {
            $cookie = $options['prefix'] . $name;
            $data = [
                'name' => (string)$cookie,
                'value' => (string)$value,
                'expire' => (int)$expire,
                'path' => (string)$options['path'],
                'domain' => (string)$options['domain'],
                'secure' => (bool)$options['secure'],
                'httpOnly' => (bool)$options['httpOnly'],
            ];

            // Per RFC 6265, key is name + domain + path
            $key = "{$data['name']}\n{$data['domain']}\n{$data['path']}";

            // If this cookie name was in the request, fake an entry in
            // self::$setCookies for it so the deleting check works right.
            if ( isset( $_COOKIE[$cookie] ) && !array_key_exists( $key, self::$setCookies ) ) {
                self::$setCookies[$key] = [];
            }

            // PHP deletes if value is the empty string; also, a past expiry is deleting
            $deleting = ( $data['value'] === '' || $data['expire'] > 0 && $data['expire'] <= time() );

            if ( $deleting && !isset( self::$setCookies[$key] ) ) { // isset( null ) is false
                wfDebugLog( 'cookie', 'already deleted ' . $func . ': "' . implode( '", "', $data ) . '"' );
            } elseif ( !$deleting && isset( self::$setCookies[$key] ) &&
                self::$setCookies[$key] === [ $func, $data ]
            ) {
                wfDebugLog( 'cookie', 'already set ' . $func . ': "' . implode( '", "', $data ) . '"' );
            } else {
                wfDebugLog( 'cookie', $func . ': "' . implode( '", "', $data ) . '"' );
                if ( call_user_func_array( $func, array_values( $data ) ) ) {
                    self::$setCookies[$key] = $deleting ? null : [ $func, $data ];
                }
            }
        }
    }

    public function clearCookie( $name, $options = [] ) {
        $this->setCookie( $name, '', time() - 31536000 /* 1 year */, $options );
    }

    public function hasCookies() {
        return (bool)self::$setCookies;
    }
}

class FauxResponse extends WebResponse {
    private $headers;
    private $cookies = [];
    private $code;

    public function header( $string, $replace = true, $http_response_code = null ) {
        if ( substr( $string, 0, 5 ) == 'HTTP/' ) {
            $parts = explode( ' ', $string, 3 );
            $this->code = intval( $parts[1] );
        } else {
            list( $key, $val ) = array_map( 'trim', explode( ":", $string, 2 ) );

            $key = strtoupper( $key );

            if ( $replace || !isset( $this->headers[$key] ) ) {
                $this->headers[$key] = $val;
            }
        }

        if ( $http_response_code !== null ) {
            $this->code = intval( $http_response_code );
        }
    }

    public function statusHeader( $code ) {
        $this->code = intval( $code );
    }

    public function headersSent() {
        return false;
    }

    public function getHeader( $key ) {
        $key = strtoupper( $key );

        if ( isset( $this->headers[$key] ) ) {
            return $this->headers[$key];
        }
        return null;
    }

    public function getStatusCode() {
        return $this->code;
    }

    public function setCookie( $name, $value, $expire = 0, $options = [] ) {
        global $wgCookiePath, $wgCookiePrefix, $wgCookieDomain;
        global $wgCookieSecure, $wgCookieExpiration, $wgCookieHttpOnly;

        if ( !is_array( $options ) ) {
            // Backwards compatibility
            $options = [ 'prefix' => $options ];
            if ( func_num_args() >= 5 ) {
                $options['domain'] = func_get_arg( 4 );
            }
            if ( func_num_args() >= 6 ) {
                $options['secure'] = func_get_arg( 5 );
            }
        }
        $options = array_filter( $options, function ( $a ) {
            return $a !== null;
        } ) + [
            'prefix' => $wgCookiePrefix,
            'domain' => $wgCookieDomain,
            'path' => $wgCookiePath,
            'secure' => $wgCookieSecure,
            'httpOnly' => $wgCookieHttpOnly,
            'raw' => false,
        ];

        if ( $expire === null ) {
            $expire = 0; // Session cookie
        } elseif ( $expire == 0 && $wgCookieExpiration != 0 ) {
            $expire = time() + $wgCookieExpiration;
        }

        $this->cookies[$options['prefix'] . $name] = [
            'value' => (string)$value,
            'expire' => (int)$expire,
            'path' => (string)$options['path'],
            'domain' => (string)$options['domain'],
            'secure' => (bool)$options['secure'],
            'httpOnly' => (bool)$options['httpOnly'],
            'raw' => (bool)$options['raw'],
        ];
    }

    public function getCookie( $name ) {
        if ( isset( $this->cookies[$name] ) ) {
            return $this->cookies[$name]['value'];
        }
        return null;
    }

    public function getCookieData( $name ) {
        if ( isset( $this->cookies[$name] ) ) {
            return $this->cookies[$name];
        }
        return null;
    }

    public function getCookies() {
        return $this->cookies;
    }
}