Merge branch 'po/remote-set-head-usage'
[git] / builtin / stripspace.c
1 #include "builtin.h"
2 #include "cache.h"
3
4 /*
5  * Returns the length of a line, without trailing spaces.
6  *
7  * If the line ends with newline, it will be removed too.
8  */
9 static size_t cleanup(char *line, size_t len)
10 {
11         while (len) {
12                 unsigned char c = line[len - 1];
13                 if (!isspace(c))
14                         break;
15                 len--;
16         }
17
18         return len;
19 }
20
21 /*
22  * Remove empty lines from the beginning and end
23  * and also trailing spaces from every line.
24  *
25  * Turn multiple consecutive empty lines between paragraphs
26  * into just one empty line.
27  *
28  * If the input has only empty lines and spaces,
29  * no output will be produced.
30  *
31  * If last line does not have a newline at the end, one is added.
32  *
33  * Enable skip_comments to skip every line starting with comment
34  * character.
35  */
36 void stripspace(struct strbuf *sb, int skip_comments)
37 {
38         int empties = 0;
39         size_t i, j, len, newlen;
40         char *eol;
41
42         /* We may have to add a newline. */
43         strbuf_grow(sb, 1);
44
45         for (i = j = 0; i < sb->len; i += len, j += newlen) {
46                 eol = memchr(sb->buf + i, '\n', sb->len - i);
47                 len = eol ? eol - (sb->buf + i) + 1 : sb->len - i;
48
49                 if (skip_comments && len && sb->buf[i] == comment_line_char) {
50                         newlen = 0;
51                         continue;
52                 }
53                 newlen = cleanup(sb->buf + i, len);
54
55                 /* Not just an empty line? */
56                 if (newlen) {
57                         if (empties > 0 && j > 0)
58                                 sb->buf[j++] = '\n';
59                         empties = 0;
60                         memmove(sb->buf + j, sb->buf + i, newlen);
61                         sb->buf[newlen + j++] = '\n';
62                 } else {
63                         empties++;
64                 }
65         }
66
67         strbuf_setlen(sb, j);
68 }
69
70 static void comment_lines(struct strbuf *buf)
71 {
72         char *msg;
73         size_t len;
74
75         msg = strbuf_detach(buf, &len);
76         strbuf_add_commented_lines(buf, msg, len);
77         free(msg);
78 }
79
80 static const char *usage_msg = "\n"
81 "  git stripspace [-s | --strip-comments] < input\n"
82 "  git stripspace [-c | --comment-lines] < input";
83
84 int cmd_stripspace(int argc, const char **argv, const char *prefix)
85 {
86         struct strbuf buf = STRBUF_INIT;
87         int strip_comments = 0;
88         enum { INVAL = 0, STRIP_SPACE = 1, COMMENT_LINES = 2 } mode = STRIP_SPACE;
89
90         if (argc == 2) {
91                 if (!strcmp(argv[1], "-s") ||
92                     !strcmp(argv[1], "--strip-comments")) {
93                         strip_comments = 1;
94                 } else if (!strcmp(argv[1], "-c") ||
95                            !strcmp(argv[1], "--comment-lines")) {
96                         mode = COMMENT_LINES;
97                 } else {
98                         mode = INVAL;
99                 }
100         } else if (argc > 1) {
101                 mode = INVAL;
102         }
103
104         if (mode == INVAL)
105                 usage(usage_msg);
106
107         if (strip_comments || mode == COMMENT_LINES)
108                 git_config(git_default_config, NULL);
109
110         if (strbuf_read(&buf, 0, 1024) < 0)
111                 die_errno("could not read the input");
112
113         if (mode == STRIP_SPACE)
114                 stripspace(&buf, strip_comments);
115         else
116                 comment_lines(&buf);
117
118         write_or_die(1, buf.buf, buf.len);
119         strbuf_release(&buf);
120         return 0;
121 }