API: Making a bunch of state-changing modules require POST requests.
[lhc/web/wiklou.git] / includes / api / ApiMain.php
index fda6f88..4c294a4 100644 (file)
@@ -53,11 +53,26 @@ class ApiMain extends ApiBase {
         */
        private static $Modules = array (
                'login' => 'ApiLogin',
+               'logout' => 'ApiLogout',
                'query' => 'ApiQuery',
+               'expandtemplates' => 'ApiExpandTemplates',
+               'parse' => 'ApiParse',
                'opensearch' => 'ApiOpenSearch',
                'feedwatchlist' => 'ApiFeedWatchlist',
                'help' => 'ApiHelp',
        );
+       
+       private static $WriteModules = array (
+               'rollback' => 'ApiRollback',
+               'delete' => 'ApiDelete',
+               'undelete' => 'ApiUndelete',
+               'protect' => 'ApiProtect',
+               'block' => 'ApiBlock',
+               'unblock' => 'ApiUnblock',
+               'move' => 'ApiMove',
+               #'changerights' => 'ApiChangeRights'
+               # Disabled for now
+       );
 
        /**
         * List of available formats: format name => format class
@@ -77,7 +92,7 @@ class ApiMain extends ApiBase {
        );
 
        private $mPrinter, $mModules, $mModuleNames, $mFormats, $mFormatNames;
-       private $mResult, $mShowVersions, $mEnableWrite, $mRequest, $mInternalMode, $mSquidMaxage;
+       private $mResult, $mAction, $mShowVersions, $mEnableWrite, $mRequest, $mInternalMode, $mSquidMaxage;
 
        /**
        * Constructs an instance of ApiMain that utilizes the module and format specified by $request.
@@ -92,7 +107,26 @@ class ApiMain extends ApiBase {
                // Special handling for the main module: $parent === $this
                parent :: __construct($this, $this->mInternalMode ? 'main_int' : 'main');
 
-               $this->mModules = self :: $Modules;
+               if (!$this->mInternalMode) {
+                       
+                       // Impose module restrictions.
+                       // If the current user cannot read, 
+                       // Remove all modules other than login
+                       global $wgUser;
+                       if (!$wgUser->isAllowed('read')) {
+                               self::$Modules = array(
+                                       'login'  => self::$Modules['login'],
+                                       'logout' => self::$Modules['logout'],
+                                       'help'   => self::$Modules['help'],
+                                       ); 
+                       }
+               }
+
+               global $wgAPIModules, $wgEnableWriteAPI; // extension modules
+               $this->mModules = $wgAPIModules + self :: $Modules;
+               if($wgEnableWriteAPI)
+                       $this->mModules += self::$WriteModules;
+
                $this->mModuleNames = array_keys($this->mModules); // todo: optimize
                $this->mFormats = self :: $Formats;
                $this->mFormatNames = array_keys($this->mFormats); // todo: optimize
@@ -106,6 +140,13 @@ class ApiMain extends ApiBase {
                $this->mSquidMaxage = 0;
        }
 
+       /**
+        * Return true if the API was started by other PHP code using FauxRequest
+        */
+       public function isInternalMode() {
+               return $this->mInternalMode;
+       }
+
        /**
         * Return the request object that contains client's request
         */
@@ -126,7 +167,7 @@ class ApiMain extends ApiBase {
        public function requestWriteMode() {
                if (!$this->mEnableWrite)
                        $this->dieUsage('Editing of this site is disabled. Make sure the $wgEnableWriteAPI=true; ' .
-                       'statement is included in the site\'s LocalSettings.php file', 'readonly');
+                       'statement is included in the site\'s LocalSettings.php file', 'noapiwrite');
        }
 
        /**
@@ -174,9 +215,43 @@ class ApiMain extends ApiBase {
                        // handler will process and log it.
                        //
 
+                       $errCode = $this->substituteResultWithError($e);
+
                        // Error results should not be cached
                        $this->setCacheMaxAge(0);
 
+                       $headerStr = 'MediaWiki-API-Error: ' . $errCode;
+                       if ($e->getCode() === 0)
+                               header($headerStr, true);
+                       else
+                               header($headerStr, true, $e->getCode());
+
+                       // Reset and print just the error message
+                       ob_clean();
+
+                       // If the error occured during printing, do a printer->profileOut()
+                       $this->mPrinter->safeProfileOut();
+                       $this->printResult(true);
+               }
+
+               // Set the cache expiration at the last moment, as any errors may change the expiration.
+               // if $this->mSquidMaxage == 0, the expiry time is set to the first second of unix epoch
+               $expires = $this->mSquidMaxage == 0 ? 1 : time() + $this->mSquidMaxage;
+               header('Expires: ' . wfTimestamp(TS_RFC2822, $expires));
+               header('Cache-Control: s-maxage=' . $this->mSquidMaxage . ', must-revalidate, max-age=0');
+
+               if($this->mPrinter->getIsHtml())
+                       echo wfReportTime();
+
+               ob_end_flush();
+       }
+
+       /**
+        * Replace the result data with the information about an exception.
+        * Returns the error code 
+        */
+       protected function substituteResultWithError($e) {
+       
                        // Printer may not be initialized if the extractRequestParams() fails for the main module
                        if (!isset ($this->mPrinter)) {
                                // The printer has not been created yet. Try to manually get formatter value.
@@ -194,10 +269,11 @@ class ApiMain extends ApiBase {
                                // User entered incorrect parameters - print usage screen
                                //
                                $errMessage = array (
-                               'code' => $e->getCodeString(), 'info' => $e->getMessage());
+                               'code' => $e->getCodeString(),
+                               'info' => $e->getMessage());
                                
                                // Only print the help message when this is for the developer, not runtime
-                               if ($this->mPrinter->getIsHtml())
+                               if ($this->mPrinter->getIsHtml() || $this->mAction == 'help')
                                        ApiResult :: setContent($errMessage, $this->makeHelpMsg());
 
                        } else {
@@ -205,54 +281,50 @@ class ApiMain extends ApiBase {
                                // Something is seriously wrong
                                //
                                $errMessage = array (
-                                       'code' => 'internal_api_error',
+                                       'code' => 'internal_api_error_'. get_class($e),
                                        'info' => "Exception Caught: {$e->getMessage()}"
                                );
                                ApiResult :: setContent($errMessage, "\n\n{$e->getTraceAsString()}\n\n");
                        }
 
-                       $headerStr = 'MediaWiki-API-Error: ' . $errMessage['code'];
-                       if ($e->getCode() === 0)
-                               header($headerStr, true);
-                       else
-                               header($headerStr, true, $e->getCode());
-
-                       // Reset and print just the error message
-                       ob_clean();
                        $this->getResult()->reset();
                        $this->getResult()->addValue(null, 'error', $errMessage);
 
-                       // If the error occured during printing, do a printer->profileOut()
-                       $this->mPrinter->safeProfileOut();
-                       $this->printResult(true);
-               }
-
-               // Set the cache expiration at the last moment, as any errors may change the expiration.
-               // if $this->mSquidMaxage == 0, the expiry time is set to the first second of unix epoch
-               $expires = $this->mSquidMaxage == 0 ? 1 : time() + $this->mSquidMaxage;
-               header('Expires: ' . wfTimestamp(TS_RFC2822, $expires));
-               header('Cache-Control: s-maxage=' . $this->mSquidMaxage . ', must-revalidate, max-age=0');
-
-               if($this->mPrinter->getIsHtml())
-                       echo wfReportTime();
-
-               ob_end_flush();
+               return $errMessage['code'];
        }
 
        /**
         * Execute the actual module, without any error handling
         */
        protected function executeAction() {
-               
+
                $params = $this->extractRequestParams();
-               
+
                $this->mShowVersions = $params['version'];
-               $action = $params['action'];
+               $this->mAction = $params['action'];
 
                // Instantiate the module requested by the user
-               $module = new $this->mModules[$action] ($this, $action);
-
+               $module = new $this->mModules[$this->mAction] ($this, $this->mAction);
+               
+               if( $module->shouldCheckMaxlag() && isset( $params['maxlag'] ) ) {
+                       // Check for maxlag
+                       global $wgLoadBalancer, $wgShowHostnames;
+                       $maxLag = $params['maxlag'];
+                       list( $host, $lag ) = $wgLoadBalancer->getMaxLag();
+                       if ( $lag > $maxLag ) {
+                               if( $wgShowHostnames ) {
+                                       ApiBase :: dieUsage( "Waiting for $host: $lag seconds lagged", 'maxlag' );
+                               } else {
+                                       ApiBase :: dieUsage( "Waiting for a database server: $lag seconds lagged", 'maxlag' );
+                               }
+                               return;
+                       }
+               }
+               
                if (!$this->mInternalMode) {
+                       // Ignore mustBePosted() for internal calls
+                       if($module->mustBePosted() && !$this->mRequest->wasPosted())
+                               $this->dieUsage("The {$this->mAction} module requires a POST request", 'mustbeposted');
 
                        // See if custom printer is used
                        $this->mPrinter = $module->getCustomPrinter();
@@ -282,7 +354,16 @@ class ApiMain extends ApiBase {
        protected function printResult($isError) {
                $printer = $this->mPrinter;
                $printer->profileIn();
+       
+               /* If the help message is requested in the default (xmlfm) format,
+                * tell the printer not to escape ampersands so that our links do
+                * not break. */
+               $params = $this->extractRequestParams();
+               $printer->setUnescapeAmps ( ( $this->mAction == 'help' || $isError ) 
+                               && $params['format'] == ApiMain::API_DEFAULT_FORMAT );
+
                $printer->initPrinter($isError);
+
                $printer->execute();
                $printer->closePrinter();
                $printer->profileOut();
@@ -301,7 +382,10 @@ class ApiMain extends ApiBase {
                                ApiBase :: PARAM_DFLT => 'help',
                                ApiBase :: PARAM_TYPE => $this->mModuleNames
                        ),
-                       'version' => false
+                       'version' => false,
+                       'maxlag'  => array (
+                               ApiBase :: PARAM_TYPE => 'integer'
+                       ),
                );
        }
 
@@ -312,7 +396,8 @@ class ApiMain extends ApiBase {
                return array (
                        'format' => 'The format of the output',
                        'action' => 'What action you would like to perform',
-                       'version' => 'When showing help, include version for each module'
+                       'version' => 'When showing help, include version for each module',
+                       'maxlag' => 'Maximum lag'
                );
        }
 
@@ -322,14 +407,28 @@ class ApiMain extends ApiBase {
        protected function getDescription() {
                return array (
                        '',
-                       'This API allows programs to access various functions of MediaWiki software.',
-                       'For more details see API Home Page @ http://www.mediawiki.org/wiki/API',
                        '',
-                       'Status: ALPHA -- all features shown on this page should be working,',
-                       '                 but the API is still in active development, and  may change at any time.',
-                       '                 Make sure you monitor changes to this page, wikitech-l mailing list,',
-                       '                 or the source code in the includes/api directory for any changes.',
-                       ''
+                       '******************************************************************',
+                       '**                                                              **',
+                       '**  This is an auto-generated MediaWiki API documentation page  **',
+                       '**                                                              **',
+                       '**                  Documentation and Examples:                 **',
+                       '**               http://www.mediawiki.org/wiki/API              **',
+                       '**                                                              **',
+                       '******************************************************************',
+                       '',
+                       'Status:          All features shown on this page should be working, but the API',
+                       '                 is still in active development, and  may change at any time.',
+                       '                 Make sure to monitor our mailing list for any updates.',
+                       '',
+                       'Documentation:   http://www.mediawiki.org/wiki/API',
+                       'Mailing list:    http://lists.wikimedia.org/mailman/listinfo/mediawiki-api',
+                       'Bugs & Requests: http://bugzilla.wikimedia.org/buglist.cgi?component=API&bug_status=NEW&bug_status=ASSIGNED&bug_status=REOPENED&order=bugs.delta_ts',
+                       '',
+                       '',
+                       '',
+                       '',
+                       '',
                );
        }
        
@@ -347,6 +446,8 @@ class ApiMain extends ApiBase {
         * Override the parent to generate help messages for all available modules.
         */
        public function makeHelpMsg() {
+               
+               $this->mPrinter->setHelp();
 
                // Use parent to make default message for the main module
                $msg = parent :: makeHelpMsg();
@@ -379,19 +480,20 @@ class ApiMain extends ApiBase {
        }
 
        public static function makeHelpMsgHeader($module, $paramName) {
-               $paramPrefix = $module->getParamPrefix();
-               if (!empty($paramPrefix))
-                       $paramPrefix = "($paramPrefix) "; 
+               $modulePrefix = $module->getModulePrefix();
+               if (!empty($modulePrefix))
+                       $modulePrefix = "($modulePrefix) "; 
                
-               return "* $paramName={$module->getModuleName()} $paramPrefix*";
+               return "* $paramName={$module->getModuleName()} $modulePrefix*";
        } 
 
        private $mIsBot = null;
-       
        private $mIsSysop = null;
+       private $mCanApiHighLimits = null;
        
        /**
         * Returns true if the currently logged in user is a bot, false otherwise
+        * OBSOLETE, use canApiHighLimits() instead
         */
        public function isBot() {
                if (!isset ($this->mIsBot)) {
@@ -404,16 +506,25 @@ class ApiMain extends ApiBase {
        /**
         * Similar to isBot(), this method returns true if the logged in user is
         * a sysop, and false if not.
+        * OBSOLETE, use canApiHighLimits() instead
         */
        public function isSysop() {
                if (!isset ($this->mIsSysop)) {
                        global $wgUser;
-                       $this->mIsSysop = in_array( 'sysop',
-                               $wgUser->getGroups());
+                       $this->mIsSysop = in_array( 'sysop', $wgUser->getGroups());
                }
 
                return $this->mIsSysop;
        }
+       
+       public function canApiHighLimits() {
+               if (!isset($this->mCanApiHighLimits)) {
+                       global $wgUser;
+                       $this->mCanApiHighLimits = $wgUser->isAllowed('apihighlimits');
+               }
+
+               return $this->mCanApiHighLimits;
+       }
 
        public function getShowVersions() {
                return $this->mShowVersions;
@@ -433,6 +544,31 @@ class ApiMain extends ApiBase {
                $vers[] = ApiFormatFeedWrapper :: getVersion(); // not accessible with format=xxx
                return $vers;
        }
+
+       /**
+        * Add or overwrite a module in this ApiMain instance. Intended for use by extending
+        * classes who wish to add their own modules to their lexicon or override the 
+        * behavior of inherent ones.
+        *
+        * @access protected
+        * @param $mdlName String The identifier for this module.
+        * @param $mdlClass String The class where this module is implemented.
+        */
+       protected function addModule( $mdlName, $mdlClass ) {
+               $this->mModules[$mdlName] = $mdlClass;
+       }
+
+       /**
+        * Add or overwrite an output format for this ApiMain. Intended for use by extending
+        * classes who wish to add to or modify current formatters.
+        *
+        * @access protected
+        * @param $fmtName The identifier for this format.
+        * @param $fmtClass The class implementing this format.
+        */
+       protected function addFormat( $fmtName, $fmtClass ) {
+               $this->mFormats[$fmtName] = $fmtClass;
+       }
 }
 
 /**
@@ -457,3 +593,4 @@ class UsageException extends Exception {
        }
 }
 
+