uri-resolver/src/main/java/org/gcube/datatransfer/resolver/services/StorageHub.java

package org.gcube.datatransfer.resolver.services;

import javax.enterprise.context.RequestScoped;
import javax.servlet.http.HttpServletRequest;
import javax.ws.rs.GET;
import javax.ws.rs.Path;
import javax.ws.rs.PathParam;
import javax.ws.rs.core.Context;
import javax.ws.rs.core.Response;
import javax.ws.rs.core.Response.Status;

import org.gcube.common.storagehub.client.StreamDescriptor;
import org.gcube.common.storagehub.client.dsl.StorageHubClient;
import org.gcube.datatransfer.resolver.services.exceptions.BadRequestException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

@Path("shub/{id}")
public class StorageHub {

	/**
	 *
	 */
	public static final String STORAGE_HUB_ID = "id";

	private static Logger logger = LoggerFactory.getLogger(StorageHub.class);

	@RequestScoped
	@PathParam(STORAGE_HUB_ID)
	String id;

	@GET
	@Path("")
	public Response download(@Context HttpServletRequest httpRequest) {
		StorageHubClient shc = new StorageHubClient();

		//Checking mandatory parameter id
		if(id==null || id.isEmpty()){
			logger.error("Path Parameter "+STORAGE_HUB_ID+" not found");
			throw new BadRequestException(httpRequest, Status.NOT_ACCEPTABLE, "Missing mandatory path parameter "+STORAGE_HUB_ID, StorageHub.class);
		}

		StreamDescriptor descriptor = shc.open(id).asFile().download();

		return Response
				.ok(descriptor.getStream())
				.header("content-disposition","attachment; filename = \""+descriptor.getFileName()+"\"").build();
	}

	@GET
	@Path("{version}")
	public Response downloadVersion(@Context HttpServletRequest httpRequest, @PathParam("version") String version) {
		StorageHubClient shc = new StorageHubClient();

		//Checking mandatory parameter id
		if(id==null || id.isEmpty()){
			logger.error("Path Parameter "+STORAGE_HUB_ID+" not found");
			throw new BadRequestException(httpRequest, Status.NOT_ACCEPTABLE, "Missing mandatory path parameter "+STORAGE_HUB_ID, StorageHub.class);
		}

		//Checking mandatory parameter id
		if(version==null || version.isEmpty()){
			logger.error("Parameter 'version' not found");
			throw new BadRequestException(httpRequest, Status.NOT_ACCEPTABLE, "Missing mandatory parameter 'version'", StorageHub.class);
		}

		StreamDescriptor descriptor = shc.open(id).asFile().downloadSpecificVersion(version);
		return Response
				.ok(descriptor.getStream())
				.header("content-disposition","attachment; filename = \""+descriptor.getFileName()+"\"").build();
	}
}