using Semmle.Extraction.CSharp;
using Semmle.Util.Logging;
using Semmle.Autobuild.Shared;
using Semmle.Util;
using System.Linq;
using System.Collections.Generic;

namespace Semmle.Autobuild.CSharp
{
    /// <summary>
    /// Encapsulates C# build options.
    /// </summary>
    public class CSharpAutobuildOptions : AutobuildOptionsShared
    {
        private const string extractorOptionPrefix = "CODEQL_EXTRACTOR_CSHARP_OPTION_";

        public bool Buildless { get; }

        public override Language Language => Language.CSharp;


        /// <summary>
        /// Reads options from environment variables.
        /// Throws ArgumentOutOfRangeException for invalid arguments.
        /// </summary>
        public CSharpAutobuildOptions(IBuildActions actions) : base(actions)
        {
            Buildless = actions.GetEnvironmentVariable(lgtmPrefix + "BUILDLESS").AsBool("buildless", false) ||
                actions.GetEnvironmentVariable(extractorOptionPrefix + "BUILDLESS").AsBool("buildless", false);
        }
    }

    public class CSharpAutobuilder : Autobuilder<CSharpAutobuildOptions>
    {
        private const string buildCommandDocsUrl =
            "https://docs.github.com/en/code-security/code-scanning/automatically-scanning-your-code-for-vulnerabilities-and-errors/configuring-the-codeql-workflow-for-compiled-languages";

        private readonly AutoBuildRule autoBuildRule;

        private readonly DiagnosticClassifier diagnosticClassifier;

        protected override DiagnosticClassifier DiagnosticClassifier => diagnosticClassifier;

        public CSharpAutobuilder(IBuildActions actions, CSharpAutobuildOptions options) : base(actions, options)
        {
            this.autoBuildRule = new AutoBuildRule(this);
            this.diagnosticClassifier = new CSharpDiagnosticClassifier();
        }

        public override BuildScript GetBuildScript()
        {
            var attempt = BuildScript.Failure;
            switch (GetCSharpBuildStrategy())
            {
                case CSharpBuildStrategy.CustomBuildCommand:
                    attempt = new BuildCommandRule(DotNetRule.WithDotNet).Analyse(this, false) & CheckExtractorRun(true);
                    break;
                case CSharpBuildStrategy.Buildless:
                    // No need to check that the extractor has been executed in buildless mode
                    attempt = new StandaloneBuildRule().Analyse(this, false);
                    break;
                case CSharpBuildStrategy.MSBuild:
                    attempt = new MsBuildRule().Analyse(this, false) & CheckExtractorRun(true);
                    break;
                case CSharpBuildStrategy.DotNet:
                    attempt = new DotNetRule().Analyse(this, false) & CheckExtractorRun(true);
                    break;
                case CSharpBuildStrategy.Auto:
                    attempt =
                        // Attempt a few different build strategies to see if one works
                        this.autoBuildRule.Analyse(this, true) |
                        // All attempts failed: print message
                        AutobuildFailure();
                    break;
            }

            return attempt;
        }

        /// <summary>
        /// A script that checks that the C# extractor has been executed.
        /// </summary>
        public BuildScript CheckExtractorRun(bool warnOnFailure) =>
            BuildScript.Create(actions =>
            {
                if (actions.FileExists(Extractor.GetCSharpLogPath()))
                    return 0;

                if (warnOnFailure)
                    Log(Severity.Error, "No C# code detected during build.");

                return 1;
            });

