How to customize WP_Error (REST JWT authentication plugin) [closed]

✨ Solution

I’m not sure this is the most efficient way to use rest_post_dispatch (documentation) for this purpose (maybe overkill but I didn’t found other way).

It works as expected.

add_filter("rest_post_dispatch", "rest_customize_result", 10, 3);

function rest_customize_result(
  WP_REST_Response $result,
  WP_REST_Server $server,
  WP_REST_Request $request
) {
  if ( $request->get_route() === "/jwt-auth/v1/token"
       && $result->get_status() === 403 ) {
    $data = $result->get_data();
    $data["code"] = "jwt_auth_failed";
    $data["message"] = __("Invalid Credentials.", "theme");
    $result->set_data($data);
  }
  return $result;
}