feat: UserTools handler

2026-05-08 03:15:44 +00:00
parent c8653e7493
commit c0ec50bd42
1 changed files with 58 additions and 0 deletions
@@ -0,0 +1,58 @@
+<?php
+
+namespace AskTamMCP\Tools;
+
+defined('ABSPATH') || exit;
+
+class UserTools {
+    public static function listUsers(array $args): array {
+        $qArgs = ['number' => (int) ($args['per_page'] ?? 20), 'paged' => (int) ($args['page'] ?? 1)];
+        if (!empty($args['role']))   $qArgs['role']   = $args['role'];
+        if (!empty($args['search'])) $qArgs['search'] = '*' . $args['search'] . '*';
+        $query = new \WP_User_Query($qArgs);
+        $users = [];
+        foreach ($query->get_results() as $u) {
+            $users[] = ['id' => $u->ID, 'username' => $u->user_login, 'email' => $u->user_email, 'name' => $u->display_name, 'roles' => $u->roles, 'registered' => $u->user_registered];
+        }
+        return ['users' => $users, 'total' => $query->get_total()];
+    }
+
+    public static function getUser(array $args): array {
+        $user = get_userdata((int) $args['user_id']);
+        if (!$user) throw new \RuntimeException("User not found: {$args['user_id']}");
+        return ['id' => $user->ID, 'username' => $user->user_login, 'email' => $user->user_email, 'first_name' => $user->first_name, 'last_name' => $user->last_name, 'display_name' => $user->display_name, 'roles' => $user->roles, 'registered' => $user->user_registered, 'capabilities' => array_keys(array_filter($user->allcaps))];
+    }
+
+    public static function createUser(array $args): array {
+        $userId = wp_create_user($args['username'], $args['password'], $args['email']);
+        if (is_wp_error($userId)) throw new \RuntimeException($userId->get_error_message());
+        $user = new \WP_User($userId);
+        if (!empty($args['role']))       $user->set_role($args['role']);
+        if (!empty($args['first_name'])) update_user_meta($userId, 'first_name', $args['first_name']);
+        if (!empty($args['last_name']))  update_user_meta($userId, 'last_name', $args['last_name']);
+        return ['id' => $userId, 'username' => $args['username'], 'email' => $args['email']];
+    }
+
+    public static function updateUser(array $args): array {
+        $userId = (int) $args['user_id'];
+        $data = ['ID' => $userId];
+        if (isset($args['email']))    $data['user_email'] = $args['email'];
+        if (isset($args['password'])) $data['user_pass']  = $args['password'];
+        $result = wp_update_user($data);
+        if (is_wp_error($result)) throw new \RuntimeException($result->get_error_message());
+        if (!empty($args['first_name'])) update_user_meta($userId, 'first_name', $args['first_name']);
+        if (!empty($args['last_name']))  update_user_meta($userId, 'last_name', $args['last_name']);
+        if (!empty($args['role'])) { $user = new \WP_User($userId); $user->set_role($args['role']); }
+        if (!empty($args['meta'])) foreach ($args['meta'] as $k => $v) update_user_meta($userId, $k, $v);
+        return ['id' => $userId, 'updated' => true];
+    }
+
+    public static function deleteUser(array $args): array {
+        require_once ABSPATH . 'wp-admin/includes/user.php';
+        $userId = (int) $args['user_id'];
+        $reassign = isset($args['reassign']) ? (int) $args['reassign'] : null;
+        $result = wp_delete_user($userId, $reassign);
+        if (!$result) throw new \RuntimeException("Failed to delete user: {$userId}");
+        return ['id' => $userId, 'deleted' => true];
+    }
+}