        protected override void AutobuildFailureDiagnostic()
        {
            // if `ScriptPath` is not null here, the `BuildCommandAuto` rule was
            // run and found at least one script to execute
            if (this.autoBuildRule.BuildCommandAutoRule.ScriptPath is not null)
            {
                DiagnosticMessage message;

                // if we found multiple build scripts in the project directory, then we can say
                // as much to indicate that we may have picked the wrong one;
                // otherwise, we just report that the one script we found didn't work
                if (this.autoBuildRule.BuildCommandAutoRule.CandidatePaths.Count() > 1)
                {
                    message = MakeDiagnostic("multiple-build-scripts", "There are multiple potential build scripts");
                    message.MarkdownMessage =
                        "CodeQL found multiple potential build scripts for your project and " +
                        $"attempted to run `{autoBuildRule.BuildCommandAutoRule.ScriptPath}`, which failed. " +
                        "This may not be the right build script for your project. " +
                        $"Set up a [manual build command]({buildCommandDocsUrl}).";
                }
                else
                {
                    message = MakeDiagnostic("script-failure", "Unable to build project using build script");
                    message.MarkdownMessage =
                        "CodeQL attempted to build your project using a script located at " +
                        $"`{autoBuildRule.BuildCommandAutoRule.ScriptPath}`, which failed. " +
                        $"Set up a [manual build command]({buildCommandDocsUrl}).";
                }

                message.Severity = DiagnosticMessage.TspSeverity.Error;
                AddDiagnostic(message);
            }

            // both dotnet and msbuild builds require project or solution files; if we haven't found any
            // then neither of those rules would've worked
            if (this.ProjectsOrSolutionsToBuild.Count == 0)
            {
                var message = MakeDiagnostic("no-projects-or-solutions", "No project or solutions files found");
                message.PlaintextMessage =
                    "CodeQL could not find any project or solution files in your repository. " +
                    $"Set up a [manual build command]({buildCommandDocsUrl}).";
                message.Severity = DiagnosticMessage.TspSeverity.Error;

                AddDiagnostic(message);
            }
            else if (autoBuildRule.DotNetRule.NotDotNetProjects.Any())
            {
                var message = MakeDiagnostic("dotnet-incompatible-projects", "Some projects are incompatible with .NET Core");
                message.MarkdownMessage =
                    "CodeQL found some projects which cannot be built with .NET Core:\n" +
                    autoBuildRule.DotNetRule.NotDotNetProjects.ToMarkdownList(5);
                message.Severity = DiagnosticMessage.TspSeverity.Warning;

                AddDiagnostic(message);
            }

            // report any projects that failed to build with .NET Core
            if (autoBuildRule.DotNetRule.FailedProjectsOrSolutions.Any())
            {
                var message = MakeDiagnostic("dotnet-build-failure", "Some projects or solutions failed to build using .NET Core");
                message.MarkdownMessage =
                    "CodeQL was unable to build the following projects using .NET Core:\n" +
                    autoBuildRule.DotNetRule.FailedProjectsOrSolutions.ToMarkdownList(10) +
                    $"\nSet up a [manual build command]({buildCommandDocsUrl}).";
                message.Severity = DiagnosticMessage.TspSeverity.Error;

                AddDiagnostic(message);
            }

            // report any projects that failed to build with MSBuild
            if (autoBuildRule.MsBuildRule.FailedProjectsOrSolutions.Any())
            {
                var message = MakeDiagnostic("msbuild-build-failure", "Some projects or solutions failed to build using MSBuild");
                message.MarkdownMessage =
                    "CodeQL was unable to build the following projects using MSBuild:\n" +
                    autoBuildRule.MsBuildRule.FailedProjectsOrSolutions.ToMarkdownList(10) +
                    $"\nSet up a [manual build command]({buildCommandDocsUrl}).";
                message.Severity = DiagnosticMessage.TspSeverity.Error;

                AddDiagnostic(message);
            }
        }

        /// <summary>
        /// Gets the build strategy that the autobuilder should apply, based on the
        /// options in the `lgtm.yml` file.
        /// </summary>
        private CSharpBuildStrategy GetCSharpBuildStrategy()
        {
            if (Options.BuildCommand is not null)
                return CSharpBuildStrategy.CustomBuildCommand;

            if (Options.Buildless)
                return CSharpBuildStrategy.Buildless;

            if (Options.MsBuildArguments is not null
                || Options.MsBuildConfiguration is not null
                || Options.MsBuildPlatform is not null
                || Options.MsBuildTarget is not null)
            {
                return CSharpBuildStrategy.MSBuild;
            }

            if (Options.DotNetArguments is not null || Options.DotNetVersion is not null)
                return CSharpBuildStrategy.DotNet;

            return CSharpBuildStrategy.Auto;
        }

        private enum CSharpBuildStrategy
        {
            CustomBuildCommand,
            Buildless,
            MSBuild,
            DotNet,
            Auto
        }
    }
}