NominalAttributeResource.java

/*
 * To change this template, choose Tools | Templates
 * and open the template in the editor.
 */
package qa.qcri.aidr.predictui.api;

import java.util.List;

import javax.ejb.EJB;
import javax.ejb.Stateless;
import javax.ws.rs.Consumes;
import javax.ws.rs.DELETE;
import javax.ws.rs.GET;
import javax.ws.rs.POST;
import javax.ws.rs.PUT;
import javax.ws.rs.Path;
import javax.ws.rs.PathParam;
import javax.ws.rs.Produces;
import javax.ws.rs.QueryParam;
import javax.ws.rs.core.Context;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Response;
import javax.ws.rs.core.UriInfo;

import org.apache.log4j.Logger;

import qa.qcri.aidr.common.exception.PropertyNotSetException;
import qa.qcri.aidr.common.wrapper.CollectionBriefInfo;
import qa.qcri.aidr.dbmanager.dto.CrisisAttributesDTO;
import qa.qcri.aidr.dbmanager.dto.NominalAttributeDTO;
import qa.qcri.aidr.predictui.facade.CollectionResourceFacade;
import qa.qcri.aidr.predictui.facade.NominalAttributeFacade;
import qa.qcri.aidr.predictui.util.ResponseWrapper;
import qa.qcri.aidr.predictui.util.TaggerAPIConfigurationProperty;
import qa.qcri.aidr.predictui.util.TaggerAPIConfigurator;

/**
 * REST Web Service
 *
 * @author Imran
 */
@Path("/attribute")
@Stateless
public class NominalAttributeResource {

	@Context
	private UriInfo context;
	@EJB
	private NominalAttributeFacade attributeLocalEJB;
	
	@EJB
	private CollectionResourceFacade collectionResourceFacade;

	public NominalAttributeResource() {
	}

	private static Logger logger = Logger.getLogger(NominalAttributeResource.class);

	@GET
	@Produces("application/json")
	@Path("/all")
	public Response getAllNominalAttributes() throws PropertyNotSetException {
		List<NominalAttributeDTO> attributeList = attributeLocalEJB.getAllAttributes();
		ResponseWrapper response = new ResponseWrapper();
		response.setNominalAttributes(attributeList);
		return Response.ok(response).build();
	}

	@GET
	@Produces("application/json")
	@Path("/crisis/all")
	public Response getAllNominalAttributesWithCrisis(@QueryParam("exceptCrisis") Long crisisID) throws PropertyNotSetException {
		List<CrisisAttributesDTO> attributeList = attributeLocalEJB.getAllAttributesExceptCrisis(crisisID);
		ResponseWrapper response = new ResponseWrapper();
		if (attributeList.isEmpty() || attributeList == null) {
			response.setMessage("No attribute left.");
			return Response.ok(response).build();
		}
		response.setCrisisAttributes(attributeList);
		return Response.ok(response).build();
	}

	@GET
	@Produces("application/json")
	@Path("{attributeID}")
	public Response getAttributesnLabelByAttrID(@PathParam("attributeID") Long attributeID) throws PropertyNotSetException {
		NominalAttributeDTO attribute = attributeLocalEJB.getAttributeByID(attributeID);
		ResponseWrapper response = new ResponseWrapper();
		if (attribute != null) {
			return Response.ok(attribute).build();
		}
		response.setMessage("no attribute found with the given id.");
		return Response.ok(response).build();
	}

	@POST
	@Consumes("application/json")
	@Produces(MediaType.APPLICATION_JSON)
	public Response addAttribute(NominalAttributeDTO attribute) throws PropertyNotSetException {
		try {
			NominalAttributeDTO dto = attributeLocalEJB.addAttribute(attribute);
			if (dto != null) {
				return Response.ok(dto).build();
			} else {
				return Response.ok(new ResponseWrapper(TaggerAPIConfigurator.getInstance().getProperty(TaggerAPIConfigurationProperty.STATUS_CODE_FAILED))).build();
			}
		} catch (Exception e) {
			logger.error("failed to attribute attribute: " + attribute.getCode());
			return Response.ok(new ResponseWrapper(TaggerAPIConfigurator.getInstance().getProperty(TaggerAPIConfigurationProperty.STATUS_CODE_FAILED), e.getCause().getCause().getMessage())).build();
		}
	}

	@PUT
	@Consumes("application/json")
	@Produces(MediaType.APPLICATION_JSON)
	public Response editAttribute(NominalAttributeDTO attribute) throws PropertyNotSetException {
		try {
			attribute = attributeLocalEJB.editAttribute(attribute);
		} catch (RuntimeException e) {
			logger.error("failed to edit attribute: " + attribute.getCode());
			return Response.ok(new ResponseWrapper(TaggerAPIConfigurator.getInstance().getProperty(TaggerAPIConfigurationProperty.STATUS_CODE_FAILED), e.getCause().getCause().getMessage())).build();
		}
		return Response.ok(attribute).build();
	}

	@DELETE
	@Path("{id}")
	@Produces(MediaType.APPLICATION_JSON)
	public Response deleteAttribute(@PathParam("id") Long id) throws PropertyNotSetException {
		
		boolean response = attributeLocalEJB.deleteNominalAttributeData(id);
		return response == true ? Response.ok(new ResponseWrapper(TaggerAPIConfigurator.getInstance().getProperty(TaggerAPIConfigurationProperty.STATUS_CODE_FAILED))).build() : Response.ok(new ResponseWrapper(TaggerAPIConfigurator.getInstance().getProperty(TaggerAPIConfigurationProperty.STATUS_CODE_SUCCESS))).build();
	}

	@GET
	@Produces("application/json")
	@Path("/code/{code}")
	public Response isAttributeExists(@PathParam("code") String code) throws PropertyNotSetException {
		Long attributeID = attributeLocalEJB.isAttributeExists(code);
		if (attributeID == null) {
			attributeID = 0l;
		}
		String response = "{\"code\":\"" + code + "\", \"nominalAttributeID\":\"" + attributeID + "\"}";
		return Response.ok(response).build();
	}
	
	@GET
    @Produces("application/json")
    @Path("/{attributeID}/collections")
    public Response getCrisisForNominalAttributeById(@PathParam("attributeID") Integer attributeID, 
    		@QueryParam("crisisType") Integer crisisType, @QueryParam("langFilters") String languageFilters
    		,@QueryParam("collectionId") Integer collectionId) {
     
		Long val = null;
		if(collectionId != null) {
			val = collectionId.longValue();
		}
		List<CollectionBriefInfo> crisisList = collectionResourceFacade.getCrisisForNominalAttributeById
				(attributeID, crisisType, languageFilters, val);
        return Response.ok(crisisList).build();
    }
}