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,30);
    JTextField tf1 = new JTextField(10);
    JTextField tf2 = new JTextField(29);
    JButton b1 = new JButton("登錄");

    JButton b3 = new JButton("發(fā)送");
    JLabel lab1 = new JLabel("用戶名：");
    JLabel lab2 = new JLabel("信息： ");
    List list = new List(12);
    JRadioButton rb1 = new JRadioButton("發(fā)送",true);

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

          p2.add(lab2);
          p2.add(tf2);
          p2.add(b3);
          p3.setLayout(new BorderLayout());
          p3.add(list,BorderLayout.NORTH);
          p3.add(rb1,BorderLayout.CENTER);

          bg.add(rb1);

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

          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);
                          //啟動(dòng)客戶端線程，監(jiān)聽(tīng)服務(wù)器端傳來(lái)的信息
                          Thread t = new ClientListener(socket);
                          t.start();
                      }
                  }
                  catch(IOException ex)
                  {
                      ta.append("Error: "+ex+"\n");
                  }
              }
          });
          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ù)器端傳來(lái)各種信息的處理
                  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();
                      }
                      else
                      {
                          closeSocket();
                          b1.setEnabled(true);

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