duke@1: /*
jjg@1733:  * Copyright (c) 2006, 2013, Oracle and/or its affiliates. All rights reserved.
duke@1:  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
duke@1:  *
duke@1:  * This code is free software; you can redistribute it and/or modify it
duke@1:  * under the terms of the GNU General Public License version 2 only, as
duke@1:  * published by the Free Software Foundation.
duke@1:  *
duke@1:  * This code is distributed in the hope that it will be useful, but WITHOUT
duke@1:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
duke@1:  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
duke@1:  * version 2 for more details (a copy is included in the LICENSE file that
duke@1:  * accompanied this code).
duke@1:  *
duke@1:  * You should have received a copy of the GNU General Public License version
duke@1:  * 2 along with this work; if not, write to the Free Software Foundation,
duke@1:  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
duke@1:  *
ohair@554:  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
ohair@554:  * or visit www.oracle.com if you need additional information or have any
ohair@554:  * questions.
duke@1:  */
duke@1: 
duke@1: import java.io.File;
duke@1: import java.io.IOException;
duke@1: import java.nio.charset.Charset;
jjg@1733: import java.util.*;
duke@1: import javax.tools.*;
duke@1: 
duke@1: import static javax.tools.StandardLocation.CLASS_PATH;
duke@1: import static javax.tools.StandardLocation.SOURCE_PATH;
duke@1: import static javax.tools.StandardLocation.CLASS_OUTPUT;
duke@1: 
duke@1: public class ToolTester {
duke@1:     public final File test_src     = new File(System.getProperty("test.src", "."));
duke@1:     public final File test_classes = new File(System.getProperty("test.classes", "."));
jjg@1733:     public final List<File> test_class_path = pathToFiles(System.getProperty("test.class.path"),
jjg@1733:                                      Arrays.asList(test_classes));
duke@1:     public final JavaCompiler tool = ToolProvider.getSystemJavaCompiler();
duke@1:     public final StandardJavaFileManager fm = getFileManager(tool, null, null);
duke@1:     public JavaCompiler.CompilationTask task = null;
duke@1:     public final StandardJavaFileManager getFileManager(JavaCompiler tool,
duke@1:                                                         DiagnosticListener<JavaFileObject> dl,
duke@1:                                                         Charset encoding) {
duke@1:         StandardJavaFileManager fm = tool.getStandardFileManager(dl, null, encoding);
duke@1:         try {
duke@1:             fm.setLocation(SOURCE_PATH,  Arrays.asList(test_src));
jjg@1733:             fm.setLocation(CLASS_PATH,   test_class_path);
duke@1:             fm.setLocation(CLASS_OUTPUT, Arrays.asList(test_classes));
duke@1:         } catch (IOException e) {
duke@1:             throw new AssertionError(e);
duke@1:         }
duke@1:         return fm;
duke@1:     }
jjg@1733: 
jjg@1733:     protected List<File> pathToFiles(String path, List<File> defaultPath) {
jjg@1733:         List<File> files = new ArrayList<>();
jjg@1733:         for (String f: path.split(File.pathSeparator)) {
jjg@1733:             if (f.isEmpty())
jjg@1733:                 continue;
jjg@1733:             File file = new File(f);
jjg@1733:             if (file.exists())
jjg@1733:                 files.add(file);
jjg@1733:         }
jjg@1733:         if (files.isEmpty())
jjg@1733:             files.addAll(defaultPath);
jjg@1733:         return files;
jjg@1733:     }
jjg@1733: 
jjg@1733:     protected <T> List<T> join(List<T> a, List<T> b) {
jjg@1733:         if (a.isEmpty())
jjg@1733:             return b;
jjg@1733:         if (b.isEmpty())
jjg@1733:             return a;
jjg@1733:         List<T> result = new ArrayList<>();
jjg@1733:         result.addAll(a);
jjg@1733:         result.addAll(b);
jjg@1733:         return result;
jjg@1733:     }
duke@1: }