<?php
/**
 * WP Force Logout Process File.
 *
 * @package    WP Force Logout
 * @author     Sanjeev Aryal
 * @since      1.0.0
 * @license    GPL-3.0+
 */

if ( ! defined( 'ABSPATH' ) ) {
	exit;
	// Exit if accessed directly.
}

/**
 * WP Force Logout Process Class.
 *
 * @class WP_Force_Logout_Process
 *
 * @since  1.0.0
 */
class WP_Force_Logout_Process {

	/**
	 * Constructor.
	 */
	public function __construct() {

		global $pagenow;

		add_action( 'init', array( $this, 'update_online_users_status' ) );
		add_action( 'init', array( $this, 'update_last_login' ) );
		add_action( 'wp_ajax_wp_force_logout_dismiss_review_notice', array( $this, 'dismiss_review_notice' ) );

		// Return if it's not the users page and if user do not have capability to force logout.
		if ( 'users.php' !== $pagenow || ! $this->user_has_cap() ) {
			return;
		}

		add_action( 'admin_enqueue_scripts', array( $this, 'enqueue_scripts' ) );
		add_filter( 'manage_users_columns', array( $this, 'add_column_title' ) );
		add_filter( 'manage_users_custom_column', array( $this, 'add_column_value' ), 10, 3 );
		add_filter( 'manage_users_sortable_columns', array( $this, 'sortable_login_activity' ) );
		add_filter( 'users_list_table_query_args', array( $this, 'sortby_login_activity' ) );

		// add_action( 'load-users.php', array( $this, 'add_last_login' ) );	// Commented since 1.3.0.
		add_action( 'load-users.php', array( $this, 'trigger_query_actions' ) );
		add_action( 'load-users.php', array( $this, 'trigger_bulk_actions' ) );
		add_filter( 'bulk_actions-users', array( $this, 'add_bulk_action' ) );
		add_action( 'restrict_manage_users', array( $this, 'add_all_users_logout' ), 1000 );
		add_action( 'in_admin_header', array( $this, 'review_notice' ), 100 );
	}

	/**
	 * Enqueue Scripts
	 */
	public function enqueue_scripts() {

		wp_enqueue_style( 'wp-force-logout', plugins_url( 'assets/css/wp-force-logout.css', WP_FORCE_LOGOUT_PLUGIN_FILE ), array(), WPFL_VERSION, $media = 'all' );
		wp_enqueue_script( 'wp-force-logout-js', plugins_url( 'assets/js/script.js', WP_FORCE_LOGOUT_PLUGIN_FILE ), array( 'jquery' ), WPFL_VERSION, false );
		wp_localize_script(
			'wp-force-logout-js',
			'wpfl_plugins_params',
			array(
				'ajax_url'     => admin_url( 'admin-ajax.php' ),
				'review_nonce' => wp_create_nonce( 'review-notice' ),
			)
		);
	}

	/**
	 * Add the column title for the login activity column
	 *
	 * @param array $columns Columns.
	 *
	 * @return array
	 */
	public function add_column_title( $columns ) {

		$new_columns['wpfl'] = esc_html__( 'Login Activity', 'wp-force-logout' );

		return $this->custom_insert_after_helper( $columns, $new_columns, 'cb' );
	}

	/**
	 * Insert Login Activity after first checkbox column
	 *
	 * @param  array  $columns     WP_User_List_Table columns.
	 * @param  array  $new_column  New Columns to insert.
	 * @param  string $after       Position of new column after.
	 * @return array  Columns.
	 */
	public function custom_insert_after_helper( $columns, $new_columns, $after ) {

		// Search for the item position and +1 since is after the selected item key.
		$position = array_search( $after, array_keys( $columns ) ) + 1;

		// Insert the new item.
		$return_columns  = array_slice( $columns, 0, $position, true );
		$return_columns += $new_columns;
		$return_columns += array_slice( $columns, $position, count( $columns ) - $position, true );

		return $return_columns;
	}

	/**
	 * Set the value for login activity column for each user in the users list
	 *
	 * @param string $value       Value to display
	 * @param string $column_name Column Name.
	 * @param int    $user_id        User ID.
	 *
	 * @return string
	 */
	public function add_column_value( $value, $column_name, $user_id ) {

		if ( $column_name == 'wpfl' ) {
			$user = wp_get_current_user();

			$is_user_online = $this->is_user_online( $user_id );

			if ( $is_user_online ) {
				$logout_link = add_query_arg(
					array(
						'action' => 'wpfl-logout',
						'user'   => $user_id,
					)
				);
				$logout_link = remove_query_arg( array( 'new_role' ), $logout_link );
				$logout_link = wp_nonce_url( $logout_link, 'wpfl-logout' );

				$value  = '<span class="online-circle">' . esc_html__( 'Online', 'wp-force-logout' ) . '</span>';
				$value .= ' ';
				$value .= '<a style="color:red" href="' . esc_url( $logout_link ) . '">' . _x( 'Logout', 'The action on users list page', 'wp-force-logout' ) . '</a>';
			} else {
				$last_login = $this->get_last_login( $user_id );
				$value      = '<span class="offline-circle">' . esc_html__( 'Offline ', 'wp-force-logout' );
				$value     .= '</br>' . esc_html__( 'Last Login: ', 'wp-force-logout' );
				$value     .= ! empty( $last_login ) ? $last_login . ' ago' : esc_html__( 'Never', 'wp-force-logout' ) . '</span>';
			}
		}//end if

		return $value;
	}

