Ejemplo n.º 1
0
  /**
   * Returns the contents of an attachment
   *
   * @param assetId The ID of the asset owning the attachment
   * @param attachmentId The ID of the attachment
   * @return The input stream for the attachment
   * @throws IOException
   * @throws RequestFailureException
   */
  public InputStream getAttachment(String assetId, String attachmentId)
      throws IOException, BadVersionException, RequestFailureException {
    // Get the URL for the attachment
    Attachment attachment = getAttachmentMetaData(assetId, attachmentId);

    // accept license for type CONTENT
    HttpURLConnection connection;
    if (attachment.getType() == Attachment.Type.CONTENT) {
      connection = createHttpURLConnection(attachment.getUrl() + "?license=agree");
    } else {
      connection = createHttpURLConnection(attachment.getUrl());
    }

    // If the attachment was a link and we have a basic auth userid + password specified
    // we are attempting to access the files staged from a protected site so authorise for it
    if (attachment.getLinkType() == LinkType.DIRECT) {
      if ((loginInfo.getAttachmentBasicAuthUserId() != null)
          && (loginInfo.getAttachmentBasicAuthPassword() != null)) {
        String userpass =
            loginInfo.getAttachmentBasicAuthUserId()
                + ":"
                + loginInfo.getAttachmentBasicAuthPassword();
        String basicAuth =
            "Basic "
                + javax.xml.bind.DatatypeConverter.printBase64Binary(
                    userpass.getBytes(Charset.forName("UTF-8")));
        connection.setRequestProperty("Authorization", basicAuth);
      }
    }

    connection.setRequestMethod("GET");
    testResponseCode(connection);
    return connection.getInputStream();
  }
Ejemplo n.º 2
0
  public Attachment addAttachment(final String assetId, final AttachmentSummary attSummary)
      throws IOException, BadVersionException, RequestFailureException {
    final Attachment attach = attSummary.getAttachment();
    final String name = attSummary.getName();
    // Info about the attachment goes into the URL
    String urlString = "/assets/" + assetId + "/attachments?name=" + name;
    if (attach.getType() != null) {
      urlString = urlString + "&type=" + attach.getType().toString();
    }

    HttpURLConnection connection = createHttpURLConnectionToMassive(urlString);
    if (attSummary.getURL() == null) {
      writeMultiPart(assetId, attSummary, connection);
    } else {
      writeSinglePart(assetId, attSummary, connection);
    }

    ByteArrayOutputStream baos = new ByteArrayOutputStream();
    testResponseCode(connection);
    InputStream is = connection.getInputStream();
    int len = 0;
    while ((len = is.read()) != -1) {
      baos.write((byte) len);
    }
    is.close();
    baos.close();

    Attachment attachment =
        JSONAssetConverter.readValue(
            new ByteArrayInputStream(baos.toByteArray()), Attachment.class);

    return attachment;
  }
Ejemplo n.º 3
0
 /**
  * This will delete an asset and all its attachments
  *
  * @param assetId The id of the asset
  * @return <code>true</code> if the asset and all its attachments are deleted. Note this is not
  *     atomic so some attachments may be deleted and <code>false</code> returned.
  * @throws IOException
  * @throws RequestFailureException
  */
 public void deleteAssetAndAttachments(String assetId)
     throws IOException, RequestFailureException {
   Asset ass = getUnverifiedAsset(assetId);
   List<Attachment> attachments = ass.getAttachments();
   if (attachments != null) {
     for (Attachment attachment : attachments) {
       deleteAttachment(assetId, attachment.get_id());
     }
   }
   // Now delete the asset
   deleteAsset(assetId);
 }
Ejemplo n.º 4
0
  private byte[] getStartBytes(final AttachmentSummary attSummary, String boundary)
      throws IOException {
    final String name = attSummary.getName();
    final File fileToWrite = attSummary.getFile();
    final Attachment attach = attSummary.getAttachment();
    ByteArrayOutputStream startOutputStream = new ByteArrayOutputStream();

    try {
      OutputStreamWriter writer =
          new OutputStreamWriter(startOutputStream, Charset.forName("UTF-8"));
      writer.write("--" + boundary + NEWLINE);
      writer.write("Content-Disposition: form-data; name=\"attachmentInfo\"" + NEWLINE);
      writer.write("Content-Type: application/json" + NEWLINE);
      writer.write(NEWLINE);
      writer.flush();
      JSONAssetConverter.writeValue(startOutputStream, attach);
      writer.write(NEWLINE);

      writer.write("--" + boundary + NEWLINE);
      writer.write(
          "Content-Disposition: form-data; name=\""
              + name
              + "\"; filename=\""
              + fileToWrite.getName()
              + "\""
              + NEWLINE);
      String uploadType = "application/binary"; // default value
      if (attach.getType() == null) {
        // Attachments should have a Type specified
        throw new IllegalArgumentException("Attachments must have a Type specified");
      } else {
        switch (attach.getType()) {
          case LICENSE:
            uploadType = "text/html";
            break;
          case LICENSE_AGREEMENT:
          case LICENSE_INFORMATION:
            uploadType = "text/plain";
            break;
          default:
            break;
        }
      }
      writer.write("Content-Type: " + uploadType + NEWLINE);
      writer.write(NEWLINE);
      writer.close();
    } finally {
      if (startOutputStream != null) {
        startOutputStream.close();
      }
    }
    return startOutputStream.toByteArray();
  }
Ejemplo n.º 5
0
  /**
   * Returns the meta data about an attachment
   *
   * @param assetId The ID of the asset owning the attachment
   * @param attachmentId The ID of the attachment
   * @return The attachment meta data
   * @throws IOException
   * @throws RequestFailureException
   */
  public Attachment getAttachmentMetaData(String assetId, String attachmentId)
      throws IOException, BadVersionException, RequestFailureException {
    // At the moment can only get all attachments
    Asset ass = getAsset(assetId);
    List<Attachment> allAttachments = ass.getAttachments();
    for (Attachment attachment : allAttachments) {
      if (attachmentId.equals(attachment.get_id())) {
        return attachment;
      }
    }

    // Didn't find it so just return null
    return null;
  }
Ejemplo n.º 6
0
  /**
   * This method will update an existing attachment on an asset. Note that Massive currently doesn't
   * support update attachment so this will do a delete and an add.
   *
   * @param assetId The ID of the asset that the attachment is attached to
   * @param name The name of the attachment to update
   * @param file The file to attach
   * @param attach Attachment metadata
   * @return
   * @throws IOException
   * @throws RequestFailureException
   */
  public Attachment updateAttachment(String assetId, AttachmentSummary summary)
      throws IOException, BadVersionException, RequestFailureException {
    // First find the attachment to update
    // TODO update when massive supports proper updates to attachments
    Asset ass = getAsset(assetId);
    List<Attachment> attachments = ass.getAttachments();

    if (attachments != null) {
      for (Attachment attachment : attachments) {
        if (attachment.getName().equals(summary.getName())) {
          this.deleteAttachment(assetId, attachment.get_id());
          break;
        }
      }
    }
    return this.addAttachment(assetId, summary);
  }