// Copyright (c) 2026 Progmasoft.
// SPDX-License-Identifier: MPL-2.0 WITH AdditionRef-Progmasoft-Exception-2.0
#pragma once

#include <Progmasoft/Catch3/Detail/TypeTraits.hpp>
#include <Progmasoft/Catch3/Generator.hpp>
#include <Progmasoft/Catch3/ToString.hpp>
#include <cstddef>
#include <cstdint>
#include <exception>
#include <functional>
#include <optional>
#include <sstream>
#include <string>
#include <tuple>
#include <type_traits>
#include <utility>
#include <vector>

namespace Progmasoft::Catch3
{

    /// Controls a property run without changing the compatibility runner's CLI.
    struct PropertyOptions final
    {
        std::size_t Trials = 100;
        std::uint64_t Seed = 0xC47C4A53ULL;
        std::size_t MaxShrinkSteps = 1000;
        /// When set, runs exactly one trial with this seed instead of
        /// deriving Trials seeds from Seed. Use the "replay seed" of a
        /// reported failure to reproduce that single trial, including its
        /// shrinking, without rerunning the trials before it.
        std::optional<std::uint64_t> ReplayTrialSeed = std::nullopt;

        // The fields below are read by ForAll and the other strategy-based
        // checks; CheckProperty with a Generator ignores them.

        /// A replay token from a failure report: runs exactly the reported,
        /// minimized counterexample and nothing else.
        std::optional<std::string> Replay = std::nullopt;
        /// Trials that may be discarded, by a filter that found nothing or
        /// by CATCH3_PROPERTY_ASSUME, per requested trial before the
        /// property gives up.
        std::size_t MaxDiscardRatio = 10;
        /// Extra trials that search for a higher score after the regular
        /// ones, when the property calls Target. 0 disables the search.
        std::size_t TargetSteps = 500;
        /// Directory of the example database. Failing examples are stored
        /// there and replayed first on the next run. Empty: the directory
        /// named by CATCH3_PROPERTY_DATABASE, or no database when that is
        /// unset.
        std::string DatabasePath;
        /// Names the property in the database. The CATCH3_ macros fill it
        /// with the test case name and the source line.
        std::string Name;
        /// Adds the Classify and Cover statistics to a passing report.
        bool ShowStatistics = false;
    };

    enum class PropertyStatus : std::uint8_t
    {
        kPassed,
        kFailed,
        kErrored,
        kInvalidConfiguration,
        /// A Cover requirement was not met.
        kInsufficientCoverage,
        /// Too many trials were discarded to trust the result.
        kGaveUp,
    };

    /// How often one Classify or Cover label held.
    struct LabelStatistic final
    {
        std::string Label;
        std::size_t Count = 0;
        double Percentage = 0.0;
        /// The percentage a Cover call requires; 0 for Classify.
        double Required = 0.0;
    };

    struct PropertyFailure final
    {
        std::uint64_t RootSeed = 0;
        std::uint64_t TrialSeed = 0;
        std::size_t Trial = 0;
        std::size_t ShrinkSteps = 0;
        std::string Counterexample;
        std::string Reason;
        /// Replay token of the minimized counterexample; empty for
        /// CheckProperty with a Generator, whose replay is TrialSeed.
        std::string Replay;
        /// What the property recorded with Note for the counterexample.
        std::vector<std::string> Notes;
    };

    /// Stable, machine-readable outcome with a deterministic human-readable
    /// report.
    struct PropertyResult final
    {
        PropertyStatus Status = PropertyStatus::kPassed;
        std::size_t TrialsRun = 0;
        std::optional<PropertyFailure> Failure;
        std::string Error;
        std::size_t Discarded = 0;
        std::vector<LabelStatistic> Statistics;
        /// Set when Statistics should be part of a passing report.
        bool ShowStatistics = false;

        [[nodiscard]] bool
        Succeeded() const noexcept
        {
            return Status == PropertyStatus::kPassed;
        }

        [[nodiscard]] std::string
        Describe() const
        {
            std::ostringstream output;
            switch (Status)
            {
                case PropertyStatus::kPassed:
                    output << "property passed: " << TrialsRun << " trials";
                    if (Discarded != 0)
                    {
                        output << ", " << Discarded << " discarded";
                    }
                    if (ShowStatistics)
                    {
                        DescribeStatistics(output);
                    }
                    return output.str();
                case PropertyStatus::kInsufficientCoverage:
                    output << "property coverage insufficient after "
                           << TrialsRun << " trials: " << Error;
                    DescribeStatistics(output);
                    return output.str();
                case PropertyStatus::kGaveUp:
                    output << "property gave up after " << TrialsRun
                           << " trials: " << Error;
                    return output.str();
                case PropertyStatus::kInvalidConfiguration:
                    output << "invalid property configuration: " << Error;
                    return output.str();
                case PropertyStatus::kErrored:
                    output << "property execution errored after " << TrialsRun
                           << " trials: " << Error;
                    break;
                case PropertyStatus::kFailed:
                    output << "property failed after " << TrialsRun
                           << " trials";
                    break;
            }

            if (Failure.has_value() && Failure->Replay.empty())
            {
                output << "; failing trial " << Failure->Trial << ", root seed "
                       << Failure->RootSeed << ", replay seed "
                       << Failure->TrialSeed << ", shrink checks "
                       << Failure->ShrinkSteps
                       << "\ncounterexample: " << Failure->Counterexample
                       << "\nreason: " << Failure->Reason
                       << "\nreplay: set PropertyOptions::ReplayTrialSeed to "
                       << Failure->TrialSeed;
            }
            else if (Failure.has_value())
            {
                output << "; failing trial " << Failure->Trial << ", root seed "
                       << Failure->RootSeed << ", shrink checks "
                       << Failure->ShrinkSteps
                       << "\ncounterexample: " << Failure->Counterexample
                       << "\nreason: " << Failure->Reason;
                for (const auto &note : Failure->Notes)
                {
                    output << "\nnote: " << note;
                }
                output << "\nreplay: set PropertyOptions::Replay to \""
                       << Failure->Replay << '"';
            }
            return output.str();
        }

