MAW-875: session plugin to must-use

2024-02-16 16:58:08 +00:00
parent 9169f08384
commit c9dfa772f7
10 changed files with 11 additions and 0 deletions
--- a/wp/wp-content/mu-plugins/wp-native-php-sessions/inc/class-cli-command.php
+++ b/wp/wp-content/mu-plugins/wp-native-php-sessions/inc/class-cli-command.php
@@ -0,0 +1,116 @@
+<?php
+/**
+ * CLI interface to interact with Pantheon sessions.
+ *
+ * @package WPNPS
+ */
+
+namespace Pantheon_Sessions;
+
+use WP_CLI;
+
+/**
+ * Interact with Pantheon Sessions
+ */
+class CLI_Command extends \WP_CLI_Command {
+
+	/**
+	 * List all registered sessions.
+	 *
+	 * [--format=<format>]
+	 * : Accepted values: table, csv, json, count, ids. Default: table
+	 *
+	 * @subcommand list
+	 */
+	public function list_( $args, $assoc_args ) {
+		global $wpdb;
+
+		if ( ! PANTHEON_SESSIONS_ENABLED ) {
+			WP_CLI::error( 'Pantheon Sessions is currently disabled.' );
+		}
+
+		$defaults   = [
+			'format' => 'table',
+			'fields' => 'session_id,user_id,datetime,ip_address,data',
+		];
+		$assoc_args = array_merge( $defaults, $assoc_args );
+
+		$sessions = [];
+		foreach ( new \WP_CLI\Iterators\Query( "SELECT * FROM {$wpdb->pantheon_sessions} ORDER BY datetime DESC" ) as $row ) {
+			$sessions[] = $row;
+		}
+
+		\WP_CLI\Utils\Format_Items( $assoc_args['format'], $sessions, $assoc_args['fields'] );
+	}
+
+	/**
+	 * Delete one or more sessions.
+	 *
+	 * [<session-id>...]
+	 * : One or more session IDs
+	 *
+	 * [--all]
+	 * : Delete all sessions.
+	 *
+	 * @subcommand delete
+	 */
+	public function delete( $args, $assoc_args ) {
+		global $wpdb;
+
+		if ( ! PANTHEON_SESSIONS_ENABLED ) {
+			WP_CLI::error( 'Pantheon Sessions is currently disabled.' );
+		}
+
+		if ( isset( $assoc_args['all'] ) ) {
+			$args = $wpdb->get_col( "SELECT session_id FROM {$wpdb->pantheon_sessions}" );
+			if ( empty( $args ) ) {
+				WP_CLI::warning( 'No sessions to delete.' );
+			}
+		}
+
+		foreach ( $args as $session_id ) {
+			$session = \Pantheon_Sessions\Session::get_by_sid( $session_id );
+			if ( $session ) {
+				$session->destroy();
+				WP_CLI::log( sprintf( 'Session destroyed: %s', $session_id ) );
+			} else {
+				WP_CLI::warning( sprintf( "Session doesn't exist: %s", $session_id ) );
+			}
+		}
+	}
+
+	/**
+	 * Set id as primary key in the Native PHP Sessions plugin table.
+	 *
+	 * @subcommand add-index
+	 */
+	public function add_index( $args, $assoc_args ) {
+		$pantheon_session = new \Pantheon_Sessions();
+		$resume_point = isset( $assoc_args['start_point'] ) ? $assoc_args['start_point'] : 0;
+		$pantheon_session->add_index( $resume_point );
+	}
+
+	/**
+	 * Finalizes the creation of a primary key by deleting the old data.
+	 *
+	 * @subcommand primary-key-finalize
+	 */
+	public function primary_key_finalize( $args, $assoc_args ) {
+		$pantheon_session = new \Pantheon_Sessions();
+		$resume_point = isset( $assoc_args['start_point'] ) ? $assoc_args['start_point'] : 0;
+		$pantheon_session->primary_key_finalize( $resume_point );
+	}
+
+	/**
+	 * Reverts addition of primary key.
+	 *
+	 * @subcommand primary-key-revert
+	 */
+	public function primary_key_revert( $args, $assoc_args ) {
+		$pantheon_session = new \Pantheon_Sessions();
+		$resume_point = isset( $assoc_args['start_point'] ) ? $assoc_args['start_point'] : 0;
+		$pantheon_session->primary_key_revert( $resume_point );
+	}
+}
+
+\WP_CLI::add_command( 'pantheon session', '\Pantheon_Sessions\CLI_Command' );