com.lob.net.MultipartProcessor Maven / Gradle / Ivy

Go to download
package com.lob.net;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.net.URLConnection;
import java.util.Random;

public class MultipartProcessor {
    private final String boundary;
    private static final String LINE_BREAK = "\r\n";
    private OutputStream outputStream;
    private PrintWriter writer;
    private String charset;
    private java.net.HttpURLConnection conn;

    public static String getBoundary() {
        Random random = new Random();
        Long positiveRandomLong = Math.abs(random.nextLong());
        return String.valueOf(positiveRandomLong);
    }

    public MultipartProcessor(java.net.HttpURLConnection conn, String boundary, String charset)
            throws IOException {
        this.boundary = boundary;
        this.charset = charset;
        this.conn = conn;

        this.outputStream = conn.getOutputStream();
        this.writer = new PrintWriter(new OutputStreamWriter(outputStream,
                charset), true);
    }

    public void addFormField(String name, String value) {
        writer.append("--" + boundary);
        writer.append(LINE_BREAK);
        writer.append("Content-Disposition: form-data; name=\"" + name + "\"");
        writer.append(LINE_BREAK);
        writer.append(LINE_BREAK);
        writer.append(value);
        writer.append(LINE_BREAK);
        writer.flush();
    }

    public void addFileField(String name, File file) throws IOException {
        String fileName = file.getName();
        writer.append("--" + boundary).append(LINE_BREAK);
        writer.append(
                "Content-Disposition: form-data; name=\"" + name
                        + "\"; filename=\"" + fileName + "\"").append(
                LINE_BREAK);

        String probableContentType = URLConnection.guessContentTypeFromName(fileName);
        writer.append("Content-Type: " + probableContentType);
        writer.append(LINE_BREAK);
        writer.append("Content-Transfer-Encoding: binary");
        writer.append(LINE_BREAK);
        writer.append(LINE_BREAK);
        writer.flush();

        FileInputStream inputStream = new FileInputStream(file);
        try {
            byte[] buffer = new byte[4096];
            int bytesRead = -1;
            while ((bytesRead = inputStream.read(buffer)) != -1) {
                outputStream.write(buffer, 0, bytesRead);
            }
            outputStream.flush();
        } finally {
            inputStream.close();
        }

        writer.append(LINE_BREAK);
        writer.flush();
    }

    public void finish() throws IOException {
        writer.append("--" + boundary + "--").append(LINE_BREAK);
        writer.flush();
        writer.close();
        outputStream.flush();
        outputStream.close();
    }

}