From cf7fe88f59ab48f40e307075879c22b5d52c9975 Mon Sep 17 00:00:00 2001
From: Ankit Gade <ankit.gade@10up.com>
Date: Mon, 16 Sep 2024 12:50:36 +0530
Subject: [PATCH] Add filter for row actions

---
 .../js/components/UsersTable/RowActions.jsx   |  22 ++-
 assets/js/data/utils.js                       |   7 +
 package-lock.json                             |  13 +-
 package.json                                  |   1 +
 src/Container.php                             |  25 +++-
 src/Plugin.php                                |  22 +--
 src/REST/Base.php                             |  97 +++++++++++++
 src/REST/Manager.php                          |  56 ++++++++
 src/REST/Settings.php                         | 130 ++++++++++++++++++
 src/REST/TempUser.php                         |  14 +-
 10 files changed, 354 insertions(+), 33 deletions(-)
 create mode 100644 src/REST/Base.php
 create mode 100644 src/REST/Manager.php
 create mode 100644 src/REST/Settings.php

diff --git a/assets/js/components/UsersTable/RowActions.jsx b/assets/js/components/UsersTable/RowActions.jsx
index a711fcc..63de640 100644
--- a/assets/js/components/UsersTable/RowActions.jsx
+++ b/assets/js/components/UsersTable/RowActions.jsx
@@ -12,6 +12,7 @@ import { useDispatch } from '@wordpress/data';
 /**
  * Internal dependencies
  */
