mchung@1472: /*
jjg@1648:  * Copyright (c) 2012, 2013, Oracle and/or its affiliates. All rights reserved.
mchung@1472:  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
mchung@1472:  *
mchung@1472:  * This code is free software; you can redistribute it and/or modify it
mchung@1472:  * under the terms of the GNU General Public License version 2 only, as
mchung@1472:  * published by the Free Software Foundation.  Oracle designates this
mchung@1472:  * particular file as subject to the "Classpath" exception as provided
mchung@1472:  * by Oracle in the LICENSE file that accompanied this code.
mchung@1472:  *
mchung@1472:  * This code is distributed in the hope that it will be useful, but WITHOUT
mchung@1472:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
mchung@1472:  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
mchung@1472:  * version 2 for more details (a copy is included in the LICENSE file that
mchung@1472:  * accompanied this code).
mchung@1472:  *
mchung@1472:  * You should have received a copy of the GNU General Public License version
mchung@1472:  * 2 along with this work; if not, write to the Free Software Foundation,
mchung@1472:  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
mchung@1472:  *
mchung@1472:  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
mchung@1472:  * or visit www.oracle.com if you need additional information or have any
mchung@1472:  * questions.
mchung@1472:  */
mchung@1472: package com.sun.tools.jdeps;
mchung@1472: 
mchung@1472: import java.io.File;
mchung@1472: import java.io.IOException;
mchung@1472: import java.nio.file.FileVisitResult;
mchung@1472: import java.nio.file.Files;
mchung@1472: import java.nio.file.Path;
mchung@1472: import java.nio.file.SimpleFileVisitor;
mchung@1472: import java.nio.file.attribute.BasicFileAttributes;
mchung@1472: import java.util.*;
mchung@1472: 
mchung@1472: /**
mchung@1472:  * ClassPath for Java SE and JDK
mchung@1472:  */
mchung@1472: class PlatformClassPath {
mchung@1472:     private final static List<Archive> javaHomeArchives = init();
mchung@1472:     static List<Archive> getArchives() {
mchung@1472:         return javaHomeArchives;
mchung@1472:     }
mchung@1472: 
mchung@1472:     static boolean contains(Archive archive) {
mchung@1472:         return javaHomeArchives.contains(archive);
mchung@1472:     }
mchung@1472: 
mchung@1472:     private static List<Archive> init() {
mchung@1472:         List<Archive> result = new ArrayList<Archive>();
mchung@1472:         String javaHome = System.getProperty("java.home");
mchung@1472:         File jre = new File(javaHome, "jre");
mchung@1472:         File lib = new File(javaHome, "lib");
mchung@1472: 
mchung@1472:         try {
mchung@1472:             if (jre.exists() && jre.isDirectory()) {
mchung@1472:                 result.addAll(addJarFiles(new File(jre, "lib")));
mchung@1472:                 result.addAll(addJarFiles(lib));
mchung@1472:             } else if (lib.exists() && lib.isDirectory()) {
mchung@1472:                 // either a JRE or a jdk build image
mchung@1472:                 File classes = new File(javaHome, "classes");
mchung@1472:                 if (classes.exists() && classes.isDirectory()) {
mchung@1472:                     // jdk build outputdir
mchung@1472:                     result.add(new Archive(classes, ClassFileReader.newInstance(classes)));
mchung@1472:                 }
mchung@1472:                 // add other JAR files
mchung@1472:                 result.addAll(addJarFiles(lib));
mchung@1472:             } else {
mchung@1472:                 throw new RuntimeException("\"" + javaHome + "\" not a JDK home");
mchung@1472:             }
mchung@1472:         } catch (IOException e) {
mchung@1472:             throw new RuntimeException(e);
mchung@1472:         }
mchung@1472:         return result;
mchung@1472:     }
mchung@1472: 
mchung@1472:     private static List<Archive> addJarFiles(File f) throws IOException {
mchung@1472:         final List<Archive> result = new ArrayList<Archive>();
mchung@1472:         final Path root = f.toPath();
mchung@1472:         final Path ext = root.resolve("ext");
mchung@1472:         Files.walkFileTree(root, new SimpleFileVisitor<Path>() {
mchung@1472:             @Override
mchung@1472:             public FileVisitResult preVisitDirectory(Path dir, BasicFileAttributes attrs)
mchung@1472:                 throws IOException
mchung@1472:             {
mchung@1472:                 if (dir.equals(root) || dir.equals(ext)) {
mchung@1472:                     return FileVisitResult.CONTINUE;
mchung@1472:                 } else {
mchung@1472:                     // skip other cobundled JAR files
mchung@1472:                     return FileVisitResult.SKIP_SUBTREE;
mchung@1472:                 }
mchung@1472:             }
mchung@1472:             @Override
mchung@1472:             public FileVisitResult visitFile(Path file, BasicFileAttributes attrs)
mchung@1472:                 throws IOException
mchung@1472:             {
mchung@1472:                 File f = file.toFile();
mchung@1472:                 String fn = f.getName();
mchung@1472:                 if (fn.endsWith(".jar") && !fn.equals("alt-rt.jar")) {
mchung@1472:                     result.add(new Archive(f, ClassFileReader.newInstance(f)));
mchung@1472:                 }
mchung@1472:                 return FileVisitResult.CONTINUE;
mchung@1472:             }
mchung@1472:         });
mchung@1472:         return result;
mchung@1472:     }
mchung@1472: }