3/28/2020, 10:36HEADmaster

1 files changed, 64 insertions, 0 deletions

diff --git a/includes/class-mysql.php b/includes/class-mysql.php
new file mode 100644
index 0000000..65e7acd
--- /dev/null
+++ b/includes/class-mysql.php
@@ -0,0 +1,64 @@
+<?php
+
+/**
+ * Connect to DB
+ *
+ * @since 1.0
+ */
+function yourls_db_connect() {
+	global $ydb;
+
+	if (   !defined( 'YOURLS_DB_USER' )
+		or !defined( 'YOURLS_DB_PASS' )
+		or !defined( 'YOURLS_DB_NAME' )
+		or !defined( 'YOURLS_DB_HOST' )
+	) yourls_die ( yourls__( 'Incorrect DB config, or could not connect to DB' ), yourls__( 'Fatal error' ), 503 );
+
+    $dbhost = YOURLS_DB_HOST;
+    $user   = YOURLS_DB_USER;
+    $pass   = YOURLS_DB_PASS;
+    $dbname = YOURLS_DB_NAME;
+
+    // This action is deprecated
+    yourls_do_action( 'set_DB_driver', 'deprecated' );
+
+    // Get custom port if any
+    if ( false !== strpos( $dbhost, ':' ) ) {
+        list( $dbhost, $dbport ) = explode( ':', $dbhost );
+        $dbhost = sprintf( '%1$s;port=%2$d', $dbhost, $dbport );
+    }
+
+    $charset = yourls_apply_filter( 'db_connect_charset', 'utf8' );
+
+    /**
+     * Data Source Name (dsn) used to connect the DB
+     *
+     * DSN with PDO is something like:
+     * 'mysql:host=123.4.5.6;dbname=test_db;port=3306'
+     * 'sqlite:/opt/databases/mydb.sq3'
+     * 'pgsql:host=192.168.13.37;port=5432;dbname=omgwtf'
+     */
+    $dsn = sprintf( 'mysql:host=%s;dbname=%s;charset=%s', $dbhost, $dbname, $charset );
+    $dsn = yourls_apply_filter( 'db_connect_custom_dsn', $dsn );
+
+    /**
+     * PDO driver options and attributes
+
+     * The PDO constructor is something like:
+     *   new PDO( string $dsn, string $username, string $password [, array $options ] )
+     * The driver options are passed to the PDO constructor, eg array(PDO::ATTR_ERRMODE => PDO::ERRMODE_EXCEPTION)
+     * The attribute options are then set in a foreach($attr as $k=>$v){$db->setAttribute($k, $v)} loop
+     */
+    $driver_options = yourls_apply_filter( 'db_connect_driver_option', array() ); // driver options as key-value pairs
+    $attributes     = yourls_apply_filter( 'db_connect_attributes',    array() ); // attributes as key-value pairs
+
+    $ydb = new \YOURLS\Database\YDB( $dsn, $user, $pass, $driver_options, $attributes );
+    $ydb->init();
+
+    // Past this point, we're connected
+    yourls_debug_log(sprintf('Connected to database %s on %s ', $dbname, $dbhost));
+
+    yourls_debug_mode(YOURLS_DEBUG);
+
+	return $ydb;
+}