1 /*
2 * Copyright © 2017, Saleforce.com, Inc
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions are met:
7 * * Redistributions of source code must retain the above copyright
8 * notice, this list of conditions and the following disclaimer.
9 * * Redistributions in binary form must reproduce the above copyright
10 * notice, this list of conditions and the following disclaimer in the
11 * documentation and/or other materials provided with the distribution.
12 * * Neither the name of the <organization> nor the
13 * names of its contributors may be used to endorse or promote products
14 * derived from this software without specific prior written permission.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
17 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
18 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
19 * DISCLAIMED. IN NO EVENT SHALL <COPYRIGHT HOLDER> BE LIABLE FOR ANY
20 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
21 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
22 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
23 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
25 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26 */
27 package com.salesforce.apt.graph.model.storage.classpath;
28
29 import java.io.File;
30 import java.io.IOException;
31 import java.net.MalformedURLException;
32 import java.net.URL;
33 import java.net.URLClassLoader;
34 import java.security.AccessController;
35 import java.security.PrivilegedAction;
36 import java.util.ArrayList;
37 import java.util.Enumeration;
38 import java.util.List;
39
40 import com.salesforce.apt.graph.model.storage.Resource;
41 import com.salesforce.apt.graph.model.storage.ResourceLoader;
42
43 public class ClasspathUrlResourceLoader implements ResourceLoader {
44
45 private ClassLoader classloader;
46
47 public ClasspathUrlResourceLoader() {
48 classloader = Thread.currentThread().getContextClassLoader();
49 }
50
51 public ClasspathUrlResourceLoader(ClassLoader classLoader) {
52 classloader = classLoader;
53 }
54
55 /**
56 * If the file exists, use it a root of the UrlClassLoader
57 * @param root file system directory to include as part of the classloader.
58 */
59 public ClasspathUrlResourceLoader(File root) {
60 AccessController.doPrivileged(new PrivilegedAction<Void>() {
61 public Void run() {
62 try {
63 classloader = new URLClassLoader(new URL[] {
64 root.toURI().toURL()},
65 Thread.currentThread().getContextClassLoader());
66 return null;
67 } catch (MalformedURLException | NullPointerException ex) {
68 throw new IllegalArgumentException("Invalide File, it's not a url.", ex);
69 }
70 }
71 });
72 }
73
74 /**
75 * Finds all possible resources that match the name of the DefinitionMode we wish to find.
76 *
77 * @param name the resources to search for with in the class loader.
78 * @return the list of found resources in class loader.
79 */
80 public List<Resource> getEntries(String name) {
81 List<Resource> output = new ArrayList<>();
82 try {
83 Enumeration<URL> resources = classloader.getResources(FileStore.STANDARD.getPath() + "/"
84 + name + ".json");
85 while (resources.hasMoreElements()) {
86 output.add(new UrlResource(resources.nextElement()));
87 }
88 } catch (IOException ioe) {
89 throw new IllegalStateException("Could not read classpath while looking for ForceInject models of name: " + name, ioe);
90 }
91 return output;
92 }
93
94 }