com.jgeppert.struts2.jquery.richtext.actions.CkeditorUpload Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of struts2-jquery-richtext-plugin Show documentation
There is a newer version: 5.0.6
Show newest version
package com.jgeppert.struts2.jquery.richtext.actions;

import com.opensymphony.xwork2.Action;
import com.opensymphony.xwork2.ActionSupport;
import com.opensymphony.xwork2.inject.Inject;
import org.apache.commons.lang.ArrayUtils;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.apache.struts2.ServletActionContext;

import java.io.*;
import java.util.UUID;

public class CkeditorUpload extends ActionSupport {

    private static final long serialVersionUID = 8163586984026522679L;
    private String uploadContentType;
    private String uploadFileName;
    private String uploadFolder = "/uploads";
    private String CKEditorFuncNum;
    private String CKEditor;
    private String langCode;
    private File upload;
    private boolean allowUploads = false;
    private String allowedFileExtensions = "";

    private final static Logger log = LogManager.getLogger(CkeditorUpload.class);

    @Override
    public String execute() throws Exception {

        if (log.isDebugEnabled()) {
            log.debug("Upload is allowed: {}", this.allowUploads);
            log.debug("Allowed File Extensions: {}", this.allowedFileExtensions);
            log.debug("Upload Folder : {}", this.uploadFolder);
        }

        if (!allowUploads) {
            addActionError(getText("struts.ckeditor.error.notallowed", "Uploads are not allowed!"));
            return Action.ERROR;
        }

        String strPath = ServletActionContext.getServletContext().getRealPath(uploadFolder);
        File path = new File(strPath);
        if (!path.exists()) {
            path.mkdirs();
        }
        String uuid = UUID.randomUUID().toString();
        String rt[] = uploadFileName.split("\\.");

        String uploadFileExtension = "";
        if (rt.length >= 2) {
            uploadFileExtension = rt[rt.length - 1];
            uploadFileName = uuid + "." + uploadFileExtension;
        }

        boolean isAllowed = ArrayUtils.contains(allowedFileExtensions.split(","), uploadFileExtension.toLowerCase());

        if (log.isDebugEnabled()) {
            log.debug("Upload Content Type: {}", this.uploadContentType);
            log.debug("Upload File Name: {}", uploadFileName);
            log.debug("Upload File Extension: {}", uploadFileExtension);
            log.debug("File Extension is allowed: {}", isAllowed);
        }

        if (!isAllowed) {
            addActionError(getText("struts.ckeditor.error.invalid.extension", "Invalid File Extension!"));
            return Action.ERROR;
        }

        try (
                InputStream is = new FileInputStream(this.upload);
                OutputStream os = new FileOutputStream(new File(strPath + File.separator + this.uploadFileName))
        ) {
            int len;
            byte[] buffer = new byte[1024];
            while ((len = is.read(buffer)) > 0) {
                os.write(buffer, 0, len);
            }
        } catch (Exception e) {
            log.error(e.getLocalizedMessage(), e);
            addActionError(getText("struts.ckeditor.error.general", "Error occurred! " + e.getLocalizedMessage()));
            return Action.ERROR;
        }
        PrintWriter out = ServletActionContext.getResponse().getWriter();
        out.write("");
        return Action.NONE;
    }

    /**
     * @param uploadFolder folder for uploads. default is "/uploads".
     */
    @Inject(value = "struts.ckeditor.uploadFolder", required = false)
    public void setUploadFolder(String uploadFolder) {
        this.uploadFolder = uploadFolder;
    }

    /**
     * @param allowedFileExtensions allowed file extensions. default is "".
     */
    @Inject(value = "struts.ckeditor.allowedFileExtensions", required = false)
    public void setAllowedFileExtensions(String allowedFileExtensions) {
        this.allowedFileExtensions = allowedFileExtensions;
    }

    /**
     * @param allowUploads allow file uploads. default is false.
     */
    @Inject(value = "struts.ckeditor.allowUploads", required = false)
    public void setAllowUploads(String allowUploads) {
        this.allowUploads = allowUploads.equalsIgnoreCase("true");
    }

    /**
     * @return the upload
     */
    public File getUpload() {
        return upload;
    }

    /**
     * @param upload the upload to set
     */
    public void setUpload(File upload) {
        this.upload = upload;
    }

    /**
     * @return the cKEditorFuncNum
     */
    public String getCKEditorFuncNum() {
        return CKEditorFuncNum;
    }

    /**
     * @param cKEditorFuncNum the cKEditorFuncNum to set
     */
    public void setCKEditorFuncNum(String cKEditorFuncNum) {
        CKEditorFuncNum = cKEditorFuncNum;
    }

    /**
     * @return the cKEditor
     */
    public String getCKEditor() {
        return CKEditor;
    }

    /**
     * @param cKEditor the cKEditor to set
     */
    public void setCKEditor(String cKEditor) {
        CKEditor = cKEditor;
    }

    /**
     * @return the langCode
     */
    public String getLangCode() {
        return langCode;
    }

    /**
     * @param langCode the langCode to set
     */
    public void setLangCode(String langCode) {
        this.langCode = langCode;
    }

    /**
     * @return the uploadContentType
     */
    public String getUploadContentType() {
        return uploadContentType;
    }

    /**
     * @param uploadContentType the uploadContentType to set
     */
    public void setUploadContentType(String uploadContentType) {
        this.uploadContentType = uploadContentType;
    }

    /**
     * @return the uploadFileName
     */
    public String getUploadFileName() {
        return uploadFileName;
    }

    /**
     * @param uploadFileName the uploadFileName to set
     */
    public void setUploadFileName(String uploadFileName) {
        this.uploadFileName = uploadFileName;
    }
}