/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */
package org.apache.maven.plugins.enforcer;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.Optional;
import java.util.concurrent.ConcurrentHashMap;
import java.util.stream.Collectors;

import org.apache.maven.enforcer.rule.api.AbstractEnforcerRule;
import org.apache.maven.enforcer.rule.api.AbstractEnforcerRuleConfigProvider;
import org.apache.maven.enforcer.rule.api.EnforcerLevel;
import org.apache.maven.enforcer.rule.api.EnforcerRule;
import org.apache.maven.enforcer.rule.api.EnforcerRuleBase;
import org.apache.maven.enforcer.rule.api.EnforcerRuleError;
import org.apache.maven.enforcer.rule.api.EnforcerRuleException;
import org.apache.maven.enforcer.rule.api.EnforcerRuleHelper;
import org.apache.maven.execution.MavenSession;
import org.apache.maven.plugin.AbstractMojo;
import org.apache.maven.plugin.MojoExecution;
import org.apache.maven.plugin.MojoExecutionException;
import org.apache.maven.plugin.PluginParameterExpressionEvaluator;
import org.apache.maven.plugin.logging.Log;
import org.apache.maven.plugins.annotations.Component;
import org.apache.maven.plugins.annotations.LifecyclePhase;
import org.apache.maven.plugins.annotations.Mojo;
import org.apache.maven.plugins.annotations.Parameter;
import org.apache.maven.plugins.annotations.ResolutionScope;
import org.apache.maven.plugins.enforcer.internal.DefaultEnforcementRuleHelper;
import org.apache.maven.plugins.enforcer.internal.EnforcerRuleCache;
import org.apache.maven.plugins.enforcer.internal.EnforcerRuleDesc;
import org.apache.maven.plugins.enforcer.internal.EnforcerRuleManager;
import org.apache.maven.plugins.enforcer.internal.EnforcerRuleManagerException;
import org.apache.maven.project.MavenProject;
import org.codehaus.plexus.PlexusContainer;
import org.codehaus.plexus.configuration.DefaultPlexusConfiguration;
import org.codehaus.plexus.configuration.PlexusConfiguration;
import org.codehaus.plexus.configuration.xml.XmlPlexusConfiguration;
import org.codehaus.plexus.util.StringUtils;

/**
 * This goal executes the defined enforcer-rules once per module.
 *
 * @author <a href="mailto:brianf@apache.org">Brian Fox</a>
 */
@Mojo(
        name = "enforce",
        defaultPhase = LifecyclePhase.VALIDATE,
        requiresDependencyCollection = ResolutionScope.TEST,
        threadSafe = true)
public class EnforceMojo extends AbstractMojo {
    /**
     * This is a static variable used to persist the cached results across plugin invocations.
     */
    protected static Map<String, EnforcerRule> cache = new ConcurrentHashMap<>();

    /**
     * MojoExecution needed by the ExpressionEvaluator
     */
    @Parameter(defaultValue = "${mojoExecution}", readonly = true, required = true)
    protected MojoExecution mojoExecution;

    /**
     * The MavenSession
     */
    @Parameter(defaultValue = "${session}", readonly = true, required = true)
    protected MavenSession session;

    /**
     * POM
     */
    @Parameter(defaultValue = "${project}", readonly = true, required = true)
    protected MavenProject project;

    /**
     * Flag to easily skip all checks
     */
    @Parameter(property = "enforcer.skip", defaultValue = "false")
    protected boolean skip = false;

    /**
     * Flag to fail the build if at least one check fails.
     */
    @Parameter(property = "enforcer.fail", defaultValue = "true")
    private boolean fail = true;

    /**
     * Fail on the first rule that doesn't pass
     */
    @Parameter(property = "enforcer.failFast", defaultValue = "false")
    private boolean failFast = false;

    /**
     * Flag to fail the build if no rules are present
     *
     * @since 3.2.0
     */
    @Parameter(property = "enforcer.failIfNoRules", defaultValue = "true")
    private boolean failIfNoRules = true;

    /**
     * Rules configuration to execute as XML.
     * Each first level tag represents rule name to execute.
     * Inner tags are configurations for rule.
     * Eg:
     * <pre>
     *     &lt;rules&gt;
     *         &lt;alwaysFail/&gt;
     *         &lt;alwaysPass&gt;
     *             &lt;message&gt;message for rule&lt;/message&gt;
     *         &lt;/alwaysPass&gt;
     *         &lt;myRule implementation="org.example.MyRule"/&gt;
     *     &lt;/rules&gt;
     * </pre>
     *
     * @since 1.0.0
     */
    @Parameter
    private PlexusConfiguration rules;

