don't just assume we get a valid title object
[lhc/web/wiklou.git] / includes / SpecialSearch.php
index d390fb3..0af2925 100644 (file)
  * @subpackage SpecialPage
  */
 
+/** */
 require_once( 'SearchEngine.php' );
+require_once( 'Revision.php' );
 
-function wfSpecialSearch( $par='' ) {
+/**
+ * Entry point
+ *
+ * @param string $par (default '')
+ */
+function wfSpecialSearch( $par = '' ) {
        global $wgRequest, $wgUser;
        
        $search = $wgRequest->getText( 'search', $par );
@@ -39,8 +46,13 @@ function wfSpecialSearch( $par='' ) {
        }
 }
 
-
+/**
+ * @todo document
+ * @package MediaWiki
+ * @subpackage SpecialPage
+ */
 class SpecialSearch {
+
        /**
         * Set up basic search parameters from the request and user settings.
         * Typically you'll pass $wgRequest and $wgUser.
@@ -101,8 +113,7 @@ class SpecialSearch {
                                $editurl = $t->escapeLocalURL( 'action=edit' );
                        }
                }
-               # FIXME: HTML in wiki message
-               $wgOut->addHTML( '<p>' . wfMsg('nogomatch', $editurl, htmlspecialchars( $term ) ) . "</p>\n" );
+               $wgOut->addWikiText( wfMsg('nogomatch', ":$term" ) );
 
                return $this->showResults( $term );
        }
@@ -132,20 +143,33 @@ class SpecialSearch {
                
                global $wgDisableTextSearch;
                if ( $wgDisableTextSearch ) {
+                       global $wgForwardSearchUrl;
+                       if( $wgForwardSearchUrl ) {
+                               $url = str_replace( '$1', urlencode( $term ), $wgForwardSearchUrl );
+                               $wgOut->redirect( $url );
+                               return;
+                       }
                        global $wgInputEncoding;
                        $wgOut->addHTML( wfMsg( 'searchdisabled' ) );
-                       $wgOut->addHTML( wfMsg( 'googlesearch',
-                               htmlspecialchars( $term ),
-                               htmlspecialchars( $wgInputEncoding ) ) );
+                       $wgOut->addHTML(
+                               wfMsg( 'googlesearch',
+                                       htmlspecialchars( $term ),
+                                       htmlspecialchars( $wgInputEncoding ),
+                                       htmlspecialchars( wfMsg( 'search' ) )
+                               )
+                       );
                        wfProfileOut( $fname );
                        return;
                }
 
-               $search =& $this->getSearchEngine();
+               $search =& SearchEngine::create();
+               $search->setLimitOffset( $this->limit, $this->offset );
+               $search->setNamespaces( $this->namespaces );
                $titleMatches = $search->searchTitle( $term );
                $textMatches = $search->searchText( $term );
                
-               $num = $titleMatches->numRows() + $textMatches->numRows();
+               $num = ( $titleMatches ? $titleMatches->numRows() : 0 )
+                       + ( $textMatches ? $textMatches->numRows() : 0);
                if ( $num >= $this->limit ) {
                        $top = wfShowingResults( $this->offset, $this->limit );
                } else {
@@ -162,21 +186,23 @@ class SpecialSearch {
                        $wgOut->addHTML( "<br />{$prevnext}\n" );
                }
 
-               $terms = implode( '|', $search->termMatches() );
-               
-               if( $titleMatches->numRows() ) {
-                       $wgOut->addWikiText( '==' . wfMsg( 'titlematches' ) . "==\n" );
-                       $wgOut->addHTML( $this->showMatches( $titleMatches, $terms ) );
-               } else {
-                       $wgOut->addWikiText( '==' . wfMsg( 'notitlematches' ) . "==\n" );
+               if( $titleMatches ) {
+                       if( $titleMatches->numRows() ) {
+                               $wgOut->addWikiText( '==' . wfMsg( 'titlematches' ) . "==\n" );
+                               $wgOut->addHTML( $this->showMatches( $titleMatches ) );
+                       } else {
+                               $wgOut->addWikiText( '==' . wfMsg( 'notitlematches' ) . "==\n" );
+                       }
                }
                
-               if( $textMatches->numRows() ) {
-                       $wgOut->addWikiText( '==' . wfMsg( 'textmatches' ) . "==\n" );
-                       $wgOut->addHTML( $this->showMatches( $textMatches, $terms ) );
-               } elseif( $num == 0 ) {
-                       # Don't show the 'no text matches' if we received title matches
-                       $wgOut->addWikiText( '==' . wfMsg( 'notextmatches' ) . "==\n" );
+               if( $textMatches ) {
+                       if( $textMatches->numRows() ) {
+                               $wgOut->addWikiText( '==' . wfMsg( 'textmatches' ) . "==\n" );
+                               $wgOut->addHTML( $this->showMatches( $textMatches ) );
+                       } elseif( $num == 0 ) {
+                               # Don't show the 'no text matches' if we received title matches
+                               $wgOut->addWikiText( '==' . wfMsg( 'notextmatches' ) . "==\n" );
+                       }
                }
                
                if ( $num == 0 ) {
@@ -198,39 +224,10 @@ class SpecialSearch {
        function setupPage( $term ) {
                global $wgOut;
                $wgOut->setPageTitle( wfMsg( 'searchresults' ) );
-               $wgOut->setSubtitle( wfMsg( 'searchquery', htmlspecialchars( $term ) ) );
+               $wgOut->setSubtitle( wfMsg( 'searchquery', $term ) );
                $wgOut->setArticleRelated( false );
                $wgOut->setRobotpolicy( 'noindex,nofollow' );
        }
-
-       /**
-        * Load up the appropriate search engine class for the currently
-        * active database backend, and return a configured instance.
-        *
-        * @return SearchEngine
-        * @access private
-        */
-       function &getSearchEngine() {
-               global $wgDBtype, $wgDBmysql4, $wgSearchType;
-               if( $wgDBtype == 'mysql' ) {
-                       if( $wgDBmysql4 ) {
-                               $class = 'SearchMySQL4';
-                               require_once( 'SearchMySQL4.php' );
-                       } else {
-                               $class = 'SearchMysql3';
-                               require_once( 'SearchMySQL3.php' );
-                       }
-               } else if ( $wgDBtype == 'PostgreSQL' ) {
-                       $class = 'SearchTsearch2';
-                       require_once( 'SearchTsearch2.php' );
-               } else {
-                       $class = 'SearchEngineDummy';
-               }
-               $search = new $class( wfGetDB( DB_SLAVE ) );
-               $search->setLimitOffset( $this->limit, $this->offset );
-               $search->setNamespaces( $this->namespaces );
-               return $search;
-       }
        
        /**
         * Extract default namespaces to search from the given user's
@@ -284,36 +281,44 @@ class SpecialSearch {
        }
        
        /**
-        * @param ResultWrapper $matches
+        * @param SearchResultSet $matches
         * @param string $terms partial regexp for highlighting terms
         */
-       function showMatches( &$matches, $terms ) {
+       function showMatches( &$matches ) {
                $fname = 'SpecialSearch::showMatches';
                wfProfileIn( $fname );
                
+               global $wgContLang;
+               $tm = $wgContLang->convertForSearchResult( $matches->termMatches() );
+               $terms = implode( '|', $tm );
+               
                global $wgOut;
                $off = $this->offset + 1;
                $out = "<ol start='{$off}'>\n";
 
-               while( $row = $matches->fetchObject() ) {
-                       $out .= $this->showHit( $row, $terms );
+               while( $result = $matches->next() ) {
+                       $out .= $this->showHit( $result, $terms );
                }
                $out .= "</ol>\n";
+
+               // convert the whole thing to desired language variant
+               global $wgContLang;
+               $out = $wgContLang->convert( $out );
                wfProfileOut( $fname );
                return $out;
        }
        
        /**
         * Format a single hit result
-        * @param object $row
+        * @param SearchResult $result
         * @param string $terms partial regexp for highlighting terms
         */
-       function showHit( $row, $terms ) {
+       function showHit( $result, $terms ) {
                $fname = 'SpecialSearch::showHit';
                wfProfileIn( $fname );
                global $wgUser, $wgContLang;
 
-               $t = Title::makeTitle( $row->cur_namespace, $row->cur_title );
+               $t = $result->getTitle();
                if( is_null( $t ) ) {
                        wfProfileOut( $fname );
                        return "<!-- Broken link in search result -->\n";
@@ -325,12 +330,16 @@ class SpecialSearch {
                $contextchars = $wgUser->getOption( 'contextchars' );
                if ( '' == $contextchars ) { $contextchars = 50; }
 
-               $link = $sk->makeKnownLinkObj( $t, '' );
-               $size = wfMsg( 'nbytes', strlen( $row->cur_text ) );
+               $link = $sk->makeKnownLinkObj( $t );
+               $revision = Revision::newFromTitle( $t );
+               $text = $revision->getText();
+               $size = wfMsg( 'nbytes', strlen( $text ) );
+
+               $lines = explode( "\n", $text );
 
-               $lines = explode( "\n", $row->cur_text );
                $max = IntVal( $contextchars ) + 1;
                $pat1 = "/(.*)($terms)(.{0,$max})/i";
+
                $lineno = 0;
                
                $extract = '';