Java.libraries.feign.FormAwareEncoder.mustache Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of swagger-codegen Show documentation
The newest version!
package {{invokerPackage}};

import java.io.*;
import java.lang.reflect.Type;
import java.net.URLEncoder;
import java.net.URLConnection;
import java.nio.charset.Charset;
import java.util.*;

import java.text.DateFormat;
import java.text.SimpleDateFormat;

import feign.codec.EncodeException;
import feign.codec.Encoder;
import feign.RequestTemplate;

{{>generatedAnnotation}}
public class FormAwareEncoder implements Encoder {
  public static final String UTF_8 = "utf-8";
  private static final String LINE_FEED = "\r\n";
  private static final String TWO_DASH = "--";
  private static final String BOUNDARY = "----------------314159265358979323846";

  private byte[] lineFeedBytes;
  private byte[] boundaryBytes;
  private byte[] twoDashBytes;
  private byte[] atBytes;
  private byte[] eqBytes;

  private final Encoder delegate;
  private final DateFormat dateFormat;

  public FormAwareEncoder(Encoder delegate) {
    this.delegate = delegate;
    // Use RFC3339 format for date and datetime.
    // See http://xml2rfc.ietf.org/public/rfc/html/rfc3339.html#anchor14
    this.dateFormat = new SimpleDateFormat("yyyy-MM-dd'T'HH:mm:ss.SSSXXX");

    // Use UTC as the default time zone.
    this.dateFormat.setTimeZone(TimeZone.getTimeZone("UTC"));
    try {
      this.lineFeedBytes = LINE_FEED.getBytes(UTF_8);
      this.boundaryBytes = BOUNDARY.getBytes(UTF_8);
      this.twoDashBytes = TWO_DASH.getBytes(UTF_8);
      this.atBytes = "&".getBytes(UTF_8);
      this.eqBytes = "=".getBytes(UTF_8);
    } catch (UnsupportedEncodingException e) {
      e.printStackTrace();
    }
  }

  public void encode(Object object, Type bodyType, RequestTemplate template) throws EncodeException {
    if (object instanceof Map) {
      try {
        encodeFormParams(template, (Map) object);
      } catch (IOException e) {
        throw new EncodeException("Failed to create request", e);
      }
    } else {
      delegate.encode(object, bodyType, template);
    }
  }

  private void encodeFormParams(RequestTemplate template, Map formParams) throws IOException {
    ByteArrayOutputStream baos = new ByteArrayOutputStream();

    boolean isMultiPart = isMultiPart(formParams);
    boolean isFirstField = true;
    for (Map.Entry param : formParams.entrySet()) {
      String keyStr = param.getKey();
      if (param.getValue() instanceof File) {
          addFilePart(baos, keyStr, (File) param.getValue());
      } else {
        String valueStr = parameterToString(param.getValue());
        if (isMultiPart) {
          addMultiPartFormField(baos, keyStr, valueStr);
        } else {
          addEncodedFormField(baos, keyStr, valueStr, isFirstField);
          isFirstField = false;
        }
      }
    }

    if (isMultiPart) {
      baos.write(lineFeedBytes);
      baos.write(twoDashBytes);
      baos.write(boundaryBytes);
      baos.write(twoDashBytes);
      baos.write(lineFeedBytes);
    }

    String contentType = isMultiPart ? "multipart/form-data; boundary=" + BOUNDARY : "application/x-www-form-urlencoded";
    template.header("Content-type");
    template.header("Content-type", contentType);
    template.header("MIME-Version", "1.0");
    template.body(baos.toByteArray(), Charset.forName(UTF_8));
  }

  /*
   * Currently only supports text files
   */
  private void addFilePart(ByteArrayOutputStream baos, String fieldName, File uploadFile) throws IOException {
    String fileName = uploadFile.getName();
    baos.write(twoDashBytes);
    baos.write(boundaryBytes);
    baos.write(lineFeedBytes);

    String contentDisposition = "Content-Disposition: form-data; name=\"" + fieldName
            + "\"; filename=\"" + fileName + "\"";
    baos.write(contentDisposition.getBytes(UTF_8));
    baos.write(lineFeedBytes);
    String contentType = "Content-Type: " + URLConnection.guessContentTypeFromName(fileName);
    baos.write(contentType.getBytes(UTF_8));
    baos.write(lineFeedBytes);
    baos.write(lineFeedBytes);

    BufferedReader reader = new BufferedReader(new FileReader(uploadFile));
    InputStream input = new FileInputStream(uploadFile);
    byte[] bytes = new byte[4096];
    int len = bytes.length;
    while ((len = input.read(bytes)) != -1) {
      baos.write(bytes, 0, len);
      baos.write(lineFeedBytes);
    }

    baos.write(lineFeedBytes);
  }

  private void addEncodedFormField(ByteArrayOutputStream baos, String name, String value, boolean isFirstField) throws IOException {
    if (!isFirstField) {
      baos.write(atBytes);
    }

    String encodedName = URLEncoder.encode(name, UTF_8);
    String encodedValue = URLEncoder.encode(value, UTF_8);
    baos.write(encodedName.getBytes(UTF_8));
    baos.write("=".getBytes(UTF_8));
    baos.write(encodedValue.getBytes(UTF_8));
  }

  private void addMultiPartFormField(ByteArrayOutputStream baos, String name, String value) throws IOException {
    baos.write(twoDashBytes);
    baos.write(boundaryBytes);
    baos.write(lineFeedBytes);

    String contentDisposition = "Content-Disposition: form-data; name=\"" + name + "\"";
    String contentType = "Content-Type: text/plain; charset=utf-8";

    baos.write(contentDisposition.getBytes(UTF_8));
    baos.write(lineFeedBytes);
    baos.write(contentType.getBytes(UTF_8));
    baos.write(lineFeedBytes);
    baos.write(lineFeedBytes);
    baos.write(value.getBytes(UTF_8));
    baos.write(lineFeedBytes);
  }

  private boolean isMultiPart(Map formParams) {
    boolean isMultiPart = false;
    for (Map.Entry entry : formParams.entrySet()) {
        if (entry.getValue() instanceof File) {
            isMultiPart = true;
            break;
        }
    }
    return isMultiPart;
  }

  /**
   * Format the given parameter object into string.
   */
  public String parameterToString(Object param) {
    if (param == null) {
      return "";
    } else if (param instanceof Date) {
      return formatDate((Date) param);
    } else if (param instanceof Collection) {
      StringBuilder b = new StringBuilder();
      for(Object o : (Collection)param) {
        if(b.length() > 0) {
          b.append(",");
        }
        b.append(String.valueOf(o));
      }
      return b.toString();
    } else {
      return String.valueOf(param);
    }
  }

  /**
   * Format the given Date object into string.
   */
  public String formatDate(Date date) {
    return dateFormat.format(date);
  }
}