CashPointSearchHandler.java

/*
 * The contents of this file are subject to the OpenMRS Public License
 * Version 2.0 (the "License"); you may not use this file except in
 * compliance with the License. You may obtain a copy of the License at
 * http://license.openmrs.org
 *
 * Software distributed under the License is distributed on an "AS IS"
 * basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See
 * the License for the specific language governing rights and
 * limitations under the License.
 *
 * Copyright (C) OpenHMIS.  All Rights Reserved.
 */
package org.openmrs.module.webservices.rest.search;

import java.util.Arrays;
import java.util.List;

import org.apache.commons.lang.StringUtils;
import org.openmrs.Location;
import org.openmrs.api.LocationService;
import org.openmrs.api.context.Context;
import org.openmrs.module.openhmis.cashier.api.ICashPointService;
import org.openmrs.module.openhmis.cashier.api.model.CashPoint;
import org.openmrs.module.openhmis.cashier.web.CashierRestConstants;
import org.openmrs.module.openhmis.commons.api.PagingInfo;
import org.openmrs.module.openhmis.inventory.api.model.Department;
import org.openmrs.module.webservices.rest.resource.AlreadyPagedWithLength;
import org.openmrs.module.webservices.rest.resource.PagingUtil;
import org.openmrs.module.webservices.rest.web.RequestContext;
import org.openmrs.module.webservices.rest.web.resource.api.PageableResult;
import org.openmrs.module.webservices.rest.web.resource.api.SearchConfig;
import org.openmrs.module.webservices.rest.web.resource.api.SearchHandler;
import org.openmrs.module.webservices.rest.web.resource.api.SearchQuery;
import org.springframework.stereotype.Component;

/**
 * Search handler for {@link CashPoint}s.
 */
@Component
public class CashPointSearchHandler implements SearchHandler {
	private final SearchConfig searchConfig = new SearchConfig("default", CashierRestConstants.CASH_POINT_RESOURCE,
	        Arrays.asList("*"), Arrays.asList(new SearchQuery.Builder(
	                "Find a cashpoint by its name, optionally filtering by location").withRequiredParameters("q")
	                .withOptionalParameters("location_uuid").build()));

	@Override
	public PageableResult search(RequestContext context) {
		String query = context.getParameter("q");
		String locationUuid = context.getParameter("location_uuid");
		query = query.isEmpty() ? null : query;
		locationUuid = StringUtils.isEmpty(locationUuid) ? null : locationUuid;

		ICashPointService service = Context.getService(ICashPointService.class);
		LocationService locationService = Context.getLocationService();
		Location location = locationService.getLocationByUuid(locationUuid);
		PagingInfo pagingInfo = PagingUtil.getPagingInfoFromContext(context);

		List<CashPoint> cashpoints = null;
		PageableResult results = null;

		if (locationUuid == null) {
			// Do a name search
			cashpoints = service.getByNameFragment(query, context.getIncludeAll(), pagingInfo);
		} else if (query == null) {
			//performs the location search
			cashpoints = service.getCashPointsByLocation(location, context.getIncludeAll(), pagingInfo);
		} else {
			// Do a name & location search
			cashpoints = service.getCashPointsByLocationAndName(location, query, context.getIncludeAll(), pagingInfo);
		}

		results =
		        new AlreadyPagedWithLength<CashPoint>(context, cashpoints, pagingInfo.hasMoreResults(),
		                pagingInfo.getTotalRecordCount());
		return results;
	}

	@Override
	public SearchConfig getSearchConfig() {
		return searchConfig;
	}
}