package org.kne.cloud.network.klalb;

import java.io.File;
import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.channels.ServerSocketChannel;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Iterator;
import java.util.List;
import java.util.NoSuchElementException;
import java.util.Scanner;

import org.kne.cloud.network.MultipurposeSocketAddress;
import org.kne.cloud.network.SocketChannelListener;
import org.kne.cloud.network.ipv6.IPv6NetworkLink;
import org.kne.cloud.network.ipv6.RouteItem;
import org.kne.cloud.network.klalb.ui.KLALBStateGUI3;
import org.kne.cloud.network.klalb.ui.UIEnv;
import org.kne.cloud.network.perf.Kperf;
import org.kne.cloud.network.perf.NodeBenchmark;
import org.kne.debug.TimeDebugger;
import org.kne.membandboost.MembandBenchmark;
import org.kne.membandboost.ObjectPoolBandwidthTest;

public class KLALBMain {
	public static KLALBStateGUI3 ksg;
	public static void main(String[] args) throws IOException {
		try {
			UIEnv.inituie();
		}catch(Throwable e) {
			e.printStackTrace();
		}
		System.out.println(CONST.klalb+" V"+CONST.klalbver);
		Scanner scn=new Scanner(System.in);
		TimeDebugger dtb=new TimeDebugger();
		dtb.putTime("Start");
		File configJson=new File("klalb-config.json");
		
		KLALBProxySystem kpcje=new KLALBProxySystem();
		dtb.putTime("Create");
		kpcje.loadConfigJson(configJson);
		dtb.putTime("Load");
		System.out.println("=".repeat(65));
		System.out.println("  - IPv6 Address / End SID: "+kpcje.getKlalbController().getSelf().getAddress());
		System.out.println("=".repeat(65));
		try {
			if(!kpcje.getControllerConfig().isNogui())
		openGUI(kpcje); 
		}catch(Throwable e) {
			e.printStackTrace();
		}
		dtb.putTime("UI");
		//dtb.print();
		/*MultipurposeSocketAddress mpa=new MultipurposeSocketAddress("127.9.9.9", 49573);
		kpcje.enableRemoteManagement(mpa);
		System.out.println("远程管理端口已在"+mpa+"端口上开启");*/
		/*if(true)
		return;*/
		ServerSocketChannel kpsvr=KLALBVirtualServerSocketChannel.open(kpcje.getKlalbController());
		kpsvr.bind(new InetSocketAddress("::0", 4564));
		SocketChannelListener stlr=new SocketChannelListener(kpsvr);
		stlr.setCon((scl)->{
			try {
				new Kperf(new StreamChannelKLALBPacketLink(scl,false)).startPerfing();
			} catch (IOException e) {
				e.printStackTrace();
			}
		});
		System.out.println("Done!");
		while(true) {
			System.out.print("KLALB>");
			try {
		String s=scn.nextLine();
		String tri=s.trim();
		if(tri.isEmpty()) {
			continue;
		}
		String[]sc=tri.split(" ");
		switch(sc[0]) {
		case "?":
		case "help":
			System.out.println("  help / ?: see help");
			System.out.println("  monitor: show monitor GUI");
			System.out.println("  links-state: query link states");
			System.out.println("  links-add <addr:port>: add link");
			System.out.println("  links-remove <addr:port>: remove link");
			System.out.println("  links-reconnect: reconnect all link");
			System.out.println("  kltp-state: query kltp states");
			System.out.println("  route: display internal route table");
			System.out.println("  kperf <addr:port>: performance benchmark");
			System.out.println("  exit: Exit");
		break;
		
		case "monitor":
			try {
			openGUI(kpcje);
			}catch(Throwable e) {
				e.printStackTrace();
			}
			break;
		case "links-state":
			System.out.println("links state：");
			//System.out.println("状态\t上传流量\t下载流量\t上传速度\t下载速度\t上传延迟\t下载延迟\t上传抖动\t下载抖动");
			System.out.println("state\tspd↑\tspd↓\tPPS↑\tPPS↓\tdelay↑\tdelay↓\tjitter↑\tjitter↓");
			synchronized (kpcje.getKlalbController().getLines()) {
			for (Iterator<IPv6NetworkLink> iterator = kpcje.getKlalbController().getLines().iterator(); iterator.hasNext();) {
				IPv6NetworkLink link=iterator.next();
				if(link instanceof KLALBRemoteLink) {
				KLALBRemoteLink hostPort =  (KLALBRemoteLink) link;
				System.out.println(hostPort .toString());
				}
			}
			}
			break;
		case "links-add":
			if(sc.length>=2) {
			MultipurposeSocketAddress mpsa=new MultipurposeSocketAddress(sc[1]);
			List<KLALBRemoteLink>addl=kpcje.getKlalbController().addRemoteLines(mpsa);
			if(addl.isEmpty()) {
				//System.out.println("添加失败，线路已存在！");
				System.out.println("link already exist!");
			}else {
				for (Iterator<KLALBRemoteLink> iterator = addl.iterator(); iterator.hasNext();) {
					KLALBRemoteLink klalbRemoteLine = (KLALBRemoteLink) iterator.next();
					
			//System.out.println("添加成功："+klalbRemoteLine.getMonitor().getName());
					System.out.println("Successfully added: "+klalbRemoteLine.getName());
				}
			}
			}else {
			//	System.out.println("请输入要添加地址:端口!");
				System.out.println("Input Address:port");
			}
			break;
		case "links-remove":
			if(sc.length>=2) {
			MultipurposeSocketAddress mpsa=new MultipurposeSocketAddress(sc[1]);
			List<KLALBRemoteLink>rmvl=kpcje.getKlalbController().removeRemoteLines(mpsa);
			if(rmvl.isEmpty()) {
				System.out.println("No matched link has been found.");
			}else {
				for (Iterator iterator = rmvl.iterator(); iterator.hasNext();) {
					KLALBRemoteLink klalbRemoteLine = (KLALBRemoteLink) iterator.next();
					
			System.out.println("Successfully removed: "+klalbRemoteLine.getName());
				}
			}
			}else {
				System.out.println("Invalid arguments.");
			}
			break;
		case "exit":
			System.out.println("Exited.");
			System.exit(0);
			break;
		case "links-reconnect":
			System.out.println("Trying to reconnect all links.");
			kpcje.getKlalbController().reconnectImmediately();
			break;
		case "kltp-state":
			System.out.println(kpcje.getKlalbController().getKLTPregister().toString());
			break;
		case "route":
			List<RouteItem> lri=new ArrayList<>( kpcje.getKlalbController().getIpv6Router().getCurrentRouteTabel());
			Collections.sort(lri);
			System.out.println("Route Table:");
			System.out.println("Prefix\tProtocol\tPref\tCost\tFlag\tNext Hop\tInterface");
			for (Iterator<RouteItem> iterator = lri.iterator(); iterator.hasNext();) {
				RouteItem routeItem = (RouteItem) iterator.next();
				System.out.println(routeItem.getDestination()+"\t"+routeItem.getProto()+"\t"+routeItem.getPre()+"\t"+routeItem.getCost()+"\t"+routeItem.getFlag()+"\t"+routeItem.getNexthop()+"\t"+routeItem.getDestlink().getName());
			}
			
			break;
		case "kperf":
			if(sc.length>=2) {
				MultipurposeSocketAddress mpsa=new MultipurposeSocketAddress(sc[1]);
				Kperf kp=new Kperf(mpsa);
				kp.startPerfing();
			}else {
				System.out.println("Invalid arguments.");
			}
				
			break;
		case "membandbenchmark":
			MembandBenchmark.main(args);
			break;
		case "memqueuebenchmark":

			ObjectPoolBandwidthTest.main(args);
			break;
		//case "$$SYSTEM:":
			//System.out.println();
			//break;
		case "nodebenchmark":
			int nodenumber=100;
			if(sc.length>=2) {
				nodenumber=Integer.parseInt(sc[1]);
			}
			NodeBenchmark nbc=new NodeBenchmark(kpcje.getKlalbController(),nodenumber);
			nbc.startPerfing();
			break;
		default:
			System.out.println("Unknown command. See \"help\" or \"?\".");
		}
			}catch (NoSuchElementException err) {
				System.out.println("Exited.");
				System.exit(0);
			}catch(RuntimeException e) {
				System.out.println("Exception Occured!");
				e.printStackTrace();
			}
		}
		
	}
	private static void openGUI(KLALBProxySystem kpcje) throws RuntimeException{
		Thread t=new Thread(()->{
		if(ksg==null) {
			ksg=kpcje.getKLALBGUI();
		}
		ksg.setVisible(true);
		});
		t.start();
	}

}