Improve ApiFormatJson test coverage
[lhc/web/wiklou.git] / includes / api / ApiFormatJson.php
1 <?php
2 /**
3 * Copyright © 2006 Yuri Astrakhan "<Firstname><Lastname>@gmail.com"
4 *
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation; either version 2 of the License, or
8 * (at your option) any later version.
9 *
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU General Public License for more details.
14 *
15 * You should have received a copy of the GNU General Public License along
16 * with this program; if not, write to the Free Software Foundation, Inc.,
17 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
18 * http://www.gnu.org/copyleft/gpl.html
19 *
20 * @file
21 */
22
23 /**
24 * API JSON output formatter
25 * @ingroup API
26 */
27 class ApiFormatJson extends ApiFormatBase {
28
29 private $isRaw;
30
31 public function __construct( ApiMain $main, $format ) {
32 parent::__construct( $main, $format );
33 $this->isRaw = ( $format === 'rawfm' );
34
35 if ( $this->getMain()->getCheck( 'callback' ) ) {
36 # T94015: jQuery appends a useless '_' parameter in jsonp mode.
37 # Mark the parameter as used in that case to avoid a warning that's
38 # outside the control of the end user.
39 # (and do it here because ApiMain::reportUnusedParams() gets called
40 # before our ::execute())
41 $this->getMain()->markParamsUsed( '_' );
42 }
43 }
44
45 public function getMimeType() {
46 $params = $this->extractRequestParams();
47 // callback:
48 if ( isset( $params['callback'] ) ) {
49 return 'text/javascript';
50 }
51
52 return 'application/json';
53 }
54
55 public function execute() {
56 $params = $this->extractRequestParams();
57
58 $opt = 0;
59 if ( $this->isRaw ) {
60 $opt |= FormatJson::ALL_OK;
61 $transform = [];
62 } else {
63 switch ( $params['formatversion'] ) {
64 case 1:
65 $opt |= $params['utf8'] ? FormatJson::ALL_OK : FormatJson::XMLMETA_OK;
66 $transform = [
67 'BC' => [],
68 'Types' => [ 'AssocAsObject' => true ],
69 'Strip' => 'all',
70 ];
71 break;
72
73 case 2:
74 case 'latest':
75 $opt |= $params['ascii'] ? FormatJson::XMLMETA_OK : FormatJson::ALL_OK;
76 $transform = [
77 'Types' => [ 'AssocAsObject' => true ],
78 'Strip' => 'all',
79 ];
80 break;
81
82 default:
83 // Should have been caught during parameter validation
84 // @codeCoverageIgnoreStart
85 $this->dieDebug( __METHOD__, 'Unknown value for \'formatversion\'' );
86 // @codeCoverageIgnoreEnd
87 }
88 }
89 $data = $this->getResult()->getResultData( null, $transform );
90 $json = FormatJson::encode( $data, $this->getIsHtml(), $opt );
91
92 // T68776: OutputHandler::mangleFlashPolicy() avoids a nasty bug in
93 // Flash, but what it does isn't friendly for the API, so we need to
94 // work around it.
95 if ( preg_match( '/\<\s*cross-domain-policy(?=\s|\>)/i', $json ) ) {
96 $json = preg_replace(
97 '/\<(\s*cross-domain-policy(?=\s|\>))/i', '\\u003C$1', $json
98 );
99 }
100
101 if ( isset( $params['callback'] ) ) {
102 $callback = preg_replace( "/[^][.\\'\\\"_A-Za-z0-9]/", '', $params['callback'] );
103 # Prepend a comment to try to avoid attacks against content
104 # sniffers, such as T70187.
105 $this->printText( "/**/$callback($json)" );
106 } else {
107 $this->printText( $json );
108 }
109 }
110
111 public function getAllowedParams() {
112 if ( $this->isRaw ) {
113 return parent::getAllowedParams();
114 }
115
116 $ret = parent::getAllowedParams() + [
117 'callback' => [
118 ApiBase::PARAM_HELP_MSG => 'apihelp-json-param-callback',
119 ],
120 'utf8' => [
121 ApiBase::PARAM_DFLT => false,
122 ApiBase::PARAM_HELP_MSG => 'apihelp-json-param-utf8',
123 ],
124 'ascii' => [
125 ApiBase::PARAM_DFLT => false,
126 ApiBase::PARAM_HELP_MSG => 'apihelp-json-param-ascii',
127 ],
128 'formatversion' => [
129 ApiBase::PARAM_TYPE => [ '1', '2', 'latest' ],
130 ApiBase::PARAM_DFLT => '1',
131 ApiBase::PARAM_HELP_MSG => 'apihelp-json-param-formatversion',
132 ],
133 ];
134 return $ret;
135 }
136 }