#include "pip.h"
#include "pifiletransfer.h"
#include "pidatatransfer.h"
#include "piscreen.h"
#include "piscreentiles.h"


TileSimple pmt;


using namespace PICoutManipulators;

class UDPFileTransfer: public PITimer {
	PIOBJECT_SUBCLASS(UDPFileTransfer, PITimer)
public:
	UDPFileTransfer(const PIString &src_ip_port, const PIString &dst_ip_port, bool test = false) {
		quet_ = false;
		test_ = test;
		eth.setReadAddress(src_ip_port);
		eth.setSendAddress(dst_ip_port);
		if (test_) {
			testt.setCRCEnabled(false);
			//testt.setPacketSize(1280);
			testt.setName("TEST");
			//testt.setTimeout(0.1);
			CONNECTU(&testt, sendRequest, this, ftsend);
		} else {
			ft.setPacketSize(65000);
			ft.setName("PIFT");
			CONNECTU(&ft, sendRequest, this, ftsend);
			CONNECTU(&ft, sendFilesStarted, this, ftevent);
			CONNECTU(&ft, receiveFilesStarted, this, ftevent);
			CONNECTU(&ft, sendFilesFinished, this, ftevent);
			CONNECTU(&ft, receiveFilesFinished, this, ftevent);
		}
		CONNECTU(&eth, threadedReadEvent, this, received);
		start(50);
		eth.setParameter(PIEthernet::SeparateSockets);
		eth.startThreadedRead();
	}

	void setQuet(bool quet) {quet_ = quet;}

	void startSend(const PIStringList &file) {
		ft.send(file);
	}

	void startTest() {
		PIByteArray ba(1024*1024*8);
		testt.send(ba);
	}

	PIFileTransfer ft;
	PIDataTransfer testt;
	bool test_;

private:
	PIEthernet eth;
	bool quet_;

	void tick(void *, int) {
		if (ft.isStarted()) {
			ftevent();
			updatePMT();
			if (PIKbdListener::exiting) {
				ft.stopSend();
				ft.stopReceive();
			}
			//			piCout << (int)ft.diagnostic().quality();
		}
		if (testt.isSending() || testt.isReceiving()) {
			ftevent();
			updatePMT();
		}
	}

	EVENT_HANDLER(void, ftevent) {
		if (test_) {

#ifdef WINDOWS
			piCout
		#else
			PICout(AddSpaces) << ClearLine
					 #endif
							  << testt.stateString()
							  << testt.diagnostic().receiveSpeed()
							  << testt.diagnostic().sendSpeed()
							  << "("
							  << PIString::readableSize(testt.bytesCur()) << "/" << PIString::readableSize(testt.bytesAll())
							  << ")"
					 #ifndef WINDOWS
							  << Flush
					 #endif
								 ;
			return;
		}
		if (quet_) return;

#ifdef WINDOWS
		piCout
		#else
		PICout(AddSpaces) << ClearLine
					 #endif
						  << ft.stateString() << ft.curFile()
						  << ft.diagnostic().receiveSpeed()
						  << ft.diagnostic().sendSpeed()
						  << "(" << PIString::readableSize(ft.bytesFileCur()) << "/" << PIString::readableSize(ft.bytesFileAll()) << ", "
						  << PIString::readableSize(ft.bytesCur()) << "/" << PIString::readableSize(ft.bytesAll()) << ")"
						  << "ETA" << (ft.diagnostic().state().received_bytes_per_sec > 0 ?
										   PIString::fromNumber(PISystemTime::fromSeconds((ft.bytesAll() - ft.bytesCur()) /
																						  ft.diagnostic().state().received_bytes_per_sec).toSeconds())
										 : PIString("unknown"))
					 #ifndef WINDOWS
						  << Flush
					 #endif
							 ;
	}

	EVENT_HANDLER1(void, ftsend, PIByteArray &, data) {
		eth.send(data);
	}

	EVENT_HANDLER2(void, received, uchar * , readed, int, size) {
		PIByteArray ba(readed, size);
		if(test_) {
			testt.received(ba);
			return;
		}
		ft.received(ba);
	}


