org.openmrs.contrib.metadatarepository.webapp.controller.PackageFormController.java Source code

Introduction

Here is the source code for org.openmrs.contrib.metadatarepository.webapp.controller.PackageFormController.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.contrib.metadatarepository.webapp.controller;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.util.Locale;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.commons.io.IOUtils;
import org.openmrs.contrib.metadatarepository.model.MetadataPackage;
import org.openmrs.contrib.metadatarepository.model.User;
import org.openmrs.contrib.metadatarepository.service.PackageManager;
import org.openmrs.contrib.metadatarepository.service.UserManager;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.ui.ModelMap;
import org.springframework.validation.BindingResult;
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.servlet.ModelAndView;

@Controller
@RequestMapping("/packageform*")
public class PackageFormController extends BaseFormController {

    private PackageManager packageManager = null;
    private UserManager userManager = null;

    @Autowired
    public void setUserManager(UserManager userManager) {
        this.userManager = userManager;
    }

    @Autowired
    public void setPackageManager(PackageManager packageManager) {
        this.packageManager = packageManager;
    }

    public PackageFormController() {
        setCancelView("redirect:mainMenu");
        setSuccessView("redirect:/mainMenu");
    }

    @ModelAttribute
    @RequestMapping(method = { RequestMethod.GET, RequestMethod.POST })
    protected MetadataPackage showForm(@RequestParam(required = false) final Long id) throws Exception {
        if (id != null) {
            return packageManager.get(id);
        }

        return new MetadataPackage();
    }

    @RequestMapping(value = "/packagedownload")
    public void downloadPackage(@RequestParam(required = false) final Long id, HttpServletResponse response)
            throws IOException {
        MetadataPackage p = packageManager.loadPackage(id);

        response.setContentType("application/x-zip-compressed");
        response.setHeader("Content-disposition", "attatchment;filename=\"" + id + ".zip");
        IOUtils.copy(new ByteArrayInputStream(p.getFile()), response.getOutputStream());
        Long i = p.getDownloadCount();
        log.debug("download count:" + i);
        i = i + 1;
        p.setDownloadCount(i);
        log.debug("download count:" + p.getDownloadCount());
        packageManager.save(p);
        response.flushBuffer();

    }

    @RequestMapping(value = "/viewPackage*")
    public ModelAndView viewPackage(MetadataPackage pkg, ModelMap map) {

        map.addAttribute("metadataPackage", packageManager.get(pkg.getId()));

        return new ModelAndView("/viewPackage");
    }

    @RequestMapping(method = RequestMethod.POST)
    public String onSubmit(MetadataPackage pkg, BindingResult errors, HttpServletRequest request,
            HttpServletResponse response, ModelMap model) throws Exception {
        if (request.getParameter("cancel") != null) {
            return getCancelView();
        }

        log.debug("entering 'onSubmit' method...");

        String success = getSuccessView();
        Locale locale = request.getLocale();

        if (request.getParameter("delete") != null) {
            packageManager.remove(pkg.getId());
            packageManager.deleteFile(pkg.getId() + "");
            saveMessage(request, getText("package.deleted", locale));
        } else {

            saveMessage(request, getText("package.saved", locale));
            MetadataPackage metadataPackage = packageManager.get(pkg.getId());
            metadataPackage.setName(pkg.getName());
            metadataPackage.setDescription(pkg.getDescription());

            packageManager.save(metadataPackage);
            model.addAttribute("metadataPackage", metadataPackage);

        }

        return success + "?id=" + pkg.getId();
    }

}