diff --git a/WebContent/WEB-INF/jsp/planoTrabalho/planoTrabalho.jsp b/WebContent/WEB-INF/jsp/planoTrabalho/planoTrabalho.jsp index e0f761e..7c5768f 100644 --- a/WebContent/WEB-INF/jsp/planoTrabalho/planoTrabalho.jsp +++ b/WebContent/WEB-INF/jsp/planoTrabalho/planoTrabalho.jsp @@ -33,7 +33,7 @@



diff --git a/src/br/com/controller/PlanoTrabalhoController.java b/src/br/com/controller/PlanoTrabalhoController.java index d5d7571..9656dc2 100644 --- a/src/br/com/controller/PlanoTrabalhoController.java +++ b/src/br/com/controller/PlanoTrabalhoController.java @@ -14,59 +14,69 @@ import br.com.model.entity.Questionario; @Resource public class PlanoTrabalhoController { - + private Validator validator; private Result result; private ServletContext application; private String path; - - public PlanoTrabalhoController(Result result, Validator validator, ServletContext application) - { + + public PlanoTrabalhoController(Result result, Validator validator, ServletContext application) { this.validator = validator; this.application = application; this.path = this.application.getRealPath("") + "/WEB-INF/conexao.properties"; this.result = result; } - + @Path("/planoTrabalho-site") - public void planoTrabalho() - { + public void planoTrabalho() { } - + @Path("/plano-trabalho-confirmacao") - public void planoTrabalhoConfirmacao() - { - + public void planoTrabalhoConfirmacao() { + } - + @Path("/salvar-plano-trabalho") - public void enviarPlanoTrabalho(String valorHash, UploadedFile file) - { - HashDao hashDao = new HashDao(); - Questionario questionario = new Questionario(); - questionario = hashDao.verificarHash(path, valorHash); + public void enviarPlanoTrabalho(String valorHash, UploadedFile file) { + String tipoArquivo = file.getContentType(); String mensagem = "Não foi possível enviar o plano de trabalho!"; boolean status = false; - - if(questionario != null && questionario.getDtInicioQuestionario() != null) - { - PlanoTrabalhoDao planoTrabalhoDao = new PlanoTrabalhoDao(); - - if(planoTrabalhoDao.salvarPlanoTrabalho(valorHash, file.getFile(), questionario)) - { - mensagem = "Plano de trabalho enviado com sucesso!"; - status = true; - } - } - else - { - mensagem = "Favor verificar o Código Hash!"; + + if (!tipoArquivo.equalsIgnoreCase("application/msword") + && !tipoArquivo.equalsIgnoreCase("application/vnd.ms-excel") + && !tipoArquivo.equalsIgnoreCase("application/vnd.oasis.opendocument.text") + && !tipoArquivo.equalsIgnoreCase("text/plain") + && !tipoArquivo.equalsIgnoreCase("application/pdf") + && !tipoArquivo.equalsIgnoreCase("image/png") + && !tipoArquivo.equalsIgnoreCase("image/gif") + && !tipoArquivo.equalsIgnoreCase("image/jpeg")) { + + mensagem = "Formato de Arquivo inválido!"; + status = false; + + } else { + HashDao hashDao = new HashDao(); + Questionario questionario = new Questionario(); + questionario = hashDao.verificarHash(path, valorHash); + + + if (questionario != null && questionario.getDtInicioQuestionario() != null) { + PlanoTrabalhoDao planoTrabalhoDao = new PlanoTrabalhoDao(); + + if (planoTrabalhoDao.salvarPlanoTrabalho(valorHash, file.getFile(), questionario)) { + mensagem = "Plano de trabalho enviado com sucesso!"; + status = true; + } + + else { + mensagem = "Favor verificar o Código Hash!"; + } + } } - result.include("mensagem", mensagem); - result.include("status", status); - this.validator.add(new ValidationMessage("", "")); - this.validator.onErrorUsePageOf(PlanoTrabalhoController.class).planoTrabalhoConfirmacao(); + result.include("mensagem", mensagem); + result.include("status", status); + this.validator.add(new ValidationMessage("", "")); + this.validator.onErrorUsePageOf(PlanoTrabalhoController.class).planoTrabalhoConfirmacao(); } - } -- libgit2 0.21.2