	/**
	 * Make login activity column sortable
	 *
	 * @param  array $columns Sortable columns.
	 * @since  1.2.2
	 *
	 * @return array
	 */
	public function sortable_login_activity( $columns ) {
		$columns['wpfl'] = 'wpfl';

		return $columns;
	}

	/**
	 *
	 * Sort users by login activity.
	 *
	 * @param array $args.
	 *
	 * @since 1.2.2
	 *
	 * @return array
	 */
	public function sortby_login_activity( $args ) {

		if ( isset( $args['orderby'] ) && 'wpfl' == $args['orderby'] ) {

			$order = isset( $args['order'] ) && $args['order'] === 'asc' ? 'desc' : 'asc';
			// Wierd way of reversing the order. Still works.

			$args = array_merge(
				$args,
				array(
					'meta_key' => 'last_login',
					'orderby'  => 'meta_value',
					'order'    => $order,
				)
			);
		}

		return $args;
	}

	/**
	 * Checks if the current user is online or not.
	 *
	 * @param  int $user_id    User ID.
	 * @return boolean
	 */
	public function is_user_online( $user_id ) {

		// Get the online users list
		$logged_in_users = get_transient( 'online_status' );

		 // Online, if (s)he is in the list and last activity was less than 60 seconds ago
		return isset( $logged_in_users[ $user_id ] ) && ( $logged_in_users[ $user_id ] > ( time() - ( 1 * 60 ) ) );
	}

	/**
	 * Update online users status. Store in transient.
	 *
	 * @link  https://wordpress.stackexchange.com/a/34434/126847
	 * @return void.
	 */
	public function update_online_users_status() {

		// Get the user online status list.
		$logged_in_users = get_transient( 'online_status' ) ?: array();

		// Get current user ID
		$user = wp_get_current_user();

		// Check if the current user needs to update his online status;
		// Needs if user is not in the list.
		$no_need_to_update = isset( $logged_in_users[ $user->ID ] )

			// And if his "last activity" was less than let's say ...6 seconds ago
			&& $logged_in_users[ $user->ID ] > ( time() - ( 1 * 60 ) );

		// Update the list if needed
		if ( ! $no_need_to_update ) {
			$logged_in_users[ $user->ID ] = time();
			set_transient( 'online_status', $logged_in_users, $expire_in = ( 60 * 60 ) );
			// 60 mins
		}
	}

	/**
	 * Store last login info in usermeta table.
	 *
	 * @since  1.1.0
	 *
	 * @return void.
	 */
	public function update_last_login() {

		$user_id = get_current_user_id();
		update_user_meta( $user_id, 'last_login', time() );
	}

	/**
	 * Add last_login meta key for all users with a random number to later compare.
	 *
	 * @todo :: this might gets slower for sites with large number of users.
	 *
	 * @since  1.2.2
	 *
	 * @return  void.
	 */
	public function add_last_login() {
		$users = get_users();
		foreach ( $users as $user ) {
			$last_login = get_user_meta( $user->ID, 'last_login', true );
			if ( empty( $last_login ) ) {
				update_user_meta( $user->ID, 'last_login', 0.00058373 );
				// Store random number to identify. Because meta_ley last_login is required for the sortby login activitiy feature to display all previous users too. Currently, the feature is disabled as it seems to effect performance.
			}
		}
	}

	/**
	 * Get last login time.
	 *
	 * @since  1.1.0
	 *
	 * @return string
	 */
	public function get_last_login( $user_id ) {
		$last_login     = get_user_meta( $user_id, 'last_login', true );
		$the_login_date = '';

		if ( ! empty( $last_login ) && 0.00058373 != $last_login ) {
			// Backwards compatibility for v1.3.0-
			$the_login_date = human_time_diff( $last_login );
		}

		return $the_login_date;
	}

	/**
	 * Trigger the action query logout
	 */
	public function trigger_query_actions() {

		if ( isset( $_REQUEST['action'] ) && $_REQUEST['action'] === 'force_logout_all' ) {
			check_admin_referer( 'wp-force-logout-nonce' );
			$this->force_all_users_logout();

			// Redirect to users/same page after logout.
			wp_safe_redirect( admin_url( 'users.php ' ) );
			exit();
		}

		$action = isset( $_REQUEST['action'] ) ? sanitize_key( $_REQUEST['action'] ) : false;
		$mode   = isset( $_POST['mode'] ) ? $_POST['mode'] : false;

		// If this is a multisite, bulk request, stop now!
		if ( 'list' == $mode ) {
			return;
		}

		if ( ! empty( $action ) && 'wpfl-logout' === $action && ! isset( $_GET['new_role'] ) ) {

			check_admin_referer( 'wpfl-logout' );

			$redirect = admin_url( 'users.php' );
			$user_id  = isset( $_GET['user'] ) ? absint( $_GET['user'] ) : 0;

			if ( $action == 'wpfl-logout' ) {
				// Get all sessions for user with ID $user_id
				$sessions = WP_Session_Tokens::get_instance( $user_id );

				// We have got the sessions, destroy them all!
				$sessions->destroy_all();
			}

			wp_redirect( $redirect );
			exit;
		}
	}

