<?php
/**
 * AI crawler control.
 *
 * Injects per-bot Allow/Disallow directives into the virtual robots.txt
 * served by WordPress. Supports the major AI crawlers as of 2026.
 *
 * @package AEOmatic
 */

if ( ! defined( 'ABSPATH' ) ) {
	exit;
}

class AEOMATIC_Crawler_Control {

	public function register(): void {
		// Modern WP (5.3+) uses robots_txt filter.
		add_filter( 'robots_txt', array( $this, 'append_directives' ), 10, 2 );
	}

	/**
	 * Append AI bot directives to robots.txt output.
	 *
	 * @param string $output  Existing robots.txt body.
	 * @param string $public  '1' if site is public.
	 */
	public function append_directives( string $output, string $public ): string {
		$crawlers   = AEOMATIC_Settings::get( 'crawlers' );
		$block_pdfs = (bool) AEOMATIC_Settings::get( 'robots', 'block_pdfs' );

		if ( empty( $crawlers ) && ! $block_pdfs ) {
			return $output;
		}

		$lines = array();

		if ( $block_pdfs ) {
			$lines[] = '';
			$lines[] = '# Block PDF files — managed by AEOmatic';
			$lines[] = 'User-agent: *';
			$lines[] = 'Disallow: /*.pdf$';
			$lines[] = '';
		}

		if ( ! empty( $crawlers ) ) {
			$lines[] = '';
			$lines[] = '# AI crawler directives — managed by AEOmatic';

			foreach ( $crawlers as $bot => $allowed ) {
				$lines[] = 'User-agent: ' . $bot;
				$lines[] = $allowed ? 'Allow: /' : 'Disallow: /';
				$lines[] = '';
			}
		}

		// Hint to llms.txt for compatible AI crawlers.
		//
		// Deliberately a comment, not a `Sitemap:` directive: Sitemap expects an
		// XML sitemap or a plain list of URLs, and llms.txt is Markdown. Pointing
		// Sitemap at it makes conforming crawlers report a malformed sitemap and
		// can suppress the site's real sitemap. The canonical discovery routes for
		// llms.txt are the well-known path itself and the Link: rel="ai-context"
		// response header AEOMATIC_AI_Headers emits.
		$llms_enabled = AEOMATIC_Settings::get( 'llms_txt', 'enabled' );
		if ( $llms_enabled ) {
			$lines[] = '# AI context file (llms.txt spec: https://llmstxt.org/)';
			$lines[] = '# ' . home_url( '/llms.txt' );
			$lines[] = '';
		}

		return $output . "\n" . implode( "\n", $lines );
	}
}