* (bug 1970) Don't show move tab for immobile pages
[lhc/web/wiklou.git] / includes / SpecialImport.php
index be4208a..3903de0 100644 (file)
 <?php
-# Copyright (C) 2003 Brion Vibber <brion@pobox.com>
-# http://www.mediawiki.org/
-# 
-# This program is free software; you can redistribute it and/or modify
-# it under the terms of the GNU General Public License as published by
-# the Free Software Foundation; either version 2 of the License, or 
-# (at your option) any later version.
-# 
-# This program is distributed in the hope that it will be useful,
-# but WITHOUT ANY WARRANTY; without even the implied warranty of
-# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
-# GNU General Public License for more details.
-# 
-# You should have received a copy of the GNU General Public License along
-# with this program; if not, write to the Free Software Foundation, Inc.,
-# 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
-# http://www.gnu.org/copyleft/gpl.html
+/**
+ * MediaWiki page data importer
+ * Copyright (C) 2003,2005 Brion Vibber <brion@pobox.com>
+ * http://www.mediawiki.org/
+ * 
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or 
+ * (at your option) any later version.
+ * 
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ * 
+ * You should have received a copy of the GNU General Public License along
+ * with this program; if not, write to the Free Software Foundation, Inc.,
+ * 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
+ * http://www.gnu.org/copyleft/gpl.html
+ *
+ * @package MediaWiki
+ * @subpackage SpecialPage
+ */
 
