#include "git-compat-util.h"

#include "protocol-caps.h"

#include "gettext.h"

#include "hex.h"

#include "pkt-line.h"

#include "hash.h"

#include "hex.h"

#include "object.h"

#include "object-store-ll.h"

#include "repository.h"

#include "string-list.h"

#include "strbuf.h"

struct requested_info {

  unsigned size : 1;

};

/*

 * Parses oids from the given line and collects them in the given

 * oid_str_list. Returns 1 if parsing was successful and 0 otherwise.

 */

static int parse_oid(const char *line, struct string_list *oid_str_list)

{

  const char *arg;

  if (!skip_prefix(line, "oid ", &arg))

    return 0;

  string_list_append(oid_str_list, arg);

  return 1;

}

/*

 * Validates and send requested info back to the client. Any errors detected

 * are returned as they are detected.

 */

static void send_info(struct repository *r, struct packet_writer *writer,

          struct string_list *oid_str_list,

          struct requested_info *info)

{

  struct string_list_item *item;

  struct strbuf send_buffer = STRBUF_INIT;

  if (!oid_str_list->nr)

    return;

  if (info->size)

    packet_writer_write(writer, "size");

  for_each_string_list_item (item, oid_str_list) {

    const char *oid_str = item->string;

    struct object_id oid;

    unsigned long object_size;

    if (get_oid_hex_algop(oid_str, &oid, r->hash_algo) < 0) {

      packet_writer_error(

        writer,

        "object-info: protocol error, expected to get oid, not '%s'",

        oid_str);

      continue;

    }

    strbuf_addstr(&send_buffer, oid_str);

    if (info->size) {

      if (oid_object_info(r, &oid, &object_size) < 0) {

        strbuf_addstr(&send_buffer, " ");

      } else {

        strbuf_addf(&send_buffer, " %lu", object_size);

      }

    }

    packet_writer_write(writer, "%s", send_buffer.buf);

    strbuf_reset(&send_buffer);

  }

  strbuf_release(&send_buffer);

}

int cap_object_info(struct repository *r, struct packet_reader *request)

{

  struct requested_info info = { 0 };

  struct packet_writer writer;

  struct string_list oid_str_list = STRING_LIST_INIT_DUP;

  packet_writer_init(&writer, 1);

  while (packet_reader_read(request) == PACKET_READ_NORMAL) {

    if (!strcmp("size", request->line)) {

      info.size = 1;

      continue;

    }

    if (parse_oid(request->line, &oid_str_list))

      continue;

    packet_writer_error(&writer,

            "object-info: unexpected line: '%s'",

            request->line);

  }

  if (request->status != PACKET_READ_FLUSH) {

    packet_writer_error(

      &writer, "object-info: expected flush after arguments");

    die(_("object-info: expected flush after arguments"));

  }

  send_info(r, &writer, &oid_str_list, &info);

  string_list_clear(&oid_str_list, 1);

  packet_flush(1);

  return 0;

}