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

Go to download
Show more of this group Show more artifacts with this name
Show all versions of eligible-java Show documentation
Java bindings for Eligible APIs (https://eligible.com).
There is a newer version: 1.13.13
package com.eligible.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 static com.eligible.util.NetworkUtil.moveContent;

/**
 * Implementation for making HTTP multipart requests.
 */
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;

    /**
     * Create a HTTP Multipart Processor.
     *
     * @param conn     HTTP connection
     * @param boundary boundary
     * @param charset  charset
     * @throws IOException on network issue
     */
    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);
    }

    /**
     * Add for data into request.
     *
     * @param name  field name
     * @param value field value
     */
    public void addFormField(String name, String value) {
        writer.append("--" + boundary).append(LINE_BREAK);
        writer.append("Content-Disposition: form-data; name=\"" + name + "\"")
                .append(LINE_BREAK);
        writer.append(LINE_BREAK);
        writer.append(value).append(LINE_BREAK);
        writer.flush();
    }

    /**
     * Add file to be uploading into request.
     *
     * @param name field name
     * @param file file object
     * @throws IOException on network issue
     */
    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)
                .append(LINE_BREAK);
        writer.append("Content-Transfer-Encoding: binary").append(LINE_BREAK);
        writer.append(LINE_BREAK);
        writer.flush();

        try (FileInputStream inputStream = new FileInputStream(file)) {
            moveContent(inputStream, outputStream);
        }

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

    /**
     * Flush streams and cleanup.
     *
     * @throws IOException on network issue
     */
    public void finish() throws IOException {
        writer.append("--" + boundary + "--").append(LINE_BREAK);
        writer.flush();
        writer.close();
        outputStream.flush();
        outputStream.close();
    }

}