com.thoughtworks.gauge.ClasspathHelper Maven / Gradle / Ivy

/*----------------------------------------------------------------
 *  Copyright (c) ThoughtWorks, Inc.
 *  Licensed under the Apache License, Version 2.0
 *  See LICENSE.txt in the project root for license information.
 *----------------------------------------------------------------*/
package com.thoughtworks.gauge;

import java.net.URL;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.List;

public class ClasspathHelper {

    public static Collection getUrls() {
        final String packagesToScan = System.getenv(GaugeConstant.PACKAGE_TO_SCAN);
        if (packagesToScan != null) {
            Collection urls = new ArrayList<>();
            final List packages = Arrays.asList(packagesToScan.split(","));
            for (String packageToScan : packages) {
                urls.addAll(org.reflections.util.ClasspathHelper.forPackage(packageToScan.trim()));
            }
            return urls;
        }
        return org.reflections.util.ClasspathHelper.forJavaClassPath();
    }

}