+import { tempAccessHooks } from '../../data/utils';
 import { STORE_NAME, UI_STORE_NAME } from '../../datastores/constants';
 
 export default function RowActions({ user }) {
@@ -44,26 +45,37 @@ export default function RowActions({ user }) {
 		setUserForDeletion(user);
 	};
 
-	return (
-		<div className="tempuser-actions">
+	const actions = tempAccessHooks.applyFilters(
+		'tempAccess.rowActions',
+		[
 			<Button
 				icon="edit"
 				title={__('Edit', 'passwordless-temporary-login')}
 				isPressed={false}
 				onClick={onEdit}
-			/>
+				key={'row-action-edit'}
+			/>,
 			<Button
 				icon="trash"
 				title={__('Delete', 'passwordless-temporary-login')}
 				isPressed={false}
 				onClick={onDelete}
-			/>
+				key={'row-action-delete'}
+			/>,
 			<Button
 				icon="admin-links"
 				title={__('Copy link', 'passwordless-temporary-login')}
 				isPressed={false}
 				onClick={() => onCopyLink(user)}
-			/>
+				key={'row-action-copy-link'}
+			/>,
+		],
+		user
+	);
+
+	return (
+		<div className="tempuser-actions">
+			{actions.map((action) => ({ ...action }))}
 		</div>
 	);
 }
diff --git a/assets/js/data/utils.js b/assets/js/data/utils.js
index e9a264c..d8a380e 100644
--- a/assets/js/data/utils.js
+++ b/assets/js/data/utils.js
@@ -7,6 +7,13 @@
  */
 import invariant from 'invariant';
 
+/**
+ * WordPress dependencies
+ */
+import { createHooks } from '@wordpress/hooks';
+
+export const tempAccessHooks = createHooks();
+
 /**
  * Collects and combines multiple objects of similar shape.
  *
diff --git a/package-lock.json b/package-lock.json
index d9553bf..46d644c 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -21,6 +21,7 @@
         "@wordpress/date": "^5.5.0",
         "@wordpress/dom-ready": "^4.3.0",
         "@wordpress/element": "^6.3.0",
+        "@wordpress/hooks": "^4.7.0",
         "@wordpress/i18n": "^5.3.0",
         "@wordpress/url": "^4.3.0",
         "classnames": "^2.5.1",
@@ -8731,9 +8732,9 @@
       }
     },
     "node_modules/@wordpress/hooks": {
-      "version": "4.5.0",
-      "resolved": "https://registry.npmjs.org/@wordpress/hooks/-/hooks-4.5.0.tgz",
-      "integrity": "sha512-wr1l1WM1yobyNGFLRgbLNBGtYIkzAGfmbT2e9zIOvqllRJT4wprYTqqhgovOKHlET5ij/TKdv4tApkGLUIXTsA==",
+      "version": "4.7.0",
+      "resolved": "https://registry.npmjs.org/@wordpress/hooks/-/hooks-4.7.0.tgz",
+      "integrity": "sha512-EGHMsNCt+PyStm3o1JWujaTA+HKcTxuEXdSHBBFDavzsgOF13bxTf1LpDYgTZJT3K9TSMP983IwfckP5t66pDw==",
       "dependencies": {
         "@babel/runtime": "^7.16.0"
       },
@@ -34389,9 +34390,9 @@
       }
     },
     "@wordpress/hooks": {
-      "version": "4.5.0",
-      "resolved": "https://registry.npmjs.org/@wordpress/hooks/-/hooks-4.5.0.tgz",
-      "integrity": "sha512-wr1l1WM1yobyNGFLRgbLNBGtYIkzAGfmbT2e9zIOvqllRJT4wprYTqqhgovOKHlET5ij/TKdv4tApkGLUIXTsA==",
+      "version": "4.7.0",
+      "resolved": "https://registry.npmjs.org/@wordpress/hooks/-/hooks-4.7.0.tgz",
+      "integrity": "sha512-EGHMsNCt+PyStm3o1JWujaTA+HKcTxuEXdSHBBFDavzsgOF13bxTf1LpDYgTZJT3K9TSMP983IwfckP5t66pDw==",
       "requires": {
         "@babel/runtime": "^7.16.0"
       }
diff --git a/package.json b/package.json
index 304a4d7..fe136da 100644
--- a/package.json
+++ b/package.json
@@ -75,6 +75,7 @@
     "@wordpress/date": "^5.5.0",
     "@wordpress/dom-ready": "^4.3.0",
     "@wordpress/element": "^6.3.0",
+    "@wordpress/hooks": "^4.7.0",
     "@wordpress/i18n": "^5.3.0",
     "@wordpress/url": "^4.3.0",
     "classnames": "^2.5.1",
diff --git a/src/Container.php b/src/Container.php
index 5d3a861..28bc5f9 100644
--- a/src/Container.php
+++ b/src/Container.php
@@ -11,11 +11,13 @@
 namespace Ankit\TemporaryAccess;
 
 use Ankit\TemporaryAccess\Interfaces\Container as ContainerInterface;
+use Ankit\TemporaryAccess\REST\Manager;
 use Ankit\TemporaryAccess\UI\Settings;
 use Ankit\TemporaryAccess\UI\UsersTable;
 use InvalidArgumentException;
 use Pimple\Container as PimpleContainer;
 use Ankit\TemporaryAccess\REST\TempUser;
+use Ankit\TemporaryAccess\REST\Settings as SettingsEndpoint;
 
 /**
  * Class Container
@@ -25,7 +27,6 @@
  */
 class Container implements ContainerInterface {
 
-
 	/**
 	 * Pimple container.
 	 *
@@ -103,6 +104,28 @@ public function define_services(): void {
 			return new TempUser( $c['user_manager'] );
 		};
 
+		/**
+		 * Settings endpoint.
+		 *
+		 * @param PimpleContainer $c Pimple container object.
+		 *
+		 * @return TempUser
+		 */
+		$this->container['settings_endpoint'] = function ( PimpleContainer $c ) {
+			return new SettingsEndpoint();
+		};
+
+		/**
+		 * REST API manager.
+		 * 
+		 * @param PimpleContainer $c Pimple container object.
+		 * 
+		 * @return Manager
+		 */
+		$this->container['rest_manager'] = function ( PimpleContainer $c ) {
+			return new Manager();
+		};
+
 		/**
 		 * Settings service.
 		 *
diff --git a/src/Plugin.php b/src/Plugin.php
index 1f1a2df..79962c6 100644
--- a/src/Plugin.php
+++ b/src/Plugin.php
@@ -12,7 +12,7 @@
 
 use Ankit\TemporaryAccess\Interfaces\Container as ContainerInterface;
 use Ankit\TemporaryAccess\Interfaces\UserManagement;
-use Ankit\TemporaryAccess\REST\TempUser;
+use Ankit\TemporaryAccess\REST\Manager;
 
 /**
  * Class Plugin
@@ -53,16 +53,16 @@ class Plugin {
 	/**
 	 * User manager.
 	 *
-	 * @var UserManagement
+	 * @var Manager
 	 */
-	private $user_manager;
+	private $rest_manager;
 
 	/**
-	 * Endpoint service.
-	 *
-	 * @var TempUser
+	 * User manager.
+	 * 
+	 * @var UserManagement
 	 */
-	private $user_endpoint;
+	private $user_manager;
 
 	/**
 	 * Authentication service.
@@ -103,7 +103,7 @@ public function run(): void {
 
 		$settings            = $this->container->get( 'settings' );
 		$this->user_manager  = $this->container->get( 'user_manager' );
-		$this->user_endpoint = $this->container->get( 'temp_user_endpoint' );
+		$this->rest_manager  = $this->container->get( 'rest_manager' );
 		$this->authenticator = $this->container->get( 'authenticator' );
 		$users_table         = $this->container->get( 'users_table' );
 
@@ -111,9 +111,11 @@ public function run(): void {
 		$settings->init();
 		$users_table->init();
 
-		add_action( 'init', array( $this->user_manager, 'init' ) );
-		add_action( 'rest_api_init', array( $this->user_endpoint, 'register' ) );
+		add_action( 'rest_api_init', array( $this->rest_manager, 'register' ) );
+
 		add_action( 'tempaccess.user_authenticated', array( $this->user_manager, 'post_login_actions' ) );
+		add_action( 'init', array( $this->user_manager, 'init' ) );
+
 		add_action( 'init', array( $this, 'load_translations' ) );
 	}
 
diff --git a/src/REST/Base.php b/src/REST/Base.php
new file mode 100644
index 0000000..67cb371
--- /dev/null
+++ b/src/REST/Base.php
@@ -0,0 +1,97 @@
+<?php
+/**
+ * Base class for REST endpoints.
+ */
+
+declare(strict_types=1);
+
+namespace Ankit\TemporaryAccess\REST;
+
+use WP_REST_Server;
+
+/**
+ * Class Base
+ *
+ * @package Ankit\TemporaryAccess\REST
+ */
+
+abstract class Base {
+
+	/**
+	 * Namespace for the controller.
+	 */
+	const NAMESPACE = 'tempaccess/v1';
+
+	/**
+	 * Register the routes for the objects of the controller.
+	 * 
+	 * @return void
+	 */
+	abstract public function register(): void;
+
+	/**
+	 * Register the routes for the objects of the controller.
+	 *
+	 * @param string $route Route.
+	 * @param array  $args  Args.
+	 */
+	protected function register_route( string $route, array $args ) {
+		$default = array(
+			'permission_callback' => $this->get_permission_callback(),
+			'args'                => $this->get_args(),
+		);
+
+		$route_args = wp_parse_args( $args, $default );
+
+		register_rest_route(
+			self::NAMESPACE,
+			$route,
+			$route_args
+		);
+	}
+
+	/**
+	 * Get the namespace for the controller.
+	 *
+	 * @return string
+	 */
+	public function get_namespace(): string {
+		return self::NAMESPACE;
+	}
+
+	/**
+	 * Get the version for the controller.
+	 *
+	 * @return string
+	 */
+	public function get_version(): string {
+		return '1';
+	}
+
+	/**
+	 * Get the method for the controller.
+	 *
+	 * @return string
+	 */
+	public function get_method(): string {
+		return WP_REST_Server::READABLE;
+	}
+
+	/**
+	 * Get the permission callback for the controller.
+	 *
+	 * @return string
+	 */
+	public function get_permission_callback(): bool {
+		return current_user_can( 'edit_users' );
+	}
+
+	/**
+	 * Get the args for the controller.
+	 *
+	 * @return array
+	 */
+	public function get_args(): array {
+		return array();
+	}
+}
diff --git a/src/REST/Manager.php b/src/REST/Manager.php
new file mode 100644
index 0000000..df09f2b
--- /dev/null
+++ b/src/REST/Manager.php
@@ -0,0 +1,56 @@
+<?php
+/**
+ * Manager class for REST endpoints.
+ */
+
+declare(strict_types=1);
+
+namespace Ankit\TemporaryAccess\REST;
+
+use Ankit\TemporaryAccess\Container;
+use Ankit\TemporaryAccess\REST\Base;
+use function Ankit\TemporaryAccess\plugin;
+
+/**
+ * Class Manager
+ *
+ * @package Ankit\TemporaryAccess\REST
+ */
+class Manager {
+
+    /**
+     * Container instance.
+     *
+     * @var Container
+     */
+    private $container;
+
+    /**
+     * List of REST endpoints.
+     * 
+     * @var Base[]
+     */
+    private $endpoints;
+
+	/**
+	 * Manager constructor.
+	 *
+	 * @param array $endpoints List of REST endpoints.
+	 */
+	public function __construct() {
+        $this->container = plugin()->container();
+        $this->endpoints = [
+            'tempuser' => $this->container->get( 'temp_user_endpoint' ),
+            'settings' => $this->container->get( 'settings_endpoint' ),
+        ];
+	}
+
+	/**
+	 * Register the REST endpoints.
+	 */
+	public function register() {
+		foreach ( $this->endpoints as $endpoint ) {
+			$endpoint->register();
+		}
+	}
+}
diff --git a/src/REST/Settings.php b/src/REST/Settings.php
new file mode 100644
index 0000000..5a4628f
--- /dev/null
+++ b/src/REST/Settings.php
@@ -0,0 +1,130 @@
+<?php
+/**
+ * Endpoint to manage temporary user settings.
+ *
+ * @package Ankit\TemporaryAccess
+ * @since   1.0.0
+ */
+
+declare(strict_types=1);
+
+namespace Ankit\TemporaryAccess\REST;
+
+use WP_REST_Response;
+use WP_REST_Server;
+
+/**
+ * Class Settings
+ *
+ * @package Ankit\TemporaryAccess\REST
+ */
+class Settings extends Base {
+
+	/**
+	 * Settings name.
+	 */
+	const SETTINGS_NAME = 'temporary_access_settings';
+
+	/**
+	 * Get settings.
+	 *
+	 * @param \WP_REST_Request $request Request object.
+	 *
+	 * @return void
+	 */
+	public function register(): void {
+		$this->register_route(
+			'/settings',
+			array(
+				'methods'             => WP_REST_Server::READABLE,
+				'callback'            => array( $this, 'get_settings' ),
+				'permission_callback' => array( $this, 'get_permission_callback' ),
+			)
+		);
+
+		$this->register_route(
+			'/settings',
+			array(
+				'methods'             => WP_REST_Server::EDITABLE,
+				'callback'            => array( $this, 'update_settings' ),
+				'permission_callback' => array( $this, 'get_permission_callback' ),
+			)
+		);
+	}
+
+	/**
+	 * Get settings.
+	 */
+	protected function get_allowed_settings(): array {
+		return array(
+			'default_expiry'           => array(
+				'type'        => 'integer',
+				'description' => __( 'Default expiry time for temporary users in seconds.', 'temporary-access' ),
+			),
+			'default_role'             => array(
+				'type'        => 'string',
+				'description' => __( 'Default role for temporary users.', 'temporary-access' ),
+				'default'     => 'administrator',
+			),
+			'default_redirect'         => array(
+				'type'        => 'string',
+				'description' => __( 'Default redirect URL after login.', 'temporary-access' ),
+				'default'     => home_url(),
+			),
+			'delete_data_on_uninstall' => array(
+				'type'        => 'boolean',
+				'description' => __( 'Delete all plugin data on uninstall.', 'temporary-access' ),
+				'default'     => false,
+			),
+			'track_usage'              => array(
+				'type'        => 'boolean',
+				'description' => __( 'Help us to track the usage of plugin anonymously to make it better.', 'temporary-access' ),
+				'default'     => false,
+			),
+		);
+	}
+
+	/**
+	 * Get settings.
+	 *
+	 * @param \WP_REST_Request $request Request object.
+	 *
+	 * @return WP_REST_Response
+	 */
+	public function get_settings( \WP_REST_Request $request ): WP_REST_Response {
+		// Check if settings is available in database, if not then return default values of allowed settings.
+		$settings = get_option( self::SETTINGS_NAME, array() );
+
+		if ( empty( $settings ) ) {
+			$settings = array();
+			foreach ( $this->get_allowed_settings() as $key => $setting ) {
+				$settings[ $key ] = $setting['default'] ?? null;
+			}
+		}
+
+		return new WP_REST_Response( $settings, 200 );
+	}
+
+	/**
+	 * Update settings.
+	 *
+	 * @param \WP_REST_Request $request Request object.
+	 *
+	 * @return WP_REST_Response
+	 */
+	public function update_settings( \WP_REST_Request $request ): WP_REST_Response {
+		$settings = get_option( self::SETTINGS_NAME, array() );
+
+		$allowed_settings = $this->get_allowed_settings();
+
+		foreach ( $allowed_settings as $key => $setting ) {
+			if ( $request->get_param( $key ) ) {
+				$settings[ $key ] = $request->get_param( $key );
+			}
+		}
+
+		update_option( self::SETTINGS_NAME, $settings );
+
+		return new WP_REST_Response( $settings, 200 );
+	}
+}
diff --git a/src/REST/TempUser.php b/src/REST/TempUser.php
index 4e8e32f..391021f 100644
--- a/src/REST/TempUser.php
+++ b/src/REST/TempUser.php
@@ -23,8 +23,7 @@
  *
  * @package Ankit\TemporaryAccess\REST
  */
-class TempUser {
-
+class TempUser extends Base {
 
 	/**
 	 * Endpoint namespace.
@@ -64,9 +63,7 @@ public function permission_check(): bool {
 	 * @return void
 	 */
 	public function register(): void {
-
-		register_rest_route(
-			self::NAMESPACE,
+		$this->register_route(
 			'/users',
 			array(
 				array(
@@ -77,13 +74,10 @@ public function register(): void {
 					'methods'  => WP_REST_Server::CREATABLE,
 					'callback' => array( $this, 'create_user' ),
 				),
-				'permission_callback' => array( $this, 'permission_check' ),
-				'args'                => array(),
 			)
 		);
 
-		register_rest_route(
-			self::NAMESPACE,
+		$this->register_route(
 			'/users/(?P<ID>\d+)',
 			array(
 				array(
@@ -98,8 +92,6 @@ public function register(): void {
 					'methods'  => WP_REST_Server::DELETABLE,
 					'callback' => array( $this, 'delete_user' ),
 				),
-				'permission_callback' => array( $this, 'permission_check' ),
-				'args'                => array(),
 			)
 		);
 	}