Allow using the new driver for initial database creation

Author: JanJakes

wp-includes/sqlite/install-functions.php

@@ -18,6 +18,8 @@
  * @throws PDOException If the database connection fails.
  */
 function sqlite_make_db_sqlite() {
+	global $wpdb;
+
 	include_once ABSPATH . 'wp-admin/includes/schema.php';
 
 	$table_schemas = wp_get_db_schema();
@@ -31,8 +33,17 @@ function sqlite_make_db_sqlite() {
 		wp_die( $message, 'Database Error!' );
 	}
 
-	$translator = new WP_SQLite_Translator( $pdo );
-	$query      = null;
+	if ( defined( 'WP_SQLITE_AST_DRIVER' ) && WP_SQLITE_AST_DRIVER ) {
+		$translator = new WP_SQLite_Driver(
+			array(
+				'database'   => $wpdb->dbname,
+				'connection' => $pdo,
+			)
+		);
+	} else {
+		$translator = new WP_SQLite_Translator( $pdo );
+	}
+	$query = null;
 
 	try {
 		$translator->begin_transaction();