builtin / stripspace.con commit Merge branch 'nd/tree-walk-with-repo' (1eb0a12)
   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}