de.micromata.genome.gwiki.pagetemplates_1_0.editor.PtWikiUploadEditor.java Source code

Introduction

Here is the source code for de.micromata.genome.gwiki.pagetemplates_1_0.editor.PtWikiUploadEditor.java

Source

//
// Copyright (C) 2010-2016 Roger Rene Kommer & Micromata GmbH
//
// Licensed under the Apache License, Version 2.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://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//

package de.micromata.genome.gwiki.pagetemplates_1_0.editor;

import java.io.IOException;

import org.apache.commons.fileupload.FileItem;
import org.apache.commons.io.IOUtils;
import org.apache.commons.io.output.ByteArrayOutputStream;
import org.apache.commons.lang3.StringUtils;

import de.micromata.genome.gwiki.model.GWikiArtefakt;
import de.micromata.genome.gwiki.model.GWikiElement;
import de.micromata.genome.gwiki.model.GWikiPropKeys;
import de.micromata.genome.gwiki.model.GWikiWebUtils;
import de.micromata.genome.gwiki.model.logging.GWikiLog;
import de.micromata.genome.gwiki.page.GWikiContext;
import de.micromata.genome.gwiki.page.impl.GWikiBinaryAttachmentArtefakt;

/**
 * @author Christian Claus (c.claus@micromata.de)
 */
public abstract class PtWikiUploadEditor extends PtWikiTextEditorBase {
    private static final long serialVersionUID = 5901053792188232570L;

    protected FileItem dataFile;

    private String maxFileSize;

    private byte[] data;

    public PtWikiUploadEditor(GWikiElement element, String sectionName, String editor, String hint,
            String maxFileSize) {
        super(element, sectionName, editor, hint);

        this.maxFileSize = maxFileSize;
    }

    /**
     * @param ctx
     * @return
     */
    public String saveContent(final GWikiContext ctx) {
        String parentPageId = uploadFile(ctx);

        if (parentPageId == null) {
            return null;
        }

        String pageId = parentPageId + "/" + dataFile.getName();

        return pageId;
    }

    /**
     * @param ctx
     * @param maxWidthInPx
     * @return
     */
    private String uploadFile(final GWikiContext ctx) {
        String parentPageId = ctx.getRequestParameter("pageId");

        if (dataFile == null) {
            dataFile = ctx.getFileItem(sectionName);
        }

        String name = dataFile.getName();
        String pageIdFromTitle = GWikiContext.getPageIdFromTitle(dataFile.getName());

        if (!StringUtils.equals(name, pageIdFromTitle)) {
            ctx.addSimpleValidationError(ctx.getTranslated("gwiki.editor.upload.allowdSymbols"));
            return null;
        }

        String pageId = parentPageId + "/" + pageIdFromTitle;

        if (ctx.getWikiWeb().findElement(pageId) != null) {
            ctx.addSimpleValidationError(ctx.getTranslated("gwiki.editor.upload.fileExists"));
            return null;
        }

        String title = ctx.getRequest().getParameter("title");

        try {
            ByteArrayOutputStream bout = new ByteArrayOutputStream();
            IOUtils.copy(dataFile.getInputStream(), bout);
            data = bout.toByteArray();

            String metaTemplateId = "admin/templates/FileWikiPageMetaTemplate";
            GWikiElement element = GWikiWebUtils.createNewElement(ctx, pageId, metaTemplateId, parentPageId);
            GWikiArtefakt<?> art = element.getMainPart();
            GWikiBinaryAttachmentArtefakt att = (GWikiBinaryAttachmentArtefakt) art;
            att.setStorageData(data);

            if (data.length > 0) {
                element.getElementInfo().getProps().setIntValue(GWikiPropKeys.SIZE, data.length);
            } else {
                ctx.addSimpleValidationError(ctx.getTranslated("gwiki.editor.empty"));
                return null;
            }

            if (!StringUtils.isEmpty(title)) {
                element.getElementInfo().getProps().setStringValue(GWikiPropKeys.TITLE, title);
            } else {
                element.getElementInfo().getProps().setStringValue(GWikiPropKeys.TITLE, pageIdFromTitle);
            }

            element.getElementInfo().getProps().setStringValue(GWikiPropKeys.PARENTPAGE, parentPageId);

            if (maxFileSize != null && StringUtils.isNotEmpty(maxFileSize)) {
                try {
                    long maxSize = Long.parseLong(StringUtils.trimToEmpty(maxFileSize)) * 1024;
                    long currSize = dataFile.getSize();

                    if (maxSize < currSize) {
                        ctx.addSimpleValidationError(
                                ctx.getTranslated("gwiki.editor.upload.filesize") + maxFileSize + " kB");
                    } else {
                        ctx.getWikiWeb().saveElement(ctx, element, false);
                    }
                } catch (Exception e) {
                    GWikiLog.error(e.getMessage(), e);
                }
            } else {
                ctx.getWikiWeb().saveElement(ctx, element, false);
            }
        } catch (IOException ex) {
            ctx.addValidationError("gwiki.edit.EditPage.attach.message.uploadfailed", ex.getMessage());
        }

        return parentPageId;
    }

    /*
     * (non-Javadoc)
     * 
     * @see de.micromata.genome.gwiki.page.impl.GWikiEditorArtefakt#getTabTitle()
     */
    public String getTabTitle() {
        return "";
    }

}