viewcommon.java

一个论坛程序的简单实现

/*
 * Copyright (c) 2003, Rafael Steil
 * All rights reserved.
 * 
 * Redistribution and use in source and binary forms, 
 * with or without modification, are permitted provided 
 * that the following conditions are met:
 * 
 * 1) Redistributions of source code must retain the above 
 * copyright notice, this list of conditions and the 
 * following  disclaimer.
 * 2)  Redistributions in binary form must reproduce the 
 * above copyright notice, this list of conditions and 
 * the following disclaimer in the documentation and/or 
 * other materials provided with the distribution.
 * 3) Neither the name of "Rafael Steil" nor 
 * the names of its contributors may be used to endorse 
 * or promote products derived from this software without 
 * specific prior written permission.
 * 
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT 
 * HOLDERS AND CONTRIBUTORS "AS IS" AND ANY 
 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, 
 * BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF 
 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR 
 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL 
 * THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE 
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 
 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES 
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, 
 * OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER 
 * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER 
 * IN CONTRACT, STRICT LIABILITY, OR TORT 
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN 
 * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF 
 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE
 * 
 * This file creation date: 02/04/2004 - 20:31:35
 * The JForum Project
 * http://www.jforum.net
 */
package net.jforum.view.forum;

import java.awt.image.BufferedImage;
import java.io.BufferedInputStream;
import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.security.NoSuchAlgorithmException;
import java.util.ArrayList;
import java.util.List;

import org.apache.log4j.Logger;

import net.jforum.JForum;
import net.jforum.SessionFacade;
import net.jforum.entities.User;
import net.jforum.model.DataAccessDriver;
import net.jforum.model.UserModel;
import net.jforum.util.I18n;
import net.jforum.util.MD5;
import net.jforum.util.SafeHtml;
import net.jforum.util.image.ImageUtils;
import net.jforum.util.preferences.ConfigKeys;
import net.jforum.util.preferences.SystemGlobals;

/**
 * @author Rafael Steil
 * @version $Id: ViewCommon.java,v 1.18 2004/10/22 04:23:47 rafaelsteil Exp $
 */
public final class ViewCommon
{
	private static final Logger logger = Logger.getLogger(ViewCommon.class);
	
	/**
	 * Updates the user information
	 * 
	 * @param userId The user id we are saving
	 * @throws Exception
	 */
	public static List saveUser(int userId) throws Exception
	{
		UserModel um = DataAccessDriver.getInstance().newUserModel();
		User u = um.selectById(userId);
		
		String username = JForum.getRequest().getParameter("username");
		if (username != null) {
			u.setUsername(username);
		}
		
		u.setId(userId);
		u.setEmail(SafeHtml.makeSafe(JForum.getRequest().getParameter("email")));
		u.setIcq(SafeHtml.makeSafe(JForum.getRequest().getParameter("icq")));
		u.setAim(SafeHtml.makeSafe(JForum.getRequest().getParameter("aim")));
		u.setMsnm(SafeHtml.makeSafe(JForum.getRequest().getParameter("msn")));
		u.setYim(SafeHtml.makeSafe(JForum.getRequest().getParameter("yim")));
		u.setFrom(SafeHtml.makeSafe(JForum.getRequest().getParameter("location")));
		u.setOccupation(SafeHtml.makeSafe(JForum.getRequest().getParameter("occupation")));
		u.setInterests(SafeHtml.makeSafe(JForum.getRequest().getParameter("interests")));
		u.setSignature(SafeHtml.makeSafe(JForum.getRequest().getParameter("signature")));
		u.setViewEmailEnabled(JForum.getRequest().getParameter("viewemail").equals("1"));
		u.setViewOnlineEnabled(JForum.getRequest().getParameter("hideonline").equals("0"));
		u.setNotifyPrivateMessagesEnabled(JForum.getRequest().getParameter("notifypm").equals("1"));
		u.setNotifyOnMessagesEnabled(JForum.getRequest().getParameter("notifyreply").equals("1"));
		u.setAttachSignatureEnabled(JForum.getRequest().getParameter("attachsig").equals("1"));
		u.setHtmlEnabled(JForum.getRequest().getParameter("allowhtml").equals("1"));
		u.setLang(JForum.getRequest().getParameter("language"));
		
		String website = SafeHtml.makeSafe(JForum.getRequest().getParameter("website"));
		if (website != null && !"".equals(website.trim()) && !website.toLowerCase().startsWith("http://")) {
			website = "http://" + website;
		}

		u.setWebSite(website);
		
		if (JForum.getRequest().getParameter("new_password") != null && JForum.getRequest().getParameter("new_password").length() > 0) {
			u.setPassword(MD5.crypt(JForum.getRequest().getParameter("new_password")));
		}
		
		if (JForum.getRequest().getParameter("avatardel") != null) {
			File f = new File(SystemGlobals.getApplicationPath() +"images/avatar/"+ u.getAvatar());
			f.delete();
			
			u.setAvatar(null);
		}

		List warns = new ArrayList();
		if (JForum.getRequest().getObjectParameter("avatar") != null) {
			try {
				handleAvatar(u);
			}
			catch (Exception e) {
				logger.warn("Problems while uploading the avatar: " + e);
				warns.add(I18n.getMessage("User.avatarUploadError"));
			}
		}
		else {
			String avatarUrl = JForum.getRequest().getParameter("avatarUrl");
			if (avatarUrl != null && !"".equals(avatarUrl.trim())) {
				if (avatarUrl.toLowerCase().startsWith("http://")) {
					u.setAvatar(avatarUrl);
				}
				else {
					warns.add(I18n.getMessage("User.avatarUrlShouldHaveHttp"));
				}
			}
			else {
				u.setAvatar(null);
			}
		}
		
		um.update(u);
		
		SessionFacade.getUserSession().setLang(u.getLang());
		
		return warns;
	}

