/**

 * Progress bar to give updates for operations in gfs2-utils.

 * Adapted from the simple progress bar in e2fsprogs progress.c

 *

 */

#include <stdio.h>

#include <string.h>

#include <inttypes.h>

#include <unistd.h>

#include <time.h>

#include "progress.h"

static char spaces[44], backspaces[44];

static time_t last_update;

static int number_of_digits(int value)

{

	int digits = 0;

	do {

		value /= 10;

		digits++;

	} while (value != 0);

	return digits;

}

void gfs2_progress_init(struct gfs2_progress_bar *progress, uint64_t max, const char *message, int quiet)

{

	/**

	 * NOTE:

	 *

	 * Default operation is to output the progress indication

	 * in full. Although we will honor the quiet flag in the

	 * application, if this is set we skip progress bar any

	 * update operations and output.

	 *

	 */

	memset(spaces, ' ', sizeof(spaces)-1);

	spaces[sizeof(spaces)-1] = 0;

	memset(backspaces, '\b', sizeof(backspaces)-1);

	backspaces[sizeof(backspaces)-1] = 0;

	memset(progress, 0, sizeof(*progress));

	if (quiet) {

		progress->skip_progress++;

		return;

	}

	progress->max = max;

	progress->max_digits = number_of_digits(max);

	if (message) {

		fputs(message, stdout);

		fflush(stdout);

	}

	last_update = 0;

}

void gfs2_progress_update(struct gfs2_progress_bar *progress, uint64_t value)

{

	time_t current_time;

	if (progress->skip_progress || (!isatty(STDOUT_FILENO)))

		return;

	current_time = time(0);

	if (current_time == last_update)

		return;

	last_update = current_time;

	printf("[%*"PRIu64"/%*"PRIu64"]", progress->max_digits, value,

		progress->max_digits, progress->max);

	fflush(stdout);

	fprintf(stdout, "%.*s", (2 * progress->max_digits) + 3, backspaces);

}

void gfs2_progress_close(struct gfs2_progress_bar *progress, const char *message)

{

	if (progress->skip_progress)

		return;

	if (isatty(STDOUT_FILENO)) {

		fprintf(stdout, "%.*s", (2 * progress->max_digits) + 3, spaces);

		fprintf(stdout, "%.*s", (2 * progress->max_digits) + 3, backspaces);

	}

	if (message)

		fputs(message, stdout);

}