blame: simplify 'setup_blame_bloom_data' interface
[git] / ls-refs.c
1 #include "cache.h"
2 #include "repository.h"
3 #include "refs.h"
4 #include "remote.h"
5 #include "strvec.h"
6 #include "ls-refs.h"
7 #include "pkt-line.h"
8 #include "config.h"
9
10 /*
11  * Check if one of the prefixes is a prefix of the ref.
12  * If no prefixes were provided, all refs match.
13  */
14 static int ref_match(const struct strvec *prefixes, const char *refname)
15 {
16         int i;
17
18         if (!prefixes->nr)
19                 return 1; /* no restriction */
20
21         for (i = 0; i < prefixes->nr; i++) {
22                 const char *prefix = prefixes->v[i];
23
24                 if (starts_with(refname, prefix))
25                         return 1;
26         }
27
28         return 0;
29 }
30
31 struct ls_refs_data {
32         unsigned peel;
33         unsigned symrefs;
34         struct strvec prefixes;
35 };
36
37 static int send_ref(const char *refname, const struct object_id *oid,
38                     int flag, void *cb_data)
39 {
40         struct ls_refs_data *data = cb_data;
41         const char *refname_nons = strip_namespace(refname);
42         struct strbuf refline = STRBUF_INIT;
43
44         if (ref_is_hidden(refname_nons, refname))
45                 return 0;
46
47         if (!ref_match(&data->prefixes, refname_nons))
48                 return 0;
49
50         strbuf_addf(&refline, "%s %s", oid_to_hex(oid), refname_nons);
51         if (data->symrefs && flag & REF_ISSYMREF) {
52                 struct object_id unused;
53                 const char *symref_target = resolve_ref_unsafe(refname, 0,
54                                                                &unused,
55                                                                &flag);
56
57                 if (!symref_target)
58                         die("'%s' is a symref but it is not?", refname);
59
60                 strbuf_addf(&refline, " symref-target:%s",
61                             strip_namespace(symref_target));
62         }
63
64         if (data->peel) {
65                 struct object_id peeled;
66                 if (!peel_ref(refname, &peeled))
67                         strbuf_addf(&refline, " peeled:%s", oid_to_hex(&peeled));
68         }
69
70         strbuf_addch(&refline, '\n');
71         packet_write(1, refline.buf, refline.len);
72
73         strbuf_release(&refline);
74         return 0;
75 }
76
77 static int ls_refs_config(const char *var, const char *value, void *data)
78 {
79         /*
80          * We only serve fetches over v2 for now, so respect only "uploadpack"
81          * config. This may need to eventually be expanded to "receive", but we
82          * don't yet know how that information will be passed to ls-refs.
83          */
84         return parse_hide_refs_config(var, value, "uploadpack");
85 }
86
87 int ls_refs(struct repository *r, struct strvec *keys,
88             struct packet_reader *request)
89 {
90         struct ls_refs_data data;
91
92         memset(&data, 0, sizeof(data));
93
94         git_config(ls_refs_config, NULL);
95
96         while (packet_reader_read(request) == PACKET_READ_NORMAL) {
97                 const char *arg = request->line;
98                 const char *out;
99
100                 if (!strcmp("peel", arg))
101                         data.peel = 1;
102                 else if (!strcmp("symrefs", arg))
103                         data.symrefs = 1;
104                 else if (skip_prefix(arg, "ref-prefix ", &out))
105                         strvec_push(&data.prefixes, out);
106         }
107
108         if (request->status != PACKET_READ_FLUSH)
109                 die(_("expected flush after ls-refs arguments"));
110
111         head_ref_namespaced(send_ref, &data);
112         for_each_namespaced_ref(send_ref, &data);
113         packet_flush(1);
114         strvec_clear(&data.prefixes);
115         return 0;
116 }