#include "git-compat-util.h"
#include "gettext.h"
#include "hex.h"
#include "pkt-line.h"
#include "connect.h"
#include "oid-array.h"
#include "object-store-ll.h"
#include "fetch-object-info.h"
#include "string-list.h"

/**
 * send_object_info_request sends git-cat-file object-info command and its
 * arguments into the request buffer.
 */
static void send_object_info_request(const int fd_out, struct object_info_args *args)
{
	struct strbuf req_buf = STRBUF_INIT;

	write_command_and_capabilities(&req_buf, "object-info", args->server_options);

	if (unsorted_string_list_has_string(args->object_info_options, "size"))
		packet_buf_write(&req_buf, "size");

	if (args->oids) {
		for (size_t i = 0; i < args->oids->nr; i++)
			packet_buf_write(&req_buf, "oid %s", oid_to_hex(&args->oids->oid[i]));
	}

	packet_buf_flush(&req_buf);
	if (write_in_full(fd_out, req_buf.buf, req_buf.len) < 0)
		die_errno(_("unable to write request to remote"));

	strbuf_release(&req_buf);
}

/**
 * fetch_object_info sends git-cat-file object-info command into the request buf
 * and read the results from packets.
 */
int fetch_object_info(const enum protocol_version version, struct object_info_args *args,
					  struct packet_reader *reader, struct object_info *object_info_data,
					  const int stateless_rpc, const int fd_out)
{
	int size_index = -1;

	switch (version) {
	case protocol_v2:
		if (!server_supports_v2("object-info"))
			return -1;
		if (unsorted_string_list_has_string(args->object_info_options, "size")
				&& !server_supports_feature("object-info", "size", 0))
			return -1;
		send_object_info_request(fd_out, args);
		break;
	case protocol_v1:
	case protocol_v0:
		die(_("wrong protocol version. expected v2"));
	case protocol_unknown_version:
		BUG("unknown protocol version");
	}

	for (size_t i = 0; i < args->object_info_options->nr; i++) {
		if (packet_reader_read(reader) != PACKET_READ_NORMAL) {
			check_stateless_delimiter(stateless_rpc, reader, "stateless delimiter expected");
			return -1;
		}
		if (unsorted_string_list_has_string(args->object_info_options, reader->line)) {
			if (!strcmp(reader->line, "size")) {
				size_index = i;
				for (size_t j = 0; j < args->oids->nr; j++)
					object_info_data[j].sizep = xcalloc(1, sizeof(long));
			}
			continue;
		}
		return -1;
	}

	for (size_t i = 0; packet_reader_read(reader) == PACKET_READ_NORMAL && i < args->oids->nr; i++){
		struct string_list object_info_values = STRING_LIST_INIT_DUP;

		string_list_split(&object_info_values, reader->line, ' ', -1);
		if (0 <= size_index) {
			if (!strcmp(object_info_values.items[1 + size_index].string, ""))
				die("object-info: not our ref %s",
					object_info_values.items[0].string);

			*object_info_data[i].sizep = strtoul(object_info_values.items[1 + size_index].string, NULL, 10);
		}

		string_list_clear(&object_info_values, 0);
	}
	check_stateless_delimiter(stateless_rpc, reader, "stateless delimiter expected");

	return 0;
}