#include <git2.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include "common.h"

// This could be run in the main loop whilst the application waits for
// the indexing to finish in a worker thread
int index_cb(const git_indexer_stats *stats, void *data)
{
  printf("\rProcessing %d of %d", stats->processed, stats->total);
}

int index_pack(git_repository *repo, int argc, char **argv)
{
  git_indexer_stream *idx;
  git_indexer_stats stats = {0, 0};
  int error, fd;
  char hash[GIT_OID_HEXSZ + 1] = {0};
  ssize_t read_bytes;
  char buf[512];

  if (argc < 2) {
	  fprintf(stderr, "I need a packfile\n");
	  return EXIT_FAILURE;
  }

  if (git_indexer_stream_new(&idx, ".git") < 0) {
	  puts("bad idx");
	  return -1;
  }

  if ((fd = open(argv[1], 0)) < 0) {
	  perror("open");
	  return -1;
  }

  do {
	  read_bytes = read(fd, buf, sizeof(buf));
	  if (read_bytes < 0)
		  break;

	  if ((error = git_indexer_stream_add(idx, buf, read_bytes, &stats)) < 0)
		  goto cleanup;

	  printf("\rIndexing %d of %d", stats.processed, stats.total);
  } while (read_bytes > 0);

  if (read_bytes < 0) {
	  error = -1;
	  perror("failed reading");
	  goto cleanup;
  }

  if ((error = git_indexer_stream_finalize(idx, &stats)) < 0)
	  goto cleanup;

  printf("\rIndexing %d of %d\n", stats.processed, stats.total);

  git_oid_fmt(hash, git_indexer_stream_hash(idx));
  puts(hash);

cleanup:
  close(fd);
  git_indexer_stream_free(idx);
  return error;
}

int index_pack_old(git_repository *repo, int argc, char **argv)
{
  git_indexer *indexer;
  git_indexer_stats stats;
  int error;
  char hash[GIT_OID_HEXSZ + 1] = {0};

  if (argc < 2) {
    fprintf(stderr, "I need a packfile\n");
    return EXIT_FAILURE;
  }

  // Create a new indexer
  error = git_indexer_new(&indexer, argv[1]);
  if (error < GIT_SUCCESS)
    return error;

  // Index the packfile. This function can take a very long time and
  // should be run in a worker thread.
  error = git_indexer_run(indexer, &stats);
  if (error < GIT_SUCCESS)
    return error;

  // Write the information out to an index file
  error = git_indexer_write(indexer);

  // Get the packfile's hash (which should become it's filename)
  git_oid_fmt(hash, git_indexer_hash(indexer));
  puts(hash);

  git_indexer_free(indexer);

  return GIT_SUCCESS;
}