-function wfSpecialImport( $page = "" ) {
+require_once( 'WikiError.php' );
+
+/**
+ * Constructor
+ */
+function wfSpecialImport( $page = '' ) {
        global $wgOut, $wgLang, $wgRequest, $wgTitle;
+       global $wgImportSources;
+       
+       ###
+#      $wgOut->addWikiText( "Special:Import is not ready for this beta release, sorry." );
+#      return;
+       ###
        
        if( $wgRequest->wasPosted() && $wgRequest->getVal( 'action' ) == 'submit') {
-               $importer = new WikiImporter;
-               if( $importer->setupFromUpload( "xmlimport" ) ) {
-                       $importer->setRevisionHandler( "wfImportRevision" );
-                       if( $importer->doImport() ) {
+               $importer = new WikiImporter();
+               
+               switch( $wgRequest->getVal( "source" ) ) {
+               case "upload":
+                       $result = $importer->setupFromUpload( "xmlimport" );
+                       break;
+               case "interwiki":
+                       $result = $importer->setupFromInterwiki(
+                               $wgRequest->getVal( "interwiki" ),
+                               $wgRequest->getText( "frompage" ) );
+                       break;
+               default:
+                       $result = new WikiError( "Unknown import source type" );
+               }
+               
+               if( WikiError::isError( $result ) ) {
+                       $wgOut->addWikiText( htmlspecialchars( $result->toString() ) );
+               } else {
+                       $importer->setRevisionHandler( "wfImportOldRevision" );
+                       $result = $importer->doImport();
+                       if( WikiError::isError( $result ) ) {
+                               $wgOut->addHTML( "<p>" . wfMsg( "importfailed",
+                                       htmlspecialchars( $result->toString() ) ) . "</p>" );
+                       } else {
                                # Success!
                                $wgOut->addHTML( "<p>" . wfMsg( "importsuccess" ) . "</p>" );
-                       } else {
-                               $wgOut->addHTML( "<p>" . wfMsg( "importfailed",
-                                       htmlspecialchars( $importer->getError() ) ) . "</p>" );
                        }
-               } else {
-                       $wgOut->addWikiText( htmlspecialchars( $importer->getError() ) );
                }
        }
        
        $wgOut->addWikiText( "<p>" . wfMsg( "importtext" ) . "</p>" );
-       $action = $wgTitle->escapeLocalUrl();
+       $action = $wgTitle->escapeLocalUrl( 'action=submit' );
        $wgOut->addHTML( "
-<form enctype='multipart/form-data' method='post' action=\"$action\">
-       <input type='hidden' name='action' value='submit' />
-       <input type='hidden' name='MAX_FILE_SIZE' value='200000' />
-       <input type='file' name='xmlimport' value='' size='40' /><br />
-       <input type='submit' value='" . htmlspecialchars( wfMsg( "uploadbtn" ) ) . "'/>
-</form>
+<fieldset>
+       <legend>Upload XML</legend>
+       <form enctype='multipart/form-data' method='post' action=\"$action\">
+               <input type='hidden' name='action' value='submit' />
+               <input type='hidden' name='source' value='upload' />
+               <input type='hidden' name='MAX_FILE_SIZE' value='2000000' />
+               <input type='file' name='xmlimport' value='' size='30' />
+               <input type='submit' value='" . htmlspecialchars( wfMsg( "uploadbtn" ) ) . "'/>
+       </form>
+</fieldset>
 " );
+
+       if( !empty( $wgImportSources ) ) {
+               $wgOut->addHTML( "
+<fieldset>
+       <legend>Interwiki import</legend>
+       <form method='post' action=\"$action\">
+               <input type='hidden' name='action' value='submit' />
+               <input type='hidden' name='source' value='interwiki' />
+               <select name='interwiki'>
+" );
+               foreach( $wgImportSources as $interwiki ) {
+                       $iw = htmlspecialchars( $interwiki );
+                       $wgOut->addHTML( "<option value=\"$iw\">$iw</option>\n" );
+               }
+               $wgOut->addHTML( "
+               </select>
+               <input name='frompage' />
+               <input type='submit' />
+       </form>
+</fieldset>
+" );
+       }
 }
 
-function wfImportRevision( $revision ) {
-       $fname = "wfImportRevision";
-       
-       # Sneak a single revision into place
-       $ns = $revision->title->getNamespace();
-       $t = wfStrencode( $revision->title->getDBkey() );
-       $text = wfStrencode( $revision->text );
-       $ts = wfStrencode( $revision->timestamp );
-       $its = wfStrencode( wfInvertTimestamp( $revision->timestamp ) ) ;
-       $minor = 0; # ??
-       $flags = "";
-       
-       # Make sure it doesn't already exist
-       $res = wfQuery( "SELECT COUNT(*) FROM old WHERE old_namespace=$ns AND old_title='$t' AND old_timestamp='$ts'", DB_WRITE, $fname );
-       $numrows = wfNumRows( $res );
-       wfFreeResult( $res );
-       if( $numrows > 0 ) {
-               return "gaaah";
-       }
-       
-       $res = wfQuery( "INSERT INTO old " .
-         "(old_namespace,old_title,old_text,old_comment,old_user,old_user_text," .
-         "old_timestamp,inverse_timestamp,old_minor_edit,old_flags) " .
-         "VALUES ($ns,'$t','$text','$comment',$user,'$user_text','$ts','$its',$minor,'$flags')",
-         DB_WRITE, $fname );
-       
-       return true;
+function wfImportOldRevision( &$revision ) {
+       $dbw =& wfGetDB( DB_MASTER );
+       $dbw->deadlockLoop( array( &$revision, 'importOldRevision' ) );
 }
 
+/**
+ *
+ * @package MediaWiki
+ * @subpackage SpecialPage
+ */
 class WikiRevision {
        var $title = NULL;
        var $timestamp = "20010115000000";
        var $user = 0;
        var $user_text = "";
        var $text = "";
+       var $comment = "";
        
        function setTitle( $text ) {
-               $text = $this->fixEncoding( $text );
                $this->title = Title::newFromText( $text );
        }
        
@@ -95,25 +134,19 @@ class WikiRevision {
        }
        
        function setUsername( $user ) {
-               $this->user_text = $this->fixEncoding( $user );
+               $this->user_text = $user;
        }
        
        function setUserIP( $ip ) {
-               $this->user_text = $this->fixEncoding( $ip );
+               $this->user_text = $ip;
        }
        
        function setText( $text ) {
-               $this->text = $this->fixEncoding( $text );
+               $this->text = $text;
        }
        
-       function fixEncoding( $data ) {
-               global $wgLang, $wgInputEncoding;
-               
-               if( strcasecmp( $wgInputEncoding, "utf-8" ) == 0 ) {
-                       return $data;
-               } else {
-                       return $wgLang->iconv( "utf-8", $wgInputEncoding, $data );
-               }
+       function setComment( $text ) {
+               $this->comment = $text;
        }
        
        function getTitle() {
@@ -131,12 +164,58 @@ class WikiRevision {
        function getText() {
                return $this->text;
        }
+       
+       function getComment() {
+               return $this->comment;
+       }
+
+       function importOldRevision() {
+               $fname = "WikiImporter::importOldRevision";
+               $dbw =& wfGetDB( DB_MASTER );
+               
+               # Sneak a single revision into place
+               $user = User::newFromName( $this->getUser() );
+
+               $article = new Article( $this->title );
+               $pageId = $article->getId();
+               if( $pageId == 0 ) {
+                       # must create the page...
+                       $pageId = $article->insertOn( $dbw );
+               }
+               
+               # FIXME: Check for exact conflicts
+               # FIXME: Use original rev_id optionally
+               # FIXME: blah blah blah
+               
+               #if( $numrows > 0 ) {
+               #       return wfMsg( "importhistoryconflict" );
+               #}
+               
+               # Insert the row
+               $revision = new Revision( array(
+                       'page'       => $pageId,
+                       'text'       => $this->getText(),
+                       'comment'    => $this->getComment(),
+                       'user'       => IntVal( $user->getId() ),
+                       'user_text'  => $user->getName(),
+                       'timestamp'  => $this->timestamp,
+                       'minor_edit' => 0
+                       ) );
+               $revId = $revision->insertOn( $dbw );
+               $article->updateIfNewerOn( $dbw, $revision );
+               
+               return true;
+       }
+
 }
 
+/**
+ *
+ * @package MediaWiki
+ * @subpackage SpecialPage
+ */
 class WikiImporter {
        var $mSource = NULL;
-       var $mError = "";
-       var $mXmlError = XML_ERROR_NONE;
        var $mRevisionHandler = NULL;
        var $lastfield;
        
@@ -144,25 +223,15 @@ class WikiImporter {
                $this->setRevisionHandler( array( &$this, "defaultRevisionHandler" ) );
        }
        
-       function setError( $err ) {
-               $this->mError = $err;
-               return false;
-       }
-       
-       function getError() {
-               if( $this->mXmlError == XML_ERROR_NONE ) {
-                       return $this->mError;
-               } else {
-                       return xml_error_string( $this->mXmlError );
-               }
-       }
-       
        function throwXmlError( $err ) {
                $this->debug( "FAILURE: $err" );
        }
        
        function setupFromFile( $filename ) {
-               $this->mSource = file_get_contents( $filename );
+               $this->mSource = @file_get_contents( $filename );
+               if( $this->mSource === false ) {
+                       return new WikiError( "Couldn't open import file" );
+               }
                return true;
        }
 
@@ -172,27 +241,44 @@ class WikiImporter {
                $upload =& $_FILES[$fieldname];
                
                if( !isset( $upload ) ) {
-                       return $this->setError( wfMsg( "importnofile" ) );
+                       return new WikiErrorMsg( 'importnofile' );
                }
                if( !empty( $upload['error'] ) ) {
-                       return $this->setError( wfMsg( "importuploaderror", $upload['error'] ) );
+                       return new WikiErrorMsg( 'importuploaderror', $upload['error'] );
                }
                $fname = $upload['tmp_name'];
                if( is_uploaded_file( $fname ) ) {
                        return $this->setupFromFile( $fname );
                } else {
-                       return $this->setError( wfMsg( "importnofile" ) );
+                       return new WikiErrorMsg( 'importnofile' );
                }
        }
        
        function setupFromURL( $url ) {
-               # FIXME
-               wfDebugDieBacktrace( "Not yet implemented." );
+               # fopen-wrappers are normally turned off for security.
+               ini_set( "allow_url_fopen", true );
+               $ret = $this->setupFromFile( $url );
+               ini_set( "allow_url_fopen", false );
+               return $ret;
+       }
+       
+       function setupFromInterwiki( $interwiki, $page ) {
+               $base = Title::getInterwikiLink( $interwiki );
+               if( empty( $base ) ) {
+                       return new WikiError( 'Bad interwiki link' );
+               } else {
+                       $import = wfUrlencode( "Special:Export/$page" );
+                       $url = str_replace( "$1", $import, $base );
+                       $this->notice( "Importing from $url" );
+                       return $this->setupFromURL( $url );
+               }
        }
        
+       # --------------
+       
        function doImport() {
                if( empty( $this->mSource ) ) {
-                       return $this->setError( wfMsg( "importnotext" ) );
+                       return new WikiErrorMsg( "importnotext" );
                }
                
                $parser = xml_parser_create( "UTF-8" );
@@ -204,10 +290,7 @@ class WikiImporter {
                xml_set_element_handler( $parser, "in_start", "" );
                
                if( !xml_parse( $parser, $this->mSource, true ) ) {
-                       # return error message
-                       $this->mXmlError = xml_get_error_code( $parser );
-                       xml_parser_free( $parser );
-                       return false;
+                       return new WikiXmlError( $parser );
                }
                xml_parser_free( $parser );
                
@@ -215,15 +298,24 @@ class WikiImporter {
        }
        
        function debug( $data ) {
-               # global $wgOut;
-               # $wgOut->addHTML( htmlspecialchars( $data ) . "<br>\n" );
+               #$this->notice( "DEBUG: $data\n" );
+       }
+       
+       function notice( $data ) {
+               global $wgCommandLineMode;
+               if( $wgCommandLineMode ) {
+                       print "$data\n";
+               } else {
+                       global $wgOut;
+                       $wgOut->addHTML( "<li>$data</li>\n" );
+               }
        }
        
        function setRevisionHandler( $functionref ) {
                $this->mRevisionHandler = $functionref;
        }
        
-       function defaultRevisionHandler( $revision ) {
+       function defaultRevisionHandler( &$revision ) {
                $this->debug( "Got revision:" );
                if( is_object( $revision->title ) ) {
                        $this->debug( "-- Title: " . $revision->title->getPrefixedText() );
@@ -232,6 +324,7 @@ class WikiImporter {
                }
                $this->debug( "-- User: " . $revision->user_text );
                $this->debug( "-- Timestamp: " . $revision->timestamp );
+               $this->debug( "-- Comment: " . $revision->comment );
                $this->debug( "-- Text: " . $revision->text );
        }
        
@@ -272,6 +365,7 @@ class WikiImporter {
                case "title":
                case "restrictions":
                        $this->appendfield = $name;
+                       $this->appenddata = "";
                        $this->parenttag = "page";
                        xml_set_element_handler( $parser, "in_nothing", "out_append" );
                        xml_set_character_data_handler( $parser, "char_append" );
@@ -328,7 +422,10 @@ class WikiImporter {
                case "timestamp":
                        $this->workRevision->setTimestamp( $this->appenddata );
                        break;
-               default;
+               case "comment":
+                       $this->workRevision->setComment( $this->appenddata );
+                       break;
+               default:
                        $this->debug( "Bad append: {$this->appendfield}" );
                }
                $this->appendfield = "";
@@ -354,6 +451,7 @@ class WikiImporter {
                        return $this->throwXMLerror( "Element <$name> not allowed in a <revision>." );
                }
        }
+       
        function out_revision( $parser, $name ) {
                $this->debug( "out_revision $name" );
                if( $name != "revision" ) {
@@ -361,7 +459,11 @@ class WikiImporter {
                }
                xml_set_element_handler( $parser, "in_page", "out_page" );
                
-               call_user_func( $this->mRevisionHandler, $this->workRevision );
+               $out = call_user_func( $this->mRevisionHandler, &$this->workRevision, &$this );
+               if( !empty( $out ) ) {
+                       global $wgOut;
+                       $wgOut->addHTML( "<li>" . $out . "</li>\n" );
+               }
        }
        
        function in_contributor( $parser, $name, $attribs ) {
@@ -386,6 +488,7 @@ class WikiImporter {
                }
                xml_set_element_handler( $parser, "in_revision", "out_revision" );
        }
+
 }