includes/class-mysql.php

   1 <?php
   2
   3 /**
   4  * Pick the right DB class and return an instance
   5  *
   6  * @since 1.7
   7  * @param string $extension Optional: user defined choice
   8  * @return class $ydb DB class instance
   9  */
  10 function yourls_set_DB_driver( ) {
  11
  12         // Auto-pick the driver. Priority: user defined, then PDO, then mysqli, then mysql
  13         if ( defined( 'YOURLS_DB_DRIVER' ) ) {
  14                 $driver = strtolower( YOURLS_DB_DRIVER ); // accept 'MySQL', 'mySQL', etc
  15         } elseif ( extension_loaded( 'pdo_mysql' ) ) {
  16                 $driver = 'pdo';
  17         } elseif ( extension_loaded( 'mysqli' ) ) {
  18                 $driver = 'mysqli';
  19         } elseif ( extension_loaded( 'mysql' ) ) {
  20                 $driver = 'mysql';
  21         } else {
  22                 $driver = '';
  23         }
  24
  25         // Set the new driver
  26         if ( in_array( $driver, array( 'mysql', 'mysqli', 'pdo' ) ) ) {
  27                 require_once( YOURLS_INC . '/ezSQL/ez_sql_core.php' );
  28                 require_once( YOURLS_INC . '/ezSQL/ez_sql_core_yourls.php' );
  29                 require_once( YOURLS_INC . '/ezSQL/ez_sql_' . $driver . '.php' );
  30                 require_once( YOURLS_INC . '/ezSQL/ez_sql_' . $driver . '_yourls.php' );
  31         }
  32         $class = 'ezSQL_' . $driver . '_yourls';
  33
  34         global $ydb;
  35
  36         if ( !class_exists( $class, false ) ) {
  37                 $ydb = new stdClass();
  38                 yourls_die(
  39                         yourls__( 'YOURLS requires the mysql, mysqli or pdo_mysql PHP extension. No extension found. Check your server config, or contact your host.' ),
  40                         yourls__( 'Fatal error' ),
  41                         503
  42                 );
  43         }
  44
  45         yourls_do_action( 'set_DB_driver', $driver );
  46
  47         $ydb = new $class( YOURLS_DB_USER, YOURLS_DB_PASS, YOURLS_DB_NAME, YOURLS_DB_HOST );
  48
  49         yourls_debug_log( "DB driver: $driver" );
  50 }
  51
  52 /**
  53  * Connect to DB
  54  *
  55  * @since 1.0
  56  */
  57 function yourls_db_connect() {
  58         global $ydb;
  59
  60         if (   !defined( 'YOURLS_DB_USER' )
  61                 or !defined( 'YOURLS_DB_PASS' )
  62                 or !defined( 'YOURLS_DB_NAME' )
  63                 or !defined( 'YOURLS_DB_HOST' )
  64         ) yourls_die ( yourls__( 'Incorrect DB config, or could not connect to DB' ), yourls__( 'Fatal error' ), 503 );
  65
  66         // Are we standalone or in the WordPress environment?
  67         if ( class_exists( 'wpdb', false ) ) {
  68                 /* TODO: should we deprecate this? Follow WP dev in that area */
  69                 $ydb =  new wpdb( YOURLS_DB_USER, YOURLS_DB_PASS, YOURLS_DB_NAME, YOURLS_DB_HOST );
  70         } else {
  71                 yourls_set_DB_driver();
  72         }
  73
  74         // Check if connection attempt raised an error. It seems that only PDO does, though.
  75         if ( $ydb->last_error )
  76                 yourls_die( $ydb->last_error, yourls__( 'Fatal error' ), 503 );
  77
  78
  79         return $ydb;
  80 }
  81
  82