pkt-line: introduce struct packet_writer
[git] / ls-refs.c
1 #include "cache.h"
2 #include "repository.h"
3 #include "refs.h"
4 #include "remote.h"
5 #include "argv-array.h"
6 #include "ls-refs.h"
7 #include "pkt-line.h"
8
9 /*
10  * Check if one of the prefixes is a prefix of the ref.
11  * If no prefixes were provided, all refs match.
12  */
13 static int ref_match(const struct argv_array *prefixes, const char *refname)
14 {
15         int i;
16
17         if (!prefixes->argc)
18                 return 1; /* no restriction */
19
20         for (i = 0; i < prefixes->argc; i++) {
21                 const char *prefix = prefixes->argv[i];
22
23                 if (starts_with(refname, prefix))
24                         return 1;
25         }
26
27         return 0;
28 }
29
30 struct ls_refs_data {
31         unsigned peel;
32         unsigned symrefs;
33         struct argv_array prefixes;
34 };
35
36 static int send_ref(const char *refname, const struct object_id *oid,
37                     int flag, void *cb_data)
38 {
39         struct ls_refs_data *data = cb_data;
40         const char *refname_nons = strip_namespace(refname);
41         struct strbuf refline = STRBUF_INIT;
42
43         if (!ref_match(&data->prefixes, refname))
44                 return 0;
45
46         strbuf_addf(&refline, "%s %s", oid_to_hex(oid), refname_nons);
47         if (data->symrefs && flag & REF_ISSYMREF) {
48                 struct object_id unused;
49                 const char *symref_target = resolve_ref_unsafe(refname, 0,
50                                                                &unused,
51                                                                &flag);
52
53                 if (!symref_target)
54                         die("'%s' is a symref but it is not?", refname);
55
56                 strbuf_addf(&refline, " symref-target:%s", symref_target);
57         }
58
59         if (data->peel) {
60                 struct object_id peeled;
61                 if (!peel_ref(refname, &peeled))
62                         strbuf_addf(&refline, " peeled:%s", oid_to_hex(&peeled));
63         }
64
65         strbuf_addch(&refline, '\n');
66         packet_write(1, refline.buf, refline.len);
67
68         strbuf_release(&refline);
69         return 0;
70 }
71
72 int ls_refs(struct repository *r, struct argv_array *keys,
73             struct packet_reader *request)
74 {
75         struct ls_refs_data data;
76
77         memset(&data, 0, sizeof(data));
78
79         while (packet_reader_read(request) != PACKET_READ_FLUSH) {
80                 const char *arg = request->line;
81                 const char *out;
82
83                 if (!strcmp("peel", arg))
84                         data.peel = 1;
85                 else if (!strcmp("symrefs", arg))
86                         data.symrefs = 1;
87                 else if (skip_prefix(arg, "ref-prefix ", &out))
88                         argv_array_push(&data.prefixes, out);
89         }
90
91         head_ref_namespaced(send_ref, &data);
92         for_each_namespaced_ref(send_ref, &data);
93         packet_flush(1);
94         argv_array_clear(&data.prefixes);
95         return 0;
96 }