summaryrefslogtreecommitdiff
path: root/src/fetch.c
blob: 21f52fc5ee547268ea4eb9e32e63ac5acb86bc41 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
/*
 * This file is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License, version 2,
 * as published by the Free Software Foundation.
 *
 * In addition to the permissions in the GNU General Public License,
 * the authors give you unlimited permission to link the compiled
 * version of this file into combinations with other programs,
 * and to distribute those combinations without any restriction
 * coming from the use of this file.  (The General Public License
 * restrictions do apply in other respects; for example, they cover
 * modification of the file, and distribution when not linked into
 * a combined executable.)
 *
 * This file is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; see the file COPYING.  If not, write to
 * the Free Software Foundation, 51 Franklin Street, Fifth Floor,
 * Boston, MA 02110-1301, USA.
 */

#include "git2/remote.h"
#include "git2/oid.h"
#include "git2/refs.h"

#include "common.h"
#include "transport.h"
#include "remote.h"
#include "refspec.h"

/*
 * Don't forget that this depends on the enum being correctly set
 */
static int whn_cmp(const void *a, const void *b)
{
	git_remote_head *heada = *(git_remote_head **)(a);
	git_remote_head *headb = *(git_remote_head **)(b);

	return headb->type - heada->type;
}

/* FIXME: we assume that the transport has been connected, enforce that somehow */
int git_fetch_list_want(git_headarray *whn_list, git_repository *repo, git_remote *remote)
{
	git_vector list;
	git_headarray refs, lrefs;
	git_transport *t = remote->transport;
	const git_refspec *spec;
	int error, i;

	error = git_vector_init(&list, 16, whn_cmp);
	if (error < GIT_SUCCESS)
		return error;

	error = git_transport_ls(t, &refs);
	if (error < GIT_SUCCESS) {
		error = git__rethrow(error, "Failed to list local refs");
		goto cleanup;
	}

	spec = git_remote_fetchspec(remote);
	if (spec == NULL) {
		error = git__throw(GIT_ERROR, "The remote has to fetchspec");
		goto cleanup;
	}

	for (i = 0; i < refs.len; ++i) {
		char local[1024];
		git_reference *ref;
		git_remote_head *head = refs.heads[i];

		/* If it doesn't match the refpec, we don't want it */
		error = git_refspec_src_match(spec, head->name);
		if (error == GIT_ENOMATCH)
			continue;
		if (error < GIT_SUCCESS) {
			error = git__rethrow(error, "Error matching remote ref name");
			goto cleanup;
		}

		/* If the local ref is the same, we don't want it either */
		error = git_refspec_transform(local, sizeof(local), spec, head->name);
		if (error < GIT_SUCCESS) {
			error = git__rethrow(error, "Error transforming ref name");
			goto cleanup;
		}

		error = git_reference_lookup(&ref, repo, local);
		/* If we don't have it locally, it's new, so we want it */
		if (error < GIT_SUCCESS && error != GIT_ENOTFOUND) {
			error = git__rethrow(error, "Error looking up local ref");
			goto cleanup;
		}

		if (ref != NULL) {
			if (!git_oid_cmp(&head->oid, git_reference_oid(ref)))
				continue;
		}

		/*
		 * Now we know we want to have that ref, so add it as a "want"
		 * to the list.
		 */
		head->type = GIT_WHN_WANT;
		error = git_vector_insert(&list, head);
		if (error < GIT_SUCCESS)
			goto cleanup;
	}

	git_vector_sort(&list);
	whn_list->len = list.length;
	whn_list->heads = (git_remote_head **) list.contents;

	return GIT_SUCCESS;

cleanup:
	git_vector_free(&list);
	return error;
}