Ignore:
Timestamp:
Nov 14, 2011, 11:17:15 PM (13 years ago)
Author:
dj3c1t
Message:

passage a Fluxbb 1.4.7

File:
1 edited

Legend:

Unmodified
Added
Removed

  • branches/rsr.v5.1.dev/web/punbb/include/dblayer/sqlite.php

    r1 r3  
    11<?php
    2 /***********************************************************************
    3 
    4   Copyright (C) 2002-2005  Rickard Andersson (rickard@punbb.org)
    5 
    6   This file is part of PunBB.
    7 
    8   PunBB is free software; you can redistribute it and/or modify it
    9   under the terms of the GNU General Public License as published
    10   by the Free Software Foundation; either version 2 of the License,
    11   or (at your option) any later version.
    12 
    13   PunBB is distributed in the hope that it will be useful, but
    14   WITHOUT ANY WARRANTY; without even the implied warranty of
    15   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    16   GNU General Public License for more details.
    17 
    18   You should have received a copy of the GNU General Public License
    19   along with this program; if not, write to the Free Software
    20   Foundation, Inc., 59 Temple Place, Suite 330, Boston,
    21   MA  02111-1307  USA
    22 
    23 ************************************************************************/
    24 
     2
     3/**
     4 * Copyright (C) 2008-2011 FluxBB
     5 * based on code by Rickard Andersson copyright (C) 2002-2008 PunBB
     6 * License: http://www.gnu.org/licenses/gpl.html GPL version 2 or higher
     7 */
    258
    269// Make sure we have built in support for SQLite
    … …  
    4225        var $error_msg = 'Unknown';
    4326
     27        var $datatype_transformations = array(
     28                '%^SERIAL$%'                                                                                                                    =>      'INTEGER',
     29                '%^(TINY|SMALL|MEDIUM|BIG)?INT( )?(\\([0-9]+\\))?( )?(UNSIGNED)?$%i'    =>      'INTEGER',
     30                '%^(TINY|MEDIUM|LONG)?TEXT$%i'                                                                                  =>      'TEXT'
     31        );
     32
    4433
    4534        function DBLayer($db_host, $db_username, $db_password, $db_name, $db_prefix, $p_connect)
    … …  
    135124
    136125
    137         function result($query_id = 0, $row = 0)
     126        function result($query_id = 0, $row = 0, $col = 0)
    138127        {
    139128                if ($query_id)
    140129                {
    141                         if ($row != 0)
    142                                 @sqlite_seek($query_id, $row);
    143 
    144                         return @current(@sqlite_current($query_id));
     130                        if ($row !== 0 && @sqlite_seek($query_id, $row) === false)
     131                                return false;
     132
     133                        $cur_row = @sqlite_current($query_id);
     134                        if ($cur_row === false)
     135                                return false;
     136
     137                        return $cur_row[$col];
    145138                }
    146139                else
    … …  
    157150                        {
    158151                                // Horrible hack to get rid of table names and table aliases from the array keys
    159                                 while (list($key, $value) = @each($cur_row))
     152                                foreach ($cur_row as $key => $value)
    160153                                {
    161                                     $dot_spot = strpos($key, '.');
    162                                     if ($dot_spot !== false)
    163                                     {
    164                                         unset($cur_row[$key]);
    165                                         $key = substr($key, $dot_spot+1);
    166                                         $cur_row[$key] = $value;
    167                                     }
     154                                        $dot_spot = strpos($key, '.');
     155                                        if ($dot_spot !== false)
     156                                        {
     157                                                unset($cur_row[$key]);
     158                                                $key = substr($key, $dot_spot+1);
     159                                                $cur_row[$key] = $value;
     160                                        }
    168161                                }
    169162                        }
    … …  
    190183        function affected_rows()
    191184        {
    192                 return ($this->query_result) ? @sqlite_changes($this->query_result) : false;
     185                return ($this->link_id) ? @sqlite_changes($this->link_id) : false;
    193186        }
    194187
    … …  
    220213        function escape($str)
    221214        {
    222                 return sqlite_escape_string($str);
     215                return is_array($str) ? '' : sqlite_escape_string($str);
    223216        }
    224217
    … …  
    251244                        return false;
    252245        }
     246
     247
     248        function get_names()
     249        {
     250                return '';
     251        }
     252
     253
     254        function set_names($names)
     255        {
     256                return true;
     257        }
     258
     259
     260        function get_version()
     261        {
     262                return array(
     263                        'name'          => 'SQLite',
     264                        'version'       => sqlite_libversion()
     265                );
     266        }
     267
     268
     269        function table_exists($table_name, $no_prefix = false)
     270        {
     271                $result = $this->query('SELECT 1 FROM sqlite_master WHERE name = \''.($no_prefix ? '' : $this->prefix).$this->escape($table_name).'\' AND type=\'table\'');
     272                return $this->num_rows($result) > 0;
     273        }
     274
     275
     276        function field_exists($table_name, $field_name, $no_prefix = false)
     277        {
     278                $result = $this->query('SELECT sql FROM sqlite_master WHERE name = \''.($no_prefix ? '' : $this->prefix).$this->escape($table_name).'\' AND type=\'table\'');
     279                if (!$this->num_rows($result))
     280                        return false;
     281
     282                return preg_match('%[\r\n]'.preg_quote($field_name, '%').' %', $this->result($result));
     283        }
     284
     285
     286        function index_exists($table_name, $index_name, $no_prefix = false)
     287        {
     288                $result = $this->query('SELECT 1 FROM sqlite_master WHERE tbl_name = \''.($no_prefix ? '' : $this->prefix).$this->escape($table_name).'\' AND name = \''.($no_prefix ? '' : $this->prefix).$this->escape($table_name).'_'.$this->escape($index_name).'\' AND type=\'index\'');
     289                return $this->num_rows($result) > 0;
     290        }
     291
     292
     293        function create_table($table_name, $schema, $no_prefix = false)
     294        {
     295                if ($this->table_exists($table_name, $no_prefix))
     296                        return true;
     297
     298                $query = 'CREATE TABLE '.($no_prefix ? '' : $this->prefix).$table_name." (\n";
     299
     300                // Go through every schema element and add it to the query
     301                foreach ($schema['FIELDS'] as $field_name => $field_data)
     302                {
     303                        $field_data['datatype'] = preg_replace(array_keys($this->datatype_transformations), array_values($this->datatype_transformations), $field_data['datatype']);
     304
     305                        $query .= $field_name.' '.$field_data['datatype'];
     306
     307                        if (!$field_data['allow_null'])
     308                                $query .= ' NOT NULL';
     309
     310                        if (isset($field_data['default']))
     311                                $query .= ' DEFAULT '.$field_data['default'];
     312
     313                        $query .= ",\n";
     314                }
     315
     316                // If we have a primary key, add it
     317                if (isset($schema['PRIMARY KEY']))
     318                        $query .= 'PRIMARY KEY ('.implode(',', $schema['PRIMARY KEY']).'),'."\n";
     319
     320                // Add unique keys
     321                if (isset($schema['UNIQUE KEYS']))
     322                {
     323                        foreach ($schema['UNIQUE KEYS'] as $key_name => $key_fields)
     324                                $query .= 'UNIQUE ('.implode(',', $key_fields).'),'."\n";
     325                }
     326
     327                // We remove the last two characters (a newline and a comma) and add on the ending
     328                $query = substr($query, 0, strlen($query) - 2)."\n".')';
     329
     330                $result = $this->query($query) ? true : false;
     331
     332                // Add indexes
     333                if (isset($schema['INDEXES']))
     334                {
     335                        foreach ($schema['INDEXES'] as $index_name => $index_fields)
     336                                $result &= $this->add_index($table_name, $index_name, $index_fields, false, $no_prefix);
     337                }
     338
     339                return $result;
     340        }
     341
     342
     343        function drop_table($table_name, $no_prefix = false)
     344        {
     345                if (!$this->table_exists($table_name, $no_prefix))
     346                        return true;
     347
     348                return $this->query('DROP TABLE '.($no_prefix ? '' : $this->prefix).$table_name) ? true : false;
     349        }
     350
     351
     352        function rename_table($old_name, $new_name, $no_prefix = false)
     353        {
     354                // If there new table exists and the old one doesn't, then we're happy
     355                if ($this->table_exists($new_table, $no_prefix) && !$this->table_exists($old_table, $no_prefix))
     356                        return true;
     357
     358                $table = $this->get_table_info($old_name, $no_prefix);
     359
     360                // Create new table
     361                $newtable = str_replace('CREATE TABLE '.($no_prefix ? '' : $this->prefix).$this->escape($old_name).' (', 'CREATE TABLE '.($no_prefix ? '' : $this->prefix).$this->escape($new_name).' (', $table['sql']);
     362                $result = $this->query($newtable) ? true : false;
     363
     364                // Recreate indexes
     365                if (!empty($table['indices']))
     366                {
     367                        foreach ($table['indices'] as $cur_index)
     368                                $result &= $this->query($cur_index) ? true : false;
     369                }
     370
     371                // Copy content across
     372                $result &= $this->query('INSERT INTO '.($no_prefix ? '' : $this->prefix).$this->escape($new_name).' SELECT * FROM '.($no_prefix ? '' : $this->prefix).$this->escape($old_name)) ? true : false;
     373
     374                // Drop old table
     375                $result &= $this->drop_table(($no_prefix ? '' : $this->prefix).$this->escape($table_name));
     376
     377                return $result;
     378        }
     379
     380
     381        function get_table_info($table_name, $no_prefix = false)
     382        {
     383                // Grab table info
     384                $result = $this->query('SELECT sql FROM sqlite_master WHERE tbl_name = \''.($no_prefix ? '' : $this->prefix).$this->escape($table_name).'\' ORDER BY type DESC') or error('Unable to fetch table information', __FILE__, __LINE__, $this->error());
     385                $num_rows = $this->num_rows($result);
     386
     387                if ($num_rows == 0)
     388                        return;
     389
     390                $table = array();
     391                $table['indices'] = array();
     392                while ($cur_index = $this->fetch_assoc($result))
     393                {
     394                        if (empty($cur_index['sql']))
     395                                continue;
     396
     397                        if (!isset($table['sql']))
     398                                $table['sql'] = $cur_index['sql'];
     399                        else
     400                                $table['indices'][] = $cur_index['sql'];
     401                }
     402
     403                // Work out the columns in the table currently
     404                $table_lines = explode("\n", $table['sql']);
     405                $table['columns'] = array();
     406                foreach ($table_lines as $table_line)
     407                {
     408                        $table_line = trim($table_line, " \t\n\r,"); // trim spaces, tabs, newlines, and commas
     409                        if (substr($table_line, 0, 12) == 'CREATE TABLE')
     410                                continue;
     411                        else if (substr($table_line, 0, 11) == 'PRIMARY KEY')
     412                                $table['primary_key'] = $table_line;
     413                        else if (substr($table_line, 0, 6) == 'UNIQUE')
     414                                $table['unique'] = $table_line;
     415                        else if (substr($table_line, 0, strpos($table_line, ' ')) != '')
     416                                $table['columns'][substr($table_line, 0, strpos($table_line, ' '))] = trim(substr($table_line, strpos($table_line, ' ')));
     417                }
     418
     419                return $table;
     420        }
     421
     422
     423        function add_field($table_name, $field_name, $field_type, $allow_null, $default_value = null, $after_field = 0, $no_prefix = false)
     424        {
     425                if ($this->field_exists($table_name, $field_name, $no_prefix))
     426                        return true;
     427
     428                $table = $this->get_table_info($table_name, $no_prefix);
     429
     430                // Create temp table
     431                $now = time();
     432                $tmptable = str_replace('CREATE TABLE '.($no_prefix ? '' : $this->prefix).$this->escape($table_name).' (', 'CREATE TABLE '.($no_prefix ? '' : $this->prefix).$this->escape($table_name).'_t'.$now.' (', $table['sql']);
     433                $result = $this->query($tmptable) ? true : false;
     434                $result &= $this->query('INSERT INTO '.($no_prefix ? '' : $this->prefix).$this->escape($table_name).'_t'.$now.' SELECT * FROM '.($no_prefix ? '' : $this->prefix).$this->escape($table_name)) ? true : false;
     435
     436                // Create new table sql
     437                $field_type = preg_replace(array_keys($this->datatype_transformations), array_values($this->datatype_transformations), $field_type);
     438                $query = $field_type;
     439                if (!$allow_null)
     440                        $query .= ' NOT NULL';
     441                if ($default_value === null || $default_value === '')
     442                        $default_value = '\'\'';
     443
     444                $query .= ' DEFAULT '.$default_value;
     445
     446                $old_columns = array_keys($table['columns']);
     447                array_insert($table['columns'], $after_field, $query, $field_name);
     448
     449                $new_table = 'CREATE TABLE '.($no_prefix ? '' : $this->prefix).$this->escape($table_name).' (';
     450
     451                foreach ($table['columns'] as $cur_column => $column_details)
     452                        $new_table .= "\n".$cur_column.' '.$column_details.',';
     453
     454                if (isset($table['unique']))
     455                        $new_table .= "\n".$table['unique'].',';
     456
     457                if (isset($table['primary_key']))
     458                        $new_table .= "\n".$table['primary_key'].',';
     459
     460                $new_table = trim($new_table, ',')."\n".');';
     461
     462                // Drop old table
     463                $result &= $this->drop_table(($no_prefix ? '' : $this->prefix).$this->escape($table_name));
     464
     465                // Create new table
     466                $result &= $this->query($new_table) ? true : false;
     467
     468                // Recreate indexes
     469                if (!empty($table['indices']))
     470                {
     471                        foreach ($table['indices'] as $cur_index)
     472                                $result &= $this->query($cur_index) ? true : false;
     473                }
     474
     475                // Copy content back
     476                $result &= $this->query('INSERT INTO '.($no_prefix ? '' : $this->prefix).$this->escape($table_name).' ('.implode(', ', $old_columns).') SELECT * FROM '.($no_prefix ? '' : $this->prefix).$this->escape($table_name).'_t'.$now) ? true : false;
     477
     478                // Drop temp table
     479                $result &= $this->drop_table(($no_prefix ? '' : $this->prefix).$this->escape($table_name).'_t'.$now);
     480
     481                return $result;
     482        }
     483
     484
     485        function alter_field($table_name, $field_name, $field_type, $allow_null, $default_value = null, $after_field = 0, $no_prefix = false)
     486        {
     487                // Unneeded for SQLite
     488                return true;
     489        }
     490
     491
     492        function drop_field($table_name, $field_name, $no_prefix = false)
     493        {
     494                if (!$this->field_exists($table_name, $field_name, $no_prefix))
     495                        return true;
     496
     497                $table = $this->get_table_info($table_name, $no_prefix);
     498
     499                // Create temp table
     500                $now = time();
     501                $tmptable = str_replace('CREATE TABLE '.($no_prefix ? '' : $this->prefix).$this->escape($table_name).' (', 'CREATE TABLE '.($no_prefix ? '' : $this->prefix).$this->escape($table_name).'_t'.$now.' (', $table['sql']);
     502                $result = $this->query($tmptable) ? true : false;
     503                $result &= $this->query('INSERT INTO '.($no_prefix ? '' : $this->prefix).$this->escape($table_name).'_t'.$now.' SELECT * FROM '.($no_prefix ? '' : $this->prefix).$this->escape($table_name)) ? true : false;
     504
     505                // Work out the columns we need to keep and the sql for the new table
     506                unset($table['columns'][$field_name]);
     507                $new_columns = array_keys($table['columns']);
     508
     509                $new_table = 'CREATE TABLE '.($no_prefix ? '' : $this->prefix).$this->escape($table_name).' (';
     510
     511                foreach ($table['columns'] as $cur_column => $column_details)
     512                        $new_table .= "\n".$cur_column.' '.$column_details.',';
     513
     514                if (isset($table['unique']))
     515                        $new_table .= "\n".$table['unique'].',';
     516
     517                if (isset($table['primary_key']))
     518                        $new_table .= "\n".$table['primary_key'].',';
     519
     520                $new_table = trim($new_table, ',')."\n".');';
     521
     522                // Drop old table
     523                $result &= $this->drop_table(($no_prefix ? '' : $this->prefix).$this->escape($table_name));
     524
     525                // Create new table
     526                $result &= $this->query($new_table) ? true : false;
     527
     528                // Recreate indexes
     529                if (!empty($table['indices']))
     530                {
     531                        foreach ($table['indices'] as $cur_index)
     532                                if (!preg_match('%\('.preg_quote($field_name, '%').'\)%', $cur_index))
     533                                        $result &= $this->query($cur_index) ? true : false;
     534                }
     535
     536                // Copy content back
     537                $result &= $this->query('INSERT INTO '.($no_prefix ? '' : $this->prefix).$this->escape($table_name).' SELECT '.implode(', ', $new_columns).' FROM '.($no_prefix ? '' : $this->prefix).$this->escape($table_name).'_t'.$now) ? true : false;
     538
     539                // Drop temp table
     540                $result &= $this->drop_table(($no_prefix ? '' : $this->prefix).$this->escape($table_name).'_t'.$now);
     541
     542                return $result;
     543        }
     544
     545
     546        function add_index($table_name, $index_name, $index_fields, $unique = false, $no_prefix = false)
     547        {
     548                if ($this->index_exists($table_name, $index_name, $no_prefix))
     549                        return true;
     550
     551                return $this->query('CREATE '.($unique ? 'UNIQUE ' : '').'INDEX '.($no_prefix ? '' : $this->prefix).$table_name.'_'.$index_name.' ON '.($no_prefix ? '' : $this->prefix).$table_name.'('.implode(',', $index_fields).')') ? true : false;
     552        }
     553
     554
     555        function drop_index($table_name, $index_name, $no_prefix = false)
     556        {
     557                if (!$this->index_exists($table_name, $index_name, $no_prefix))
     558                        return true;
     559
     560                return $this->query('DROP INDEX '.($no_prefix ? '' : $this->prefix).$table_name.'_'.$index_name) ? true : false;
     561        }
     562
     563        function truncate_table($table_name, $no_prefix = false)
     564        {
     565                return $this->query('DELETE FROM '.($no_prefix ? '' : $this->prefix).$table_name) ? true : false;
     566        }
    253567}
Note: See TracChangeset for help on using the changeset viewer.