    private:
        void
        DescribeStatistics(std::ostringstream &output) const
        {
            for (const auto &statistic : Statistics)
            {
                output << "\n  " << statistic.Label << ": " << statistic.Count
                       << " ("
                       << static_cast<double>(static_cast<long long>(
                              statistic.Percentage * 10.0 + 0.5))
                              / 10.0
                       << "%";
                if (statistic.Required > 0.0)
                {
                    output << ", required " << statistic.Required << "%";
                }
                output << ')';
            }
        }
    };

    namespace Detail
    {

        /// The failure of one trial of CheckProperty with a Generator, which
        /// has no replay token and no notes of its own.
        [[nodiscard]] inline PropertyFailure
        TrialFailure(std::uint64_t rootSeed,
                     std::uint64_t trialSeed,
                     std::size_t trial,
                     std::size_t shrinkSteps,
                     std::string counterexample,
                     std::string reason)
        {
            PropertyFailure failure;
            failure.RootSeed = rootSeed;
            failure.TrialSeed = trialSeed;
            failure.Trial = trial;
            failure.ShrinkSteps = shrinkSteps;
            failure.Counterexample = std::move(counterexample);
            failure.Reason = std::move(reason);
            return failure;
        }

        template<typename ValueType>
        [[nodiscard]] std::string
        StringifyPropertyValue(const ValueType &value);

        /// Renders one counterexample. Scalars use Catch2's StringMaker, so
        /// user specializations apply. Tuples, pairs and vectors are
        /// rendered here, element by element, because Catch2 only prints
        /// tuples when a configuration macro is defined before its headers
        /// and a counterexample must not degrade to "{?}" depending on
        /// include order.
        template<typename ValueType>
        struct PropertyValuePrinter final
        {
            [[nodiscard]] static std::string
            Print(const ValueType &value)
            {
                return Catch::StringMaker<ValueType>::convert(value);
            }
        };

        template<typename... ValueTypes>
        struct PropertyValuePrinter<std::tuple<ValueTypes...>> final
        {
            [[nodiscard]] static std::string
            Print(const std::tuple<ValueTypes...> &value)
            {
                std::string text = "{";
                bool first = true;
                std::apply(
                    [&text, &first](const ValueTypes &...component) {
                        const auto append = [&text, &first](std::string item) {
                            text += first ? " " : ", ";
                            text += item;
                            first = false;
                        };
                        (append(StringifyPropertyValue(component)), ...);
                    },
                    value);
                text += first ? "}" : " }";
                return text;
            }
        };

        template<typename FirstType, typename SecondType>
        struct PropertyValuePrinter<std::pair<FirstType, SecondType>> final
        {
            [[nodiscard]] static std::string
            Print(const std::pair<FirstType, SecondType> &value)
            {
                return "{ " + StringifyPropertyValue(value.first) + ", "
                       + StringifyPropertyValue(value.second) + " }";
            }
        };

        template<typename ElementType, typename AllocatorType>
        struct PropertyValuePrinter<
            std::vector<ElementType, AllocatorType>> final
        {
            [[nodiscard]] static std::string
            Print(const std::vector<ElementType, AllocatorType> &value)
            {
                std::string text = "{";
                bool first = true;
                for (const auto &element : value)
                {
                    text += first ? " " : ", ";
                    // The cast turns std::vector<bool>'s proxy into a value.
                    text += StringifyPropertyValue(
                        static_cast<ElementType>(element));
                    first = false;
                }
                text += first ? "}" : " }";
                return text;
            }
        };

        template<typename ValueType>
        [[nodiscard]] std::string
        StringifyPropertyValue(const ValueType &value)
        {
            return PropertyValuePrinter<RemoveCvRef<ValueType>>::Print(value);
        }

        template<typename Predicate, typename ValueType>
        [[nodiscard]] std::optional<std::string>
        EvaluateProperty(Predicate &predicate, const ValueType &value)
        {
#if defined(__cpp_exceptions)
            try
            {
#endif
                if (static_cast<bool>(std::invoke(predicate, value)))
                {
                    return std::nullopt;
                }
                return "predicate returned false";
#if defined(__cpp_exceptions)
            }
            catch (const std::exception &exception)
            {
                return std::string("predicate threw std::exception: ")
                       + exception.what();
            }
            catch (...)
            {
                return "predicate threw a non-standard exception";
            }
#endif
        }

