package mainManager;

import java.awt.BorderLayout;
import java.awt.Color;
import java.awt.GridLayout;
import java.awt.HeadlessException;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.event.WindowAdapter;
import java.awt.event.WindowEvent;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.ObjectOutputStream;
import java.net.DatagramPacket;
import java.net.DatagramSocket;
import java.net.InetAddress;
import java.net.SocketException;
import java.net.UnknownHostException;
import java.util.Arrays;
import javax.swing.JButton;
import javax.swing.JCheckBox;
import javax.swing.JFrame;
import javax.swing.JLabel;
import javax.swing.JPanel;
import javax.swing.JScrollPane;
import javax.swing.JTextArea;
import javax.swing.JTextField;
import totalThreads.ForwardThread;
import totalThreads.ReceiveThread;
import totalThreads.SendThread;
import transportPacket.SendDataPacket;
import totalThreads.TotalPacket;
import entity.Constant;
import entity.NetMap;
import entity.Router;
import entity.RouterTable;

/*每一个路由器都有的路由器线程*/
public class RouterThread extends JFrame implements Runnable {
    /*产生全局唯一的序列化的实体ID*/
    private static final long serialVersionUID = -9094445007188117205L;
    /*每一个路由器线程一个路由器实体*/
    private Router router;
    /*每一个路由器实体一个接受路由表线程*/
    private DatagramSocket receiveSocket =null;
    /*每一个路由器实体一个接受转发的数据线程*/
    private DatagramSocket receiveDataSocket =null;
    /*每个路由器线程一个接受线程*/
    private ReceiveThread receiveThread=null;
    /*每个路由器线程一个发送线程*/
    private SendThread sendThread=null;
    /*每个路由器线程一个转发线程*/
    private ForwardThread forwardThread=null;
    /*每个路由器线程一个主窗口*/
    private JFrame routerFrame;
    /*有三个显示路由器信息的 JTextArea，一个路由表，一个路由表的更新信息，一个是发送数据的显示信息*/
    private JTextArea routerAreaLeft,routerAreaRightNorth,routerAreaRightSouth; //构造路由器窗口的组件。
    /*提示信息三个请输 (目的路由 #数据 )，该路由器路由表信息，路由转发和数据转发的信息*/
    private JLabel commandPrompt,showMsgLink,showMsgForward;
    /*选择是否挂起该路由的提示*/
    private JLabel suspended;
    /*写出要达到的目的地和要发送的数据*/
    private JTextField enterField;
    /*提交数据*/
    private JButton submit;
    /*选择是否挂起该路由*/
    private JCheckBox selectSuspend;
    /*是否挂起该路由的内置变量布尔值*/
    public boolean suspendFlag = false;
    /*显示几个 panal,用于承载所有的组件。*/
    private JPanel firstLine, secondLine,centerArea,centerAreaLeft,centerAreaRight,centerAreaRightBorderCenter;
    /*默认的构造方法*/
    public RouterThread() {
        super();
    }
    /*进行本路由器注入的构造方法*/
    public RouterThread(Router router) {
        try {
            this.router = router;
            routerFrame = new JFrame("路由器算法模拟实现");
            commandPrompt = new JLabel("请输入(目的路由，数据 ):");
            suspended = new JLabel("选择是否使该路由发生挂起: ");
            enterField = new JTextField("输入样例2003，11",100);
            submit=new JButton("确认");
            submit.setSize(30,15);
            selectSuspend = new JCheckBox("开始挂起...");
            firstLine = new JPanel();
            secondLine = new JPanel();
            centerArea = new JPanel();
            centerAreaLeft = new JPanel();
            centerAreaRight = new JPanel();
            centerAreaRightBorderCenter = new JPanel();
            firstLine.setLayout(new GridLayout(1, 2)); secondLine.setLayout(new GridLayout(1, 2));
            centerArea.setLayout(new GridLayout(1, 2)); centerAreaLeft.setLayout(new BorderLayout());
            centerAreaRight.setLayout(new BorderLayout());
            centerAreaRightBorderCenter.setLayout(new GridLayout(2,1));
            firstLine.add(commandPrompt);
            firstLine.add(enterField);
            firstLine.add(submit);
            secondLine.add(suspended);
            secondLine.add(selectSuspend);
            enterField.setCaretPosition(enterField.getText().length());
            showMsgLink = new JLabel("以下是该路由器路由表信息 : ");
            showMsgLink.setOpaque(true);
            showMsgLink.setBackground(Color.red);
            showMsgForward = new JLabel("以下是路由转发和数据转发的信息 :");
            showMsgForward.setOpaque(true);
            showMsgForward.setBackground(Color.red);
            routerAreaLeft=new JTextArea();
            routerAreaRightNorth=new JTextArea();
            routerAreaRightSouth=new JTextArea();
            routerAreaLeft.setBackground(Color.white);
            routerAreaRightNorth.setBackground(Color.black);
            routerAreaRightSouth.setBackground(Color.white);
            JScrollPane scrollPaneLeft=new JScrollPane(routerAreaLeft);
            JScrollPane scrollPaneRightNorth=new JScrollPane(routerAreaRightNorth);
            JScrollPane scrollPaneRightSouth=new JScrollPane(routerAreaRightSouth);
            centerAreaLeft.add(showMsgLink,BorderLayout.NORTH); centerAreaLeft.add(scrollPaneLeft,BorderLayout.CENTER);
            centerAreaRight.add(showMsgForward,BorderLayout.NORTH);
            centerAreaRightBorderCenter.add(scrollPaneRightNorth);
            centerAreaRightBorderCenter.add(scrollPaneRightSouth);
            centerAreaRight.add(centerAreaRightBorderCenter,BorderLayout.CENTER);
            centerArea.add(centerAreaLeft);
            centerArea.add(centerAreaRight);
            routerFrame.add(firstLine, BorderLayout.NORTH);
            routerFrame.add(centerArea,BorderLayout.CENTER);
            routerFrame.add(secondLine, BorderLayout.SOUTH);
            routerFrame.setVisible(true);
            routerFrame.setBounds(200, 200, 420, 300);
            showTable(routerAreaLeft);
            receiveSocket = new DatagramSocket(router.getPort());
            receiveDataSocket = new DatagramSocket(router.getPort()+10);
            sendThread=new SendThread(receiveSocket,router);
            forwardThread=new ForwardThread(routerAreaLeft,routerAreaRightNorth,routerAreaRightSouth,receiveDataSocket,router);
            receiveThread=new ReceiveThread(routerAreaLeft,routerAreaRightNorth,routerAreaRightSouth,receiveSocket,router);
            routerFrame.addWindowListener(new WindowAdapter(){ public void windowClosing(WindowEvent e) {
                routerFrame.dispose();
            }
            });
            selectSuspend.addActionListener(new ActionListener(){ public void actionPerformed(ActionEvent event){
                suspendFlag = !suspendFlag;
                if (suspendFlag) {
                    sendThread.suspend();
                    receiveThread.suspend();
                    forwardThread.suspend();
                }else{
                    sendThread.resume();
                    receiveThread.resume();
                    forwardThread.resume();
                    routerAreaLeft.setText("");
                    showTable(routerAreaLeft);
                }
            }
            });
       submit.addActionListener(new ActionListener(){ public void actionPerformed(ActionEvent event){
                String sendMsgTotal=enterField.getText();
                int destRouterId=0;
                byte [] msgBytes=null;
                try {
                    String [] sendMsgs=sendMsgTotal.split(",");
                    destRouterId=Integer.parseInt(sendMsgs[0]);
                    msgBytes=sendMsgs[1].getBytes();
                    routerAreaRightSouth.setText("");
                    sendData(destRouterId,msgBytes);
                    enterField.setText("如2003&AAA");
                } catch (NumberFormatException e) { routerAreaRightSouth.setText(""); routerAreaRightSouth.append("输入格式错误，请重新输入 ....");
                }
            }
            });
        } catch (HeadlessException e) { e.printStackTrace();
        } catch (SocketException e) { e.printStackTrace();
        }
    }
    /*继承 Thread 函数的 run 方法，进行开启每个路由器的两个监听线程（转发数据和接收路由表），一个发送线程*/
    @Override
    public synchronized void run() {
        forwardThread.start();
        receiveThread.start();
        sendThread.start();
    }
    /*展示本路由表信息的方法*/
    public synchronized void showTable(JTextArea routerAreaLeft){
        RouterTable routerTable=router.getRouterTable();
        String distances=Arrays.toString(routerTable.getDistance());
        String nextHops=Arrays.toString(routerTable.getNextHop());
        routerAreaLeft.setText("");
        routerAreaLeft.append("目的路由：\n[2001,2002,2003,2004,2005,2006]\n");
        routerAreaLeft.append("下一跳是：\n"+nextHops+"\n");
        routerAreaLeft.append("距离是：\n"+distances+"\n");
    }
    /*点击发送按钮，来向  destRouterId发送 msgBytes 数据。*/
    public synchronized boolean sendData(int destRouterId,byte [] msgBytes){ boolean sendDataSuccess=false;
        DatagramSocket sendSocket = null;
        ByteArrayOutputStream bos = null;
        ObjectOutputStream oos = null;
        int severReceivePort=0;
        DatagramPacket datagramPacket = null;
        byte[] bufferOut = new byte[1024 * 1024];
        try {
            sendSocket=new DatagramSocket();
            TotalPacket totalPacket=new TotalPacket("sendData");
            SendDataPacket sendDataPacket=new SendDataPacket();
            sendDataPacket.setSourceRouterId(router.getRouterId());
            sendDataPacket.setDestRouterId(destRouterId);
            sendDataPacket.setHops(16);
            int forwardNextHopRouterId=0;
            for(int i=0;i<new NetMap().getInitInternetMap().length;i++){
                if(i==(destRouterId-Constant.getRouterIdBasic())){
                forwardNextHopRouterId=router.getRouterTable().getNextHop()[i];
                routerAreaRightSouth.append("我 最 终 的 目 的 是 "+destRouterId+"\n");
                routerAreaRightSouth.append("我 要 转 发 这 个 数 据 包 给 "+forwardNextHopRouterId+"\n");
                routerAreaRightSouth.append("要 传 送 的 数 据 是 :"+new String(msgBytes)+"\n");
                sendDataPacket.setData(msgBytes);
                totalPacket.setSendDataPacket(sendDataPacket);
                InetAddress serverInetAddress =InetAddress.getByName("localhost");
                bos = new ByteArrayOutputStream();
                oos = new ObjectOutputStream(bos);
                oos.writeObject(totalPacket);
                bufferOut = bos.toByteArray();
                severReceivePort=forwardNextHopRouterId-Constant.getRouterIdBasic()+Constant.getPortBasic()+10;
                datagramPacket = new DatagramPacket(bufferOut, bufferOut.length,serverInetAddress, severReceivePort);
                sendSocket.send(datagramPacket);
                routerAreaRightSouth.append("转发完毕 \n");
                break;
            }
            }
            sendSocket.close();
            sendDataSuccess=true;
        } catch (UnknownHostException e) {
                    e.printStackTrace();
        } catch (IOException e) { e.printStackTrace();
        }finally{
            sendSocket.close();
        }
        return sendDataSuccess;
    }
}