<?php

/**
 * This script is added via the `WP_CLI_REQUIRE` environment variable to the WP-CLI commands executed by the Behat test runner.
 * It starts coverage collection right away and registers a shutdown hook to complete it
 * after the respective WP-CLI command has finished.
 */

use SebastianBergmann\CodeCoverage\CodeCoverage;
use SebastianBergmann\CodeCoverage\Driver\Selector;
use SebastianBergmann\CodeCoverage\Driver\Xdebug;
use SebastianBergmann\CodeCoverage\Filter;
use SebastianBergmann\CodeCoverage\Report\PHP as PHPReport;

/*
 * The names of the current feature and scenario are passed on from the Behat test runner
 * to this script through environment variables.
 */
$feature   = getenv( 'BEHAT_FEATURE_TITLE' );
$scenario  = getenv( 'BEHAT_SCENARIO_TITLE' );
$step_line = (int) getenv( 'BEHAT_STEP_LINE' );
$name      = "{$feature} - {$scenario} - {$step_line}";

/*
 * Do not run coverage if they are empty, which means we are running some command
 * during test preparation, e.g. the `wp core download` in `FeatureContext::prepare()`.
 */
if ( empty( $feature ) | empty( $scenario ) ) {
	return;
}

$project_dir = (string) getenv( 'TEST_RUN_DIR' );

if ( ! class_exists( 'SebastianBergmann\CodeCoverage\Filter' ) ) {
	if ( ! file_exists( $project_dir . '/vendor/autoload.php' ) ) {
		die( 'Could not load dependencies for generating code coverage' );
	}
	require "{$project_dir}/vendor/autoload.php";
}

$filtered_items = new CallbackFilterIterator(
	new DirectoryIterator( $project_dir ),
	function ( $file ) {
		// Allow directories named "php" or "src"
		if ( $file->isDir() && in_array( $file->getFilename(), [ 'php', 'src' ], true ) ) {
			return true;
		}

		// Allow top-level files ending in "-command.php"
		if ( $file->isFile() && false !== strpos( $file->getFilename(), '-command.php' ) ) {
			return true;
		}

		return false;
	}
);

$files = [];

foreach ( $filtered_items as $item ) {
	if ( $item->isDir() ) {
		foreach (
			new RecursiveIteratorIterator(
				new RecursiveDirectoryIterator( $item->getPathname(), RecursiveDirectoryIterator::SKIP_DOTS )
			) as $file
		) {
			if ( $file->isFile() && $file->getExtension() === 'php' ) {
				$files[] = $file->getPathname();
			}
		}
	} else {
		$files[] = $item->getPathname();
	}
}

$filter = new Filter();

if ( method_exists( $filter, 'includeFiles' ) ) {
	$filter->includeFiles( $files );
} else {
	$filter->addFilesToWhitelist( $files );
}

$coverage = new CodeCoverage(
	// Selector class was only added in v9.1 of the php-code-coverage library.
	class_exists( Selector::class ) ? ( new Selector() )->forLineCoverage( $filter ) : ( new Xdebug() ),
	$filter
);

$coverage->start( $name );

register_shutdown_function(
	static function () use ( $coverage, $feature, $scenario, $step_line, $name, $project_dir ) {
		$coverage->stop();

		$feature_suffix  = preg_replace( '/[^a-z0-9]+/', '-', strtolower( $feature ) );
		$scenario_suffix = preg_replace( '/[^a-z0-9]+/', '-', strtolower( $scenario ) );
		$db_type         = strtolower( getenv( 'WP_CLI_TEST_DBTYPE' ) );
		$destination     = "$project_dir/build/logs/$feature_suffix-$scenario_suffix-$step_line-$db_type.cov";

		$dir = dirname( $destination );
		if ( ! file_exists( $dir ) ) {
			mkdir( $dir, 0777, true /*recursive*/ );
		}

		( new PHPReport() )->process( $coverage, $destination );
	}
);