package com.soft.QQ;

import java.net.*;
import java.io.*;
import java.util.*;
import java.text.*;

/**
 * <p>Title: </p>
 *
 * <p>Description: </p>
 *
 * <p>Copyright: Copyright (c) 2007</p>
 *
 * <p>Company: </p>
 *
 * @author not attributable
 * @version 1.0
 */
public class Server
    implements Runnable {
  DatagramPacket mail_package;
  DatagramSocket socket;
  DatagramPacket in_package;
  boolean connected = false;
  ArrayList<People> list;
  Thread thread;
  public Server() {
    thread = new Thread(this);
    thread.start();
  }

  public void run() {
    receive();
  }

  /**
   * 群聊
   * 接收客戶端發來的消息并分析
   */
  public void receive() {
    DatagramSocket ds = null;
    try {
      ds = new DatagramSocket(5000);
      byte[] buf = new byte[1024];
      DatagramPacket dp = new DatagramPacket(buf, 1024);
      ds.receive(dp);
      ChatMessage cm = new Helper().unwrap(dp.getData());
      if (cm != null && Integer.parseInt(cm.getType()) == 1) { //這個包為消息包
        System.out.println("發送者：" + cm.getSender());
        System.out.println("時間：" + cm.getTime());
        System.out.println("消息內容：" + cm.getMessage());
        String message = cm.getSender() + cm.getTime() + cm.getMessage();
        send(cm.getSender(), cm.getMessage(), 1);
        chatRecord(message); //保存到聊天記錄
      }
      else if (cm != null && Integer.parseInt(cm.getType()) == 2) { //這個包為用戶登入信息包
        System.out.println("昵稱：" + cm.getSender());
        System.out.println("IP：" + cm.getIP());
        new People(cm.getSender(), cm.getIP());
        list.add(new People(cm.getSender(), cm.getIP())); //加入用戶
        String listtemp = null;
        for (int i = 0; i < list.size(); i++) {
          listtemp += list.get(i).getName() + ",";
        }
        sendOnlineUser(listtemp, 3);
      }
      else if (cm != null && Integer.parseInt(cm.getType()) == 3) { //這個包為在線列表用戶信息包
        System.out.println("用戶列表:" + cm.getOnlineTable());
      }
    }
    catch (Exception ex) {
      ex.printStackTrace();
    }
    finally {
      if (ds != null) {
        ds.close();
      }
    }
  }

  public void re() {
    DatagramSocket ds = null;
    DatagramPacket dp = null;
    try {
      ds = new DatagramSocket(5000);
      byte[] buf = new byte[1024];
      dp = new DatagramPacket(buf, 1024);
    }
    catch (Exception ex) {}
    while (true) {
      if (ds == null) {
        break;
      }
      else {
        try {
          ds.receive(dp);
          ChatMessage cm = new Helper().unwrap(dp.getData());
          if (cm != null && Integer.parseInt(cm.getType()) == 1) { //這個包為消息包
            System.out.println("發送者：" + cm.getSender());
            System.out.println("時間：" + cm.getTime());
            System.out.println("消息內容：" + cm.getMessage());
            String message = cm.getSender() + cm.getTime() + cm.getMessage();
            send(cm.getSender(), cm.getMessage(), 1);
            chatRecord(message); //保存到聊天記錄
          }
          else if (cm != null && Integer.parseInt(cm.getType()) == 2) { //這個包為用戶登入信息包
            System.out.println("昵稱：" + cm.getSender());
            System.out.println("IP：" + cm.getIP());
            new People(cm.getSender(), cm.getIP());
            list.add(new People(cm.getSender(), cm.getIP())); //加入用戶
            String listtemp = null;
            for (int i = 0; i < list.size(); i++) {
              listtemp += list.get(i).getName() + ",";
            }
            sendOnlineUser(listtemp, 3);
          }
          else if (cm != null && Integer.parseInt(cm.getType()) == 3) { //這個包為在線列表用戶信息包
            System.out.println("用戶列表:" + cm.getOnlineTable());
          }

        }
        catch (Exception ex) {
        }
      }
    }
  }

  /**
   * 從在線列表中獲得每個用戶的IP,轉發給所有在線用戶
   */
  public void send(String sender, String input, int i) {
    DatagramSocket ds = null;
    try {
      ds = new DatagramSocket();
      byte[] buf = new Helper().wrap(sender, input, i);
      for (int a = 0; a < list.size(); a++) {
        DatagramPacket dp = new DatagramPacket(buf,
                                               buf.length,
                                               InetAddress.getByName(
            list.get(a).getIP()),
                                               6000);
        ds.send(dp);
      }
    }
    catch (Exception ex) {
      ex.printStackTrace();
    }
    finally {
      if (ds != null) {
        ds.close();
      }
    }
  }

  /**
   * 給客戶端發送在線用戶信息
   */
  public void sendOnlineUser(String s, int i) {
    DatagramSocket ds = null;
    try {
      ds = new DatagramSocket();
      byte[] buf = new Helper().wrapOnlineUser(s, i);
      for (int a = 0; a < list.size(); a++) {
        DatagramPacket dp = new DatagramPacket(buf,
                                               buf.length,
                                               InetAddress.getByName(
            list.get(a).getIP()),
                                               6000);
        ds.send(dp);
      }
    }
    catch (Exception ex) {
      ex.printStackTrace();
    }
    finally {
      if (ds != null) {
        ds.close();
      }
    }
  }

  /**
   * 處理私聊請求
   * @param message String
   */
  public void p2p() {

  }

  /**
   * 聊天記錄
   *
   * @param mes
   */
  private void chatRecord(String message) {
    FileOutputStream fos = null;
    PrintStream ps = null;
    try {
      Date date = new Date();
      SimpleDateFormat sdf = new SimpleDateFormat("MM-dd HH:mm:ss");
      fos = new FileOutputStream("D:/ChatMessage.log", true);
      ps = new PrintStream(fos);
      ps.println(sdf.format(date) + "  " + message);
    }
    catch (Exception e) {
    }
    finally {
      try {
        ps.flush();
        ps.close();
        fos.close();
      }
      catch (Exception e) {
        e.printStackTrace();
      }
    }
  }

  public static void main(String[] args) {
    new Server();
  }
}