Initial checkin
git-svn-id: http://svn.igniterealtime.org/svn/repos/smack/trunk@1872 b35dd754-fafc-0310-a699-88a17e54d16e
This commit is contained in:
Bill Lynch
blynch
parent
7857d170be
commit
fea409a14c
|
|
@ -0,0 +1,346 @@
|
|||
/**
|
||||
* $RCSfile$
|
||||
* $Revision$
|
||||
* $Date$
|
||||
*
|
||||
* Copyright (C) 2003 CoolServlets, Inc. All rights reserved.
|
||||
*
|
||||
* This software is the proprietary information of Jive Software. Use is subject to license terms.
|
||||
*/
|
||||
|
||||
package org.jivesoftware.webchat;
|
||||
|
||||
import org.jivesoftware.smack.*;
|
||||
import org.jivesoftware.smack.util.StringUtils;
|
||||
import org.jivesoftware.smack.packet.Message;
|
||||
|
||||
import javax.servlet.http.*;
|
||||
import javax.servlet.ServletException;
|
||||
import javax.servlet.ServletConfig;
|
||||
import javax.servlet.RequestDispatcher;
|
||||
import java.io.IOException;
|
||||
import java.io.PrintWriter;
|
||||
import java.util.*;
|
||||
|
||||
/**
|
||||
* For now, it's assumed the all JSP pages are in the context root. <p>
|
||||
*
|
||||
* Init params:<ul>
|
||||
* <li> host:
|
||||
* <li> port (optional):
|
||||
* <li> SSLEnabled (optional):
|
||||
* </ul>
|
||||
*
|
||||
*
|
||||
*/
|
||||
public class ChatServlet extends HttpServlet implements HttpSessionListener {
|
||||
|
||||
private Map chatData = new HashMap();
|
||||
|
||||
private String host;
|
||||
private int port = -1;
|
||||
private boolean SSLEnabled;
|
||||
|
||||
public void init(ServletConfig config) throws ServletException {
|
||||
super.init(config);
|
||||
host = config.getInitParameter("host");
|
||||
if (host == null) {
|
||||
throw new ServletException("Init parameter \"host\" must be set.");
|
||||
}
|
||||
String portString = config.getInitParameter("port");
|
||||
if (portString != null) {
|
||||
try {
|
||||
port = Integer.parseInt(portString);
|
||||
}
|
||||
catch (NumberFormatException nfe) {
|
||||
throw new ServletException("Init parameter \"port\" must be a valid number.", nfe);
|
||||
}
|
||||
}
|
||||
SSLEnabled = Boolean.valueOf(config.getInitParameter("SSLEnabled")).booleanValue();
|
||||
}
|
||||
|
||||
public void destroy() {
|
||||
super.destroy();
|
||||
// Clean up.
|
||||
for (Iterator i=chatData.values().iterator(); i.hasNext(); ) {
|
||||
ChatData data = (ChatData)i.next();
|
||||
if (data.groupChat != null) {
|
||||
data.groupChat.leave();
|
||||
}
|
||||
data.connection.close();
|
||||
}
|
||||
}
|
||||
|
||||
protected void service(HttpServletRequest request, HttpServletResponse response)
|
||||
throws ServletException, IOException
|
||||
{
|
||||
HttpSession session = request.getSession();
|
||||
String sessionID = session.getId();
|
||||
String path = request.getContextPath();
|
||||
|
||||
String command = request.getParameter("command");
|
||||
if ("login".equals(command)) {
|
||||
String returnCode = handleLogin(request, response);
|
||||
if ("success".equals(returnCode)) {
|
||||
response.sendRedirect(path + "/chat-launcher.jsp");
|
||||
return;
|
||||
}
|
||||
else {
|
||||
// error, so go back to index page (so errors can be printed)
|
||||
RequestDispatcher rd = request.getRequestDispatcher("/index.jsp");
|
||||
rd.forward(request, response);
|
||||
return;
|
||||
}
|
||||
}
|
||||
else if ("logout".equals(command)) {
|
||||
XMPPConnection con = (XMPPConnection)chatData.get(sessionID);
|
||||
|
||||
}
|
||||
else if ("write".equals(command)) {
|
||||
String message = request.getParameter("message");
|
||||
ChatData data = (ChatData)chatData.get(sessionID);
|
||||
if (message == null) {
|
||||
writeData("Parameter \"message\" is required.", response);
|
||||
return;
|
||||
}
|
||||
else if (data == null) {
|
||||
writeData("Must login first.", response);
|
||||
return;
|
||||
}
|
||||
try {
|
||||
data.groupChat.sendMessage(message);
|
||||
StringBuffer reply = new StringBuffer();
|
||||
reply.append("<!DOCTYPE HTML PUBLIC \"-//W3C//DTD HTML 4.0 Transitional//EN\">\n");
|
||||
|
||||
reply.append("<html><head>\n");
|
||||
reply.append("<meta http-equiv=\"Content-Type\" content=\"text/html; charset=UTF-8\">\n");
|
||||
reply.append("<title>Chat Form</title>");
|
||||
reply.append("</head><body>\n");
|
||||
reply.append("<form action=\"ChatServlet\" name=\"chatform\" method=\"post\">\n");
|
||||
reply.append("<input type=\"hidden\" name=\"command\" value=\"write\">\n");
|
||||
reply.append("<input type=\"hidden\" name=\"message\" value=\"\">\n");
|
||||
reply.append("</form></body></html>");
|
||||
writeData(reply.toString(), response);
|
||||
}
|
||||
catch (XMPPException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
else if ("read".equals(command)) {
|
||||
ChatData data = (ChatData)chatData.get(sessionID);
|
||||
if (data == null) {
|
||||
writeData("Must login first.", response);
|
||||
return;
|
||||
}
|
||||
StringBuffer reply = new StringBuffer();
|
||||
reply.append("<!DOCTYPE HTML PUBLIC \"-//W3C//DTD HTML 4.0 Transitional//EN\">\n");
|
||||
reply.append("<html>\n<head>\n");
|
||||
reply.append("<title>Chat Read Page</title>\n");
|
||||
reply.append("<meta http-equiv=\"refresh\" content=\"2\">\n");
|
||||
reply.append("<script language=\"JavaScript\" type=\"text/javascript\">\n");
|
||||
reply.append("var data = window.parent.frames['data'].data;\n");
|
||||
boolean foundData = false;
|
||||
Message message = data.groupChat.pollMessage();
|
||||
while (message != null) {
|
||||
foundData = true;
|
||||
String from = message.getFrom();
|
||||
// We want the the user's nickname
|
||||
from = StringUtils.parseResource(from);
|
||||
String body = message.getBody();
|
||||
// escape HTML
|
||||
body = replace(body, "&", "&");
|
||||
body = replace(body, "<", "<");
|
||||
body = replace(body, ">", ">");
|
||||
// replace newlines in the body:
|
||||
body = replace(body, "\r", "");
|
||||
body = replace(body, "\n", "<br>");
|
||||
// escape quotes
|
||||
body = replace(body, "\"", """);
|
||||
reply.append("data[data.length] = new Array(\"" + from + "\", \"" + body + "\");\n");
|
||||
message = data.groupChat.pollMessage();
|
||||
}
|
||||
if (foundData) {
|
||||
reply.append("window.parent.frames['main'].location.reload();\n");
|
||||
}
|
||||
reply.append("</script>\n");
|
||||
reply.append("</head>\n");
|
||||
reply.append("<body></body>\n");
|
||||
reply.append("</html>");
|
||||
writeData(reply.toString(), response);
|
||||
}
|
||||
else if (command != null) {
|
||||
writeData("Invalid command.", response);
|
||||
}
|
||||
else {
|
||||
writeData("Jive Messenger Chat Servlet", response);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Handles all login logic.
|
||||
*
|
||||
* @param request
|
||||
* @param response
|
||||
*/
|
||||
private String handleLogin(HttpServletRequest request, HttpServletResponse response) {
|
||||
// The session ID - used to identify different chat sessions
|
||||
String sessionID = request.getSession().getId();
|
||||
// Get expected parameters
|
||||
String username = request.getParameter("username");
|
||||
String password = request.getParameter("password");
|
||||
String room = request.getParameter("room");
|
||||
String nickname = request.getParameter("nickname");
|
||||
String garbage = request.getParameter("garbage");
|
||||
// Validate parameters
|
||||
Map errors = new HashMap();
|
||||
if (username == null || "".equals(username.trim())) {
|
||||
errors.put("username","");
|
||||
}
|
||||
if (password == null || "".equals(password.trim())) {
|
||||
errors.put("password","");
|
||||
}
|
||||
if (room == null || "".equals(room.trim())) {
|
||||
errors.put("room","");
|
||||
}
|
||||
if (nickname == null || "".equals(nickname.trim())) {
|
||||
errors.put("nickname","");
|
||||
}
|
||||
// If there were no errors, continue
|
||||
if (errors.size() == 0) {
|
||||
ChatData data = (ChatData)chatData.get(sessionID);
|
||||
// If a connection already exists for this session, close it before creating
|
||||
// another.
|
||||
if (data != null) {
|
||||
if (data.groupChat != null) {
|
||||
data.groupChat.leave();
|
||||
}
|
||||
data.connection.close();
|
||||
}
|
||||
// Otherwise, create new a new connection.
|
||||
else {
|
||||
data = new ChatData();
|
||||
}
|
||||
try {
|
||||
// Create connection.
|
||||
if (!SSLEnabled) {
|
||||
if (port != -1) {
|
||||
data.connection = new XMPPConnection(host, port);
|
||||
}
|
||||
else {
|
||||
data.connection = new XMPPConnection(host);
|
||||
}
|
||||
}
|
||||
else {
|
||||
if (port != -1) {
|
||||
data.connection = new SSLXMPPConnection(host, port);
|
||||
}
|
||||
else {
|
||||
data.connection = new SSLXMPPConnection(host);
|
||||
}
|
||||
}
|
||||
// Login
|
||||
data.connection.login(username, password, "Jive Messenger Chat Servlet " + garbage);
|
||||
// Join groupChat room.
|
||||
data.groupChat = data.connection.createGroupChat(room);
|
||||
data.groupChat.join(nickname);
|
||||
|
||||
// Save chat data in the map.
|
||||
chatData.put(sessionID, data);
|
||||
|
||||
// Put the user's nickname in the session - this is used by the view to correctly
|
||||
// display the user's messages in a different color:
|
||||
request.getSession().setAttribute("messenger.servlet.nickname", nickname);
|
||||
}
|
||||
catch (XMPPException e) {
|
||||
errors.put("general","");
|
||||
e.printStackTrace();
|
||||
}
|
||||
catch (ClassCastException e) {
|
||||
// TODO: remove this catch - it's just a bug workaround for now
|
||||
errors.put("general","");
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
|
||||
if (errors.size() > 0) {
|
||||
// Add the error map to the request as an attribute:
|
||||
request.setAttribute("messenger.servlet.errors",errors);
|
||||
return "error";
|
||||
}
|
||||
else {
|
||||
return "success";
|
||||
}
|
||||
}
|
||||
|
||||
// HttpSessionListener Interface
|
||||
|
||||
public void sessionCreated(HttpSessionEvent event) {
|
||||
// Do nothing.
|
||||
}
|
||||
|
||||
public void sessionDestroyed(HttpSessionEvent event) {
|
||||
// Close the XMPPConnection if there is one.
|
||||
String sessionID = event.getSession().getId();
|
||||
ChatData data = (ChatData)chatData.get(sessionID);
|
||||
if (data.groupChat != null) {
|
||||
data.groupChat.leave();
|
||||
}
|
||||
data.connection.close();
|
||||
}
|
||||
|
||||
private void writeData(String data, HttpServletResponse response) {
|
||||
try {
|
||||
PrintWriter out = response.getWriter();
|
||||
response.setContentType("text/html");
|
||||
out.println(data);
|
||||
out.close();
|
||||
}
|
||||
catch (IOException ioe) {}
|
||||
}
|
||||
|
||||
private class ChatData {
|
||||
public XMPPConnection connection;
|
||||
public GroupChat groupChat;
|
||||
}
|
||||
|
||||
/**
|
||||
* Replaces all instances of oldString with newString in string.
|
||||
*
|
||||
* @param string the String to search to perform replacements on
|
||||
* @param oldString the String that should be replaced by newString
|
||||
* @param newString the String that will replace all instances of oldString
|
||||
*
|
||||
* @return a String will all instances of oldString replaced by newString
|
||||
*/
|
||||
public static final String replace(String string, String oldString, String newString) {
|
||||
if (string == null) {
|
||||
return null;
|
||||
}
|
||||
// If the newString is null just return the string since there's nothing
|
||||
// to replace.
|
||||
if (newString == null) {
|
||||
return string;
|
||||
}
|
||||
int i=0;
|
||||
// Make sure that oldString appears at least once before doing any processing.
|
||||
if ( ( i=string.indexOf(oldString, i) ) >= 0 ) {
|
||||
// Use char []'s, as they are more efficient to deal with.
|
||||
char [] string2 = string.toCharArray();
|
||||
char [] newString2 = newString.toCharArray();
|
||||
int oLength = oldString.length();
|
||||
StringBuffer buf = new StringBuffer(string2.length);
|
||||
buf.append(string2, 0, i).append(newString2);
|
||||
i += oLength;
|
||||
int j = i;
|
||||
// Replace all remaining instances of oldString with newString.
|
||||
while ( ( i=string.indexOf(oldString, i) ) > 0 ) {
|
||||
buf.append(string2, j, i-j).append(newString2);
|
||||
i += oLength;
|
||||
j = i;
|
||||
}
|
||||
buf.append(string2, j, string2.length - j);
|
||||
return buf.toString();
|
||||
}
|
||||
return string;
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue