import java.awt.Container;
import java.awt.GridLayout;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.event.WindowEvent;
import java.io.IOException;
import java.net.URI;
import java.net.URISyntaxException;
import javax.swing.JButton;
import javax.swing.JComboBox;
import javax.swing.JFrame;
import javax.swing.JMenu;
import javax.swing.JScrollPane;
import javax.swing.JTextArea;
import javax.swing.JTextField;
import net.tootallnate.websocket.Draft;
import net.tootallnate.websocket.WebSocketClient;
import net.tootallnate.websocket.drafts.Draft_10;
import net.tootallnate.websocket.drafts.Draft_17;
import net.tootallnate.websocket.drafts.Draft_75;
import net.tootallnate.websocket.drafts.Draft_76;
public class ChatClient extends JFrame implements ActionListener {
private static final long serialVersionUID = -6056260699202978657L;
private final JTextField uriField;
private final JButton connect;
private final JButton close;
private final JTextArea ta;
private final JTextField chatField;
private final JComboBox draft;
private WebSocketClient cc;
public ChatClient() {
super("WebSocket Chat Client");
Container c = getContentPane();
GridLayout layout = new GridLayout();
layout.setColumns(1);
layout.setRows(6);
c.setLayout(layout);
Draft[] drafts = { new Draft_10(), new Draft_17(), new Draft_76(), new Draft_75() };
draft = new JComboBox( drafts );
c.add(draft);
uriField = new JTextField();
uriField.setText("ws://localhost:8887");
c.add(uriField);
connect = new JButton("Connect");
connect.addActionListener(this);
c.add(connect);
close = new JButton("Close");
close.addActionListener(this);
close.setEnabled( false );
c.add(close);
JScrollPane scroll = new JScrollPane();
ta = new JTextArea();
scroll.setViewportView(ta);
c.add(scroll);
chatField = new JTextField();
chatField.setText("");
chatField.addActionListener(this);
c.add(chatField);
java.awt.Dimension d = new java.awt.Dimension(300, 400);
setPreferredSize(d);
setSize(d);
addWindowListener(new java.awt.event.WindowAdapter() {
@Override
public void windowClosing(WindowEvent e) {
if (cc != null) {
try {
cc.close();
} catch (IOException ex) {
ex.printStackTrace();
}
}
dispose();
}
});
setLocationRelativeTo(null);
setVisible(true);
}
public void actionPerformed(ActionEvent e) {
if (e.getSource() == chatField) {
if (cc != null) {
try {
cc.send(chatField.getText());
chatField.setText("");
chatField.requestFocus();
} catch (IOException ex) { ex.printStackTrace(); }
}
} else if (e.getSource() == connect) {
try {
//cc = new ChatClient(new URI(uriField.getText()), area, ( Draft ) draft.getSelectedItem() );
cc = new WebSocketClient( new URI(uriField.getText()),( Draft ) draft.getSelectedItem()) {
public void onMessage(String message) {
ta.append( "got: " + message + "\n");
ta.setCaretPosition( ta.getDocument().getLength() );
}
public void onOpen() {
ta.append("You are connected to ChatServer: " + getURI() + "\n");
ta.setCaretPosition( ta.getDocument().getLength() );
}
public void onClose() {
ta.append("You have been disconnected from: " + getURI() + "\n");
ta.setCaretPosition( ta.getDocument().getLength() );
connect.setEnabled(true);
uriField.setEditable(true);
draft.setEditable(true);
close.setEnabled( false );
}
public void onIOError(IOException ex) {
ta.append("Network problem ...\n"+ex+"\n");
ta.setCaretPosition( ta.getDocument().getLength() );
ex.printStackTrace();
connect.setEnabled(true);
uriField.setEditable(true);
draft.setEditable(true);
close.setEnabled( false );
}
};
close.setEnabled( true);
connect.setEnabled(false);
uriField.setEditable(false);
draft.setEditable(false);
cc.connect();
} catch (URISyntaxException ex) {
ta.append(uriField.getText() + " is not a valid WebSocket URI\n");
}
} else if (e.getSource() == close) {
try {
cc.close();
} catch(Exception ex) {
ex.printStackTrace();
}
}
}
public static void main(String[] args) {
new ChatClient();
}
}