/*
* Copyright (C) 2009 Mamadou Diop.
*
* Contact: Mamadou Diop <diopmamadou(at)doubango.org>
*	
* This file is part of Open Source Doubango Framework.
*
* DOUBANGO is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*	
* DOUBANGO is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
* GNU General Public License for more details.
*	
* You should have received a copy of the GNU General Public License
* along with DOUBANGO.
*
*/

/**@file tmsrp_sender.c
 * @brief MSRP sender.
 *
 * @author Mamadou Diop <diopmamadou(at)doubango.org>
 *

 */
#include "tinymsrp/session/tmsrp_sender.h"

#include "tnet_utils.h"

#include "tsk_thread.h"
#include "tsk_memory.h"
#include "tsk_string.h"
#include "tsk_time.h"
#include "tsk_debug.h"


static void* TSK_STDCALL run(void* self);


tmsrp_sender_t* tmsrp_sender_create(tmsrp_config_t* config, tnet_fd_t fd)
{
	return (tmsrp_sender_t*)tsk_object_new(tmsrp_sender_def_t, config, fd);
}

int tmsrp_sender_set_fd(tmsrp_sender_t* self, tnet_fd_t fd)
{
	if(!self){
		TSK_DEBUG_ERROR("Invalid parameter");
		return -1;
	}
	self->fd = fd;
	return 0;
}

int tmsrp_sender_start(tmsrp_sender_t* self)
{
	int ret = -1;
	
	if(!self){
		TSK_DEBUG_ERROR("Invalid parameter");
		goto bail;
	}
	
	TSK_RUNNABLE(self)->run = run;
	if((ret = tsk_runnable_start(TSK_RUNNABLE(self), tmsrp_data_out_def_t))){
		goto bail;
	}
	
bail:
	return ret;
}

int tsmrp_sender_send_data(tmsrp_sender_t* self, const void* pdata, tsk_size_t size, const char* ctype, const char* wctype)
{
	tmsrp_data_out_t* data_out;

	if(!self){
		TSK_DEBUG_ERROR("Invalid parameter");
		return -1;
	}

	if((data_out = tmsrp_data_out_create(pdata, size))){
		if(ctype){
			tsk_strupdate(&TMSRP_DATA(data_out)->ctype, ctype);
		}
		if(wctype){
			tsk_strupdate(&TMSRP_DATA(data_out)->wctype, wctype);
		}
		TSK_RUNNABLE_ENQUEUE_OBJECT(self, data_out);
		return 0;
	}
	return -2;
}

int tsmrp_sender_send_file(tmsrp_sender_t* self, const char* filepath)
{
	tmsrp_data_out_t* data_out;

	if(!self || !filepath){
		TSK_DEBUG_ERROR("Invalid parameter");
		return -1;
	}

	if((data_out = tmsrp_data_out_file_create(filepath))){
		if(TMSRP_DATA(data_out)->isOK){
			TSK_RUNNABLE_ENQUEUE_OBJECT(self, data_out);
			return 0;
		}
		else{
			TSK_OBJECT_SAFE_FREE(data_out);
			return -3;
		}
	}
	return -2;
}

int tmsrp_sender_stop(tmsrp_sender_t* self)
{
	int ret = -1;

	if(!self){
		TSK_DEBUG_ERROR("Invalid parameter");
		goto bail;
	}

	if((ret = tsk_runnable_stop(TSK_RUNNABLE(self)))){
		goto bail;
	}

bail:
	return ret;
}




