import java.util.HashMap;
import java.util.Map;

import org.slf4j.event.Level;

/**
 * Responsible for keeping state of whether the threshold of the --fail-on-severity flag has been hit.
 */
public class LogLevelRecorder {
    private static final Map<String, Level> ACCEPTED_LEVELS = new HashMap<>();

    static {
        ACCEPTED_LEVELS.put("WARN", Level.WARN);
        ACCEPTED_LEVELS.put("WARNING", Level.WARN);

 */
public class MavenFailOnSeverityLogger extends MavenSimpleLogger {
    private final LogLevelRecorder logLevelRecorder;

    MavenFailOnSeverityLogger(String name, LogLevelRecorder logLevelRecorder) {
        super(name);
        this.logLevelRecorder = logLevelRecorder;
    }

    /**
     * A simple implementation which always logs messages of level WARN
     * according to the format outlined above.
     */

    private LogLevelRecorder logLevelRecorder = null;

    public MavenLoggerFactory() {}

    @Override
    public void setLogLevelRecorder(LogLevelRecorder logLevelRecorder) {
        if (this.logLevelRecorder != null) {
            throw new IllegalStateException("LogLevelRecorder has already been set.");
        }
        this.logLevelRecorder = logLevelRecorder;
        reset();
    }

    @Test
    void createsLogLevelRecorder() {
        LogLevelRecorder logLevelRecorder = new LogLevelRecorder("WARN");
        logLevelRecorder.record(Level.ERROR);

        assertTrue(logLevelRecorder.metThreshold());
    }

    @Test
    void failsOnLowerThanWarn() {
        assertThrows(IllegalArgumentException.class, () -> new LogLevelRecorder("INFO"));
    }

    @Test

        mavenLoggerFactory.setLogLevelRecorder(new LogLevelRecorder("ERROR"));

        assertTrue(mavenLoggerFactory.getLogLevelRecorder().isPresent());
        LogLevelRecorder logLevelRecorder =
                mavenLoggerFactory.getLogLevelRecorder().get();

        MavenFailOnSeverityLogger logger = (MavenFailOnSeverityLogger) mavenLoggerFactory.getLogger("Test");
        assertFalse(logLevelRecorder.metThreshold());

/**
 * Wrapper for creating loggers which can have a log level threshold.
 */
public interface MavenSlf4jWrapperFactory extends ILoggerFactory {
    void setLogLevelRecorder(LogLevelRecorder logLevelRecorder);

    Optional<LogLevelRecorder> getLogLevelRecorder();

import org.apache.maven.execution.ExecutionEvent;
import org.apache.maven.execution.MavenExecutionResult;
import org.apache.maven.execution.MavenSession;
import org.apache.maven.logwrapper.LogLevelRecorder;
import org.apache.maven.logwrapper.MavenSlf4jWrapperFactory;
import org.apache.maven.plugin.MojoExecution;
import org.apache.maven.plugin.descriptor.MojoDescriptor;
import org.apache.maven.project.MavenProject;

            if (slf4jLoggerFactory instanceof MavenSlf4jWrapperFactory) {
                LogLevelRecorder logLevelRecorder = new LogLevelRecorder(logLevelThreshold);
                ((MavenSlf4jWrapperFactory) slf4jLoggerFactory).setLogLevelRecorder(logLevelRecorder);
                slf4jLogger.info("Enabled to break the build on log level {}.", logLevelThreshold);
            } else {