#include "tmate.h"
#include <errno.h>
#include <libssh/server.h>

void tmate_dump_exec_response(struct tmate_session *session,
			      int exit_code, const char *message)
{
	struct tmate_ssh_client *client = &session->ssh_client;

	ssh_channel_write(client->channel, message, strlen(message));
	ssh_channel_request_send_exit_status(client->channel, exit_code);

	ssh_channel_send_eof(client->channel);
	ssh_channel_close(client->channel);

	if (event_base_loopexit(session->ev_base, NULL) < 0)
		tmate_fatal("cannot stop event loop");
}

static void on_proxy_error(struct tmate_session *session, __unused short events)
{
	tmate_warn("Lost proxy connection");
	tmate_dump_exec_response(session, 1, "Internal Error\r\n");
}

void tmate_client_exec_init(struct tmate_session *session)
{
	struct tmate_ssh_client *client = &session->ssh_client;

	memset(&client->channel_cb, 0, sizeof(client->channel_cb));
	ssh_callbacks_init(&client->channel_cb);
	ssh_set_channel_callbacks(client->channel, &client->channel_cb);

	tmate_init_proxy(session, on_proxy_error);

	tmate_proxy_exec(session, client->exec_command);
}