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

Introduction

Here is the source code for org.motechproject.server.omod.web.controller.MotechModuleFormController.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.sql.Time;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.motechproject.server.model.Blackout;
import org.motechproject.server.model.TroubledPhone;
import org.motechproject.server.service.ContextService;
import org.motechproject.server.service.MotechService;
import org.motechproject.server.svc.RegistrarBean;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.stereotype.Controller;
import org.springframework.ui.ModelMap;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RequestParam;

/**
 * A spring webmvc controller, defining operations for the administrative
 * actions for the OpenMRS admin links.
 * 
 * @see org.motechproject.server.omod.extension.html.AdminList
 */
@Controller
public class MotechModuleFormController {

    protected final Log log = LogFactory.getLog(MotechModuleFormController.class);

    @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;
    }

    @RequestMapping("/module/motechmodule/viewdata")
    public String viewData(ModelMap model) {

        model.addAttribute("allLocations", registrarBean.getAllLocations());
        model.addAttribute("allStaff", registrarBean.getAllStaff());
        model.addAttribute("allPatients", registrarBean.getAllPatients());
        model.addAttribute("allPregnancies", registrarBean.getAllPregnancies());
        model.addAttribute("allScheduledMessages", registrarBean.getAllScheduledMessages());

        return "/module/motechmodule/viewdata";
    }

    @RequestMapping("/module/motechmodule/blackout")
    public String viewBlackoutSettings(ModelMap model) {

        Blackout blackout = contextService.getMotechService().getBlackoutSettings();

        if (blackout != null) {
            model.addAttribute("startTime", blackout.getStartTime());
            model.addAttribute("endTime", blackout.getEndTime());
        }

        return "/module/motechmodule/blackout";
    }

    @RequestMapping(value = "/module/motechmodule/blackout", method = RequestMethod.POST)
    public String saveBlackoutSettings(@RequestParam("startTime") String startTime,
            @RequestParam("endTime") String endTime, ModelMap model) {

        MotechService motechService = contextService.getMotechService();
        Blackout blackout = motechService.getBlackoutSettings();

        Time startTimeCvt = Time.valueOf(startTime);
        Time endTimeCvt = Time.valueOf(endTime);

        if (blackout != null) {
            blackout.setStartTime(startTimeCvt);
            blackout.setEndTime(endTimeCvt);
        } else {
            blackout = new Blackout(startTimeCvt, endTimeCvt);
        }

        motechService.setBlackoutSettings(blackout);

        model.addAttribute("startTime", blackout.getStartTime());
        model.addAttribute("endTime", blackout.getEndTime());

        return "/module/motechmodule/blackout";
    }

    @RequestMapping(value = "/module/motechmodule/troubledphone", method = RequestMethod.GET)
    public String handleTroubledPhone(@RequestParam(required = false, value = "phoneNumber") String phoneNumber,
            @RequestParam(required = false, value = "remove") Boolean remove, ModelMap model) {

        if (phoneNumber != null && !phoneNumber.trim().isEmpty()) {

            MotechService motechService = contextService.getMotechService();

            TroubledPhone troubledPhone = motechService.getTroubledPhone(phoneNumber);

            if (remove == Boolean.TRUE) {
                motechService.removeTroubledPhone(phoneNumber);
                return "redirect:/module/motechmodule/troubledphone.form";
            } else if (troubledPhone != null)
                model.addAttribute(troubledPhone);
        }

        return "/module/motechmodule/troubledphone";
    }
}