    /**
     * List of strings that matches the EnforcerRules to skip.
     *
     * @since 3.2.0
     */
    @Parameter(required = false, property = "enforcer.skipRules")
    private List<String> rulesToSkip;

    /**
     * Use this flag to disable rule result caching. This will cause all rules to execute on each project even if the
     * rule indicates it can safely be cached.
     */
    @Parameter(property = "enforcer.ignoreCache", defaultValue = "false")
    protected boolean ignoreCache = false;

    @Component
    private PlexusContainer container;

    @Component
    private EnforcerRuleManager enforcerRuleManager;

    @Component
    private EnforcerRuleCache ruleCache;

    private List<String> rulesToExecute;

    /**
     * List of strings that matches the EnforcerRules to execute. Replacement for the <code>rules</code> property.
     *
     * @param rulesToExecute a rules to execute
     * @throws MojoExecutionException when values are incorrect
     * @since 3.2.0
     */
    @Parameter(required = false, property = "enforcer.rules")
    public void setRulesToExecute(List<String> rulesToExecute) throws MojoExecutionException {
        if (rulesToExecute != null && !rulesToExecute.isEmpty()) {
            if (this.rulesToExecute != null && !this.rulesToExecute.isEmpty()) {
                throw new MojoExecutionException("Detected the usage of both '-Drules' (which is deprecated) "
                        + "and '-Denforcer.rules'. Please use only one of them, preferably '-Denforcer.rules'.");
            }
            this.rulesToExecute = rulesToExecute;
        }
    }

    /**
     * List of strings that matches the EnforcerRules to execute.
     *
     * @param rulesToExecute a rules to execute
     * @throws MojoExecutionException when values are incorrect
     * @deprecated Use <code>enforcer.rules</code> property instead
     */
    @Parameter(required = false, property = "rules")
    @Deprecated
    public void setCommandLineRules(List<String> rulesToExecute) throws MojoExecutionException {
        if (rulesToExecute != null && !rulesToExecute.isEmpty()) {
            getLog().warn(
                            "Detected the usage of property '-Drules' which is deprecated. Use '-Denforcer.rules' instead.");
        }
        setRulesToExecute(rulesToExecute);
    }

    @Override
    public void execute() throws MojoExecutionException {
        Log log = this.getLog();

        if (skip) {
            log.info("Skipping Rule Enforcement.");
            return;
        }

        Optional<PlexusConfiguration> rulesFromCommandLine = createRulesFromCommandLineOptions();
        List<EnforcerRuleDesc> rulesList;

        // current behavior - rules from command line override all other configured rules.
        List<EnforcerRuleDesc> allRules = enforcerRuleManager.createRules(rulesFromCommandLine.orElse(rules), log);
        rulesList = filterOutSkippedRules(allRules);

        List<EnforcerRuleDesc> additionalRules = processRuleConfigProviders(rulesList);
        rulesList = filterOutRuleConfigProviders(rulesList);
        rulesList.addAll(additionalRules);

        if (rulesList.isEmpty()) {
            if (failIfNoRules) {
                throw new MojoExecutionException(
                        "No rules are configured. Use the skip flag if you want to disable execution.");
            } else {
                log.warn("No rules are configured.");
                return;
            }
        }

        // create my helper
        PluginParameterExpressionEvaluator evaluator = new PluginParameterExpressionEvaluator(session, mojoExecution);
        EnforcerRuleHelper helper = new DefaultEnforcementRuleHelper(session, evaluator, log, container);

        // if we are only warning, then disable
        // failFast
        if (!fail) {
            failFast = false;
        }

        List<String> errorMessages = new ArrayList<>();

        // go through each rule
        for (int ruleIndex = 0; ruleIndex < rulesList.size(); ruleIndex++) {

            EnforcerRuleDesc ruleDesc = rulesList.get(ruleIndex);
            EnforcerLevel level = ruleDesc.getLevel();
            try {
                executeRule(ruleIndex, ruleDesc, helper);
            } catch (EnforcerRuleError e) {
                String ruleMessage = createRuleMessage(ruleIndex, ruleDesc, EnforcerLevel.ERROR, e);
                throw new MojoExecutionException(System.lineSeparator() + ruleMessage, e);
            } catch (EnforcerRuleException e) {

                String ruleMessage = createRuleMessage(ruleIndex, ruleDesc, level, e);

                if (failFast && level == EnforcerLevel.ERROR) {
                    throw new MojoExecutionException(System.lineSeparator() + ruleMessage, e);
                }

                if (level == EnforcerLevel.ERROR) {
                    errorMessages.add(ruleMessage);
                } else {
                    log.warn(ruleMessage);
                }
            }
        }

        if (!errorMessages.isEmpty()) {
            if (fail) {
                throw new MojoExecutionException(
                        System.lineSeparator() + String.join(System.lineSeparator(), errorMessages));
            } else {
                errorMessages.forEach(log::warn);
            }
        }
    }

