//super.ie 2.0聊天軟件
//實現了基本通信
//按照協議完成，部分功能在以后的更新中實現
package myprojects.client;
import java.awt.*;
import java.awt.event.*;
import java.io.*;
import java.net.*;
import javax.swing.*;
import java.util.*;
class Server extends JFrame implements ActionListener{
	ServerSocket ssock;
	JButton start=new JButton("start");
	JButton stop=new JButton("stop");
	JLabel num=new JLabel("    ");
	Hashtable clients=new Hashtable(); 
	Accept accept;
	public Server() {
	  this.getContentPane().setLayout(new FlowLayout());
	  this.getContentPane().add(start);
	  this.getContentPane().add(stop);
	  this.getContentPane().add(num);
	  stop.setEnabled(false);
	  start.addActionListener(this);
      stop.addActionListener(this);
      setSize(300, 80);
	  setTitle("Server");
	  this.addWindowListener(new WindowAdapter(){
      	public void windowClosing(WindowEvent e)
      	{ try{
      	   sengMsgToAll("serverquit");
      	   }catch(Exception ee){}
      	   dispose();
      	   System.exit(0);	
      	}
      });	  
      setVisible(true);
    }  
    public void actionPerformed(ActionEvent e)
    { if(e.getSource()==start)  
      {  try{
	       ssock=new ServerSocket(5555);//端口地址
	       stop.setEnabled(true);
	       start.setEnabled(false);
	       accept=new Accept(this); 
	     }catch(Exception ee){}
	  }
	  else
	  {
	    accept.flag=false;    	
	    Enumeration en=clients.elements();
        while(en.hasMoreElements())
        {
          ServerForClient sfClient=(ServerForClient)en.nextElement();   
    	  sfClient.flag=false;
    	}
    	stop.setEnabled(false);
	    start.setEnabled(true);	 	
	    try{
	      ssock.close();
	    }catch(Exception ee){}
	  }      
	}   
    public void sengMsgToAll(String msg)
    {
       Enumeration en=clients.elements();
       while(en.hasMoreElements())
       {
         ServerForClient sfClient=(ServerForClient)en.nextElement();   
    	 try{
    	   sfClient.out.writeUTF(msg);
         }catch(Exception ee){}
       }	
    }   
	public void sengMsgToOne(String msg,String name)
    {
       Enumeration en=clients.elements();
       while(en.hasMoreElements())
       {
         ServerForClient sfClient=(ServerForClient)en.nextElement();   
    	 if(sfClient.getClientName().equals(name))
    	 { try{
    	      sfClient.out.writeUTF(msg);
           }catch(Exception ee){}
           break;
         }
       }	
    }   
	public void refresh()
	{
	  Enumeration en=clients.keys();
	  sengMsgToAll("refresh");
	  while(en.hasMoreElements())
      {  
         sengMsgToAll("name");
         sengMsgToAll((String)en.nextElement());	
	  }	
	}
	public static void main(String args[]) {
		Server mainFrame = new Server();
    }
}
class Accept extends Thread
{ boolean flag=true; 
  Server server;
  Socket sock;
  Accept(Server server)
  {
  	this.server=server;
    this.start(); 	
  }
  public void run()
  {	while(flag)
    { try{
 	    sock=server.ssock.accept();  
	    ServerForClient sfclient=new ServerForClient(server,sock);	    
        String name=sfclient.getClientName();
        boolean f=false;
        Enumeration en=server.clients.keys();
        while(en.hasMoreElements())
        {
          if(((String)en.nextElement()).equals(name))
          {f=true;break;}
        }
        if(f==true)//驗證
        {  sfclient.out.writeUTF("equals");
           sock.close();
        }
        else
        {server.clients.put(name,sfclient);
         server.sengMsgToAll("msg");
         server.sengMsgToAll(name+"進入聊天室");
         server.refresh();
         sfclient.start();  
        }
      }catch(Exception ee){}
    }
  }	
}
class ServerForClient extends Thread
{  DataInputStream in;
   DataOutputStream out;
   Socket sock;
   Server server;
   private String name;
   boolean flag=true;
   ServerForClient(Server server,Socket sock)
   {  
      this.server=server;
   	  this.sock=sock;
      try{
        in=new DataInputStream(sock.getInputStream());
        out=new DataOutputStream(sock.getOutputStream());
        name=in.readUTF();
      }catch(Exception e){}
   }
   public String getClientName()
   {return name;}
   public void pwd() //密碼
   {
   	}
   public void time() //系統時間
   {}
   public void fun() //設置功能位
   {}
   public void Jie()//結束位
   {}
   public void Xuhao()   //序號
   {}
   public void con() //連接功能
   {}
   public void pic() //圖元
   {}
   public void intu() //插入圖元
   {}
   
   public void run() 
   { String type=null;
     String msg=null;
     String target=null;
   	 while(flag)
   	 {  try{
   	 	
   	       type=in.readUTF();
           if(type.equals("public"))//群聊
           {
           	  msg=in.readUTF();
              server.sengMsgToAll("msg");
           	  server.sengMsgToAll(name+"對大家說:"+msg);   
           }
           else if(type.equals("private"))
           {
           	  msg=in.readUTF();
           	  target=in.readUTF();
           	  server.sengMsgToOne("msg",target);
              server.sengMsgToOne(name+"對"+target+"說:"+msg,target);   
           }
           else if(type.equals("privateAll"))//私聊
           {
           	  msg=in.readUTF();
           	  target=in.readUTF();
           	  server.sengMsgToAll("msg");
              server.sengMsgToAll(name+"對"+target+"說:"+msg);   
           }
           else if(type.equals("say"))
           {
           	  msg=in.readUTF();
           	  target=in.readUTF();  
           	  server.sengMsgToAll("msg");
              server.sengMsgToAll(name+"對"+target+"說:"+msg);   
           }
           else if(type.equals("quit"))//退出
           {  
              server.clients.remove(name);
           	  server.sengMsgToAll("msg");
           	  server.sengMsgToAll(name+"已離開聊天系統");     	
              flag=false;
              server.refresh();
           }
        }catch(Exception ee){break;}
     }
     try{
       sock.close();
     }catch(Exception ee){}
   }
}