org.openstreetmap.atlas.checks.commands.AtlasChecksCommand Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of atlas-checks Show documentation

"Tool to run data checks over OSM Atlas data"

There is a newer version: 6.3.12

package org.openstreetmap.atlas.checks.commands;

import java.util.ArrayList;
import java.util.List;
import java.util.stream.Stream;

import org.openstreetmap.atlas.utilities.runtime.FlexibleCommand;
import org.openstreetmap.atlas.utilities.runtime.FlexibleSubCommand;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import io.github.classgraph.ClassGraph;
import io.github.classgraph.ClassInfoList;
import io.github.classgraph.ScanResult;

/**
 * Shell for running atlas-checks commands. Run this command with no arguments to learn more about
 * it.
 *
 * @author bbreithaupt
 */
public class AtlasChecksCommand extends FlexibleCommand
{
    private static final Logger logger = LoggerFactory.getLogger(AtlasChecksCommand.class);

    public static void main(final String... args)
    {
        final AtlasChecksCommand command = new AtlasChecksCommand(args);
        try
        {
            command.runWithoutQuitting(args);
        }
        catch (final RuntimeException e)
        {
            logger.error("Unable to runWithoutQuitting for class {} with message {}",
                    "AtlasChecksCommand", e.getMessage());
            command.printUsageAndExit(1);
        }
    }

    public AtlasChecksCommand(final String... args)
    {
        super(args);
    }

    @SuppressWarnings("unchecked")
    @Override
    protected Stream> getSupportedCommands()
    {
        final List> returnValue = new ArrayList<>();

        try (ScanResult scanResult = new ClassGraph().enableAllInfo()
                .whitelistPackages(AtlasChecksCommand.class.getPackage().getName()).scan())
        {
            final ClassInfoList classInfoList = scanResult
                    .getClassesImplementing(FlexibleSubCommand.class.getName());
            classInfoList.loadClasses()
                    .forEach(klass -> returnValue.add((Class) klass));
        }
        return returnValue.stream();
    }
}