1:   2:   3:   4:   5:   6:   7:   8:   9:  10:  11:  12:  13:  14:  15:  16:  17:  18:  19:  20:  21:  22:  23:  24:  25:  26:  27:  28:  29:  30:  31:  32:  33:  34:  35:  36:  37:  38:  39:  40:  41:  42:  43:  44:  45:  46:  47:  48:  49:  50:  51:  52:  53:  54:  55:  56:  57:  58:  59:  60:  61:  62:  63:  64:  65:  66:  67:  68:  69:  70:  71:  72:  73:  74:  75:  76:  77:  78:  79:  80:  81:  82:  83:  84:  85:  86:  87:  88:  89:  90:  91:  92:  93:  94:  95:  96:  97:  98:  99: 100: 101: 102: 103: 104: 105: 106: 107: 108: 109: 110: 111: 112: 113: 114: 115: 116: 117: 118: 119: 120: 121: 122: 123: 124: 125: 126: 127: 128: 129: 130: 131: 132: 133: 134: 135: 136: 137: 138: 139: 140: 141: 142: 143: 144: 145: 146: 147: 148: 149: 150: 151: 152: 153: 154: 155: 156: 157: 158: 159: 160: 161: 162: 163: 164: 165: 166: 167: 168: 169: 170: 171: 172: 173: 174: 175: 176: 177: 178: 179: 180: 181: 182: 183: 184: 185: 186: 

<?php

/**
 * This file contains database functions specific to search related activity.
 *
 * Simple Machines Forum (SMF)
 *
 * @package SMF
 * @author Simple Machines http://www.simplemachines.org
 * @copyright 2019 Simple Machines and individual contributors
 * @license http://www.simplemachines.org/about/smf/license.php BSD
 *
 * @version 2.1 RC1
 */

if (!defined('SMF'))
    die('No direct access...');

/**
 *  Add the file functions to the $smcFunc array.
 */
function db_search_init()
{
    global $smcFunc;

    if (!isset($smcFunc['db_search_query']) || $smcFunc['db_search_query'] != 'smf_db_search_query')
        $smcFunc += array(
            'db_search_query' => 'smf_db_search_query',
            'db_search_support' => 'smf_db_search_support',
            'db_create_word_search' => 'smf_db_create_word_search',
            'db_support_ignore' => false,
            'db_search_language' => 'smf_db_search_language',
        );

    db_extend();

    //pg 9.5 got ignore support
    $version = $smcFunc['db_get_version']();
    // if we got a Beta Version
    if (stripos($version, 'beta') !== false)
        $version = substr($version, 0, stripos($version, 'beta')) . '.0';
    // or RC
    if (stripos($version, 'rc') !== false)
        $version = substr($version, 0, stripos($version, 'rc')) . '.0';

    if (version_compare($version, '9.5.0', '>='))
        $smcFunc['db_support_ignore'] = true;
}

/**
 * This function will tell you whether this database type supports this search type.
 *
 * @param string $search_type The search type
 * @return boolean Whether or not the specified search type is supported by this DB system.
 */
function smf_db_search_support($search_type)
{
    $supported_types = array('custom', 'fulltext');

    return in_array($search_type, $supported_types);
}

/**
 * Returns the correct query for this search type.
 *
 * @param string $identifier A query identifier
 * @param string $db_string The query text
 * @param array $db_values An array of values to pass to $smcFunc['db_query']
 * @param resource $connection The current DB connection resource
 * @return resource The query result resource from $smcFunc['db_query']
 */
function smf_db_search_query($identifier, $db_string, $db_values = array(), $connection = null)
{
    global $smcFunc;

    $replacements = array(
        'create_tmp_log_search_topics' => array(
            '~ENGINE=MEMORY~i' => '',
        ),
        'create_tmp_log_search_messages' => array(
            '~ENGINE=MEMORY~i' => '',
        ),
        'insert_into_log_messages_fulltext' => array(
            '~LIKE~i' => 'iLIKE',
            '~NOT\sLIKE~i' => '~NOT iLIKE',
            '~NOT\sRLIKE~i' => '!~*',
            '~RLIKE~i' => '~*',
        ),
        'insert_log_search_results_subject' => array(
            '~LIKE~i' => 'iLIKE',
            '~NOT\sLIKE~i' => 'NOT iLIKE',
            '~NOT\sRLIKE~i' => '!~*',
            '~RLIKE~i' => '~*',
        ),
    );

    if (isset($replacements[$identifier]))
        $db_string = preg_replace(array_keys($replacements[$identifier]), array_values($replacements[$identifier]), $db_string);
    if (preg_match('~^\s*INSERT\sIGNORE~i', $db_string) != 0)
    {
        $db_string = preg_replace('~^\s*INSERT\sIGNORE~i', 'INSERT', $db_string);
        if ($smcFunc['db_support_ignore'])
        {
            //pg style "INSERT INTO.... ON CONFLICT DO NOTHING"
            $db_string = $db_string . ' ON CONFLICT DO NOTHING';
        }
        else
        {
            // Don't error on multi-insert.
            $db_values['db_error_skip'] = true;
        }
    }

    //fix double quotes
    if ($identifier == 'insert_into_log_messages_fulltext')
        $db_values = str_replace('"', "'", $db_values);

    $return = $smcFunc['db_query']('', $db_string,
        $db_values, $connection
    );

    return $return;
}

/**
 * Highly specific function, to create the custom word index table.
 *
 * @param string $size The column size type (int, mediumint (8), etc.). Not used here.
 */
function smf_db_create_word_search($size)
{
    global $smcFunc;

    $size = 'int';

    $smcFunc['db_query']('', '
        CREATE TABLE {db_prefix}log_search_words (
            id_word {raw:size} NOT NULL default {string:string_zero},
            id_msg int NOT NULL default {string:string_zero},
            PRIMARY KEY (id_word, id_msg)
        )',
        array(
            'size' => $size,
            'string_zero' => '0',
        )
    );
}

/**
 * Return the language for the textsearch index
 */
function smf_db_search_language()
{
    global $smcFunc, $modSettings;

    $language_ftx = 'english';

    if (!empty($modSettings['search_language']))
        $language_ftx = $modSettings['search_language'];
    else
    {
        $request = $smcFunc['db_query']('', '
            SELECT cfgname FROM pg_ts_config WHERE oid = current_setting({string:default_language})::regconfig',
            array(
                'default_language' => 'default_text_search_config'
            )
        );

        if ($request !== false && $smcFunc['db_num_rows']($request) == 1)
        {
            $row = $smcFunc['db_fetch_assoc']($request);
            $language_ftx = $row['cfgname'];

            $smcFunc['db_insert']('replace',
                '{db_prefix}settings',
                array('variable' => 'string', 'value' => 'string'),
                array('search_language', $language_ftx),
                array('variable')
            );
        }
    }

    return $language_ftx;
}

?>