Merge branch 'jt/use-trailer-api-in-commands'
[git] / builtin / stripspace.c
1 #include "builtin.h"
2 #include "cache.h"
3 #include "parse-options.h"
4 #include "strbuf.h"
5
6 static void comment_lines(struct strbuf *buf)
7 {
8         char *msg;
9         size_t len;
10
11         msg = strbuf_detach(buf, &len);
12         strbuf_add_commented_lines(buf, msg, len);
13         free(msg);
14 }
15
16 static const char * const stripspace_usage[] = {
17         N_("git stripspace [-s | --strip-comments]"),
18         N_("git stripspace [-c | --comment-lines]"),
19         NULL
20 };
21
22 enum stripspace_mode {
23         STRIP_DEFAULT = 0,
24         STRIP_COMMENTS,
25         COMMENT_LINES
26 };
27
28 int cmd_stripspace(int argc, const char **argv, const char *prefix)
29 {
30         struct strbuf buf = STRBUF_INIT;
31         enum stripspace_mode mode = STRIP_DEFAULT;
32
33         const struct option options[] = {
34                 OPT_CMDMODE('s', "strip-comments", &mode,
35                             N_("skip and remove all lines starting with comment character"),
36                             STRIP_COMMENTS),
37                 OPT_CMDMODE('c', "comment-lines", &mode,
38                             N_("prepend comment character and space to each line"),
39                             COMMENT_LINES),
40                 OPT_END()
41         };
42
43         argc = parse_options(argc, argv, prefix, options, stripspace_usage, 0);
44         if (argc)
45                 usage_with_options(stripspace_usage, options);
46
47         if (mode == STRIP_COMMENTS || mode == COMMENT_LINES) {
48                 setup_git_directory_gently(NULL);
49                 git_config(git_default_config, NULL);
50         }
51
52         if (strbuf_read(&buf, 0, 1024) < 0)
53                 die_errno("could not read the input");
54
55         if (mode == STRIP_DEFAULT || mode == STRIP_COMMENTS)
56                 strbuf_stripspace(&buf, mode == STRIP_COMMENTS);
57         else
58                 comment_lines(&buf);
59
60         write_or_die(1, buf.buf, buf.len);
61         strbuf_release(&buf);
62         return 0;
63 }