UploadImageFilter.java :  » Development » JdonFramework-4.1 » com » jdon » jivejdon » model » message » upload » Java Open Source

/*
 * Copyright 2003-2006 the original author or authors.
 * 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 com.jdon.jivejdon.model.message.upload;

import java.util.Collection;
import java.util.Iterator;
import java.util.List;

import com.jdon.jivejdon.model.ForumMessage;
import com.jdon.jivejdon.model.attachment.UploadFile;
import com.jdon.jivejdon.model.message.MessageRenderSpecification;
import com.jdon.jivejdon.model.message.MessageVO;
import com.jdon.util.Debug;

/**
 * @author banq(http://www.jdon.com)
 * 
 */
public class UploadImageFilter implements MessageRenderSpecification {
  private final static String module = UploadImageFilter.class.getName();
  // this imageShowUrl must modified if jivejdon is not default web name
  private String imageShowUrl = "/jivejdon/img/uploadShowAction.shtml";
  private String imageShowInHtmlUrl = "/jivejdon/imageShowInHtml.jsp";
  private UploadHelper uploadHelper = new UploadHelper();

  public String getImageShowFileName() {
    return imageShowUrl;
  }

  public void setImageShowFileName(String imageShowFileName) {
    imageShowUrl = imageShowFileName;
  }

  public ForumMessage render(ForumMessage message) {
    try {
      MessageVO messageVO = message.getMessageVO();
      String newBody = appendTags(message);
      messageVO.setBody(newBody);
    } catch (Exception e) {
      Debug.logError("" + e, module);
    }
    return message;
  }

  /**
   * return the String:
   * 
   * @param str
   * @return
   */
  public String appendTags(ForumMessage message) {
    MessageVO messageVO = message.getMessageVO();
    String str = messageVO.getBody();
    Collection uploadFiles = message.getAttachment().getUploadFiles();
    if ((uploadFiles == null) || (uploadFiles.size() == 0)) {
      return str;
    }

    if (str.indexOf("[img index=") != -1) {
      str = processImgEmbed(uploadFiles, str);
      return str;
    } else {
      StringBuffer sb = new StringBuffer(str);
      Iterator iter = uploadFiles.iterator();
      while (iter.hasNext()) {
        UploadFile uploadFile = (UploadFile) iter.next();
        // uploadFile.preloadData();
        if (uploadHelper.isImage(uploadFile.getName())) {
          sb.append(appendImgShowURL(uploadFile));
        }
      }
      return sb.toString();
    }

  }

  // see com.jdon.jivejdon.presentation.action.UploadShowAction
  private String appendImgShowURL(UploadFile uploadFile) {
    StringBuffer sb = new StringBuffer();
    try {
      sb.append("\n<p>").append(uploadFile.getDescription());
      sb.append("<br><a href='").append(imageShowInHtmlUrl).append("?id=").append(uploadFile.getId());
      sb.append("&oid=").append(uploadFile.getOid());
      sb.append("' target=_blank>");
      sb.append("<img src=\"");
      sb.append(imageShowUrl);
      sb.append("?id=").append(uploadFile.getId());
      sb.append("&oid=").append(uploadFile.getOid());
      sb.append("\" border='0' >");
      sb.append("</a></p>\n");
    } catch (Exception e) {
      e.printStackTrace();
    }
    return sb.toString();
  }

  private String processImgEmbed(Collection uploadFiles, String message) {
    String finalStr = message;
    List files = (List) uploadFiles;

    int[] index = { 1, 2, 3 };
    for (int i : index) {
      int p = message.indexOf("[img index=" + i + "]");
      try {
        if (p != -1) {
          UploadFile uploadFile = (UploadFile) files.get(i - 1);
          if (uploadHelper.isImage(uploadFile.getName()))
            finalStr = finalStr.replace("[img index=" + i + "]", appendImgShowURL(uploadFile));
        }
      } catch (IndexOutOfBoundsException e) {
        finalStr = finalStr.replace("[img index=" + i + "]", "");
      }
    }
    return finalStr;
  }

  public String getImageShowUrl() {
    return imageShowUrl;
  }

  public void setImageShowUrl(String imageShowUrl) {
    this.imageShowUrl = imageShowUrl;
  }

  public String getImageShowInHtmlUrl() {
    return imageShowInHtmlUrl;
  }

  public void setImageShowInHtmlUrl(String imageShowInHtmlUrl) {
    this.imageShowInHtmlUrl = imageShowInHtmlUrl;
  }

}