#include "usg_common.h"
#include "netdisk.h"
#include "hcnetdisk.h"
#include <jsoncpp/json/json.h>
#include <nng/nng.h>
#include <nng/protocol/reqrep0/rep.h>
#include <nng/protocol/reqrep0/req.h>
const char *url = "tcp://127.0.0.1:8899";
const char *localUrl = "tcp://127.0.0.1:9988";
void
fatal(const char *func, int rv)
{
  fprintf(stderr, "%s: %s\n", func, nng_strerror(rv));
  exit(1);
}

void doLogin(nng_socket sock) {
	char *     buf = NULL;
	int        rv;
	size_t     sz;
	Json::Value request;
	Json::Value arguments;
    arguments["loginUUID"] = "1";
    arguments["deviceType"] = "HC";
    arguments["username"] = "admin";
    arguments["password"] = "a1234567";
    arguments["host"] = "192.168.20.11";
    arguments["port"] = 8000;
    request["method"] = "login";
    request["arguments"]= arguments;

    std::string str = request.toStyledString();

    if ((rv = nng_send(sock, strdup(str.c_str()), str.length(), 0)) != 0) {
		fatal("nng_send", rv);
	}
	if ((rv = nng_recv(sock, &buf, &sz, NNG_FLAG_ALLOC)) != 0) {
		fatal("nng_recv", rv);
	}
	std::cout << buf;
	nng_free(buf, sz);
	
	 
}

void doDownloadByTime(nng_socket sock) {
	char *     buf = NULL;
	int        rv;
	size_t     sz;
	Json::Value request;
	Json::Value arguments;
    arguments["loginUUID"] = "1";
    Json::Value start;
    start["year"] = 2020;
    start["mon"] = 4;
    start["day"] = 10;
    start["hour"] = 14;
	start["min"] = 20;
    start["sec"] = 0;
    arguments["start"] = start;
    Json::Value end;
    end["year"] = 2020;
    end["mon"] = 4;
    end["day"] = 10;
    end["hour"] = 15;
	end["min"] = 20;
    end["sec"] = 0;
    arguments["end"] = end;
    
    arguments["channel"] = 1;
    arguments["destpath"] = "/home/basic/data/Downloads/test";
    arguments["host"] = "192.168.20.11";
    arguments["port"] = 8000;

    request["method"] = "downloadByTime";
    request["arguments"]= arguments;

    std::string str = request.toStyledString();

    if ((rv = nng_send(sock, strdup(str.c_str()), str.length(), 0)) != 0) {
		fatal("nng_send", rv);
	}
	if ((rv = nng_recv(sock, &buf, &sz, NNG_FLAG_ALLOC)) != 0) {
		fatal("nng_recv", rv);
	}
	std::cout << buf << std::endl;
	// Json::Reader jsonreader;
	// Json::Value response;
	// jsonreader.parse(buf, response);
	// std::cout << response["payload"]["filelist"][3];
	nng_free(buf, sz);
	
	 
}

int
client(const char *method)
{
	nng_socket sock;
	int        rv;

	if ((rv = nng_req0_open(&sock)) != 0) {
		fatal("nng_socket", rv);
	}
	if ((rv = nng_dial(sock, url, NULL, 0)) != 0) {
		fatal("nng_dial", rv);
	}
	printf("CLIENT: SENDING DATE REQUEST\n");

	if (strcmp(method, "login") == 0) {
		doLogin(sock);
	} else if (strcmp(method, "download") == 0) {
		doDownloadByTime(sock);
	} else {
		printf("Don't surpport %s \n", method);
	}
	
	nng_close(sock);
	return (0);
}

int
server()
{
	nng_socket sock;
	int        rv;

	if ((rv = nng_rep0_open(&sock)) != 0) {
		fatal("nng_rep0_open", rv);
	}
	if ((rv = nng_listen(sock, localUrl, NULL, 0)) != 0) {
		fatal("nng_listen", rv);
	}
	for (;;) {
		char *   buf = NULL;
		size_t   sz;
		if ((rv = nng_recv(sock, &buf, &sz, NNG_FLAG_ALLOC)) != 0) {
			fatal("nng_recv", rv);
		}
		std::cout << buf << std::endl;

		Json::Value response;
		response["code"] = 0;
		std::string str = response.toStyledString();
		rv = nng_send(sock, strdup(str.c_str()), str.length(), NNG_FLAG_ALLOC);
		if (rv != 0) {
			fatal("nng_send", rv);
		}
		// if ((sz == sizeof(uint64_t)) &&
		//     ((GET64(buf, val)) == DATECMD)) {
		// 	time_t now;
		// 	printf("SERVER: RECEIVED DATE REQUEST\n");
		// 	now = time(&now);
		// 	printf("SERVER: SENDING DATE: ");
		// 	showdate(now);

		// 	// Reuse the buffer.  We know it is big enough.
		// 	PUT64(buf, (uint64_t) now);
			
		// 	continue;
		// }
		// Unrecognized command, so toss the buffer.
		//nng_free(buf, sz);
	}
}


int
main(const int argc, const char **argv)
{
	if ((argc > 1)) {
	  client(argv[1]);
	  server();
	}
	// std::string str("123");
	// char *str2="123";
	// printf("str length  %d, %d\n", str.length(), strlen(str2));

	fprintf(stderr, "Usage: client  <method> ...\n");
	return (1);
}