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