    private List<EnforcerRuleDesc> processRuleConfigProviders(List<EnforcerRuleDesc> rulesList) {
        return rulesList.stream()
                .filter(Objects::nonNull)
                .filter(rd -> rd.getRule() instanceof AbstractEnforcerRuleConfigProvider)
                .map(this::executeRuleConfigProvider)
                .flatMap(xml -> enforcerRuleManager.createRules(xml, getLog()).stream())
                .collect(Collectors.toList());
    }

    private List<EnforcerRuleDesc> filterOutRuleConfigProviders(List<EnforcerRuleDesc> rulesList) {
        return rulesList.stream()
                .filter(Objects::nonNull)
                .filter(rd -> !(rd.getRule() instanceof AbstractEnforcerRuleConfigProvider))
                .collect(Collectors.toList());
    }

    private XmlPlexusConfiguration executeRuleConfigProvider(EnforcerRuleDesc ruleDesc) {
        AbstractEnforcerRuleConfigProvider ruleProducer = (AbstractEnforcerRuleConfigProvider) ruleDesc.getRule();

        if (getLog().isDebugEnabled()) {
            getLog().debug(String.format("Executing Rule Config Provider %s", ruleDesc.getRule()));
        }

        try {
            XmlPlexusConfiguration configuration = new XmlPlexusConfiguration(ruleProducer.getRulesConfig());
            getLog().info(String.format("Rule Config Provider %s executed", getRuleName(ruleDesc)));

            return configuration;
        } catch (EnforcerRuleException e) {
            throw new EnforcerRuleManagerException("Rules Provider error for: " + getRuleName(ruleDesc), e);
        }
    }

    private void executeRule(int ruleIndex, EnforcerRuleDesc ruleDesc, EnforcerRuleHelper helper)
            throws EnforcerRuleException {

        if (getLog().isDebugEnabled()) {
            getLog().debug(String.format("Executing Rule %d: %s", ruleIndex, ruleDesc));
        }

        long startTime = System.currentTimeMillis();

        try {
            if (ruleDesc.getRule() instanceof EnforcerRule) {
                executeRuleOld(ruleIndex, ruleDesc, helper);
            } else if (ruleDesc.getRule() instanceof AbstractEnforcerRule) {
                executeRuleNew(ruleIndex, ruleDesc);
            }
        } finally {
            if (getLog().isDebugEnabled()) {
                long workTime = System.currentTimeMillis() - startTime;
                getLog().debug(String.format(
                        "Finish Rule %d: %s takes %d ms", ruleIndex, getRuleName(ruleDesc), workTime));
            }
        }
    }

    private void executeRuleOld(int ruleIndex, EnforcerRuleDesc ruleDesc, EnforcerRuleHelper helper)
            throws EnforcerRuleException {

        EnforcerRule rule = (EnforcerRule) ruleDesc.getRule();

        if (ignoreCache || shouldExecute(rule)) {
            rule.execute(helper);
            getLog().info(String.format("Rule %d: %s passed", ruleIndex, getRuleName(ruleDesc)));
        }
    }

    private void executeRuleNew(int ruleIndex, EnforcerRuleDesc ruleDesc) throws EnforcerRuleException {

        AbstractEnforcerRule rule = (AbstractEnforcerRule) ruleDesc.getRule();
        if (ignoreCache || !ruleCache.isCached(rule)) {
            rule.execute();
            getLog().info(String.format("Rule %d: %s passed", ruleIndex, getRuleName(ruleDesc)));
        }
    }

