1 /*
2 * Copyright (C) 2018, Markus Duft <markus.duft@ssi-schaefer.com>
3 * and other copyright owners as documented in the project's IP log.
4 *
5 * This program and the accompanying materials are made available
6 * under the terms of the Eclipse Distribution License v1.0 which
7 * accompanies this distribution, is reproduced below, and is
8 * available at http://www.eclipse.org/org/documents/edl-v10.php
9 *
10 * All rights reserved.
11 *
12 * Redistribution and use in source and binary forms, with or
13 * without modification, are permitted provided that the following
14 * conditions are met:
15 *
16 * - Redistributions of source code must retain the above copyright
17 * notice, this list of conditions and the following disclaimer.
18 *
19 * - Redistributions in binary form must reproduce the above
20 * copyright notice, this list of conditions and the following
21 * disclaimer in the documentation and/or other materials provided
22 * with the distribution.
23 *
24 * - Neither the name of the Eclipse Foundation, Inc. nor the
25 * names of its contributors may be used to endorse or promote
26 * products derived from this software without specific prior
27 * written permission.
28 *
29 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
30 * CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
31 * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
32 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
33 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
34 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
35 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
36 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
37 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
38 * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
39 * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
40 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
41 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
42 */
43 package org.eclipse.jgit.util;
44
45 import java.io.IOException;
46 import java.io.InputStream;
47 import java.io.PrintStream;
48 import java.text.MessageFormat;
49 import java.util.concurrent.Callable;
50
51 import org.eclipse.jgit.annotations.Nullable;
52 import org.eclipse.jgit.attributes.Attribute;
53 import org.eclipse.jgit.attributes.Attributes;
54 import org.eclipse.jgit.hooks.PrePushHook;
55 import org.eclipse.jgit.internal.JGitText;
56 import org.eclipse.jgit.lib.ObjectLoader;
57 import org.eclipse.jgit.lib.Repository;
58 import org.eclipse.jgit.revwalk.RevCommit;
59 import org.eclipse.jgit.treewalk.FileTreeIterator;
60 import org.eclipse.jgit.treewalk.TreeWalk;
61 import org.eclipse.jgit.treewalk.filter.PathFilter;
62
63 /**
64 * Represents an optionally present LFS support implementation
65 *
66 * @since 4.11
67 */
68 public class LfsFactory {
69
70 private static LfsFactorytml#LfsFactory">LfsFactory instance = new LfsFactory();
71
72 /**
73 * Constructor
74 */
75 protected LfsFactory() {
76 }
77
78 /**
79 * @return the current LFS implementation
80 */
81 public static LfsFactory getInstance() {
82 return instance;
83 }
84
85 /**
86 * @param instance
87 * register a {@link LfsFactory} instance as the
88 * {@link LfsFactory} implementation to use.
89 */
90 public static void setInstance(LfsFactory instance) {
91 LfsFactory.instance = instance;
92 }
93
94 /**
95 * @return whether LFS support is available
96 */
97 public boolean isAvailable() {
98 return false;
99 }
100
101 /**
102 * Apply clean filtering to the given stream, writing the file content to
103 * the LFS storage if required and returning a stream to the LFS pointer
104 * instead.
105 *
106 * @param db
107 * the repository
108 * @param input
109 * the original input
110 * @param length
111 * the expected input stream length
112 * @param attribute
113 * the attribute used to check for LFS enablement (i.e. "merge",
114 * "diff", "filter" from .gitattributes).
115 * @return a stream to the content that should be written to the object
116 * store along with the expected length of the stream. the original
117 * stream is not applicable.
118 * @throws IOException
119 * in case of an error
120 */
121 public LfsInputStream applyCleanFilter(Repository db,
122 InputStream input, long length, Attribute attribute)
123 throws IOException {
124 return new LfsInputStream(input, length);
125 }
126
127 /**
128 * Apply smudge filtering to a given loader, potentially redirecting it to a
129 * LFS blob which is downloaded on demand.
130 *
131 * @param db
132 * the repository
133 * @param loader
134 * the loader for the blob
135 * @param attribute
136 * the attribute used to check for LFS enablement (i.e. "merge",
137 * "diff", "filter" from .gitattributes).
138 * @return a loader for the actual data of a blob, or the original loader in
139 * case LFS is not applicable.
140 * @throws IOException
141 */
142 public ObjectLoader applySmudgeFilter(Repository db,
143 ObjectLoader loader, Attribute attribute) throws IOException {
144 return loader;
145 }
146
147 /**
148 * Retrieve a pre-push hook to be applied.
149 *
150 * @param repo
151 * the {@link Repository} the hook is applied to.
152 * @param outputStream
153 * @return a {@link PrePushHook} implementation or <code>null</code>
154 */
155 @Nullable
156 public PrePushHook getPrePushHook(Repository repo,
157 PrintStream outputStream) {
158 return null;
159 }
160
161 /**
162 * Retrieve an {@link LfsInstallCommand} which can be used to enable LFS
163 * support (if available) either per repository or for the user.
164 *
165 * @return a command to install LFS support.
166 */
167 @Nullable
168 public LfsInstallCommand getInstallCommand() {
169 return null;
170 }
171
172 /**
173 * @param db
174 * the repository to check
175 * @return whether LFS is enabled for the given repository locally or
176 * globally.
177 */
178 public boolean isEnabled(Repository db) {
179 return false;
180 }
181
182 /**
183 * @param db
184 * the repository
185 * @param path
186 * the path to find attributes for
187 * @return the {@link Attributes} for the given path.
188 * @throws IOException
189 * in case of an error
190 */
191 public static Attributes getAttributesForPath(Repository db, String path)
192 throws IOException {
193 try (TreeWalkeeWalk.html#TreeWalk">TreeWalk walk = new TreeWalk(db)) {
194 walk.addTree(new FileTreeIterator(db));
195 PathFilter f = PathFilter.create(path);
196 walk.setFilter(f);
197 walk.setRecursive(false);
198 Attributes attr = null;
199 while (walk.next()) {
200 if (f.isDone(walk)) {
201 attr = walk.getAttributes();
202 break;
203 } else if (walk.isSubtree()) {
204 walk.enterSubtree();
205 }
206 }
207 if (attr == null) {
208 throw new IOException(MessageFormat
209 .format(JGitText.get().noPathAttributesFound, path));
210 }
211
212 return attr;
213 }
214 }
215
216 /**
217 * Get attributes for given path and commit
218 *
219 * @param db
220 * the repository
221 * @param path
222 * the path to find attributes for
223 * @param commit
224 * the commit to inspect.
225 * @return the {@link Attributes} for the given path.
226 * @throws IOException
227 * in case of an error
228 */
229 public static Attributes getAttributesForPath(Repository db, String path,
230 RevCommit commit) throws IOException {
231 if (commit == null) {
232 return getAttributesForPath(db, path);
233 }
234
235 try (TreeWalk walk = TreeWalk.forPath(db, path, commit.getTree())) {
236 Attributes attr = walk == null ? null : walk.getAttributes();
237 if (attr == null) {
238 throw new IOException(MessageFormat
239 .format(JGitText.get().noPathAttributesFound, path));
240 }
241
242 return attr;
243 }
244 }
245
246 /**
247 * Encapsulate a potentially exchanged {@link InputStream} along with the
248 * expected stream content length.
249 */
250 public static final class LfsInputStream extends InputStream {
251 /**
252 * The actual stream.
253 */
254 private InputStream stream;
255
256 /**
257 * The expected stream content length.
258 */
259 private long length;
260
261 /**
262 * Create a new wrapper around a certain stream
263 *
264 * @param stream
265 * the stream to wrap. the stream will be closed on
266 * {@link #close()}.
267 * @param length
268 * the expected length of the stream
269 */
270 public LfsInputStream(InputStream stream, long length) {
271 this.stream = stream;
272 this.length = length;
273 }
274
275 /**
276 * Create a new wrapper around a temporary buffer.
277 *
278 * @param buffer
279 * the buffer to initialize stream and length from. The
280 * buffer will be destroyed on {@link #close()}
281 * @throws IOException
282 * in case of an error opening the stream to the buffer.
283 */
284 public LfsInputStream(TemporaryBuffer buffer) throws IOException {
285 this.stream = buffer.openInputStreamWithAutoDestroy();
286 this.length = buffer.length();
287 }
288
289 @Override
290 public void close() throws IOException {
291 stream.close();
292 }
293
294 @Override
295 public int read() throws IOException {
296 return stream.read();
297 }
298
299 @Override
300 public int read(byte b[], int off, int len) throws IOException {
301 return stream.read(b, off, len);
302 }
303
304 /**
305 * @return the length of the stream
306 */
307 public long getLength() {
308 return length;
309 }
310 }
311
312 /**
313 * A command to enable LFS. Optionally set a {@link Repository} to enable
314 * locally on the repository only.
315 */
316 public interface LfsInstallCommand extends Callable<Void> {
317 /**
318 * @param repo
319 * the repository to enable support for.
320 * @return The {@link LfsInstallCommand} for chaining.
321 */
322 public LfsInstallCommand setRepository(Repository repo);
323 }
324
325 }