        template<typename ValueType>
        [[nodiscard]] std::string
        DescribeValue(const ValueType &value) noexcept
        {
#if defined(__cpp_exceptions)
            try
            {
#endif
                return StringifyPropertyValue(value);
#if defined(__cpp_exceptions)
            }
            catch (const std::exception &exception)
            {
                return std::string("<stringification failed: ")
                       + exception.what() + ">";
            }
            catch (...)
            {
                return "<stringification failed with a non-standard exception>";
            }
#endif
        }

    } // namespace Detail

    /// Runs independently seeded trials and greedily accepts failing shrink
    /// candidates.
    template<typename ValueType,
             typename Predicate,
             std::enable_if_t<
                 std::is_invocable_r_v<bool, Predicate &, const ValueType &>,
                 int> = 0>
    [[nodiscard]] PropertyResult
    CheckProperty(const Generator<ValueType> &generator,
                  Predicate predicate,
                  PropertyOptions options = {})
    {
        if (options.Trials == 0)
        {
            PropertyResult invalid;
            invalid.Status = PropertyStatus::kInvalidConfiguration;
            invalid.Error = "Trials must be greater than zero";
            return invalid;
        }

        PropertyResult result;
        const std::size_t trials
            = options.ReplayTrialSeed.has_value() ? 1 : options.Trials;
        for (std::size_t trial = 0; trial < trials; ++trial)
        {
            result.TrialsRun = trial + 1;
            const std::uint64_t trialSeed
                = options.ReplayTrialSeed.has_value()
                      ? *options.ReplayTrialSeed
                      : Random::DeriveSeed(options.Seed, trial);
            Random random(trialSeed);
            std::optional<ValueType> sample;

#if defined(__cpp_exceptions)
            try
            {
#endif
                sample.emplace(generator.Generate(random));
#if defined(__cpp_exceptions)
            }
            catch (const std::exception &exception)
            {
                result.Status = PropertyStatus::kErrored;
                result.Error = std::string("generator threw std::exception: ")
                               + exception.what();
                result.Failure = Detail::TrialFailure(options.Seed,
                                                      trialSeed,
                                                      trial + 1,
                                                      0,
                                                      "<not generated>",
                                                      result.Error);
                return result;
            }
            catch (...)
            {
                result.Status = PropertyStatus::kErrored;
                result.Error = "generator threw a non-standard exception";
                result.Failure = Detail::TrialFailure(options.Seed,
                                                      trialSeed,
                                                      trial + 1,
                                                      0,
                                                      "<not generated>",
                                                      result.Error);
                return result;
            }
#endif

            auto reason = Detail::EvaluateProperty(predicate, *sample);
            if (!reason.has_value())
            {
                continue;
            }

            std::size_t shrinkChecks = 0;
#if defined(__cpp_exceptions)
            try
            {
#endif
                bool improved = true;
                while (improved && shrinkChecks < options.MaxShrinkSteps)
                {
                    improved = false;
                    const std::vector<ValueType> candidates
                        = generator.Shrink(*sample);
                    for (std::size_t candidateIndex = 0;
                         candidateIndex < candidates.size();
                         ++candidateIndex)
                    {
                        if (shrinkChecks >= options.MaxShrinkSteps)
                        {
                            break;
                        }
                        ++shrinkChecks;
                        const ValueType candidate = static_cast<ValueType>(
                            candidates[candidateIndex]);
                        auto candidateReason
                            = Detail::EvaluateProperty(predicate, candidate);
                        if (candidateReason.has_value())
                        {
                            sample.emplace(candidate);
                            reason = std::move(candidateReason);
                            improved = true;
                            break;
                        }
                    }
                }
#if defined(__cpp_exceptions)
            }
            catch (const std::exception &exception)
            {
                result.Status = PropertyStatus::kErrored;
                result.Error = std::string("shrinker threw std::exception: ")
                               + exception.what();
                result.Failure
                    = Detail::TrialFailure(options.Seed,
                                           trialSeed,
                                           trial + 1,
                                           shrinkChecks,
                                           Detail::DescribeValue(*sample),
                                           *reason);
                return result;
            }
            catch (...)
            {
                result.Status = PropertyStatus::kErrored;
                result.Error = "shrinker threw a non-standard exception";
                result.Failure
                    = Detail::TrialFailure(options.Seed,
                                           trialSeed,
                                           trial + 1,
                                           shrinkChecks,
                                           Detail::DescribeValue(*sample),
                                           *reason);
                return result;
            }
#endif

            result.Status = PropertyStatus::kFailed;
            result.Failure
                = Detail::TrialFailure(options.Seed,
                                       trialSeed,
                                       trial + 1,
                                       shrinkChecks,
                                       Detail::DescribeValue(*sample),
                                       *reason);
            return result;
        }

        result.Status = PropertyStatus::kPassed;
        return result;
    }

} // namespace Progmasoft::Catch3