using BenchmarkDotNet.Configs;
using BenchmarkDotNet.Jobs;
using BenchmarkDotNet.Loggers;
using BenchmarkDotNet.Running;
using BenchmarkDotNet.Toolchains.InProcess.Emit;
using BenchmarkTests;
using Perfolizer.Horology;

// Parse command-line arguments
var runHttpBenchmarks = args.Contains("--http") || args.Contains("-h");
var runInProcessBenchmarks = args.Contains("--inprocess") || args.Contains("-i");
var runAll = args.Contains("--all") || args.Contains("-a");

if (args.Contains("--pgconv-only"))
{
    BenchmarkRunner.Run<PgConverterAndWriterBenchmarks>();
    return;
}

// Default: run in-process benchmarks if no arguments provided
if (!runHttpBenchmarks && !runInProcessBenchmarks && !runAll)
{
    runInProcessBenchmarks = true;
}

// In-process benchmarks (no external server required)
if (runInProcessBenchmarks || runAll)
{
    Console.WriteLine("=== Running In-Process Benchmarks ===");
    Console.WriteLine("These benchmarks test NpgsqlRest internals without requiring a running server.\n");

    BenchmarkRunner.Run<TypeCategoryBenchmarks>();
    BenchmarkRunner.Run<ParameterParserBenchmarks>();
    BenchmarkRunner.Run<SerializationBenchmarks>();
    BenchmarkRunner.Run<PgConverterAndWriterBenchmarks>();
}

// HTTP benchmarks (require external server)
if (runHttpBenchmarks || runAll)
{
    Console.WriteLine("\n=== Running HTTP Endpoint Benchmarks ===");
    Console.WriteLine("IMPORTANT: These benchmarks require the NpgsqlRestTests server to be running.");
    Console.WriteLine("Start the server with: dotnet run --project NpgsqlRestTests/Setup\n");

    Console.WriteLine("Press ENTER to continue (or Ctrl+C to cancel)...");
    Console.ReadLine();

    var httpConfig = DefaultConfig.Instance
        .AddJob(Job.Default
            .WithToolchain(new InProcessEmitToolchain(timeout: TimeSpan.FromSeconds(30), logOutput: false))
            .WithLaunchCount(1)
            .WithWarmupCount(3)
            .WithIterationCount(20)
            .WithIterationTime(TimeInterval.FromMilliseconds(500)))
        .AddLogger(new ConsoleLogger(unicodeSupport: true, ConsoleLogger.CreateGrayScheme()))
        .WithOptions(ConfigOptions.DisableLogFile);

    // Run HTTP benchmarks
    BenchmarkRunner.Run<HttpClientTests>(httpConfig);
    BenchmarkRunner.Run<HttpScalingBenchmarks>(httpConfig);
    BenchmarkRunner.Run<HttpTypeSerializationBenchmarks>(httpConfig);
    // Concurrency benchmarks can cause connection issues, run separately if needed
    // BenchmarkRunner.Run<HttpConcurrencyBenchmarks>(httpConfig);
}

Console.WriteLine("\n=== Benchmark Complete ===");
Console.WriteLine("Results saved to BenchmarkDotNet.Artifacts/ folder.");

// Usage help
if (args.Contains("--help") || args.Contains("-?"))
{
    Console.WriteLine(@"
NpgsqlRest Benchmark Tests

Usage: dotnet run -- [options]

Options:
  -i, --inprocess   Run in-process benchmarks only (default)
                    Tests TypeCategory, ParameterParser, Serialization internals
                    No external server required

  -h, --http        Run HTTP endpoint benchmarks only
                    Requires NpgsqlRestTests server running on localhost:5000
                    Start with: dotnet run --project NpgsqlRestTests/Setup

  -a, --all         Run all benchmarks (in-process + HTTP)

  -?, --help        Show this help message

Examples:
  dotnet run                     # Run in-process benchmarks
  dotnet run -- --http           # Run HTTP benchmarks
  dotnet run -- --all            # Run all benchmarks
");
}