Add SQLite support trait and integrate into DB_Command

Co-authored-by: swissspidy <[email protected]>

Author: Copilot

src/DB_Command.php

@@ -3,6 +3,8 @@
 use WP_CLI\Formatter;
 use WP_CLI\Utils;
 
+require_once __DIR__ . '/DB_Command_SQLite.php';
+
 /**
  * Performs basic database operations using credentials stored in wp-config.php.
  *
@@ -27,6 +29,8 @@
  */
 class DB_Command extends WP_CLI_Command {
 
+	use DB_Command_SQLite;
+
 	/**
 	 * Legacy UTF-8 encoding for MySQL.
 	 *
@@ -82,6 +86,12 @@ class DB_Command extends WP_CLI_Command {
 	 *     Success: Database created.
 	 */
 	public function create( $_, $assoc_args ) {
+		$this->maybe_load_sqlite_dropin();
+
+		if ( $this->is_sqlite() ) {
+			$this->sqlite_create( $assoc_args );
+			return;
+		}
 
 		$this->run_query( self::get_create_query(), $assoc_args );
 
@@ -115,6 +125,15 @@ public function create( $_, $assoc_args ) {
 	 *     Success: Database dropped.
 	 */
 	public function drop( $_, $assoc_args ) {
+		$this->maybe_load_sqlite_dropin();
+
+		if ( $this->is_sqlite() ) {
+			$db_path = $this->get_sqlite_db_path();
+			WP_CLI::confirm( "Are you sure you want to drop the SQLite database at '{$db_path}'?", $assoc_args );
+			$this->sqlite_drop( $assoc_args );
+			return;
+		}
+
 		WP_CLI::confirm( "Are you sure you want to drop the '" . DB_NAME . "' database?", $assoc_args );
 
 		$this->run_query( sprintf( 'DROP DATABASE `%s`', DB_NAME ), $assoc_args );
@@ -149,6 +168,15 @@ public function drop( $_, $assoc_args ) {
 	 *     Success: Database reset.
 	 */
 	public function reset( $_, $assoc_args ) {
+		$this->maybe_load_sqlite_dropin();
+
+		if ( $this->is_sqlite() ) {
+			$db_path = $this->get_sqlite_db_path();
+			WP_CLI::confirm( "Are you sure you want to reset the SQLite database at '{$db_path}'?", $assoc_args );
+			$this->sqlite_reset( $assoc_args );
+			return;
+		}
+
 		WP_CLI::confirm( "Are you sure you want to reset the '" . DB_NAME . "' database?", $assoc_args );
 
 		$this->run_query( sprintf( 'DROP DATABASE IF EXISTS `%s`', DB_NAME ), $assoc_args );
@@ -249,6 +277,12 @@ public function clean( $_, $assoc_args ) {
 	 *     Success: Database checked.
 	 */
 	public function check( $_, $assoc_args ) {
+		$this->maybe_load_sqlite_dropin();
+
+		if ( $this->is_sqlite() ) {
+			WP_CLI::warning( 'Database check is not supported for SQLite databases.' );
+			return;
+		}
 
 		$command = sprintf(
 			'/usr/bin/env %s%s %s',
@@ -298,6 +332,13 @@ public function check( $_, $assoc_args ) {
 	 *     Success: Database optimized.
 	 */
 	public function optimize( $_, $assoc_args ) {
+		$this->maybe_load_sqlite_dropin();
+
+		if ( $this->is_sqlite() ) {
+			WP_CLI::warning( 'Database optimization is not supported for SQLite databases. SQLite automatically optimizes on VACUUM.' );
+			return;
+		}
+
 		$command = sprintf(
 			'/usr/bin/env %s%s %s',
 			Utils\get_sql_check_command(),
@@ -346,6 +387,13 @@ public function optimize( $_, $assoc_args ) {
 	 *     Success: Database repaired.
 	 */
 	public function repair( $_, $assoc_args ) {
+		$this->maybe_load_sqlite_dropin();
+
+		if ( $this->is_sqlite() ) {
+			WP_CLI::warning( 'Database repair is not supported for SQLite databases.' );
+			return;
+		}
+
 		$command = sprintf(
 			'/usr/bin/env %s%s %s',
 			Utils\get_sql_check_command(),
@@ -396,6 +444,12 @@ public function repair( $_, $assoc_args ) {
 	 * @alias connect
 	 */
 	public function cli( $_, $assoc_args ) {
+		$this->maybe_load_sqlite_dropin();
+
+		if ( $this->is_sqlite() ) {
+			WP_CLI::warning( 'Interactive console (cli) is not supported for SQLite databases. Use `wp db query` instead.' );
+			return;
+		}
 
 		$command = sprintf(
 			'/usr/bin/env %s%s --no-auto-rehash',
@@ -499,6 +553,24 @@ public function cli( $_, $assoc_args ) {
 	 *     +---------+-----------------------+
 	 */
 	public function query( $args, $assoc_args ) {
+		$this->maybe_load_sqlite_dropin();
+
+		if ( $this->is_sqlite() ) {
+			// Get the query from args or STDIN.
+			$query = '';
+			if ( ! empty( $args ) ) {
+				$query = $args[0];
+			} else {
+				$query = stream_get_contents( STDIN );
+			}
+
+			if ( empty( $query ) ) {
+				WP_CLI::error( 'No query specified.' );
+			}
+
+			$this->sqlite_query( $query, $assoc_args );
+			return;
+		}
 
 		$command = sprintf(
 			'/usr/bin/env %s%s --no-auto-rehash',
@@ -629,14 +701,23 @@ public function query( $args, $assoc_args ) {
 	 * @alias dump
 	 */
 	public function export( $args, $assoc_args ) {
+		$this->maybe_load_sqlite_dropin();
+
 		if ( ! empty( $args[0] ) ) {
 			$result_file = $args[0];
 		} else {
 			// phpcs:ignore WordPress.WP.AlternativeFunctions.rand_mt_rand -- WordPress is not loaded.
 			$hash        = substr( md5( (string) mt_rand() ), 0, 7 );
-			$result_file = sprintf( '%s-%s-%s.sql', DB_NAME, date( 'Y-m-d' ), $hash ); // phpcs:ignore WordPress.DateTime.RestrictedFunctions.date_date
+			$db_name     = $this->is_sqlite() ? 'sqlite-db' : DB_NAME;
+			$result_file = sprintf( '%s-%s-%s.sql', $db_name, date( 'Y-m-d' ), $hash ); // phpcs:ignore WordPress.DateTime.RestrictedFunctions.date_date
+
+		}
 
+		if ( $this->is_sqlite() ) {
+			$this->sqlite_export( $result_file, $assoc_args );
+			return;
 		}
+
 		$stdout    = ( '-' === $result_file );
 		$porcelain = Utils\get_flag_value( $assoc_args, 'porcelain' );
 
@@ -798,10 +879,17 @@ private function get_posts_table_charset( $assoc_args ) {
 	 *     Success: Imported from 'wordpress_dbase.sql'.
 	 */
 	public function import( $args, $assoc_args ) {
+		$this->maybe_load_sqlite_dropin();
+
 		if ( ! empty( $args[0] ) ) {
 			$result_file = $args[0];
 		} else {
-			$result_file = sprintf( '%s.sql', DB_NAME );
+			$result_file = $this->is_sqlite() ? 'sqlite-db.sql' : sprintf( '%s.sql', DB_NAME );
+		}
+
+		if ( $this->is_sqlite() ) {
+			$this->sqlite_import( $result_file, $assoc_args );
+			return;
 		}
 
 		// Process options to MySQL.
@@ -1080,19 +1168,27 @@ public function size( $args, $assoc_args ) {
 
 		$default_unit = ( empty( $size_format ) && ! $human_readable ) ? ' B' : '';
 
+		$is_sqlite = $this->is_sqlite();
+
 		if ( $tables || $all_tables || $all_tables_with_prefix ) {
 
 			// Add all of the table sizes.
 			foreach ( Utils\wp_get_table_names( $args, $assoc_args ) as $table_name ) {
 
 				// Get the table size.
-				$table_bytes = $wpdb->get_var(
-					$wpdb->prepare(
-						'SELECT SUM(data_length + index_length) FROM information_schema.TABLES where table_schema = %s and Table_Name = %s GROUP BY Table_Name LIMIT 1',
-						DB_NAME,
-						$table_name
-					)
-				);
+				if ( $is_sqlite ) {
+					// For SQLite, we cannot get individual table sizes easily.
+					// Just report 0 as a placeholder.
+					$table_bytes = 0;
+				} else {
+					$table_bytes = $wpdb->get_var(
+						$wpdb->prepare(
+							'SELECT SUM(data_length + index_length) FROM information_schema.TABLES where table_schema = %s and Table_Name = %s GROUP BY Table_Name LIMIT 1',
+							DB_NAME,
+							$table_name
+						)
+					);
+				}
 
 				// Add the table size to the list.
 				$rows[] = [
@@ -1104,16 +1200,22 @@ public function size( $args, $assoc_args ) {
 		} else {
 
 			// Get the database size.
-			$db_bytes = $wpdb->get_var(
-				$wpdb->prepare(
-					'SELECT SUM(data_length + index_length) FROM information_schema.TABLES where table_schema = %s GROUP BY table_schema;',
-					DB_NAME
-				)
-			);
+			if ( $is_sqlite ) {
+				$db_bytes = $this->sqlite_size();
+				$db_name  = basename( $this->get_sqlite_db_path() );
+			} else {
+				$db_bytes = $wpdb->get_var(
+					$wpdb->prepare(
+						'SELECT SUM(data_length + index_length) FROM information_schema.TABLES where table_schema = %s GROUP BY table_schema;',
+						DB_NAME
+					)
+				);
+				$db_name = DB_NAME;
+			}
 
 			// Add the database size to the list.
 			$rows[] = [
-				'Name'  => DB_NAME,
+				'Name'  => $db_name,
 				'Size'  => strtoupper( $db_bytes ) . $default_unit,
 				'Bytes' => strtoupper( $db_bytes ),
 			];