	void updatePMT() {
		PIString pm;
		if (test_) pm = testt.packetMap();
		else pm = ft.packetMap();
		int wd = 110;
		pmt.content.resize(pm.size() / wd + 1);
		for (int i=0; i<pmt.content.size_s(); i++) {
			pmt.content[i].first = pm.mid(wd*i, piMini(pm.size() - i*wd, wd));
		}
	}
};


void usage() {
	piCout << Bold << "PIP UDP file transfer";
	piCout << Cyan << "Version" << Bold << PIPVersion() << NewLine;
	piCout << Green << Bold << "Usage:" << Default << "\"pift [-hqf] -r <receive_ip> -s <send_ip> [-d <work_dir>] [-p <port>] [<path1>] [<path2>] [<path3>] [...]\"" << NewLine;
	piCout << Green << Bold << "Details:";
	piCout << "-f --fullpath              " << Green << "- full path in <receive_ip> and <send_ip>";
	piCout << "-h --help                  " << Green << "- display this message and exit";
	piCout << "-q --quet                  " << Green << "- quiet, no debug output to console";
	piCout << "-r --receive <receive_ip>  " << Green << "- set receive ip address, must be ip address of this computer";
	piCout << "-s --send <send_ip>        " << Green << "- set send ip address, address of computer which communicate with you";
	piCout << "-p --port  <port>          " << Green << "- UDP port for transfer, by default 50005";
	piCout << "-d --dir <work_dir>        " << Green << "- directory, where place received files";
	piCout << "<path>                     " << Green << "- add path to send, if no path, then \"pift\" working in receive mode";
	piCout << "-t --test                  " << Green << "- test mode";
}


int main (int argc, char * argv[]) {
	PICLI cli(argc, argv);
	cli.setOptionalArgumentsCount(-1);
	cli.addArgument("send", true);
	cli.addArgument("receive", true);
	cli.addArgument("dir", true);
	cli.addArgument("port", true);
	cli.addArgument("help");
	cli.addArgument("quet");
	cli.addArgument("fullpath");
	cli.addArgument("test", true);
	if (cli.hasArgument("test")) {
		PIString src;
		PIString dst;
		bool send_ = false;
		if (cli.argumentValue("test") == "1") {
			src = "127.0.0.1:50005";
			dst = "127.0.0.1:50006";
			send_ = true;
		} else {
			src = "127.0.0.1:50006";
			dst = "127.0.0.1:50005";
		}
		if (cli.hasArgument("send") && cli.hasArgument("receive") && cli.hasArgument("fullpath")) {
			src = cli.argumentValue("receive");
			dst = cli.argumentValue("send");
		}
		PIScreen screen(false);
		screen.enableExitCapture();
		screen.rootTile()->addTile(new TilePICout());
		screen.rootTile()->addTile(&pmt);
		UDPFileTransfer tuf(src, dst, true);
		screen.start();
		if (send_) tuf.startTest();
		screen.waitForFinish();
		return 0;
	}
	if ((!cli.hasArgument("send") || !cli.hasArgument("receive")) || cli.hasArgument("help")) {
		usage();
		return 0;
	}
	PIString src = cli.argumentValue("receive");
	PIString dst = cli.argumentValue("send");
	int port = -1;
	if (cli.hasArgument("port")) port = cli.argumentValue("port").toInt();
	if (port <=0) port = 50005;
	if (!cli.hasArgument("fullpath")) {
		src += ":"+PIString::fromNumber(port);
		dst += ":"+PIString::fromNumber(port);
	}
//	PIKbdListener kbd;
//	kbd.enableExitCapture();
	PIScreen screen(false);
	screen.enableExitCapture();
	screen.rootTile()->addTile(new TilePICout());
	screen.rootTile()->addTile(&pmt);
	screen.start();
	UDPFileTransfer f(src, dst);
	if (cli.hasArgument("dir")) f.ft.setDirectory(cli.argumentValue("dir"));
	if (cli.hasArgument("quet")) f.setQuet(true);
	piCout << "work directory" << f.ft.directory().absolutePath() << ", listen on" << src << ", send to" << dst;
	PIStringList fls = cli.optionalArguments();
	if (fls.size() > 0) {
		piCout << "send files" << fls;
		f.startSend(fls);
		PICout(0) << "\n";
		return 0;
	} else {
		piCout << "wait for receiving";
	}
	PICout(0) << "\n";
	screen.waitForFinish();
	return 0;

}