org.openmrs.module.kenyaemr.fragment.controller.PatientSearchFragmentController.java Source code

Introduction

Here is the source code for org.openmrs.module.kenyaemr.fragment.controller.PatientSearchFragmentController.java

Source

/**
 * The contents of this file are subject to the OpenMRS Public License
 * Version 1.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) OpenMRS, LLC.  All Rights Reserved.
 */

package org.openmrs.module.kenyaemr.fragment.controller;

import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;

import org.apache.commons.lang.StringUtils;
import org.openmrs.Patient;
import org.openmrs.Visit;
import org.openmrs.api.context.Context;
import org.openmrs.module.kenyaemr.KenyaEmrUiUtils;
import org.openmrs.ui.framework.SimpleObject;
import org.openmrs.ui.framework.UiUtils;
import org.openmrs.ui.framework.annotation.SpringBean;
import org.springframework.web.bind.annotation.RequestParam;

/**
 * Handles AJAX patient searches
 */
public class PatientSearchFragmentController {

    public void controller() {
    }

    public List<SimpleObject> search(@RequestParam(value = "q", required = false) String query,
            @RequestParam(value = "which", required = false) String which,
            @RequestParam(value = "age", required = false) Integer age,
            @RequestParam(value = "ageWindow", defaultValue = "5") int ageWindow, UiUtils ui,
            @SpringBean KenyaEmrUiUtils kenyaUi) {

        if ("checked-in".equals(which)) {
            return withActiveVisits(query, age, ageWindow, ui, kenyaUi);
        }
        if (StringUtils.isBlank(query)) {
            return Collections.emptyList();
        }

        List<Patient> ret = Context.getPatientService().getPatients(query);
        if (age != null) {
            List<Patient> similar = new ArrayList<Patient>();
            for (Patient p : ret) {
                if (Math.abs(p.getAge() - age) <= ageWindow)
                    similar.add(p);
            }
            ret = similar;
        }

        List<Visit> activeVisits = Context.getVisitService().getVisits(null, null, null, null, null, null, null,
                null, null, false, false);
        final Map<Integer, Visit> patientActiveVisits = new HashMap<Integer, Visit>();
        for (Visit v : activeVisits) {
            patientActiveVisits.put(v.getPatient().getPatientId(), v);
        }

        List<SimpleObject> matching = kenyaUi.simplePatients(ret, ui);

        for (SimpleObject so : matching) {
            Visit v = patientActiveVisits.get(so.get("patientId"));
            if (v != null) {
                so.put("extra", "<div class='ke-tag ke-visittag'>" + ui.format(v.getVisitType()) + "<br/><small>"
                        + ui.format(v.getStartDatetime()) + "</small></div>");
            }
        }

        return matching;
    }

    public List<SimpleObject> withActiveVisits(@RequestParam(value = "q", required = false) String query,
            @RequestParam(value = "age", required = false) Integer age,
            @RequestParam(value = "ageWindow", defaultValue = "5") int ageWindow, UiUtils ui,
            @SpringBean KenyaEmrUiUtils kenyaUi) {

        // TODO refactor so it performs faster

        List<SimpleObject> matching = search(query, null, age, ageWindow, ui, kenyaUi);

        List<Visit> activeVisits = Context.getVisitService().getVisits(null, null, null, null, null, null, null,
                null, null, false, false);

        // no query, so we start with all patients with active visits
        if (matching == null || matching.size() == 0) {
            List<Patient> ret = new ArrayList<Patient>();
            for (Visit v : activeVisits) {
                if (!ret.contains(v.getPatient()))
                    ret.add(v.getPatient());
            }
            matching = kenyaUi.simplePatients(ret, ui);
        }

        // intersect query with active visits
        Map<Integer, Visit> patientActiveVisits = new HashMap<Integer, Visit>();
        for (Visit v : activeVisits) {
            patientActiveVisits.put(v.getPatient().getPatientId(), v);
        }
        for (Iterator<SimpleObject> i = matching.iterator(); i.hasNext();) {
            SimpleObject candidate = i.next();
            Visit v = patientActiveVisits.get(candidate.get("patientId"));
            if (v == null) {
                i.remove();
            } else {
                candidate.put("extra", "<div class='ke-tag ke-visittag'>" + ui.format(v.getVisitType())
                        + "<br/><small>" + ui.format(v.getStartDatetime()) + "</small></div>");
            }
        }
        return matching;
    }
}