builtin/for-each-ref.c

   1 #include "builtin.h"
   2 #include "config.h"
   3 #include "gettext.h"
   4 #include "refs.h"
   5 #include "object.h"
   6 #include "parse-options.h"
   7 #include "ref-filter.h"
   8 #include "strbuf.h"
   9 #include "strvec.h"
  10 #include "commit-reach.h"
  11
  12 static char const * const for_each_ref_usage[] = {
  13         N_("git for-each-ref [<options>] [<pattern>]"),
  14         N_("git for-each-ref [--points-at <object>]"),
  15         N_("git for-each-ref [--merged [<commit>]] [--no-merged [<commit>]]"),
  16         N_("git for-each-ref [--contains [<commit>]] [--no-contains [<commit>]]"),
  17         NULL
  18 };
  19
  20 int cmd_for_each_ref(int argc, const char **argv, const char *prefix)
  21 {
  22         int i;
  23         struct ref_sorting *sorting;
  24         struct string_list sorting_options = STRING_LIST_INIT_DUP;
  25         int maxcount = 0, icase = 0, omit_empty = 0;
  26         struct ref_array array;
  27         struct ref_filter filter = REF_FILTER_INIT;
  28         struct ref_format format = REF_FORMAT_INIT;
  29         struct strbuf output = STRBUF_INIT;
  30         struct strbuf err = STRBUF_INIT;
  31         int from_stdin = 0;
  32         struct strvec vec = STRVEC_INIT;
  33
  34         struct option opts[] = {
  35                 OPT_BIT('s', "shell", &format.quote_style,
  36                         N_("quote placeholders suitably for shells"), QUOTE_SHELL),
  37                 OPT_BIT('p', "perl",  &format.quote_style,
  38                         N_("quote placeholders suitably for perl"), QUOTE_PERL),
  39                 OPT_BIT(0 , "python", &format.quote_style,
  40                         N_("quote placeholders suitably for python"), QUOTE_PYTHON),
  41                 OPT_BIT(0 , "tcl",  &format.quote_style,
  42                         N_("quote placeholders suitably for Tcl"), QUOTE_TCL),
  43                 OPT_BOOL(0, "omit-empty",  &omit_empty,
  44                         N_("do not output a newline after empty formatted refs")),
  45
  46                 OPT_GROUP(""),
  47                 OPT_INTEGER( 0 , "count", &maxcount, N_("show only <n> matched refs")),
  48                 OPT_STRING(  0 , "format", &format.format, N_("format"), N_("format to use for the output")),
  49                 OPT__COLOR(&format.use_color, N_("respect format colors")),
  50                 OPT_REF_FILTER_EXCLUDE(&filter),
  51                 OPT_REF_SORT(&sorting_options),
  52                 OPT_CALLBACK(0, "points-at", &filter.points_at,
  53                              N_("object"), N_("print only refs which points at the given object"),
  54                              parse_opt_object_name),
  55                 OPT_MERGED(&filter, N_("print only refs that are merged")),
  56                 OPT_NO_MERGED(&filter, N_("print only refs that are not merged")),
  57                 OPT_CONTAINS(&filter.with_commit, N_("print only refs which contain the commit")),
  58                 OPT_NO_CONTAINS(&filter.no_commit, N_("print only refs which don't contain the commit")),
  59                 OPT_BOOL(0, "ignore-case", &icase, N_("sorting and filtering are case insensitive")),
  60                 OPT_BOOL(0, "stdin", &from_stdin, N_("read reference patterns from stdin")),
  61                 OPT_END(),
  62         };
  63
  64         memset(&array, 0, sizeof(array));
  65
  66         format.format = "%(objectname) %(objecttype)\t%(refname)";
  67
  68         git_config(git_default_config, NULL);
  69
  70         parse_options(argc, argv, prefix, opts, for_each_ref_usage, 0);
  71         if (maxcount < 0) {
  72                 error("invalid --count argument: `%d'", maxcount);
  73                 usage_with_options(for_each_ref_usage, opts);
  74         }
  75         if (HAS_MULTI_BITS(format.quote_style)) {
  76                 error("more than one quoting style?");
  77                 usage_with_options(for_each_ref_usage, opts);
  78         }
  79         if (verify_ref_format(&format))
  80                 usage_with_options(for_each_ref_usage, opts);
  81
  82         sorting = ref_sorting_options(&sorting_options);
  83         ref_sorting_set_sort_flags_all(sorting, REF_SORTING_ICASE, icase);
  84         filter.ignore_case = icase;
  85
  86         if (from_stdin) {
  87                 struct strbuf line = STRBUF_INIT;
  88
  89                 if (argv[0])
  90                         die(_("unknown arguments supplied with --stdin"));
  91
  92                 while (strbuf_getline(&line, stdin) != EOF)
  93                         strvec_push(&vec, line.buf);
  94
  95                 strbuf_release(&line);
  96
  97                 /* vec.v is NULL-terminated, just like 'argv'. */
  98                 filter.name_patterns = vec.v;
  99         } else {
 100                 filter.name_patterns = argv;
 101         }
 102
 103         filter.match_as_path = 1;
 104         filter_refs(&array, &filter, FILTER_REFS_ALL);
 105         filter_ahead_behind(the_repository, &format, &array);
 106
 107         ref_array_sort(sorting, &array);
 108
 109         if (!maxcount || array.nr < maxcount)
 110                 maxcount = array.nr;
 111         for (i = 0; i < maxcount; i++) {
 112                 strbuf_reset(&err);
 113                 strbuf_reset(&output);
 114                 if (format_ref_array_item(array.items[i], &format, &output, &err))
 115                         die("%s", err.buf);
 116                 fwrite(output.buf, 1, output.len, stdout);
 117                 if (output.len || !omit_empty)
 118                         putchar('\n');
 119         }
 120
 121         strbuf_release(&err);
 122         strbuf_release(&output);
 123         ref_array_clear(&array);
 124         ref_filter_clear(&filter);
 125         ref_sorting_release(sorting);
 126         strvec_clear(&vec);
 127         return 0;
 128 }