/*
    Copyright (C) 2005-2009  Michel de Boer <michel@twinklephone.com>

    This program 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 2 of the License, or
    (at your option) any later version.

    This program 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 this program.  If not, see <https://www.gnu.org/licenses/>.
*/

#ifndef _TRANSACTION_MGR_H
#define _TRANSACTION_MGR_H

#include <map>
#include "events.h"
#include "transaction.h"
#include "user.h"
#include "parser/request.h"
#include "parser/response.h"
#include "sockets/socket.h"
#include "stun/stun_transaction.h"

using namespace std;

class t_transaction_mgr {
private:
	// Mapping from transaction id to transaction
	map<t_tid, t_trans_client *>		map_trans_client;
	map<t_tid, t_trans_server *>		map_trans_server;
	map<t_tid, t_stun_transaction *>	map_stun_trans;

	// Find existing transactions. Return NULL if not found
	t_trans_client *find_trans_client(t_response *r) const;
	t_trans_client *find_trans_client(t_tid tid) const;
	t_trans_client *find_trans_client(const string &branch, const t_method &cseq_method) const;
	t_trans_client *find_trans_client(const t_icmp_msg &icmp) const;
	t_trans_server *find_trans_server(t_request *r) const;
	t_trans_server *find_trans_server(t_tid tid) const;
	t_stun_transaction *find_stun_trans(StunMessage *r) const;
	t_stun_transaction *find_stun_trans(t_tid tid) const;
	t_stun_transaction *find_stun_trans(const t_icmp_msg &icmp) const;

	// Create new transactions.
	// Return NULL if creation failed.
	t_tc_invite *create_tc_invite(t_user *user_config, t_request *r, unsigned short tuid);
	t_tc_non_invite *create_tc_non_invite(t_user *user_config, t_request *r,
		unsigned short tuid);
	t_ts_invite *create_ts_invite(t_request *r);
	t_ts_non_invite *create_ts_non_invite(t_request *r);
	t_sip_stun_trans *create_sip_stun_trans(t_user *user_config, StunMessage *r, 
		unsigned short tuid);
	t_media_stun_trans *create_media_stun_trans(t_user *user_config, StunMessage *r, 
		unsigned short tuid, unsigned short src_port);

	// Delete transactions
	void delete_trans_client(t_trans_client *tc);
	void delete_trans_server(t_trans_server *ts);
	void delete_stun_trans(t_stun_transaction *st);

	// Handle events
	void handle_event_network(t_event_network *e);
	void handle_event_user(t_event_user *e);
	void handle_event_timeout(t_event_timeout *e);
	void handle_event_abort(t_event_abort_trans *e);
	void handle_event_stun_request(t_event_stun_request *e);
	void handle_event_stun_response(t_event_stun_response *e);
	void handle_event_icmp(t_event_icmp *e);
	void handle_event_failure(t_event_failure *e);

public:
	~t_transaction_mgr();

	// Find the target transaction for a CANCEL.
	// Return NULL if not found.
	t_trans_server *find_cancel_target(t_request *r) const;

	// Start transaction timer. Return timer id (needed for stopping)
	t_object_id start_timer(long dur, t_sip_timer tmr,
						unsigned short tid);
	t_object_id start_stun_timer(long dur, t_stun_timer tmr,
						unsigned short tid);

	// Stop timer. Pass id that is returned by start_timer
	void stop_timer(t_object_id id);

	// Main loop of the transaction manager (infinite)
	void run (void);
};

// Thread that runs the transaction manager
void *transaction_mgr_main(void *arg);

#endif