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 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 public @Nullable PrePushHook getPrePushHook(Repository repo,
156 PrintStream outputStream) {
157 return null;
158 }
159
160 /**
161 * Retrieve an {@link LfsInstallCommand} which can be used to enable LFS
162 * support (if available) either per repository or for the user.
163 *
164 * @return a command to install LFS support.
165 */
166 public @Nullable LfsInstallCommand getInstallCommand() {
167 return null;
168 }
169
170 /**
171 * @param db
172 * the repository to check
173 * @return whether LFS is enabled for the given repository locally or
174 * globally.
175 */
176 public boolean isEnabled(Repository db) {
177 return false;
178 }
179
180 /**
181 * @param db
182 * the repository
183 * @param path
184 * the path to find attributes for
185 * @return the {@link Attributes} for the given path.
186 * @throws IOException
187 * in case of an error
188 */
189 public static Attributes getAttributesForPath(Repository db, String path)
190 throws IOException {
191 try (TreeWalk walk = new TreeWalk(db)) {
192 walk.addTree(new FileTreeIterator(db));
193 PathFilter f = PathFilter.create(path);
194 walk.setFilter(f);
195 walk.setRecursive(false);
196 Attributes attr = null;
197 while (walk.next()) {
198 if (f.isDone(walk)) {
199 attr = walk.getAttributes();
200 break;
201 } else if (walk.isSubtree()) {
202 walk.enterSubtree();
203 }
204 }
205 if (attr == null) {
206 throw new IOException(MessageFormat
207 .format(JGitText.get().noPathAttributesFound, path));
208 }
209
210 return attr;
211 }
212 }
213
214 /**
215 * Get attributes for given path and commit
216 *
217 * @param db
218 * the repository
219 * @param path
220 * the path to find attributes for
221 * @param commit
222 * the commit to inspect.
223 * @return the {@link Attributes} for the given path.
224 * @throws IOException
225 * in case of an error
226 */
227 public static Attributes getAttributesForPath(Repository db, String path,
228 RevCommit commit) throws IOException {
229 if (commit == null) {
230 return getAttributesForPath(db, path);
231 }
232
233 try (TreeWalk walk = TreeWalk.forPath(db, path, commit.getTree())) {
234 Attributes attr = walk == null ? null : walk.getAttributes();
235 if (attr == null) {
236 throw new IOException(MessageFormat
237 .format(JGitText.get().noPathAttributesFound, path));
238 }
239
240 return attr;
241 }
242 }
243
244 /**
245 * Encapsulate a potentially exchanged {@link InputStream} along with the
246 * expected stream content length.
247 */
248 public static final class LfsInputStream extends InputStream {
249 /**
250 * The actual stream.
251 */
252 private InputStream stream;
253
254 /**
255 * The expected stream content length.
256 */
257 private long length;
258
259 /**
260 * Create a new wrapper around a certain stream
261 *
262 * @param stream
263 * the stream to wrap. the stream will be closed on
264 * {@link #close()}.
265 * @param length
266 * the expected length of the stream
267 */
268 public LfsInputStream(InputStream stream, long length) {
269 this.stream = stream;
270 this.length = length;
271 }
272
273 /**
274 * Create a new wrapper around a temporary buffer.
275 *
276 * @param buffer
277 * the buffer to initialize stream and length from. The
278 * buffer will be destroyed on {@link #close()}
279 * @throws IOException
280 * in case of an error opening the stream to the buffer.
281 */
282 public LfsInputStream(TemporaryBuffer buffer) throws IOException {
283 this.stream = buffer.openInputStreamWithAutoDestroy();
284 this.length = buffer.length();
285 }
286
287 @Override
288 public void close() throws IOException {
289 stream.close();
290 }
291
292 @Override
293 public int read() throws IOException {
294 return stream.read();
295 }
296
297 /**
298 * @return the length of the stream
299 */
300 public long getLength() {
301 return length;
302 }
303 }
304
305 /**
306 * A command to enable LFS. Optionally set a {@link Repository} to enable
307 * locally on the repository only.
308 */
309 public interface LfsInstallCommand extends Callable<Void> {
310 /**
311 * @param repo
312 * the repository to enable support for.
313 * @return The {@link LfsInstallCommand} for chaining.
314 */
315 public LfsInstallCommand setRepository(Repository repo);
316 }
317
318 }