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 = YOURLS_DB_DRIVER;
  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         }
  22
  23         // Get the new driver
  24         if ( in_array( $driver, array( 'mysql', 'mysqli', 'pdo' ) ) ) {
  25                 require_once( YOURLS_INC . '/ezSQL/ez_sql_core.php' );
  26                 require_once( YOURLS_INC . '/ezSQL/ez_sql_core_yourls.php' );
  27                 require_once( YOURLS_INC . '/ezSQL/ez_sql_' . $driver . '.php' );
  28                 require_once( YOURLS_INC . '/ezSQL/ez_sql_' . $driver . '_yourls.php' );
  29         }
  30         $class = 'ezSQL_' . $driver . '_yourls';
  31
  32         global $ydb;
  33
  34         if ( !class_exists( $class ) ) {
  35                 $ydb = new stdClass();
  36                 yourls_die(
  37                         yourls__( 'YOURLS requires the mysql, mysqli or pdo_mysql PHP extension. No extension found. Check your server config, or contact your host.' ),
  38                         yourls__( 'Fatal error' ),
  39                         503
  40                 );
  41         }
  42
  43         yourls_do_action( 'set_DB_driver', $driver );
  44
  45         $ydb = new $class( YOURLS_DB_USER, YOURLS_DB_PASS, YOURLS_DB_NAME, YOURLS_DB_HOST );
  46         $ydb->debug_log[] = "DB driver: $driver";
  47 }
  48