org.motechproject.server.omod.web.controller.DemoEnrollController.java Source code

Introduction

Here is the source code for org.motechproject.server.omod.web.controller.DemoEnrollController.java

Source

/**
 * MOTECH PLATFORM OPENSOURCE LICENSE AGREEMENT
 *
 * Copyright (c) 2010-11 The Trustees of Columbia University in the City of
 * New York and Grameen Foundation USA.  All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * 1. Redistributions of source code must retain the above copyright notice,
 * this list of conditions and the following disclaimer.
 *
 * 2. Redistributions in binary form must reproduce the above copyright notice,
 * this list of conditions and the following disclaimer in the documentation
 * and/or other materials provided with the distribution.
 *
 * 3. Neither the name of Grameen Foundation USA, Columbia University, or
 * their respective contributors may be used to endorse or promote products
 * derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY GRAMEEN FOUNDATION USA, COLUMBIA UNIVERSITY
 * AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING,
 * BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL GRAMEEN FOUNDATION
 * USA, COLUMBIA UNIVERSITY OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA,
 * OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
 * EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

package org.motechproject.server.omod.web.controller;

import javax.servlet.http.HttpSession;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.motechproject.server.service.ContextService;
import org.motechproject.server.omod.web.model.WebModelConverter;
import org.motechproject.server.omod.web.model.WebPatient;
import org.motechproject.server.svc.OpenmrsBean;
import org.motechproject.server.svc.RegistrarBean;
import org.openmrs.Patient;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.beans.propertyeditors.StringTrimmerEditor;
import org.springframework.stereotype.Controller;
import org.springframework.ui.ModelMap;
import org.springframework.validation.Errors;
import org.springframework.validation.ValidationUtils;
import org.springframework.web.bind.WebDataBinder;
import org.springframework.web.bind.annotation.InitBinder;
import org.springframework.web.bind.annotation.ModelAttribute;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.SessionAttributes;
import org.springframework.web.bind.support.SessionStatus;

@Controller
@RequestMapping("/module/motechmodule/demo-enrollpatient")
@SessionAttributes("enrollpatient")
public class DemoEnrollController {

    private static Log log = LogFactory.getLog(DemoEnrollController.class);

    private WebModelConverter webModelConverter;

    @Autowired
    @Qualifier("registrarBean")
    private RegistrarBean registrarBean;

    @Autowired
    @Qualifier("openmrsBean")
    private OpenmrsBean openmrsBean;

    private ContextService contextService;

    @Autowired
    public void setContextService(ContextService contextService) {
        this.contextService = contextService;
    }

    public void setOpenmrsBean(OpenmrsBean openmrsBean) {
        this.openmrsBean = openmrsBean;
    }

    public void setRegistrarBean(RegistrarBean registrarBean) {
        this.registrarBean = registrarBean;
    }

    @Autowired
    public void setWebModelConverter(WebModelConverter webModelConverter) {
        this.webModelConverter = webModelConverter;
    }

    @InitBinder
    public void initBinder(WebDataBinder binder) {
        binder.registerCustomEditor(String.class, new StringTrimmerEditor(true));
    }

    @RequestMapping(method = RequestMethod.GET)
    public void viewForm(@RequestParam(required = false) Integer id, ModelMap model) {
    }

    @ModelAttribute("enrollpatient")
    public WebPatient getWebPatient(@RequestParam(required = false) Integer id, HttpSession session) {
        WebPatient result = new WebPatient();
        if (id != null) {
            Patient patient = contextService.getPatientService().getPatient(id);

            if (patient != null) {
                webModelConverter.patientToWeb(patient, result);
            }
        }

        Integer motechId = (Integer) session.getAttribute("demoLastMotechId");
        if (motechId != null)
            result.setMotechId(motechId);

        return result;
    }

    @RequestMapping(method = RequestMethod.POST)
    public String submitForm(@ModelAttribute("enrollpatient") WebPatient webPatient, Errors errors, ModelMap model,
            SessionStatus status, HttpSession session) {

        log.debug("Enroll Demo Patient");

        ValidationUtils.rejectIfEmpty(errors, "motechId", "motechmodule.motechId.required");

        Patient patient = null;
        if (webPatient.getMotechId() != null) {
            patient = openmrsBean.getPatientByMotechId(webPatient.getMotechId().toString());
            if (patient == null) {
                errors.rejectValue("motechId", "motechmodule.motechId.notexist");
            }
        }

        if (!Boolean.TRUE.equals(webPatient.getConsent())) {
            errors.rejectValue("consent", "motechmodule.consent.required");
        }

        if (!errors.hasErrors()) {
            registrarBean.demoEnrollPatient(patient);

            model.addAttribute("successMsg", "motechmodule.Demo.Patient.enroll.success");

            status.setComplete();

            session.removeAttribute("demoLastMotechId");

            return "redirect:/module/motechmodule/demo-success.htm";
        }

        return "/module/motechmodule/demo-enrollpatient";
    }
}