package com.ruoyi.common.filter;


import javax.servlet.ReadListener;
import javax.servlet.ServletInputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;
import java.io.BufferedReader;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStreamReader;

public class CustomRequestWrapper extends HttpServletRequestWrapper {
    private final String body;

    /**
     * 构造函数，用于创建自定义请求包装器。
     *
     * @param request 原始 HTTP 请求对象
     * @param body    新的请求体内容
     * @throws IOException 如果读取或处理请求体时发生错误
     */
    public CustomRequestWrapper(HttpServletRequest request, String body) throws IOException {
        super(request);
        this.body = body;
    }

    /**
     * 重写 getInputStream 方法，返回一个新的 ServletInputStream，
     * 其中包含修改后的请求体内容。
     *
     * @return 包含修改后请求体内容的 ServletInputStream
     * @throws IOException 如果读取输入流时发生错误
     */
    @Override
    public ServletInputStream getInputStream() throws IOException {
        final ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(body.getBytes());
        return new ServletInputStream() {
            @Override
            public boolean isFinished() {
                // 返回是否已经完成读取
                return false;
            }

            @Override
            public boolean isReady() {
                // 返回是否准备好读取数据
                return false;
            }

            @Override
            public void setReadListener(ReadListener readListener) {
                // 设置读监听器，这里不需要实现
            }

            @Override
            public int read() throws IOException {
                // 从字节数组输入流中读取下一个字节
                return byteArrayInputStream.read();
            }
        };
    }

    /**
     * 重写 getReader 方法，返回一个新的 BufferedReader，
     * 其中包含修改后的请求体内容。
     *
     * @return 包含修改后请求体内容的 BufferedReader
     * @throws IOException 如果读取输入流时发生错误
     */
    @Override
    public BufferedReader getReader() throws IOException {
        return new BufferedReader(new InputStreamReader(this.getInputStream()));
    }
}