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