SensitiveMiddleware.php 2.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788
  1. <?php
  2. declare(strict_types=1);
  3. namespace App\Middleware\Auth;
  4. use App\Tools\PublicData;
  5. use Hyperf\HttpServer\Contract\RequestInterface;
  6. use Hyperf\HttpServer\Contract\ResponseInterface as HttpResponse;
  7. use Psr\Container\ContainerInterface;
  8. use Psr\Http\Message\ResponseInterface;
  9. use Psr\Http\Message\ServerRequestInterface;
  10. use Psr\Http\Server\MiddlewareInterface;
  11. use Psr\Http\Server\RequestHandlerInterface;
  12. use Hyperf\HttpMessage\Stream\SwooleStream;
  13. use Hyperf\Di\Annotation\Inject;
  14. use Hyperf\Redis\Redis;
  15. //use Swoole\Http\Request;
  16. class SensitiveMiddleware implements MiddlewareInterface
  17. {
  18. protected ContainerInterface $container;
  19. protected RequestInterface $request;
  20. protected HttpResponse $response;
  21. #[Inject]
  22. protected Redis $redis;
  23. const STREAM_URL = [
  24. '/news/addArticle',
  25. '/news/updateArticle'
  26. ];
  27. public function __construct( RequestInterface $request, HttpResponse $response)
  28. {
  29. $this->request = $request;
  30. $this->response = $response;
  31. }
  32. public function process(ServerRequestInterface $request, RequestHandlerInterface $handler): ResponseInterface
  33. {
  34. $uri = $request->getUri();
  35. $path = $uri->getPath();
  36. if (!$path || !in_array($path, self::STREAM_URL)) {
  37. return $handler->handle($request);
  38. }
  39. try {
  40. $badWords = $this->redis->sMembers('black_word');
  41. // $whiteWords = $this->redis->sMembers('white_word');
  42. $params = $this->request->all();
  43. $concatenated = "";
  44. if ($params) {
  45. foreach ($params as $value) {
  46. if (is_array($value)) {
  47. $concatenated .= json_encode($value);
  48. } else {
  49. $concatenated .= (string)$value;
  50. }
  51. }
  52. }
  53. foreach ($badWords as $badWord) {
  54. // if (in_array($badWord, $whiteWords)) {
  55. // continue;
  56. // }
  57. if (str_contains($concatenated, $badWord)) {
  58. $message = '发现违禁词: '. $badWord;
  59. return $this->response->json(
  60. [
  61. 'code' => 0,
  62. 'data' => [$message],
  63. 'message' => "该内容存在非法信息,请修改后重新发布"
  64. ]
  65. );
  66. }
  67. }
  68. return $handler->handle($request);
  69. } catch (\Exception $e) {
  70. return $this->response->raw(json_encode([
  71. 'code' => -1,
  72. 'data' => [],
  73. 'message' => '系统错误:' . $e->getMessage()
  74. ]));
  75. }
  76. }
  77. }