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

Java tutorial

Introduction

Here is the source code for org.motechproject.server.omod.web.controller.PregnancyController.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 java.text.SimpleDateFormat;
import java.util.Calendar;
import java.util.Date;

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.RegistrarBean;
import org.motechproject.server.util.MotechConstants;
import org.motechproject.ws.ContactNumberType;
import org.motechproject.ws.Gender;
import org.motechproject.ws.MediaType;
import org.openmrs.Patient;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.beans.propertyeditors.CustomDateEditor;
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/pregnancy")
@SessionAttributes("pregnancy")
public class PregnancyController {

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

    private WebModelConverter webModelConverter;

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

    private ContextService contextService;

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

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

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

    @InitBinder
    public void initBinder(WebDataBinder binder) {
        String datePattern = "dd/MM/yyyy";
        SimpleDateFormat dateFormat = new SimpleDateFormat(datePattern);
        dateFormat.setLenient(false);
        binder.registerCustomEditor(Date.class, new CustomDateEditor(dateFormat, true, datePattern.length()));

        String timePattern = MotechConstants.TIME_FORMAT_DELIVERY_TIME;
        SimpleDateFormat timeFormat = new SimpleDateFormat(timePattern);
        dateFormat.setLenient(false);
        binder.registerCustomEditor(Date.class, "timeOfDay",
                new CustomDateEditor(timeFormat, true, timePattern.length()));

        binder.registerCustomEditor(String.class, new StringTrimmerEditor(true));
    }

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

    @ModelAttribute("pregnancy")
    public WebPatient getWebPregnancy(@RequestParam(required = false) Integer id) {
        WebPatient result = new WebPatient();
        if (id != null) {
            Patient patient = contextService.getPatientService().getPatient(id);

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

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

        log.debug("Register New Pregnancy on Existing Patient");

        Patient patient = null;
        if (pregnancy.getId() != null) {
            patient = registrarBean.getPatientById(pregnancy.getId());
            if (patient == null) {
                errors.reject("motechmodule.id.notexist");
            }
        } else {
            errors.reject("motechmodule.id.required");
        }

        if (!Gender.FEMALE.equals(pregnancy.getSex())) {
            errors.reject("motechmodule.sex.female.required");
        }
        if (registrarBean.getActivePregnancy(pregnancy.getId()) != null) {
            errors.reject("motechmodule.Pregnancy.active");
        }

        ValidationUtils.rejectIfEmptyOrWhitespace(errors, "dueDate", "motechmodule.dueDate.required");
        if (pregnancy.getDueDate() != null) {
            Calendar calendar = Calendar.getInstance();
            calendar.add(Calendar.MONTH, 9);
            if (pregnancy.getDueDate().after(calendar.getTime())) {
                errors.rejectValue("dueDate", "motechmodule.dueDate.overninemonths");
            }
        }
        ValidationUtils.rejectIfEmptyOrWhitespace(errors, "dueDateConfirmed",
                "motechmodule.dueDateConfirmed.required");

        ValidationUtils.rejectIfEmptyOrWhitespace(errors, "enroll", "motechmodule.enroll.required");

        if (Boolean.TRUE.equals(pregnancy.getEnroll())) {
            if (!Boolean.TRUE.equals(pregnancy.getConsent())) {
                errors.rejectValue("consent", "motechmodule.consent.required");
            }
            ValidationUtils.rejectIfEmptyOrWhitespace(errors, "phoneType", "motechmodule.phoneType.required");
            if (pregnancy.getPhoneType() == ContactNumberType.PERSONAL
                    || pregnancy.getPhoneType() == ContactNumberType.HOUSEHOLD) {
                ValidationUtils.rejectIfEmptyOrWhitespace(errors, "phoneNumber",
                        "motechmodule.phoneNumber.required");
            }
            ValidationUtils.rejectIfEmptyOrWhitespace(errors, "mediaType", "motechmodule.mediaType.required");
            if (pregnancy.getPhoneType() == ContactNumberType.PUBLIC && pregnancy.getMediaType() != null
                    && pregnancy.getMediaType() != MediaType.VOICE) {
                errors.rejectValue("mediaType", "motechmodule.mediaType.voice");
            }
            ValidationUtils.rejectIfEmptyOrWhitespace(errors, "language", "motechmodule.language.required");
            if (pregnancy.getMediaType() == MediaType.TEXT && pregnancy.getLanguage() != null
                    && !pregnancy.getLanguage().equals("en")) {
                errors.rejectValue("language", "motechmodule.language.english");
            }
            ValidationUtils.rejectIfEmptyOrWhitespace(errors, "interestReason",
                    "motechmodule.interestReason.required");
            ValidationUtils.rejectIfEmptyOrWhitespace(errors, "howLearned", "motechmodule.howLearned.required");
        }
        if (pregnancy.getPhoneNumber() != null
                && !pregnancy.getPhoneNumber().matches(MotechConstants.PHONE_REGEX_PATTERN)) {
            errors.rejectValue("phoneNumber", "motechmodule.phoneNumber.invalid");
        }

        validateTextLength(errors, "phoneNumber", pregnancy.getPhoneNumber(),
                MotechConstants.MAX_STRING_LENGTH_OPENMRS);

        if (!errors.hasErrors()) {
            registrarBean.registerPregnancy(patient, pregnancy.getDueDate(), pregnancy.getDueDateConfirmed(),
                    pregnancy.getEnroll(), pregnancy.getConsent(), pregnancy.getPhoneNumber(),
                    pregnancy.getPhoneType(), pregnancy.getMediaType(), pregnancy.getLanguage(),
                    pregnancy.getDayOfWeek(), pregnancy.getTimeOfDay(), pregnancy.getInterestReason(),
                    pregnancy.getHowLearned());
            model.addAttribute("successMsg", "motechmodule.Pregnancy.register.success");

            status.setComplete();

            return "redirect:/module/motechmodule/viewdata.form";
        }

        return "/module/motechmodule/pregnancy";
    }

    void validateTextLength(Errors errors, String fieldname, String fieldValue, int lengthLimit) {

        if (fieldValue != null && fieldValue.length() > lengthLimit) {
            errors.rejectValue(fieldname, "motechmodule.string.maxlength", new Integer[] { lengthLimit },
                    "Specified text is longer than max characters.");
        }
    }
}