package cc.mrbird.febs.common.advise;

import cc.mrbird.febs.common.utils.AppContants;
import cn.hutool.core.io.IoUtil;
import cn.hutool.crypto.asymmetric.KeyType;
import cn.hutool.crypto.asymmetric.RSA;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpInputMessage;

import java.io.IOException;
import java.io.InputStream;

/**
 * @author wzy
 * @date 2022-06-14
 **/
public class MyHttpInputMessage implements HttpInputMessage {

    private InputStream body;

    @Override
    public InputStream getBody() throws IOException {
        return body;
    }

    @Override
    public HttpHeaders getHeaders() {
        return null;
    }

    public MyHttpInputMessage(InputStream body) {
        RSA rsa = new RSA(AppContants.REQ_PRIVATE_KEY, AppContants.REQ_PUBLIC_KEY);
        String read = IoUtil.read(body, "utf-8");
        String decrypt = rsa.decryptStr(read, KeyType.PrivateKey);
        this.body = IoUtil.toStream(decrypt.getBytes());
    }
}