CMSgov/dpc-app

package gov.cms.dpc.api.tasks;

import org.hl7.fhir.dstu3.model.Organization;

import javax.ws.rs.WebApplicationException;
import javax.ws.rs.core.Response;
import java.util.List;
import java.util.Map;

public class TasksCommon {

    private static final String ORG_PARAM = "organization";

    private TasksCommon() {
        // Not used
    }

    /**
     * Extract an {@link Organization} from the Task parameters. Specifically from the `organization` query param.
     *
     * @param parameters - {@link Map} of parameters
     * @return - {@link Organization} extracted from query params
     * @throws WebApplicationException if the organization ID is missing
     */
    public static Organization extractOrganization(Map<String, List<String>> parameters) {
        final List<String> organizationCollection = parameters.get(ORG_PARAM);

        if (organizationCollection == null || organizationCollection.isEmpty()) {
            throw new WebApplicationException("Must have organization", Response.Status.BAD_REQUEST);
        }

        final String organizationID = organizationCollection.get(0);
        final Organization orgResource = new Organization();
        orgResource.setId(organizationID);

        return orgResource;
    }
}