/*
 * ***** BEGIN LICENSE BLOCK *****
 * Version: MPL 1.1/GPL 2.0
 *
 * The contents of this file are subject to the Mozilla Public License
 * Version 1.1 (the "License"); you may not use this file except in
 * compliance with the License. You may obtain a copy of the License
 * at http://www.mozilla.org/MPL/
 *
 * Software distributed under the License is distributed on an "AS IS"
 * basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See
 * the License for the specific language governing rights and
 * limitations under the License.
 *
 * The Original Code is librabbitmq.
 *
 * The Initial Developer of the Original Code is VMware, Inc.
 * Portions created by VMware are Copyright (c) 2007-2011 VMware, Inc.
 *
 * Portions created by Tony Garnock-Jones are Copyright (c) 2009-2010
 * VMware, Inc. and Tony Garnock-Jones.
 *
 * All rights reserved.
 *
 * Alternatively, the contents of this file may be used under the terms
 * of the GNU General Public License Version 2 or later (the "GPL"), in
 * which case the provisions of the GPL are applicable instead of those
 * above. If you wish to allow use of your version of this file only
 * under the terms of the GPL, and not to allow others to use your
 * version of this file under the terms of the MPL, indicate your
 * decision by deleting the provisions above and replace them with the
 * notice and other provisions required by the GPL. If you do not
 * delete the provisions above, a recipient may use your version of
 * this file under the terms of any one of the MPL or the GPL.
 *
 * ***** END LICENSE BLOCK *****
 */

#include "config.h"

#include <stdio.h>
#include <stdlib.h>

#include "common.h"
#include "process.h"

/* Convert a amqp_bytes_t to an escaped string form for printing.  We
   use the same escaping conventions as rabbitmqctl. */
static char *stringify_bytes(amqp_bytes_t bytes)
{
	/* We will need up to 4 chars per byte, plus the terminating 0 */
	char *res = malloc(bytes.len * 4 + 1);
	uint8_t *data = bytes.bytes;
	char *p = res;
	size_t i;
	
	for (i = 0; i < bytes.len; i++) {
		if (data[i] >= 32 && data[i] != 127) {
			*p++ = data[i];
		}
		else {
			*p++ = '\\';
			*p++ = '0' + (data[i] >> 6);
			*p++ = '0' + (data[i] >> 3 & 0x7); 
			*p++ = '0' + (data[i] & 0x7);
		}
	}

	*p = 0;
	return res;
}

static amqp_bytes_t setup_queue(amqp_connection_state_t conn,
				char *queue, char *exchange,
				char *routing_key, int declare)
{
	amqp_bytes_t queue_bytes = cstring_bytes(queue);

	/* if an exchange name wasn't provided, check that we don't
	   have options that require it. */
	if (!exchange && routing_key) {
		fprintf(stderr,	"--routing-key option requires an exchange"
			" name to be provided with --exchange\n");
		exit(1);
	}

	if (!queue || exchange || declare) {
		/* Declare the queue as auto-delete.  */
		amqp_queue_declare_ok_t *res = amqp_queue_declare(conn, 1,
						    queue_bytes, 0, 0, 1, 1,
						    amqp_empty_table);
		if (!res)
			die_rpc(amqp_get_rpc_reply(conn), "queue.declare");

		if (!queue) {
			/* the server should have provided a queue name */
			char *sq;
			queue_bytes = amqp_bytes_malloc_dup(res->queue);
			sq = stringify_bytes(queue_bytes);
			fprintf(stderr, "Server provided queue name: %s\n",
				sq);
			free(sq);
		}

		/* Bind to an exchange if requested */
		if (exchange) {
			amqp_bytes_t eb = amqp_cstring_bytes(exchange);
			if (!amqp_queue_bind(conn, 1, queue_bytes, eb,
					     cstring_bytes(routing_key),
					     amqp_empty_table))
				die_rpc(amqp_get_rpc_reply(conn),
					"queue.bind");
		}
	}
	
	return queue_bytes;
}

static void do_consume(amqp_connection_state_t conn, amqp_bytes_t queue,
		       int no_ack, const char * const *argv)
{
	if (!amqp_basic_consume(conn, 1, queue, amqp_empty_bytes, 0, no_ack,
				0, amqp_empty_table))
		die_rpc(amqp_get_rpc_reply(conn), "basic.consume");

	for (;;) {
		amqp_frame_t frame;
		struct pipeline pl;
		uint64_t delivery_tag;
		int res = amqp_simple_wait_frame(conn, &frame);
		die_amqp_error(res, "waiting for header frame");

		if (frame.frame_type != AMQP_FRAME_METHOD
		    || frame.payload.method.id != AMQP_BASIC_DELIVER_METHOD)
			continue;

		amqp_basic_deliver_t *deliver
			= (amqp_basic_deliver_t *)frame.payload.method.decoded;
		delivery_tag = deliver->delivery_tag;
		
		pipeline(argv, &pl);
		copy_body(conn, pl.infd);

		if (finish_pipeline(&pl) && !no_ack)
			die_amqp_error(amqp_basic_ack(conn, 1, delivery_tag,
						      0),
				       "basic.ack");

		amqp_maybe_release_buffers(conn);
	}
}

int main(int argc, const char **argv)
{
	poptContext opts;
	amqp_connection_state_t conn;
	const char * const *cmd_argv;
	char *queue = NULL;
	char *exchange = NULL;
	char *routing_key = NULL;
	int declare = 0;
	int no_ack = 0;
	amqp_bytes_t queue_bytes;

	struct poptOption options[] = {
		INCLUDE_OPTIONS(connect_options),
		{"queue", 'q', POPT_ARG_STRING, &queue, 0,
		 "the queue to consume from", "queue"},
		{"exchange", 'e', POPT_ARG_STRING, &exchange, 0,
		 "bind the queue to this exchange", "exchange"},
		{"routing-key", 'r', POPT_ARG_STRING, &routing_key, 0,
		 "the routing key to bind with", "routing key"},
		{"declare", 'd', POPT_ARG_NONE, &declare, 0,
		 "declare an exclusive queue", NULL},
		{"no-ack", 'A', POPT_ARG_NONE, &no_ack, 0,
		 "consume in no-ack mode", NULL},
		POPT_AUTOHELP
		{ NULL, 0, 0, NULL, 0 }
	};
	
	opts = process_options(argc, argv, options,
			       "[OPTIONS]... <command> <args>");
	
	cmd_argv = poptGetArgs(opts);
	if (!cmd_argv || !cmd_argv[0]) {
		fprintf(stderr, "consuming command not specified\n");
		poptPrintUsage(opts, stderr, 0);
		goto error;
	}

	conn = make_connection();
	queue_bytes = setup_queue(conn, queue, exchange, routing_key, declare);
	do_consume(conn, queue_bytes, no_ack, cmd_argv);
	close_connection(conn);
	return 0;

error:
	poptFreeContext(opts);
	return 1;
}