diff --git a/src/Outputs/Alert.php b/src/Outputs/Alert.php index 4171a33..ae7db97 100644 --- a/src/Outputs/Alert.php +++ b/src/Outputs/Alert.php @@ -9,7 +9,7 @@ class Alert implements Output { public function output(Collection $detectedQueries, Response $response) { - if ($response->isRedirection()) { + if (stripos($response->headers->get('Content-Type'), 'text/html') !== 0 || $response->isRedirection()) { return; } diff --git a/src/Outputs/Console.php b/src/Outputs/Console.php index 00af6bb..6919952 100644 --- a/src/Outputs/Console.php +++ b/src/Outputs/Console.php @@ -1,5 +1,7 @@ isRedirection()) { + if (stripos($response->headers->get('Content-Type'), 'text/html') !== 0 || $response->isRedirection()) { return; } + $content = $response->getContent(); + $outputContent = $this->getOutputContent($detectedQueries); + $pos = strripos($content, ''); + if (false !== $pos) { $content = substr($content, 0, $pos) . $outputContent . substr($content, $pos); } else { $content = $content . $outputContent; } + // Update the new content and reset the content length $response->setContent($content); + $response->headers->remove('Content-Length'); } + protected function getOutputContent(Collection $detectedQueries) { $output = ''; + return $output; } }