    /**
     * Create rules configuration based on command line provided rules list.
     *
     * @return a configuration in case where rules list is present or empty
     */
    private Optional<PlexusConfiguration> createRulesFromCommandLineOptions() {

        if (rulesToExecute == null || rulesToExecute.isEmpty()) {
            return Optional.empty();
        }

        PlexusConfiguration configuration = new DefaultPlexusConfiguration("rules");
        for (String rule : rulesToExecute) {
            PlexusConfiguration configuredRule = null;
            // Check if there's configuration in the project for this rule and use it if so
            if (rules != null) {
                // rule names haven't been normalized yet, so check both with first character upper and lower
                String ruleLower = Character.toLowerCase(rule.charAt(0)) + rule.substring(1);
                String ruleUpper = Character.toUpperCase(rule.charAt(0)) + rule.substring(1);
                configuredRule = rules.getChild(ruleLower, false);
                if (configuredRule == null) {
                    configuredRule = rules.getChild(ruleUpper, false);
                }
            }

            if (configuredRule != null) {
                configuration.addChild(configuredRule);
            } else {
                configuration.addChild(new DefaultPlexusConfiguration(rule));
            }
        }
        return Optional.of(configuration);
    }

    /**
     * Filter out (remove) rules that have been specifically skipped via additional configuration.
     *
     * @param allRules list of enforcer rules to go through and filter
     * @return list of filtered rules
     */
    private List<EnforcerRuleDesc> filterOutSkippedRules(List<EnforcerRuleDesc> allRules) {
        if (rulesToSkip == null || rulesToSkip.isEmpty()) {
            return allRules;
        }
        return allRules.stream()
                .filter(ruleDesc -> !rulesToSkip.contains(ruleDesc.getName()))
                .collect(Collectors.toList());
    }

    /**
     * This method determines if a rule should execute based on the cache
     *
     * @param rule the rule to verify
     * @return {@code true} if rule should be executed, otherwise {@code false}
     */
    protected boolean shouldExecute(EnforcerRule rule) {
        if (rule.isCacheable()) {
            Log log = this.getLog();
            log.debug("Rule " + rule.getClass().getName() + " is cacheable.");
            String key = rule.getClass().getName() + " " + rule.getCacheId();
            if (EnforceMojo.cache.containsKey(key)) {
                log.debug("Key " + key + " was found in the cache");
                if (rule.isResultValid(cache.get(key))) {
                    log.debug("The cached results are still valid. Skipping the rule: "
                            + rule.getClass().getName());
                    return false;
                }
            }

            // add it to the cache of executed rules
            EnforceMojo.cache.put(key, rule);
        }
        return true;
    }

    /**
     * Set rule list to skip.
     *
     * @param rulesToSkip a rule list
     */
    public void setRulesToSkip(List<String> rulesToSkip) {
        if (rulesToSkip == null) {
            return;
        }
        // internally all rules begin from lowercase letter
        this.rulesToSkip = rulesToSkip.stream()
                .filter(Objects::nonNull)
                .map(StringUtils::lowercaseFirstLetter)
                .collect(Collectors.toList());
    }

    /**
     * @param fail whether to fail
     */
    public void setFail(boolean fail) {
        this.fail = fail;
    }

    /**
     * @param failFast whether to fail fast
     */
    public void setFailFast(boolean failFast) {
        this.failFast = failFast;
    }

    private String createRuleMessage(
            int ruleIndex, EnforcerRuleDesc ruleDesc, EnforcerLevel level, EnforcerRuleException e) {

        StringBuilder result = new StringBuilder();
        result.append("Rule ").append(ruleIndex).append(": ").append(getRuleName(ruleDesc));

        if (level == EnforcerLevel.ERROR) {
            result.append(" failed");
        } else {
            result.append(" warned");
        }

        if (e.getMessage() != null) {
            result.append(" with message:").append(System.lineSeparator()).append(e.getMessage());
        } else {
            result.append(" without a message");
        }

        return result.toString();
    }

    private String getRuleName(EnforcerRuleDesc ruleDesc) {

        Class<? extends EnforcerRuleBase> ruleClass = ruleDesc.getRule().getClass();

        String ruleName = ruleClass.getName();

        if (!ruleClass.getSimpleName().equalsIgnoreCase(ruleDesc.getName())) {
            ruleName += "(" + ruleDesc.getName() + ")";
        }

        return ruleName;
    }

    /**
     * @param failIfNoRules whether to fail if there are no rules
     */
    public void setFailIfNoRules(boolean failIfNoRules) {
        this.failIfNoRules = failIfNoRules;
    }
}