/*
** Authors: 
** Rama Alebouyeh ( rama[at]cs.ucsb.edu )
** Matthew Allen ( msa[at]cs.ucsb.edu )
*/

#include <stdio.h>
#include <unistd.h>
#include <sys/time.h>
#include "chimera.h"
#include "dtime.h"

#define USAGE "receiver [ -j bootstrap:port ] port key"
#define OPTSTR "j:"
extern char *optarg;
extern int optind;

#define TEST_CHAT 15
#define MSG_NUM 1000

int counter;
int recvpack;
int total;
int length;

char mylog[20000];
char *logptr;

struct timeval tv1;
struct timeval tv2;
double startTime, endTime;

void test_fwd (Key ** kp, Message ** mp, ChimeraHost ** hp)
{

    Key *k = *kp;
    Message *m = *mp;
    ChimeraHost *h = *hp;

    fprintf (stderr, "Routing %s (%s) to %s via %s:%d\n",
	     (m->type == TEST_CHAT) ? ("CHAT") : ("JOIN"), m->payload,
	     k->keystr, h->name, h->port);

}

void test_del (Key * k, Message * m)
{
    struct timeval tv;


    if (m->type == TEST_CHAT)
	{
	    gettimeofday (&tv, NULL);
	    if (counter == 0)
		{
		    tv1 = tv;
			startTime = getTime();
		    recvpack = 1;
		    logptr = mylog;
			length = strlen (m->payload);
		    fprintf (stderr, "len:%d\n", strlen (m->payload));
		    //      fprintf(stderr, "msg:%s\n",m->payload);
		}
	    else if (tv.tv_sec == tv1.tv_sec)
		{
		    //      fprintf(stderr,"-\n");
		    recvpack++;
		}
	    else if (tv.tv_sec > tv1.tv_sec)
		{
		    sprintf (logptr, "total packet recieved %d at: %s",
			     recvpack, ctime ((time_t *)&tv1));
		    logptr = logptr + strlen (logptr);
		    tv1 = tv;
		    recvpack = 1;
		}
	    if (counter == MSG_NUM - 1)
		{
			endTime = getTime();
		    sprintf (logptr, "total packet recieved %d at: %s",
			     recvpack, ctime ((time_t *)&tv1));
		    fprintf (stderr, "%s\n", mylog);
		    fprintf (stderr, "Total packet recvd: %d in %f secs\n", total++, endTime-startTime);
			fprintf (stderr, "Throughput is %f Bytes per sec\n", (length * total)/(endTime - startTime));
		    exit (0);
		}
	    total++;
	    counter++;
	}
}



void test_update (Key * k, ChimeraHost * h, int joined)
{

    if (joined)
	{
	    fprintf (stderr, "Node %s:%s:%d joined neighbor set\n", k->keystr,
		     h->name, h->port);
	}
    else
	{
	    fprintf (stderr, "Node %s:%s:%d leaving neighbor set\n",
		     k->keystr, h->name, h->port);
	}

}

int main (int argc, char **argv)
{

    int opt;
    char *hn = NULL;
    int port, joinport;
    ChimeraHost *host = NULL;
    char tmp[256];
    Key key, keyinput;
    int i, j;
    ChimeraState *state;

    while ((opt = getopt (argc, argv, OPTSTR)) != EOF)
	{
	    switch ((char) opt)
		{
		case 'j':
		    for (i = 0; optarg[i] != ':' && i < strlen (optarg); i++);
		    optarg[i] = 0;
		    hn = optarg;
		    sscanf (optarg + (i + 1), "%d", &joinport);
		    break;
		default:
		    fprintf (stderr, "invalid option %c\n", (char) opt);
		    fprintf (stderr, "usage: %s\n", USAGE);
		    exit (1);
		}
	}

    if ((argc - optind) != 2)
	{
	    fprintf (stderr, "usage: %s\n", USAGE);
	    exit (1);
	}

    counter = 0;
    recvpack = 0;
    total = 0;

    port = atoi (argv[optind]);
    str_to_key (argv[optind + 1], &keyinput);
    state = chimera_init (port);

    if (hn != NULL)
	{
	    host = host_get (state, hn, joinport);
	}
    chimera_forward (state, test_fwd);
    chimera_deliver (state, test_del);
    chimera_update (state, test_update);
    chimera_setkey (state, keyinput);
    chimera_register (state, TEST_CHAT, 1);

    log_direct (state->log, LOG_WARN, stderr);

    chimera_join (state, host);

    fprintf (stderr,
	     "** send messages to key with command <key> <message> **\n");


    while (1)
	{

	}


    /*  while(fgets(tmp, 256, stdin) != NULL) {
       if(strlen(tmp) > 2) {
       for(i = 0; tmp[i] != '\n'; i++);
       tmp[i] = 0;
       for(i = 0; tmp[i] != ' ' && i < strlen(tmp); i++);
       tmp[i] = 0;
       i++;
       str_to_key(tmp,&key);
       fprintf(stderr,"sending key:%s data:%s len:%d\n", key.keystr,tmp+i, strlen(tmp+i));
       chimera_send(state,key, TEST_CHAT, strlen(tmp+i) + 1, tmp+i);
       }
       } */

}