me.shib.java.lib.jiraclient.Attachment Maven / Gradle / Ivy

Go to download
package me.shib.java.lib.jiraclient;

import net.sf.json.JSON;
import net.sf.json.JSONObject;
import org.apache.http.HttpEntity;
import org.apache.http.HttpResponse;
import org.apache.http.client.methods.HttpGet;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.Date;
import java.util.Map;

/**
 * Represents an issue attachment.
 */
public class Attachment extends Resource {

    private User author = null;
    private String filename = null;
    private Date created = null;
    private int size = 0;
    private String mimeType = null;
    private String content = null;

    /**
     * Creates an attachment from a JSON payload.
     *
     * @param restclient REST client instance
     * @param json       JSON payload
     */
    protected Attachment(RestClient restclient, JSONObject json) {
        super(restclient);

        if (json != null)
            deserialise(json);
    }

    /**
     * Retrieves the given attachment record.
     *
     * @param restclient REST client instance
     * @param id         Internal JIRA ID of the attachment
     * @return an attachment instance
     * @throws JiraException when the retrieval fails
     */
    public static Attachment get(RestClient restclient, String id)
            throws JiraException {

        JSON result = null;

        try {
            result = restclient.get(getBaseUri() + "attachment/" + id);
        } catch (Exception ex) {
            throw new JiraException("Failed to retrieve attachment " + id, ex);
        }

        if (!(result instanceof JSONObject))
            throw new JiraException("JSON payload is malformed");

        return new Attachment(restclient, (JSONObject) result);
    }

    private void deserialise(JSONObject json) {
        Map map = json;

        self = Field.getString(map.get("self"));
        id = Field.getString(map.get("id"));
        author = Field.getResource(User.class, map.get("author"), restclient);
        filename = Field.getString(map.get("filename"));
        created = Field.getDate(map.get("created"));
        size = Field.getInteger(map.get("size"));
        mimeType = Field.getString(map.get("mimeType"));
        content = Field.getString(map.get("content"));
    }

    /**
     * Downloads attachment to byte array
     *
     * @return a byte[]
     * @throws JiraException when the download fails
     */
    public byte[] download()
            throws JiraException {
        ByteArrayOutputStream bos = new ByteArrayOutputStream();
        try {
            HttpGet get = new HttpGet(content);
            HttpResponse response = restclient.getHttpClient().execute(get);
            HttpEntity entity = response.getEntity();
            if (entity != null) {
                InputStream inputStream = entity.getContent();
                int next = inputStream.read();
                while (next > -1) {
                    bos.write(next);
                    next = inputStream.read();
                }
                bos.flush();
            }
        } catch (IOException e) {
            throw new JiraException(String.format("Failed downloading attachment from %s: %s", this.content, e.getMessage()));
        }
        return bos.toByteArray();
    }

    @Override
    public String toString() {
        return getContentUrl();
    }

    public User getAuthor() {
        return author;
    }

    public Date getCreatedDate() {
        return created;
    }

    public String getContentUrl() {
        return content;
    }

    public String getFileName() {
        return filename;
    }

    public String getMimeType() {
        return mimeType;
    }

    public int getSize() {
        return size;
    }
}