com.pamarin.income.controller.SuggestionCtrl.java Source code

Introduction

Here is the source code for com.pamarin.income.controller.SuggestionCtrl.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package com.pamarin.income.controller;

import com.google.common.io.ByteStreams;
import com.pamarin.income.App;
import com.pamarin.income.component.MailCallback;
import com.pamarin.income.component.MailSender;
import com.pamarin.income.exception.UncheckedIOException;
import com.pamarin.income.exception.UserException;
import com.pamarin.income.model.Suggestion;
import com.pamarin.income.security.SecurityUtils;
import com.pamarin.income.service.SuggestionService;
import com.pamarin.income.util.MessageNotifyCallback;
import com.pamarin.income.util.Notification;
import com.pamarin.income.util.PropertiesFileUtils;
import static com.pamarin.income.util.UploadUtils.getTempDirectory;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.text.Format;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.UUID;
import javax.annotation.PostConstruct;
import org.apache.commons.io.FilenameUtils;
import org.primefaces.model.UploadedFile;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Scope;
import org.springframework.core.io.FileSystemResource;
import org.springframework.mail.javamail.MimeMessageHelper;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;

/**
 *
 * @author jittagornp
 */
@Component
@Scope("view")
public class SuggestionCtrl {

    private static final Logger LOG = LoggerFactory.getLogger(SuggestionCtrl.class);

    private static final String EMAIL_CONFIG = "/email.properties";
    private static final String DEFAULT_TYPE = "SUGGESTION";

    private String destinationReceiveEmail;
    private Suggestion suggestion;
    @Autowired
    private SuggestionService service;
    private UploadedFile file;
    @Autowired
    private MailSender mailSender;
    @Autowired
    private App app;

    @PostConstruct
    public void postConstruct() {
        try {
            destinationReceiveEmail = PropertiesFileUtils.load(EMAIL_CONFIG).getProperty("email.username");
        } catch (IOException ex) {
            throw new UncheckedIOException("require property email.username on classpath:" + EMAIL_CONFIG);
        }
    }

    public UploadedFile getFile() {
        return file;
    }

    public void setFile(UploadedFile file) {
        this.file = file;
    }

    public Suggestion getSuggestion() {
        if (suggestion == null) {
            suggestion = new Suggestion();
            suggestion.setType(DEFAULT_TYPE);
        }

        return suggestion;
    }

    public void setSuggestion(Suggestion suggestion) {
        this.suggestion = suggestion;
    }

    private boolean isImage(String extension) {
        return "png".equals(extension) || "jpg".equals(extension) || "jpeg".equals(extension);
    }

    private void validateImageType() {
        LOG.debug("file name --> {}", file.getFileName());
        String exctension = FilenameUtils.getExtension(file.getFileName()).toLowerCase();
        if (!isImage(exctension)) {
            throw new UserException(
                    " .png, .jpg, .jpeg ");
        }
    }

    public File getTempDirectoryDateTime() throws IOException {
        Format format = new SimpleDateFormat("dd-MM-yyyy");
        File dateFile = new File(getTempDirectory(), format.format(new Date()));
        if (!dateFile.exists()) {
            dateFile.mkdirs();
        }

        return dateFile;
    }

    private File saveFile() {
        File attachFile = null;
        String errorMessage = "?";

        InputStream inputStream = null;
        OutputStream outputStream = null;

        try {
            String randomName = UUID.randomUUID().toString();
            String extension = FilenameUtils.getExtension(file.getFileName());
            inputStream = file.getInputstream();
            File parentDir = getTempDirectoryDateTime();
            attachFile = new File(parentDir, randomName + "." + extension);

            outputStream = new FileOutputStream(attachFile);
            ByteStreams.copy(inputStream, outputStream);
            getSuggestion().setImage("/" + parentDir.getName() + "/" + randomName + "." + extension);
        } catch (Exception ex) {
            LOG.warn(null, ex);
            attachFile = null;
            throw new UncheckedIOException(errorMessage);
        } finally {
            if (outputStream != null) {
                try {
                    outputStream.close();
                } catch (IOException ex) {
                    attachFile = null;
                    throw new UncheckedIOException(errorMessage);
                }
            }

            if (inputStream != null) {
                try {
                    inputStream.close();
                } catch (IOException ex) {
                    attachFile = null;
                    throw new UncheckedIOException(errorMessage);
                }
            }
        }

        return attachFile;
    }

    private void sendEmail2Admin(final File attachFile) {
        mailSender.send(new MailCallback() {

            @Override
            public void execute(MimeMessageHelper helper) throws Exception {
                if (attachFile != null) {
                    FileSystemResource file = new FileSystemResource(attachFile);
                    helper.addAttachment(attachFile.getName(), file);
                }

                helper.setSubject("?" + app.getName());
                helper.setText(getSuggestion().getType() + " : " + getSuggestion().getMessage());
                helper.setTo(destinationReceiveEmail);
            }
        });
    }

    private void sendEmail2User() {
        mailSender.send(new MailCallback() {

            @Override
            public void execute(MimeMessageHelper helper) throws Exception {
                helper.setSubject("" + app.getName());
                helper.setText(
                        " ??");
                helper.setTo(SecurityUtils.getUser().getUsername());
            }
        });
    }

    private boolean hasFile() {
        return file != null && StringUtils.hasText(file.getFileName());
    }

    public void onSave() {
        if (!SecurityUtils.isAnonymous()) {
            getSuggestion().setOwner(SecurityUtils.getUser());
        }

        Notification.notifyPhase(new MessageNotifyCallback("") {

            @Override
            public void process() throws Throwable {
                File attachFile = null;
                if (hasFile()) {
                    validateImageType();
                    attachFile = saveFile();
                }

                sendEmail2Admin(attachFile);
                if (!SecurityUtils.isAnonymous()) {
                    sendEmail2User();
                }

                service.save(suggestion);
            }

            @Override
            public String getSuccessBody() {
                return " - ";
            }

            @Override
            public void onFinally() {
                suggestion = null;
                file = null;
            }

        });
    }
}