Examples with ResourceImpl - ddf.catalog.resource.impl.ResourceImpl

Example 21 with ResourceImpl

use of ddf.catalog.resource.impl.ResourceImpl in project ddf by codice.

the class OgcUrlResourceReader method retrieveResource.

/**
 * Retrieves a {@link ddf.catalog.resource.Resource} based on a {@link URI} and provided
 * arguments. A connection is made to the {@link URI} to obtain the {@link
 * ddf.catalog.resource.Resource}'s {@link InputStream} and build a {@link ResourceResponse} from
 * that. The {@link ddf.catalog.resource.Resource}'s name gets set to the {@link URI} passed in.
 * Calls {@link URLResourceReader}, if the mime-type is "text/html" it will inject a simple script
 * to redirect to the resourceURI instead of attempting to download it.
 *
 * @param resourceURI A {@link URI} that defines what {@link Resource} to retrieve and how to do
 *     it.
 * @param properties Any additional arguments that should be passed to the {@link
 *     ddf.catalog.resource.ResourceReader}.
 * @return A {@link ResourceResponse} containing the retrieved {@link Resource}.
 * @throws ResourceNotSupportedException
 */
@Override
public ResourceResponse retrieveResource(URI resourceURI, Map<String, Serializable> properties) throws IOException, ResourceNotFoundException, ResourceNotSupportedException {
    LOGGER.debug("Calling URLResourceReader.retrieveResource()");
    ResourceResponse response = urlResourceReader.retrieveResource(resourceURI, properties);
    Resource resource = response.getResource();
    MimeType mimeType = resource.getMimeType();
    LOGGER.debug("mimeType: {}", mimeType);
    if (mimeType != null) {
        String mimeTypeStr = mimeType.toString();
        String detectedMimeType = "";
        if (UNKNOWN_MIME_TYPES.contains(mimeTypeStr)) {
            detectedMimeType = tika.detect(resourceURI.toURL());
        }
        if (StringUtils.contains(detectedMimeType, MediaType.TEXT_HTML) || StringUtils.contains(mimeTypeStr, MediaType.TEXT_HTML)) {
            LOGGER.debug("Detected \"text\\html\". Building redirect script");
            StringBuilder strBuilder = new StringBuilder();
            strBuilder.append("<html><script type=\"text/javascript\">window.location.replace(\"");
            strBuilder.append(resourceURI);
            strBuilder.append("\");</script></html>");
            return new ResourceResponseImpl(new ResourceImpl(new ByteArrayInputStream(strBuilder.toString().getBytes(StandardCharsets.UTF_8)), detectedMimeType, resource.getName()));
        }
    }
    return response;
}

Also used : ResourceImpl(ddf.catalog.resource.impl.ResourceImpl) ResourceResponse(ddf.catalog.operation.ResourceResponse) ByteArrayInputStream(java.io.ByteArrayInputStream) Resource(ddf.catalog.resource.Resource) ResourceResponseImpl(ddf.catalog.operation.impl.ResourceResponseImpl) MimeType(javax.activation.MimeType)

Aggregations

ResourceImpl (ddf.catalog.resource.impl.ResourceImpl)21 ResourceResponseImpl (ddf.catalog.operation.impl.ResourceResponseImpl)10 IOException (java.io.IOException)10 Test (org.junit.Test)10 Resource (ddf.catalog.resource.Resource)9 InputStream (java.io.InputStream)8 ResourceResponse (ddf.catalog.operation.ResourceResponse)7 FileInputStream (java.io.FileInputStream)7 Failure (org.junit.runner.notification.Failure)6 MimeType (javax.activation.MimeType)5 ResourceNotFoundException (ddf.catalog.resource.ResourceNotFoundException)4 URI (java.net.URI)4 HashMap (java.util.HashMap)4 MultivaluedHashMap (javax.ws.rs.core.MultivaluedHashMap)4 CswRecordCollection (org.codice.ddf.spatial.ogc.csw.catalog.common.CswRecordCollection)4 ResourceReader (ddf.catalog.resource.ResourceReader)3 Serializable (java.io.Serializable)3 ResourceRequestById (ddf.catalog.operation.impl.ResourceRequestById)2 ResourceNotSupportedException (ddf.catalog.resource.ResourceNotSupportedException)2 BufferedInputStream (java.io.BufferedInputStream)2