plugins/nuggie/search.php
author Dan
Thu, 31 Jan 2008 21:35:46 -0500
changeset 2 4e7762863437
child 6 c51809bdf6af
permissions -rw-r--r--
Added search module
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
2
4e7762863437 Added search module
Dan
parents:
diff changeset
     1
<?php
4e7762863437 Added search module
Dan
parents:
diff changeset
     2
4e7762863437 Added search module
Dan
parents:
diff changeset
     3
/*
4e7762863437 Added search module
Dan
parents:
diff changeset
     4
 * Nuggie
4e7762863437 Added search module
Dan
parents:
diff changeset
     5
 * Version 0.1
4e7762863437 Added search module
Dan
parents:
diff changeset
     6
 * Copyright (C) 2007 Dan Fuhry
4e7762863437 Added search module
Dan
parents:
diff changeset
     7
 *
4e7762863437 Added search module
Dan
parents:
diff changeset
     8
 * This program is Free Software; you can redistribute and/or modify it under the terms of the GNU General Public License
4e7762863437 Added search module
Dan
parents:
diff changeset
     9
 * as published by the Free Software Foundation; either version 2 of the License, or (at your option) any later version.
4e7762863437 Added search module
Dan
parents:
diff changeset
    10
 *
4e7762863437 Added search module
Dan
parents:
diff changeset
    11
 * This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied
4e7762863437 Added search module
Dan
parents:
diff changeset
    12
 * warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for details.
4e7762863437 Added search module
Dan
parents:
diff changeset
    13
 */
4e7762863437 Added search module
Dan
parents:
diff changeset
    14
4e7762863437 Added search module
Dan
parents:
diff changeset
    15
$plugins->attachHook('search_global_inner', 'nuggie_search($query, $query_phrase, $scores, $page_data, $case_sensitive, $word_list);');
4e7762863437 Added search module
Dan
parents:
diff changeset
    16
4e7762863437 Added search module
Dan
parents:
diff changeset
    17
/**
4e7762863437 Added search module
Dan
parents:
diff changeset
    18
 * Searches the forums for the specified search terms. Called from a hook.
4e7762863437 Added search module
Dan
parents:
diff changeset
    19
 * @access private
4e7762863437 Added search module
Dan
parents:
diff changeset
    20
 */
4e7762863437 Added search module
Dan
parents:
diff changeset
    21
4e7762863437 Added search module
Dan
parents:
diff changeset
    22
