1 /*
2 * Copyright (C) 2012, Google Inc.
3 * and other copyright owners as documented in the project's IP log.
4 * * This program and the accompanying materials are made available
5 * under the terms of the Eclipse Distribution License v1.0 which
6 * accompanies this distribution, is reproduced below, and is
7 * available at http://www.eclipse.org/org/documents/edl-v10.php
8 *
9 * All rights reserved.
10 *
11 * Redistribution and use in source and binary forms, with or
12 * without modification, are permitted provided that the following
13 * conditions are met:
14 *
15 * - Redistributions of source code must retain the above copyright
16 * notice, this list of conditions and the following disclaimer.
17 *
18 * - Redistributions in binary form must reproduce the above
19 * copyright notice, this list of conditions and the following
20 * disclaimer in the documentation and/or other materials provided
21 * with the distribution.
22 *
23 * - Neither the name of the Eclipse Foundation, Inc. nor the
24 * names of its contributors may be used to endorse or promote
25 * products derived from this software without specific prior
26 * written permission.
27 *
28 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
29 * CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
30 * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
31 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
32 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
33 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
34 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
35 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
36 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
37 * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
38 * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
39 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
40 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
41 */
42
43 package org.eclipse.jgit.transport;
44
45 import java.util.Map;
46 import java.util.Set;
47
48 import org.eclipse.jgit.lib.ObjectId;
49 import org.eclipse.jgit.lib.Ref;
50 import org.eclipse.jgit.lib.Repository;
51 import org.eclipse.jgit.revwalk.RevWalk;
52
53 /**
54 * Implementation of {@link AdvertiseRefsHook} that advertises the same refs for
55 * upload-pack and receive-pack.
56 *
57 * @since 2.0
58 */
59 public abstract class AbstractAdvertiseRefsHook implements AdvertiseRefsHook {
60 @Override
61 public void advertiseRefs(UploadPack uploadPack)
62 throws ServiceMayNotContinueException {
63 uploadPack.setAdvertisedRefs(getAdvertisedRefs(
64 uploadPack.getRepository(), uploadPack.getRevWalk()));
65 }
66
67 @Override
68 public void advertiseRefs(BaseReceivePack receivePack)
69 throws ServiceMayNotContinueException {
70 Map<String, Ref> refs = getAdvertisedRefs(receivePack.getRepository(),
71 receivePack.getRevWalk());
72 Set<ObjectId> haves = getAdvertisedHaves(receivePack.getRepository(),
73 receivePack.getRevWalk());
74 receivePack.setAdvertisedRefs(refs, haves);
75 }
76
77 /**
78 * Get the refs to advertise.
79 *
80 * @param repository
81 * repository instance.
82 * @param revWalk
83 * open rev walk on the repository.
84 * @return set of refs to advertise.
85 * @throws ServiceMayNotContinueException
86 * abort; the message will be sent to the user.
87 */
88 protected abstract Map<String, Ref> getAdvertisedRefs(
89 Repository repository, RevWalk revWalk)
90 throws ServiceMayNotContinueException;
91
92 /**
93 * Get the additional haves to advertise.
94 *
95 * @param repository
96 * repository instance.
97 * @param revWalk
98 * open rev walk on the repository.
99 * @return set of additional haves; see
100 * {@link ReceivePack#getAdvertisedObjects()}.
101 * @throws ServiceMayNotContinueException
102 * abort; the message will be sent to the user.
103 */
104 protected Set<ObjectId> getAdvertisedHaves(
105 Repository repository, RevWalk revWalk)
106 throws ServiceMayNotContinueException {
107 return null;
108 }
109 }