Fix new-workdir (again) to work on bare repositories
[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         if (len) {
12                 if (line[len - 1] == '\n')
13                         len--;
14
15                 while (len) {
16                         unsigned char c = line[len - 1];
17                         if (!isspace(c))
18                                 break;
19                         len--;
20                 }
21         }
22         return len;
23 }
24
25 /*
26  * Remove empty lines from the beginning and end
27  * and also trailing spaces from every line.
28  *
29  * Note that the buffer will not be NUL-terminated.
30  *
31  * Turn multiple consecutive empty lines between paragraphs
32  * into just one empty line.
33  *
34  * If the input has only empty lines and spaces,
35  * no output will be produced.
36  *
37  * If last line has a newline at the end, it will be removed.
38  *
39  * Enable skip_comments to skip every line starting with "#".
40  */
41 size_t stripspace(char *buffer, size_t length, int skip_comments)
42 {
43         int empties = -1;
44         size_t i, j, len, newlen;
45         char *eol;
46
47         for (i = j = 0; i < length; i += len, j += newlen) {
48                 eol = memchr(buffer + i, '\n', length - i);
49                 len = eol ? eol - (buffer + i) + 1 : length - i;
50
51                 if (skip_comments && len && buffer[i] == '#') {
52                         newlen = 0;
53                         continue;
54                 }
55                 newlen = cleanup(buffer + i, len);
56
57                 /* Not just an empty line? */
58                 if (newlen) {
59                         if (empties != -1)
60                                 buffer[j++] = '\n';
61                         if (empties > 0)
62                                 buffer[j++] = '\n';
63                         empties = 0;
64                         memmove(buffer + j, buffer + i, newlen);
65                         continue;
66                 }
67                 if (empties < 0)
68                         continue;
69                 empties++;
70         }
71
72         return j;
73 }
74
75 int cmd_stripspace(int argc, const char **argv, const char *prefix)
76 {
77         char *buffer;
78         unsigned long size;
79         int strip_comments = 0;
80
81         if (argc > 1 && (!strcmp(argv[1], "-s") ||
82                                 !strcmp(argv[1], "--strip-comments")))
83                 strip_comments = 1;
84
85         size = 1024;
86         buffer = xmalloc(size);
87         if (read_fd(0, &buffer, &size)) {
88                 free(buffer);
89                 die("could not read the input");
90         }
91
92         size = stripspace(buffer, size, strip_comments);
93         write_or_die(1, buffer, size);
94         if (size)
95                 putc('\n', stdout);
96
97         free(buffer);
98         return 0;
99 }