function nuggie_search(&$query, &$query_phrase, &$scores, &$page_data, &$case_sensitive, &$word_list)
4e7762863437 Added search module
Dan
parents:
diff changeset
    23
{
4e7762863437 Added search module
Dan
parents:
diff changeset
    24
  global $db, $session, $paths, $template, $plugins; // Common objects
4e7762863437 Added search module
Dan
parents:
diff changeset
    25
  
4e7762863437 Added search module
Dan
parents:
diff changeset
    26
  require_once( ENANO_ROOT . '/plugins/nuggie/postbit.php' );
4e7762863437 Added search module
Dan
parents:
diff changeset
    27
  
4e7762863437 Added search module
Dan
parents:
diff changeset
    28
  // Based on the search functions from Snapr and Decir
4e7762863437 Added search module
Dan
parents:
diff changeset
    29
  
4e7762863437 Added search module
Dan
parents:
diff changeset
    30
  // Let's do this all in one query
4e7762863437 Added search module
Dan
parents:
diff changeset
    31
  $terms = array(
4e7762863437 Added search module
Dan
parents:
diff changeset
    32
      'any' => array_merge($query['any'], $query_phrase['any']),
4e7762863437 Added search module
Dan
parents:
diff changeset
    33
      'req' => array_merge($query['req'], $query_phrase['req']),
4e7762863437 Added search module
Dan
parents:
diff changeset
    34
      'not' => $query['not']
4e7762863437 Added search module
Dan
parents:
diff changeset
    35
    );
4e7762863437 Added search module
Dan
parents:
diff changeset
    36
  $where = array('any' => array(), 'req' => array(), 'not' => array());
4e7762863437 Added search module
Dan
parents:
diff changeset
    37
  $where_any =& $where['any'];
4e7762863437 Added search module
Dan
parents:
diff changeset
    38
  $where_req =& $where['req'];
4e7762863437 Added search module
Dan
parents:
diff changeset
    39
  $where_not =& $where['not'];
4e7762863437 Added search module
Dan
parents:
diff changeset
    40
  $title_col = ( $case_sensitive ) ? 'p.post_title' : 'lcase(p.post_title)';
4e7762863437 Added search module
Dan
parents:
diff changeset
    41
  $desc_col = ( $case_sensitive ) ? 'p.post_text' : 'lcase(p.post_text)';
4e7762863437 Added search module
Dan
parents:
diff changeset
    42
  foreach ( $terms['any'] as $term )
4e7762863437 Added search module
Dan
parents:
diff changeset
    43
  {
4e7762863437 Added search module
Dan
parents:
diff changeset
    44
    $term = escape_string_like($term);
4e7762863437 Added search module
Dan
parents:
diff changeset
    45
    if ( !$case_sensitive )
4e7762863437 Added search module
Dan
parents:
diff changeset
    46
      $term = strtolower($term);
4e7762863437 Added search module
Dan
parents:
diff changeset
    47
    $where_any[] = "( $title_col LIKE '%{$term}%' OR $desc_col LIKE '%{$term}%' )";
4e7762863437 Added search module
Dan
parents:
diff changeset
    48
  }
4e7762863437 Added search module
Dan
parents:
diff changeset
    49
  foreach ( $terms['req'] as $term )
4e7762863437 Added search module
Dan
parents:
diff changeset
    50
  {
4e7762863437 Added search module
Dan
parents:
diff changeset
    51
    $term = escape_string_like($term);
4e7762863437 Added search module
Dan
parents:
diff changeset
    52
    if ( !$case_sensitive )
4e7762863437 Added search module
Dan
parents:
diff changeset
    53
      $term = strtolower($term);
4e7762863437 Added search module
Dan
parents:
diff changeset
    54
    $where_req[] = "( $title_col LIKE '%{$term}%' OR $desc_col LIKE '%{$term}%' )";
4e7762863437 Added search module
Dan
parents:
diff changeset
    55
  }
4e7762863437 Added search module
Dan
parents:
diff changeset
    56
  foreach ( $terms['not'] as $term )
4e7762863437 Added search module
Dan
parents:
diff changeset
    57
  {
4e7762863437 Added search module
Dan
parents:
diff changeset
    58
    $term = escape_string_like($term);
4e7762863437 Added search module
Dan
parents:
diff changeset
    59
    if ( !$case_sensitive )
4e7762863437 Added search module
Dan
parents:
diff changeset
    60
      $term = strtolower($term);
4e7762863437 Added search module
Dan
parents:
diff changeset
    61
    $where_not[] = "$title_col NOT LIKE '%{$term}%' AND $desc_col NOT LIKE '%{$term}%'";
4e7762863437 Added search module
Dan
parents:
diff changeset
    62
  }
4e7762863437 Added search module
Dan
parents:
diff changeset
    63
  if ( empty($where_any) )
4e7762863437 Added search module
Dan
parents:
diff changeset
    64
    unset($where_any, $where['any']);
4e7762863437 Added search module
Dan
parents:
diff changeset
    65
  if ( empty($where_req) )
4e7762863437 Added search module
Dan
parents:
diff changeset
    66
    unset($where_req, $where['req']);
4e7762863437 Added search module
Dan
parents:
diff changeset
    67
  if ( empty($where_not) )
4e7762863437 Added search module
Dan
parents:
diff changeset
    68
    unset($where_not, $where['not']);
4e7762863437 Added search module
Dan
parents:
diff changeset
    69
  
4e7762863437 Added search module
Dan
parents:
diff changeset
    70
  $where_any = '(' . implode(' OR ', $where_any) . '' . ( isset($where['req']) || isset($where['not']) ? ' OR 1 = 1' : '' ) . ')';
4e7762863437 Added search module
Dan
parents:
diff changeset
    71
  
4e7762863437 Added search module
Dan
parents:
diff changeset
    72
  if ( isset($where_req) )
4e7762863437 Added search module
Dan
parents:
diff changeset
    73
    $where_req = implode(' AND ', $where_req);
4e7762863437 Added search module
Dan
parents:
diff changeset
    74
  if ( isset($where_not) )
4e7762863437 Added search module
Dan
parents:
diff changeset
    75
  $where_not = implode( 'AND ', $where_not);
4e7762863437 Added search module
Dan
parents:
diff changeset
    76
  
4e7762863437 Added search module
Dan
parents:
diff changeset
    77
  $where = implode(' AND ', $where);
4e7762863437 Added search module
Dan
parents:
diff changeset
    78
  $sql = "SELECT p.post_id, p.post_title, p.post_title_clean, p.post_text, p.post_author, u.username, p.post_timestamp, u.username\n"
4e7762863437 Added search module
Dan
parents:
diff changeset
    79
         . "    FROM " . table_prefix . "blog_posts AS p\n"
4e7762863437 Added search module
Dan
parents:
diff changeset
    80
         . "  LEFT JOIN " . table_prefix . "users AS u\n"
4e7762863437 Added search module
Dan
parents:
diff changeset
    81
         . "    ON ( u.user_id = p.post_author )\n"
4e7762863437 Added search module
Dan
parents:
diff changeset
    82
         . "  WHERE ( $where )\n"
4e7762863437 Added search module
Dan
parents:
diff changeset
    83
         . "  GROUP BY p.post_id;";
4e7762863437 Added search module
Dan
parents:
diff changeset
    84
  
4e7762863437 Added search module
Dan
parents:
diff changeset
    85
  if ( !($q = $db->sql_unbuffered_query($sql)) )
4e7762863437 Added search module
Dan
parents:
diff changeset
    86
  {
4e7762863437 Added search module
Dan
parents:
diff changeset
    87
    $db->_die('Error is in auto-generated SQL query in the Nuggie plugin search module');
4e7762863437 Added search module
Dan
parents:
diff changeset
    88
  }
4e7762863437 Added search module
Dan
parents:
diff changeset
    89
  
4e7762863437 Added search module
Dan
parents:
diff changeset
    90
  if ( $row = $db->fetchrow() )
4e7762863437 Added search module
Dan
parents:
diff changeset
    91
  {
4e7762863437 Added search module
Dan
parents:
diff changeset
    92
    do
4e7762863437 Added search module
Dan
parents:
diff changeset
    93
    {
4e7762863437 Added search module
Dan
parents:
diff changeset
    94
      $day = enano_date('d', $row['post_timestamp']);
4e7762863437 Added search module
Dan
parents:
diff changeset
    95
      $year = enano_date('Y', $row['post_timestamp']);
4e7762863437 Added search module
Dan
parents:
diff changeset
    96
      $month = enano_date('n', $row['post_timestamp']);
4e7762863437 Added search module
Dan
parents:
diff changeset
    97
      $username = sanitize_page_id($row['username']);
4e7762863437 Added search module
Dan
parents:
diff changeset
    98
      $post_url = "{$username}/$year/$month/$day/{$row['post_title_clean']}";
4e7762863437 Added search module
Dan
parents:
diff changeset
    99
      $idstring = "ns=Blog;pid=$post_url";
4e7762863437 Added search module
Dan
parents:
diff changeset
   100
      foreach ( $word_list as $term )
4e7762863437 Added search module
Dan
parents:
diff changeset
   101
      {
4e7762863437 Added search module
Dan
parents:
diff changeset
   102
        $func = ( $case_sensitive ) ? 'strstr' : 'stristr';
4e7762863437 Added search module
Dan
parents:
diff changeset
   103
        $inc = ( $func($row['post_title'], $term) ? 1.5 : ( $func($row['post_text'], $term) ? 1 : 0 ) );
4e7762863437 Added search module
Dan
parents:
diff changeset
   104
        ( isset($scores[$idstring]) ) ? $scores[$idstring] = $scores[$idstring] + $inc : $scores[$idstring] = $inc;
4e7762863437 Added search module
Dan
parents:
diff changeset
   105
      }
4e7762863437 Added search module
Dan
parents:
diff changeset
   106
      // Generate text...
4e7762863437 Added search module
Dan
parents:
diff changeset
   107
      $post_text = highlight_and_clip_search_result($row['post_text'], $word_list);
4e7762863437 Added search module
Dan
parents:
diff changeset
   108
      $post_length = strlen($post_text);
4e7762863437 Added search module
Dan
parents:
diff changeset
   109
      
4e7762863437 Added search module
Dan
parents:
diff changeset
   110
      // Inject result
4e7762863437 Added search module
Dan
parents:
diff changeset
   111
      
4e7762863437 Added search module
Dan
parents:
diff changeset
   112
      if ( isset($scores[$idstring]) )
4e7762863437 Added search module
Dan
parents:
diff changeset
   113
      {
4e7762863437 Added search module
Dan
parents:
diff changeset
   114
        // echo('adding image "' . $row['img_title'] . '" to results<br />');
4e7762863437 Added search module
Dan
parents:
diff changeset
   115
        $page_data[$idstring] = array(
4e7762863437 Added search module
Dan
parents:
diff changeset
   116
          'page_name' => highlight_search_result(htmlspecialchars($row['post_title']), $word_list),
4e7762863437 Added search module
Dan
parents:
diff changeset
   117
          'page_text' => $post_text,
4e7762863437 Added search module
Dan
parents:
diff changeset
   118
          'score' => $scores[$idstring],
4e7762863437 Added search module
Dan
parents:
diff changeset
   119
          'page_note' => '[Blog post]',
4e7762863437 Added search module
Dan
parents:
diff changeset
   120
          'page_id' => $post_url,
4e7762863437 Added search module
Dan
parents:
diff changeset
   121
          'namespace' => 'Blog',
4e7762863437 Added search module
Dan
parents:
diff changeset
   122
          'page_length' => $post_length,
4e7762863437 Added search module
Dan
parents:
diff changeset
   123
        );
4e7762863437 Added search module
Dan
parents:
diff changeset
   124
      }
4e7762863437 Added search module
Dan
parents:
diff changeset
   125
    }
4e7762863437 Added search module
Dan
parents:
diff changeset
   126
    while ( $row = $db->fetchrow() );
4e7762863437 Added search module
Dan
parents:
diff changeset
   127
  }
4e7762863437 Added search module
Dan
parents:
diff changeset
   128
}