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();

}

}