Merge branch 'jc/test-parse-options-expect'
[git] / builtin / check-attr.c
1 #include "builtin.h"
2 #include "cache.h"
3 #include "attr.h"
4 #include "quote.h"
5 #include "parse-options.h"
6
7 static int all_attrs;
8 static int cached_attrs;
9 static int stdin_paths;
10 static const char * const check_attr_usage[] = {
11 N_("git check-attr [-a | --all | <attr>...] [--] <pathname>..."),
12 N_("git check-attr --stdin [-z] [-a | --all | <attr>...]"),
13 NULL
14 };
15
16 static int nul_term_line;
17
18 static const struct option check_attr_options[] = {
19         OPT_BOOL('a', "all", &all_attrs, N_("report all attributes set on file")),
20         OPT_BOOL(0,  "cached", &cached_attrs, N_("use .gitattributes only from the index")),
21         OPT_BOOL(0 , "stdin", &stdin_paths, N_("read file names from stdin")),
22         OPT_BOOL('z', NULL, &nul_term_line,
23                  N_("terminate input and output records by a NUL character")),
24         OPT_END()
25 };
26
27 static void output_attr(int cnt, struct git_attr_check *check,
28         const char *file)
29 {
30         int j;
31         for (j = 0; j < cnt; j++) {
32                 const char *value = check[j].value;
33
34                 if (ATTR_TRUE(value))
35                         value = "set";
36                 else if (ATTR_FALSE(value))
37                         value = "unset";
38                 else if (ATTR_UNSET(value))
39                         value = "unspecified";
40
41                 if (nul_term_line) {
42                         printf("%s%c" /* path */
43                                "%s%c" /* attrname */
44                                "%s%c" /* attrvalue */,
45                                file, 0, git_attr_name(check[j].attr), 0, value, 0);
46                 } else {
47                         quote_c_style(file, NULL, stdout, 0);
48                         printf(": %s: %s\n", git_attr_name(check[j].attr), value);
49                 }
50
51         }
52 }
53
54 static void check_attr(const char *prefix, int cnt,
55         struct git_attr_check *check, const char *file)
56 {
57         char *full_path =
58                 prefix_path(prefix, prefix ? strlen(prefix) : 0, file);
59         if (check != NULL) {
60                 if (git_check_attr(full_path, cnt, check))
61                         die("git_check_attr died");
62                 output_attr(cnt, check, file);
63         } else {
64                 if (git_all_attrs(full_path, &cnt, &check))
65                         die("git_all_attrs died");
66                 output_attr(cnt, check, file);
67                 free(check);
68         }
69         free(full_path);
70 }
71
72 static void check_attr_stdin_paths(const char *prefix, int cnt,
73         struct git_attr_check *check)
74 {
75         struct strbuf buf = STRBUF_INIT;
76         struct strbuf unquoted = STRBUF_INIT;
77         strbuf_getline_fn getline_fn;
78
79         getline_fn = nul_term_line ? strbuf_getline_nul : strbuf_getline_lf;
80         while (getline_fn(&buf, stdin) != EOF) {
81                 if (!nul_term_line && buf.buf[0] == '"') {
82                         strbuf_reset(&unquoted);
83                         if (unquote_c_style(&unquoted, buf.buf, NULL))
84                                 die("line is badly quoted");
85                         strbuf_swap(&buf, &unquoted);
86                 }
87                 check_attr(prefix, cnt, check, buf.buf);
88                 maybe_flush_or_die(stdout, "attribute to stdout");
89         }
90         strbuf_release(&buf);
91         strbuf_release(&unquoted);
92 }
93
94 static NORETURN void error_with_usage(const char *msg)
95 {
96         error("%s", msg);
97         usage_with_options(check_attr_usage, check_attr_options);
98 }
99
100 int cmd_check_attr(int argc, const char **argv, const char *prefix)
101 {
102         struct git_attr_check *check;
103         int cnt, i, doubledash, filei;
104
105         if (!is_bare_repository())
106                 setup_work_tree();
107
108         git_config(git_default_config, NULL);
109
110         argc = parse_options(argc, argv, prefix, check_attr_options,
111                              check_attr_usage, PARSE_OPT_KEEP_DASHDASH);
112
113         if (read_cache() < 0) {
114                 die("invalid cache");
115         }
116
117         if (cached_attrs)
118                 git_attr_set_direction(GIT_ATTR_INDEX, NULL);
119
120         doubledash = -1;
121         for (i = 0; doubledash < 0 && i < argc; i++) {
122                 if (!strcmp(argv[i], "--"))
123                         doubledash = i;
124         }
125
126         /* Process --all and/or attribute arguments: */
127         if (all_attrs) {
128                 if (doubledash >= 1)
129                         error_with_usage("Attributes and --all both specified");
130
131                 cnt = 0;
132                 filei = doubledash + 1;
133         } else if (doubledash == 0) {
134                 error_with_usage("No attribute specified");
135         } else if (doubledash < 0) {
136                 if (!argc)
137                         error_with_usage("No attribute specified");
138
139                 if (stdin_paths) {
140                         /* Treat all arguments as attribute names. */
141                         cnt = argc;
142                         filei = argc;
143                 } else {
144                         /* Treat exactly one argument as an attribute name. */
145                         cnt = 1;
146                         filei = 1;
147                 }
148         } else {
149                 cnt = doubledash;
150                 filei = doubledash + 1;
151         }
152
153         /* Check file argument(s): */
154         if (stdin_paths) {
155                 if (filei < argc)
156                         error_with_usage("Can't specify files with --stdin");
157         } else {
158                 if (filei >= argc)
159                         error_with_usage("No file specified");
160         }
161
162         if (all_attrs) {
163                 check = NULL;
164         } else {
165                 check = xcalloc(cnt, sizeof(*check));
166                 for (i = 0; i < cnt; i++) {
167                         const char *name;
168                         struct git_attr *a;
169                         name = argv[i];
170                         a = git_attr(name);
171                         if (!a)
172                                 return error("%s: not a valid attribute name",
173                                         name);
174                         check[i].attr = a;
175                 }
176         }
177
178         if (stdin_paths)
179                 check_attr_stdin_paths(prefix, cnt, check);
180         else {
181                 for (i = filei; i < argc; i++)
182                         check_attr(prefix, cnt, check, argv[i]);
183                 maybe_flush_or_die(stdout, "attribute to stdout");
184         }
185         return 0;
186 }