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 }