fix: sendresponse does not work properly in rest api

Author: abdul-kaioum

src/Http/Router/RouteRegister.php

@@ -34,6 +34,13 @@ final class RouteRegister
 
     private $_middleware = [];
 
+    /**
+     * Instance of rest request
+     *
+     * @var WP_REST_Response
+     */
+    private $_restResponse;
+
     /**
      * Instance of rest request
      *
@@ -314,7 +321,10 @@ public function handleRequest()
 
         $this->handleMiddleware();
         $this->handleAction($this);
-        ob_clean();
+
+        if (ob_get_level()) {
+            ob_clean();
+        }
 
         return $this->sendResponse();
     }
@@ -365,6 +375,7 @@ private function authorize()
                     ->code('NOT_AUTHORIZED')
                     ->message($message)
             );
+
             $this->sendResponse();
         }
     }
@@ -459,6 +470,11 @@ private function invokeAsReflection($class, $method, $params = [])
             $requestParams[] = $this->getParamValue($param);
         }
 
+        if (wp_is_serving_rest_request() && isset($this->_restResponse)) {
+            // maybe failed at middleware,authorization or validation
+
+            return Response::instance();
+        }
         $params = array_merge($requestParams, $params);
 
         return $reflectionMethod->invoke($reflectionMethod->isStatic() ? null : new $class(), ...$params);
@@ -515,6 +531,8 @@ private function sendApiResponse()
         $restResponse->set_status($this->_response['http_status']);
         $restResponse->set_headers($this->_response['headers']);
 
+        $this->_restResponse = $restResponse; // will USE this to return before middleware or action  excutes
+
         return $restResponse;
     }