#include "module.h"
#include "signals.h"
#include "misc.h"
#include "settings.h"

#include "silc-servers.h"

static int timeout_tag;
static SilcBool lag_event_pong(SilcClient client,
			       SilcClientConnection conn,
			       SilcCommand command,
			       SilcStatus status,
			       SilcStatus error,
			       void *context,
			       va_list ap);

static void lag_get(SILC_SERVER_REC *server)
{
	SilcBuffer idp;
	g_get_current_time(&server->lag_sent);
	server->lag_last_check = time(NULL);

	/* Send PING */
	idp = silc_id_payload_encode(&server->conn->remote_id.u.server_id,
				     SILC_ID_SERVER);
	silc_client_command_send(silc_client, server->conn,
				 SILC_COMMAND_PING, lag_event_pong, server,
				 1, 1, silc_buffer_data(idp),
				 silc_buffer_len(idp));
	silc_buffer_free(idp);
}

static SilcBool lag_event_pong(SilcClient client,
			       SilcClientConnection conn,
			       SilcCommand command,
			       SilcStatus status,
			       SilcStatus error,
			       void *context,
			       va_list ap)
{
	SILC_SERVER_REC *server = context;
	GTimeVal now;

	if (status != SILC_STATUS_OK) {
		/* if the ping failed for some reason, try it again */
		lag_get(server);
		return TRUE;
	}

	if (server->lag_sent.tv_sec == 0) {
		/* not expecting lag reply. */
		return TRUE;
	}

	g_get_current_time(&now);
	server->lag = (int) get_timeval_diff(&now, &server->lag_sent);
	memset(&server->lag_sent, 0, sizeof(server->lag_sent));

	signal_emit("server lag", 1, server);

	return TRUE;
}

static int sig_check_lag(void)
{
	GSList *tmp, *next;
	time_t now;
	int lag_check_time, max_lag;

	lag_check_time = settings_get_time("lag_check_time") / 1000;
	max_lag = settings_get_time("lag_max_before_disconnect") / 1000;

	if (lag_check_time <= 0)
		return 1;

	now = time(NULL);
	for (tmp = servers; tmp != NULL; tmp = next) {
		SILC_SERVER_REC *rec = tmp->data;

		next = tmp->next;
		if (!IS_SILC_SERVER(rec))
			continue;

		if (rec->lag_sent.tv_sec != 0) {
			/* waiting for lag reply */
			if (max_lag > 1 && now-rec->lag_sent.tv_sec > max_lag) {
				/* too much lag, disconnect */
				signal_emit("server lag disconnect", 1, rec);
				rec->connection_lost = TRUE;
				server_disconnect((SERVER_REC *) rec);
			}
		} else if (rec->lag_last_check+lag_check_time < now &&
			 rec->connected) {
			/* no commands in buffer - get the lag */
			lag_get(rec);
		}
	}

	return 1;
}

void silc_lag_init(void)
{
	/* silc-client will need those... silc-plugin uses irc defaults */
	settings_add_time("misc", "lag_check_time", "1min");
	settings_add_time("misc", "lag_max_before_disconnect", "5min");

	timeout_tag = g_timeout_add(1000, (GSourceFunc) sig_check_lag, NULL);
}

void silc_lag_deinit(void)
{
	g_source_remove(timeout_tag);
}