	/**
	 * @param u
	 * @throws NoSuchAlgorithmException
	 * @throws FileNotFoundException
	 * @throws IOException
	 */
	private static void handleAvatar(User u) throws NoSuchAlgorithmException, FileNotFoundException, IOException {
		String fileName = MD5.crypt(Integer.toString(u.getId()));
		
		// Gets file extension
		String extension = JForum.getRequest().getParameter("avatarName");
		extension = extension.substring(extension.lastIndexOf('.') + 1).toLowerCase();
		int type = -1;
		
		if (extension.equals("jpg") || extension.equals("jpeg")) {
			type = ImageUtils.IMAGE_JPEG;
		}
		else if (extension.equals("gif") || extension.equals("png")) {
			type = ImageUtils.IMAGE_PNG;
		}
		
		if (type != -1) {
			String avatarTmpFileName = SystemGlobals.getApplicationPath() +"/images/avatar/"+ fileName +"_tmp." + extension;
			
			// We cannot handle gifs
			if (extension.toLowerCase().equals("gif")) {
				extension = "png";
			}

			String avatarFinalFileName = SystemGlobals.getApplicationPath() +"/images/avatar/"+ fileName +"."+ extension;

			// Read the avatar and stores it into a temporary file
			BufferedInputStream inputStream = new BufferedInputStream((InputStream)JForum.getRequest().getObjectParameter("avatar"));
			FileOutputStream outputStream = new FileOutputStream(avatarTmpFileName);
			
			int c = 0;
			byte[] b = new byte[1024];
			while ((c = inputStream.read(b)) != -1) {
				outputStream.write(b);
			}
			
			outputStream.flush();
			outputStream.close();
			inputStream.close();
			
			// OK, time to check and process the avatar size
			int maxWidth = SystemGlobals.getIntValue(ConfigKeys.AVATAR_MAX_WIDTH);
			int maxHeight = SystemGlobals.getIntValue(ConfigKeys.AVATAR_MAX_HEIGHT);

			BufferedImage image = ImageUtils.resizeImage(avatarTmpFileName, type, maxWidth, maxHeight);
			ImageUtils.saveImage(image, avatarFinalFileName, type);

			u.setAvatar(fileName +"."+ extension);
			
			// Delete de temporary file
			new File(avatarTmpFileName).delete();
		}
	}
	
	public static void contextToLogin() 
	{
		JForum.getContext().put("moduleAction", "forum_login.htm");
		String uri = JForum.getRequest().getRequestURI();
		String query = JForum.getRequest().getQueryString();
		String path = query == null ? uri : uri + "?" + query;

		JForum.getContext().put("returnPath", path);
	}
	
	/**
	 * Returns the initial page to start fetching records from.
	 *   
	 * @return The initial page number
	 */
	public static int getStartPage()
	{
		String s = JForum.getRequest().getParameter("start");
		int start = 0;
		
		if (s == null || s.trim().equals("")) {
			start = 0;
		}
		else {
			start = Integer.parseInt(s);
			
			if (start < 0) {
				start = 0;
			}
		}
		
		return start;
	}
	
	/**
	 * Gets the forum base link.
	 * The returned link has a trailing slash
	 * @return The forum link, with the trailing slash
	 */
	public static String getForumLink()
	{
		String forumLink = SystemGlobals.getValue(ConfigKeys.FORUM_LINK);
		if (!forumLink.endsWith("/")) {
			forumLink += "/";
		}
		
		return forumLink;
	}
}