ls-refs.con commit Merge branch 'bw/protocol-v2' (5b39d49)
   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 */
  13static 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
  30struct ls_refs_data {
  31        unsigned peel;
  32        unsigned symrefs;
  33        struct argv_array prefixes;
  34};
  35
  36static 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
  72int 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}