<?php
defined( 'ABSPATH' ) || die( 'Cheatin\' uh?' );

class SQ_Models_Services_Robots extends SQ_Models_Abstract_Seo {


	public function __construct() {
		parent::__construct();
		add_filter( 'sq_robots', array( $this, 'generateRobots' ) );
		add_filter( 'sq_robots', array( $this, 'showRobots' ), 11 );
	}

	public function generateRobots( $robots = '' ) {
		$robots .= "\n";

		if ( get_option( 'blog_public' ) != 1 ) {
			$robots .= "\n# " . "Your blog is not public. Please see Site Visibility on Settings > Reading.";
		} else {

			$sq_sitemap = SQ_Classes_Helpers_Tools::getOption( 'sq_sitemap' );
			if ( SQ_Classes_Helpers_Tools::getOption( 'sq_auto_sitemap' ) == 1 ) {
				foreach ( (array) $sq_sitemap as $name => $sitemap ) {
					if ( $name == 'sitemap-product' && ! SQ_Classes_Helpers_Tools::isEcommerce() ) {
						continue;
					}
					if ( $sitemap[1] == 1 || $sitemap[1] == 2 ) {
						$robots .= "\nSitemap: " . trailingslashit( get_bloginfo( 'url' ) ) . $sitemap[0];
					}
				}
			}

			if ( empty( $sq_sitemap ) ) {
				$robots .= "\n";
			}
		}
		$robots .= "\n\n";

		$robots_permission = (array) SQ_Classes_Helpers_Tools::getOption( 'sq_robots_permission' );
		$robots_permission = array_filter( $robots_permission );

		if( empty($robots_permission) ){
			// If no custom robots permissions are set, use the default rules
			$robots_permission = array(
				'User-agent: *',
				'Disallow: */trackback/',
				'Disallow: */xmlrpc.php',
				'Disallow: /wp-*.php',
				'Disallow: /cgi-bin/',
				'Disallow: /wp-admin/',
				'Allow: */wp-content/uploads/',);
		}


		foreach (  $robots_permission as $robot_txt ) {
			if (is_string($robot_txt)){
				if ( strpos( $robot_txt, '#' ) !== false ) {
					$robots .= PHP_EOL ;
				}

				$robots .= $robot_txt . PHP_EOL;
			}
		}

		$robots .= PHP_EOL . PHP_EOL;

		return apply_filters( 'sq_custom_robots', $robots );
	}

	public function showRobots( $robots = '' ) {
		/**
		 *
		 * display robots.txt
		 */
		header( 'Status: 200 OK', true, 200 );
		header( 'Content-type: text/plain; charset=' . get_bloginfo( 'charset' ) );

		echo esc_textarea( sanitize_textarea_field( $robots ) );
		exit();
	}
}
