message.c 2.16 KB
Newer Older
1
/*
Edward Thomson committed
2
 * Copyright (C) the libgit2 contributors. All rights reserved.
3 4 5 6 7 8 9 10 11 12 13
 *
 * This file is part of libgit2, distributed under the GNU GPL v2 with
 * a Linking Exception. For full terms see the included COPYING file.
 */

#include "message.h"

static size_t line_length_without_trailing_spaces(const char *line, size_t len)
{
	while (len) {
		unsigned char c = line[len - 1];
14
		if (!git__isspace(c))
15 16 17 18 19 20 21 22 23
			break;
		len--;
	}

	return len;
}

/* Greatly inspired from git.git "stripspace" */
/* see https://github.com/git/git/blob/497215d8811ac7b8955693ceaad0899ecd894ed2/builtin/stripspace.c#L4-67 */
24
int git_message__prettify(git_buf *message_out, const char *message, int strip_comments)
25
{
Vicent Martí committed
26 27
	const size_t message_len = strlen(message);

28 29 30 31 32
	int consecutive_empty_lines = 0;
	size_t i, line_length, rtrimmed_line_length;
	char *next_newline;

	for (i = 0; i < strlen(message); i += line_length) {
Vicent Martí committed
33 34 35 36 37 38 39
		next_newline = memchr(message + i, '\n', message_len - i);

		if (next_newline != NULL) {
			line_length = next_newline - (message + i) + 1;
		} else {
			line_length = message_len - i;
		}
40 41 42 43 44 45 46 47 48 49 50 51

		if (strip_comments && line_length && message[i] == '#')
			continue;

		rtrimmed_line_length = line_length_without_trailing_spaces(message + i, line_length);

		if (!rtrimmed_line_length) {
			consecutive_empty_lines++;
			continue;
		}

		if (consecutive_empty_lines > 0 && message_out->size > 0)
52
			git_buf_putc(message_out, '\n');
53 54 55 56 57 58

		consecutive_empty_lines = 0;
		git_buf_put(message_out, message + i, rtrimmed_line_length);
		git_buf_putc(message_out, '\n');
	}

59
	return git_buf_oom(message_out) ? -1 : 0;
60
}
61 62 63 64

int git_message_prettify(char *message_out, size_t buffer_size, const char *message, int strip_comments)
{
	git_buf buf = GIT_BUF_INIT;
65
	ssize_t out_size = -1;
66

67 68
	if (message_out && buffer_size)
		*message_out = '\0';
69

70 71
	if (git_message__prettify(&buf, message, strip_comments) < 0)
		goto done;
72

73
	if (message_out && buf.size + 1 > buffer_size) { /* +1 for NUL byte */
74
		giterr_set(GITERR_INVALID, "Buffer too short to hold the cleaned message");
75
		goto done;
76 77
	}

78 79
	if (message_out)
		git_buf_copy_cstr(message_out, buffer_size, &buf);
80

81 82 83 84
	out_size = buf.size + 1;

done:
	git_buf_free(&buf);
85
	return (int)out_size;
86
}