<?php
if ( ! defined( 'ABSPATH' ) ) {
	exit;
}

/**
 * Appends per-bot allow/deny rules to the virtual robots.txt.
 * Only applies when WordPress serves robots.txt (no physical file).
 */
class ATBA_Robots {

	public function __construct() {
		add_filter( 'robots_txt', array( $this, 'append_rules' ), 20, 2 );
	}

	public function append_rules( $output, $public ) {
		if ( ! $public ) {
			return $output;
		}

		$blocked = (array) get_option( 'atba_blocked_bots', array() );
		$lines   = array( '', '# Managed by AITooler Bot Analytics' );

		foreach ( ATBA_Bots::all() as $key => $bot ) {
			if ( in_array( $key, $blocked, true ) ) {
				$lines[] = 'User-agent: ' . $bot['ua'];
				$lines[] = 'Disallow: /';
				$lines[] = '';
			}
		}

		if ( get_option( 'atba_llms_enabled', 1 ) ) {
			$lines[] = '# llms.txt: ' . home_url( '/llms.txt' );
		}

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