1#include "builtin.h"
2#include "cache.h"
3#include "parse-options.h"
4#include "strbuf.h"
5
6static 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
16static const char * const stripspace_usage[] = {
17 N_("git stripspace [-s | --strip-comments]"),
18 N_("git stripspace [-c | --comment-lines]"),
19 NULL
20};
21
22enum stripspace_mode {
23 STRIP_DEFAULT = 0,
24 STRIP_COMMENTS,
25 COMMENT_LINES
26};
27
28int 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}