package chatclient;

/**
 * <p>Title: </p>
 *
 * <p>Description: </p>
 *
 * <p>Copyright: Copyright (c) 2007</p>
 *
 * <p>Company: </p>
 *
 * @author not attributable
 * @version 1.0
 */


import javax.swing.*;
import java.awt.*;
import java.awt.event.*;
import java.net.*;
import java.io.*;
import java.util.StringTokenizer;

public class ChatClient extends JFrame
{
    JTextArea ta = new JTextArea(12,28);
    JTextField tf1 = new JTextField(10);
    JTextField tf2 = new JTextField(29);
    JButton b1 = new JButton("登錄");
    JButton b2 = new JButton("退出");
    JButton b3 = new JButton("發(fā)送");
    JLabel lab1 = new JLabel("用戶名：");
    JLabel lab2 = new JLabel("信息： ");
    List list = new List(10);
    JRadioButton rb1 = new JRadioButton("廣播發(fā)送",true);
    JRadioButton rb2 = new JRadioButton("選擇發(fā)送");
    ButtonGroup bg = new ButtonGroup();
    JScrollPane sp = new JScrollPane();
    JPanel p1 = new JPanel();
    JPanel p2 = new JPanel();
    JPanel p3 = new JPanel();
    JPanel p4 = new JPanel();

    Socket socket;
    PrintStream ps;
    String name;
    String ip;

    public ChatClient()
    {
        super("聊天室");
        p1.add(lab1);
        p1.add(tf1);
        p1.add(b1);
        p1.add(b2);
        p2.add(lab2);
        p2.add(tf2);
        p2.add(b3);
        p3.setLayout(new BorderLayout());
        p3.add(list,BorderLayout.NORTH);
        p3.add(rb1,BorderLayout.CENTER);
        p3.add(rb2,BorderLayout.SOUTH);
        bg.add(rb1);
        bg.add(rb2);
        sp.getViewport().add(ta);
        p4.add(p3);
        p4.add(sp);
        getContentPane().add(p1,BorderLayout.NORTH);
        getContentPane().add(p4,BorderLayout.CENTER);
        getContentPane().add(p2,BorderLayout.SOUTH);
        pack();
        //設(shè)置聊天室界面居中顯示
        Dimension screenSize = Toolkit.getDefaultToolkit().getScreenSize();
        Dimension frameSize = this.getSize();
        if(frameSize.width > screenSize.width)
        {
            frameSize.width = screenSize.width;
        }
        if(frameSize.height > screenSize.height)
        {
            frameSize.height = screenSize.height;
        }
        setLocation((screenSize.width - frameSize.width)/2,
                    (screenSize.height - frameSize.height)/2);
        setSize(460,330);
        setVisible(true);
        setResizable(false);
        b2.setEnabled(false);

        b1.addActionListener(new ActionListener()
        {
            public void actionPerformed(ActionEvent e)
            {
                try
                {
                    socket = new Socket(InetAddress.getLocalHost(),7000);
                    if(socket != null)
                    {
                        StringBuffer msg = new StringBuffer("CONNECT:");
                        name = tf1.getText();
                        ip = InetAddress.getLocalHost().toString();
                        msg.append(name+":"+ip);
                        ps = new PrintStream(socket.getOutputStream());
                        ps.println(msg);
                        ps.flush();
                        ta.setText("");
                        ta.append(name+"登錄成功！"+"\n");
                        b1.setEnabled(false);
                        b2.setEnabled(true);
                        //啟動客戶端線程，監(jiān)聽服務(wù)器端傳來的信息
                        Thread t = new ClientListener(socket);
                        t.start();
                    }
                }
                catch(IOException ex)
                {
                    ta.append("Error: "+ex+"\n");
                }
            }
        });
        b2.addActionListener(new ButtonListener());
        b3.addActionListener(new ButtonListener());
        addWindowListener(new WindowAdapter()
        {
            public void windowClosing(WindowEvent e)
            {
                if(socket != null)
                {
                    ps.println("QUIT:" + name);
                    ps.flush();
                    closeSocket();
                }
                System.exit(0);
            }
        });
    }

    class ClientListener extends Thread
    {
        Socket socket;
        BufferedReader br;
        PrintStream ps;
        String message;

        public ClientListener(Socket s)
        {
            try
            {
                socket = s;
                br = new BufferedReader(new InputStreamReader(socket.getInputStream()));
                ps = new PrintStream(socket.getOutputStream());
            }
            catch(IOException ex)
            {
                ta.append("Error: "+ex+"\n");
            }
        }

        public void run()
        {
            while(true)
            {
                try
                {
                    message = br.readLine();
                    if(message == null)
                        break;
                }
                catch(IOException ex)
                {
                    ta.append("Error-debug: "+ex+"\n");
                }
                StringTokenizer st = new StringTokenizer(message,":");
                String keyword = st.nextToken();

                //服務(wù)器端傳來各種信息的處理
                if(keyword.equals("PEOPLE"))
                {
                    list.removeAll();
                    while(st.hasMoreTokens())
                    {
                        String user = st.nextToken();
                        list.add(user);
                    }
                }
                else if(keyword.equals("MSG"))
                {
                    String broadcast = st.nextToken();
                    String user = st.nextToken();
                    StringBuffer sb = new StringBuffer(user+st.nextToken("\0"));
                    ta.append(sb+"\n");
                }
                else if(keyword.equals("QUIT"))
                {
                    String who = st.nextToken();
                    if(who.equals("Server"))
                    {
                        closeSocket();
                        ta.append("聊天服務(wù)器已經(jīng)關(guān)閉！" + "\n");
                        list.removeAll();
                        b2.setEnabled(false);
                    }
                    else
                    {
                        closeSocket();
                        b1.setEnabled(true);
                        b2.setEnabled(false);
                        list.removeAll();
                        ta.append(name + "退出了聊天室！" + "\n");
                    }
                    break;
                }
            }
        }
    }

    public void closeSocket()
    {
        if(socket != null)
        {
            try
            {
                socket.close();
            }
            catch (IOException ex)
            {
                ta.append("Error: " + ex + "\n");
            }
        }
    }

    class ButtonListener implements ActionListener
    {
        public void actionPerformed(ActionEvent e)
        {
            if(e.getActionCommand().equals("退出"))
            {
                ps.println("QUIT:" + name);
                ps.flush();
            }
            else if(e.getActionCommand().equals("發(fā)送"))
            {
                StringBuffer msg = new StringBuffer("MSG:");
                String info = name+":"+tf2.getText();

                if(rb1.isSelected())
                {
                    msg.append("BROAD:"+info);
                }
                else
                {
                    msg.append( list.getSelectedItem()+":"+info);
                }
                ps.println(msg);
                ps.flush();
                tf2.setText("");
            }
        }
    }

    public static void main(String[] args)
    {
        ChatClient app = new ChatClient();
    }
}