ArtifactResolver.java revision 2735:2e42fcc06344 
1/*
2 * Copyright (c) 2017, Oracle and/or its affiliates. All rights reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation.
8 *
9 * This code is distributed in the hope that it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
12 * version 2 for more details (a copy is included in the LICENSE file that
13 * accompanied this code).
14 *
15 * You should have received a copy of the GNU General Public License version
16 * 2 along with this work; if not, write to the Free Software Foundation,
17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18 *
19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20 * or visit www.oracle.com if you need additional information or have any
21 * questions.
22 */
23
24package jdk.test.lib.artifacts;
25
26import java.io.FileNotFoundException;
27import java.nio.file.Path;
28import java.util.HashMap;
29import java.util.Map;
30
31public class ArtifactResolver {
32    public static Map<String, Path> resolve(Class<?> klass) throws FileNotFoundException {
33        ArtifactManager manager = new DefaultArtifactManager();
34        try {
35            String managerName = System.getProperty("jdk.test.lib.artifacts.artifactmanager");
36            if (managerName != null) {
37                manager = (ArtifactManager) Class.forName(managerName).newInstance();
38            } else {
39                manager = JibArtifactManager.newInstance();
40            }
41        } catch (Exception e) {
42            // If we end up here, we'll use the DefaultArtifactManager
43        }
44
45        ArtifactContainer artifactContainer = klass.getAnnotation(ArtifactContainer.class);
46        HashMap<String, Path> locations = new HashMap<>();
47        Artifact[] artifacts;
48
49        if (artifactContainer == null) {
50            artifacts = new Artifact[]{klass.getAnnotation(Artifact.class)};
51        } else {
52            artifacts = artifactContainer.value();
53        }
54        for (Artifact artifact : artifacts) {
55            locations.put(artifactName(artifact), manager.resolve(artifact));
56        }
57
58        return locations;
59    }
60
61    private static String artifactName(Artifact artifact) {
62        // Format of the artifact name is <organization>.<name>-<revision>(-<classifier>)
63        String name = String.format("%s.%s-%s", artifact.organization(), artifact.name(), artifact.revision());
64        if (artifact.classifier().length() != 0) {
65            name = name +"-" + artifact.classifier();
66        }
67        return name;
68    }
69}
70