builtin/stripspace.c

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