	/**
	 * Trigger the action query logout for bulk users.
	 */
	public function trigger_bulk_actions() {

		if ( empty( $_REQUEST['users'] ) || empty( $_REQUEST['action'] ) ) {
			return;
		}

		if ( 'wpfl-bulk-logout' !== $_REQUEST['action'] ) {
			return;
		}

		$user_ids = array_map( 'absint', (array) $_REQUEST['users'] );

		foreach ( $user_ids as $user_id ) {

			// Get all sessions for user with ID $user_id
			$sessions = WP_Session_Tokens::get_instance( $user_id );

			// We have got the sessions, destroy them all!
			$sessions->destroy_all();
		}

		return admin_url( 'users.php' );
	}

	/**
	 * Add Bulk Logout Action.
	 *
	 * @param  array $actions    Current Actions.
	 * @return array    All Actions along with logout action.
	 */
	public function add_bulk_action( $actions ) {
		$actions['wpfl-bulk-logout'] = esc_html__( 'Logout', 'wp-force-logout' );

		return $actions;
	}

	/**
	 * All users logout functionality.
	 *
	 * @since  1.0.1
	 */
	public function add_all_users_logout( $which ) {
		echo '<div class="alignright">';
		$url = wp_nonce_url( 'users.php?action=force_logout_all', 'wp-force-logout-nonce' );
		echo '<a style="margin-left:5px; margin-top:0px" class="button wp-force-logout" href="' . $url . '">' . esc_html__( 'Logout All Users', 'wp-force-logout' ) . '</a>';
		echo '</div>';
	}

	/**
	 * All users logout functionality.
	 *
	 * @return void.
	 */
	public function force_all_users_logout() {

		$users = get_users();

		foreach ( $users as $user ) {

			// Get all sessions for user with ID $user_id
			$sessions = WP_Session_Tokens::get_instance( $user->ID );

			// We have got the sessions, destroy them all!
			$sessions->destroy_all();
		}
	}

	/**
	 * Check if user has capability to force logout (edit) users.
	 *
	 * @since  1.4.4
	 *
	 * @todo :: Allow site admins of single site in a multisite to force logout users.
	 *
	 * @return bool
	 */
	public function user_has_cap() {

		return current_user_can( 'edit_users' );
	}

	/**
	 * Outputs the Review notice on admin header.
	 *
	 * @since 1.2.1
	 */
	public function review_notice() {

		global $current_screen;

		// Show only to Admins
		if ( ! $this->user_has_cap() ) {
			return;
		}
		
		if ( ! empty( $current_screen->id ) && $current_screen->id !== 'users' ) {
			return;
		}

		if ( apply_filters( 'wpfl_upgrade_notice_dismiss_forever', false ) ) {
			return;
		}

		$notice_dismissed = get_option( 'wpfl_upgrade_notice_dismissed', 'no' );
		if ( $notice_dismissed !== 'no' && $notice_dismissed > strtotime( '-7 days' ) ) {
			return;
		}

		?>
			<div id="wp-force-logout-review-notice" class="notice notice-info wp-force-logout-review-notice">
				<div class="wp-force-logout-review-thumbnail">
					<img src="<?php echo plugins_url( 'assets/logo.jpg', WP_FORCE_LOGOUT_PLUGIN_FILE ); ?>" alt="">
				</div>
				<span class="button button-link notice-dismiss">x</span>
				<div class="wp-force-logout-review-text">

						<h3><?php _e( 'Whoopee! 😀', 'wp-force-logout' ); ?></h3>
						<?php
						?>
						<p><?php echo esc_html__( 'How\'s it going? WPForce Logout PRO ⭐ is now available with more powerful features. Would you like to upgrade and boost my motivation?', 'wp-force-logout' );?> </p>

					<ul class="wp-force-logout-review-ul">
						<li><a class="button button-primary" href="<?php echo admin_url( 'users.php?page=wp-force-logout-pricing' );?>"><span class=""></span><?php _e( 'Upgrade', 'wp-force-logout' ); ?></a></li>
					 </ul>
				</div>
			</div>
		<?php
	}

	/**
	 * Dismiss the reveiw notice on dissmiss click
	 *
	 * @since 1.2.1
	 */
	public function dismiss_review_notice() {

		check_admin_referer( 'review-notice', 'security' );

		if ( ! empty( $_POST['dismissed'] ) ) {
			update_option( 'wpfl_upgrade_notice_dismissed', time() );
		}
	}
}

new WP_Force_Logout_Process();