static void* TSK_STDCALL run(void* self)
{
	tsk_list_item_t *curr;
	tmsrp_sender_t *sender = (tmsrp_sender_t*)self;
	tmsrp_data_out_t *data_out;
	tsk_buffer_t* chunck, *message = tsk_buffer_create_null();
	tsk_size_t start;
	tsk_size_t end;
	tsk_size_t total;
	tsk_istr_t tid;
	int64_t __now = (int64_t)tsk_time_now();
	tsk_bool_t error = tsk_false;

	TSK_DEBUG_INFO("MSRP SENDER::run -- START");

	TSK_RUNNABLE_RUN_BEGIN(sender);

	if((curr = TSK_RUNNABLE_POP_FIRST(sender))){
		if(!(data_out = (tmsrp_data_out_t*)curr->data)){
			continue;
		}
		
		error = tsk_false;
		start = 1;
		total = data_out->size;
		
		while(TSK_RUNNABLE(self)->running && !error && (chunck = tmsrp_data_out_get(data_out))){
			tmsrp_request_t* SEND;
			// set end
			end = (start + chunck->size) - 1;
			// compute new transaction id
			tsk_itoa(++__now, &tid);
			// create SEND request
			SEND = tmsrp_request_create(tid, "SEND");
			// T-Path and From-Path (because of otherURIs)
			SEND->To = tsk_object_ref(sender->config->To_Path);
			SEND->From = tsk_object_ref(sender->config->From_Path);
			// add other headers
			tmsrp_message_add_headers(SEND,
				TMSRP_HEADER_MESSAGE_ID_VA_ARGS(TMSRP_DATA(data_out)->id),
				// TMSRP_HEADER_BYTE_RANGE_VA_ARGS(start, end, total), => See below
				TMSRP_HEADER_FAILURE_REPORT_VA_ARGS(sender->config->Failure_Report ? freport_yes : freport_no),
				TMSRP_HEADER_SUCCESS_REPORT_VA_ARGS(sender->config->Success_Report),

				tsk_null);
			// add data
			if(start == 1 && chunck->size && tsk_striequals(TMSRP_DATA(data_out)->ctype, "message/CPIM")){
				tsk_buffer_t* content_cpim = tsk_buffer_create_null();
				if(content_cpim){
					tsk_buffer_append_2(content_cpim, "Subject: %s\r\n\r\nContent-Type: %s\r\n\r\n",
						"test", TMSRP_DATA(data_out)->wctype);
					end += content_cpim->size;
					total += content_cpim->size;
					tsk_buffer_append(content_cpim, chunck->data, chunck->size);
					tmsrp_message_add_content(SEND, TMSRP_DATA(data_out)->ctype, content_cpim->data, content_cpim->size);
					TSK_OBJECT_SAFE_FREE(content_cpim);
				}
				else{
					TSK_DEBUG_ERROR("Failed to allocate new buffer");
				}
			}
			else{
				tmsrp_message_add_content(SEND, TMSRP_DATA(data_out)->ctype, chunck->data, chunck->size);
			}
			// add byte range here not before: think about message/cpim
			tmsrp_message_add_headers(SEND,
				TMSRP_HEADER_BYTE_RANGE_VA_ARGS(start, end, total),

				tsk_null);


			// set continuation flag
			SEND->end_line.cflag = (end == total) ? '$' : '+';
			// serialize and send
			if(!(tmsrp_message_serialize(SEND, message))){
				if(tnet_sockfd_send(sender->fd, message->data, message->size, 0) == 0){
					error = tsk_true;
					// abort
				}
			}
			tsk_buffer_cleanup(message);
			
			// set start
			start = (end + 1);
			// cleanup
			TSK_OBJECT_SAFE_FREE(chunck);
			TSK_OBJECT_SAFE_FREE(SEND);

			/* wait */
			if(sender->chunck_duration){
				tsk_thread_sleep(sender->chunck_duration);
			}
		}
		

		tsk_object_unref(curr);
	}

	TSK_RUNNABLE_RUN_END(self);

	TSK_OBJECT_SAFE_FREE(message);

	TSK_DEBUG_INFO("MSRP SENDER::run -- STOP");

	return 0;
}



//=================================================================================================
//	MSRP sender object definition
//
static void* tmsrp_sender_ctor(tsk_object_t * self, va_list *app)
{
	tmsrp_sender_t *sender = (tmsrp_sender_t*)self;
	if(sender){
		sender->config = (tmsrp_config_t*)tsk_object_ref(va_arg(*app, tmsrp_config_t*));
		sender->fd = va_arg(*app, tnet_fd_t);	

		sender->outgoingList = tsk_list_create();
	}
	return self;
}

static void* tmsrp_sender_dtor(tsk_object_t * self)
{ 
	tmsrp_sender_t *sender = self;
	if(sender){
		/* Stop */
		tmsrp_sender_stop(sender);

		TSK_OBJECT_SAFE_FREE(sender->config);
		TSK_OBJECT_SAFE_FREE(sender->outgoingList);
		// the FD is owned by the transport ...do not close it
	}
	return self;
}

static const tsk_object_def_t tmsrp_sender_def_s = 
{
	sizeof(tmsrp_sender_t),
	tmsrp_sender_ctor,
	tmsrp_sender_dtor,
	tsk_null, 
};
const tsk_object_def_t *tmsrp_sender_def_t = &tmsrp_sender_def_s;
