import lombok.extern.slf4j.Slf4j;
import org.springframework.context.annotation.Configuration;
import sun.misc.BASE64Decoder;

import javax.annotation.Resource;
import java.io.*;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.UUID;

/**
 * @author huqi 2021/8/12
 */
@Slf4j
@Configuration
public class UploadFile {

    @Resource
    private OpenAuthConfig openAuthConfig;

    /**
     * base64文件上传
     * @param actionUrl 请求地址
     * @param base64File base64文件
     * @param type 文件扩展名
     * @param token token
     * @return
     * @throws IOException
     */
    public String upLoadFilePost(String actionUrl,String base64File,String type, String token) throws IOException {
        String tempFileName = UUID.randomUUID()+"."+type;
        BASE64Decoder decoder = new BASE64Decoder();
        base64File = base64File.replaceAll(" ", "+");
        byte[] bfile = decoder.decodeBuffer(base64File.replace("data:image/"+type+";base64,", ""));
        for (int i = 0; i < bfile.length; ++i) {
            if (bfile[i] < 0) {
                bfile[i] += 256;
            }
        }
        actionUrl = openAuthConfig.getDomainName() + actionUrl;
        log.info("post地址"+actionUrl);
        String BOUNDARY = java.util.UUID.randomUUID().toString();
        String PREFIX = "--", LINEND = "\r\n";
        String MULTIPART_FROM_DATA = "multipart/form-data";
        String CHARSET = "UTF-8";
        URL uri = new URL(actionUrl);
        HttpURLConnection conn = (HttpURLConnection) uri.openConnection();
        conn.setReadTimeout(5 * 1000);
        conn.setDoInput(true);
        conn.setDoOutput(true);
        conn.setUseCaches(false);
        conn.setRequestMethod("POST");
        conn.setRequestProperty("connection", "keep-alive");
        conn.setRequestProperty("Charsert", "UTF-8");
        conn.setRequestProperty("Content-Type", MULTIPART_FROM_DATA
                + ";boundary=" + BOUNDARY);
        conn.setRequestProperty(IOpenApi.BASIC_HEADER_TOKEN, IOpenApi.TOKEN_BEARER + token);
        DataOutputStream outStream = new DataOutputStream(
                conn.getOutputStream());
        // 发送文件数据
        if (bfile != null && bfile.length>0) {
                StringBuilder sb1 = new StringBuilder();
                sb1.append(PREFIX);
                sb1.append(BOUNDARY);
                sb1.append(LINEND);
                sb1.append("Content-Disposition: form-data; name=\"file\"; filename=\""
                        + tempFileName + "\"" + LINEND);
                sb1.append("Content-Type: application/octet-stream; charset="
                        + CHARSET + LINEND);
                sb1.append(LINEND);
                outStream.write(sb1.toString().getBytes());
                outStream.write(bfile);
                outStream.write(LINEND.getBytes());
        }
        // 请求结束标志
        byte[] end_data = (PREFIX + BOUNDARY + PREFIX + LINEND).getBytes();
        outStream.write(end_data);
        outStream.flush();
        // 得到响应码
        int res = conn.getResponseCode();
        if (res == 200) {
            InputStream in = conn.getInputStream();
            InputStreamReader isReader = new InputStreamReader(in);
            BufferedReader bufReader = new BufferedReader(isReader);
            String line = "";
            String data = "";
            while ((line = bufReader.readLine()) != null) {
                data += line;
            }
            outStream.close();
            conn.disconnect();
            return data;
        }
        outStream.close();
        conn.disconnect();
        return null;
    }
}