<?php
/**
 * @group routing
 * @ticket BP9330
 */
class BP_Tests_Routing_Directory_Homepage extends BP_UnitTestCase {
	/**
	 * @dataProvider content_requests
	 */
	public function test_directory_homepage_preserves_content_requests( $component, $request, $permalinks ) {
		if ( 'blogs' === $component && ! is_multisite() ) {
			$this->markTestSkipped( 'The Sites directory requires multisite.' );
		}

		$pages     = bp_core_get_directory_pages();
		$old_front = get_option( 'show_on_front' );
		$old_page  = get_option( 'page_on_front' );
		$old_links = get_option( 'permalink_structure' );
		$old_user  = get_current_user_id();
		$is_draft  = in_array( $request, array( 'draft post preview', 'draft page preview' ), true );
		$is_page   = in_array( $request, array( 'page', 'draft page preview' ), true );
		$post_args = array( 'post_type' => $is_page ? 'page' : 'post' );
		if ( $is_draft ) {
			$user_id = self::factory()->user->create( array( 'role' => 'editor' ) );
			self::set_current_user( $user_id );
			$post_args['post_author'] = $user_id;
			$post_args['post_status'] = 'draft';
		}
		$post_id = self::factory()->post->create( $post_args );
		$args      = array( 'p' => $post_id );

		if ( $is_page ) {
			$args = array( 'page_id' => $post_id );
		} elseif ( 'search' === $request ) {
			$args = array( 's' => 'example' );
		}
		if ( 'preview' === $request || $is_draft ) {
			$args['preview'] = 'true';
		}

		$this->set_permalink_structure( $permalinks );
		update_option( 'show_on_front', 'page' );
		update_option( 'page_on_front', $pages->{$component}->id );

		try {
			$this->go_to( add_query_arg( $args, home_url( '/' ) ) );
			$this->assertNotEquals( 1, (int) get_query_var( buddypress()->{$component}->rewrite_ids['directory'] ), 'A content request must not become a BuddyPress directory.' );
			if ( $is_draft ) {
				$this->assertTrue( is_preview() );
				$this->assertFalse( is_404() );
			}
			if ( 'search' === $request ) {
				$this->assertTrue( is_search() );
			} else {
				$this->assertSame( $post_id, get_queried_object_id() );
			}
		} finally {
			self::set_current_user( $old_user );
			update_option( 'show_on_front', $old_front );
			update_option( 'page_on_front', $old_page );
			$this->set_permalink_structure( $old_links );
		}
	}

	public static function content_requests() {
		$cases = array();
		foreach ( array( 'activity', 'groups', 'members', 'blogs' ) as $component ) {
			foreach ( array( 'post', 'page', 'search', 'preview', 'draft post preview', 'draft page preview' ) as $request ) {
				foreach ( array( 'plain' => '', 'pretty' => '/%postname%/' ) as $name => $permalinks ) {
					$cases[ "$component $request $name" ] = array( $component, $request, $permalinks );
				}
			}
		}
		return $cases;
	}

	/**
	 * @dataProvider directory_homepage_urls
	 */
	public function test_directory_as_homepage( $component, $query_string, $permalinks ) {
		if ( 'blogs' === $component && ! is_multisite() ) {
			$this->markTestSkipped( 'The Sites directory requires multisite.' );
		}

		$pages     = bp_core_get_directory_pages();
		$old_front = get_option( 'show_on_front' );
		$old_page  = get_option( 'page_on_front' );
		$old_links = get_option( 'permalink_structure' );

		$this->set_permalink_structure( $permalinks );
		update_option( 'show_on_front', 'page' );
		update_option( 'page_on_front', $pages->{$component}->id );

		try {
			$this->go_to( home_url( '/' ) . $query_string );
			$this->assertSame( 1, (int) get_query_var( buddypress()->{$component}->rewrite_ids['directory'] ) );
			$this->assertTrue( bp_is_directory() );
			$this->assertFalse( is_404() );
			$this->assertSame( (int) $pages->{$component}->id, get_queried_object_id() );
		} finally {
			update_option( 'show_on_front', $old_front );
			update_option( 'page_on_front', $old_page );
			$this->set_permalink_structure( $old_links );
		}
	}

	public static function directory_homepage_urls() {
		$cases      = array();
		$components = array( 'activity', 'groups', 'members', 'blogs' );
		$queries    = array(
			'no query'  => '',
			'arbitrary' => '?anything=value',
			'tracking'  => '?utm_source=newsletter&ref=test',
			'encoded'   => '?anything=hello%20world',
		);
		$structures = array(
			'plain'  => '',
			'pretty' => '/%postname%/',
		);

		foreach ( $components as $component ) {
			foreach ( $structures as $structure_name => $permalinks ) {
				foreach ( $queries as $query_name => $query_string ) {
					$label           = "$component / $structure_name / $query_name";
					$cases[ $label ] = array( $component, $query_string, $permalinks );
				}
			}
		}

		return $cases;
	}
}