117 lines
2.9 KiB
PHP
117 lines
2.9 KiB
PHP
<?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' );
|