Disable attachment routing

CHANGELOG.md

@@ -2,6 +2,12 @@
 
 This library adheres to [Semantic Versioning](https://semver.org/) and [Keep a CHANGELOG](https://keepachangelog.com/en/1.0.0/).
 
+## Unreleased
+
+### Added
+
+* Added a feature to disable attachment routing.
+
 ## 2.2.1
 
 ### Added

README.md

@@ -32,6 +32,10 @@ Each feature's handle is listed below, along with a description of what it does.
 
 This feature removes selected nodes from the admin bar.
 
+### `disable_attachment_routing`
+
+This feature disables WordPress attachment pages entirely from the front end of the site.
+
 ### `disable_comments`
 
 This feature disables WordPress comments entirely, including the ability to post, view, edit, list, count, modify settings for, or access URLs that are related to comments completely.

src/alley/wp/alleyvate/features/class-disable-attachment-routing.php

@@ -0,0 +1,86 @@
+<?php
+/**
+ * Class file for Disable_Attachment_Routing
+ *
+ * (c) Alley <[email protected]>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ *
+ * @package wp-alleyvate
+ */
+
+namespace Alley\WP\Alleyvate\Features;
+
+use Alley\WP\Alleyvate\Feature;
+
+/**
+ * Disable attachment routing.
+ */
+final class Disable_Attachment_Routing implements Feature {
+	/**
+	 * Boot the feature.
+	 */
+	public function boot(): void {
+		add_filter( 'rewrite_rules_array', [ self::class, 'filter__rewrite_rules_array' ] );
+		add_filter( 'attachment_link', [ self::class, 'filter__attachment_link' ] );
+		add_action( 'pre_get_posts', [ self::class, 'action__pre_get_posts' ] );
+		add_action( 'admin_bar_menu', [ self::class, 'action__admin_bar_menu' ], 100 );
+	}
+
+	/**
+	 * Remove support for the attachment rewrite rule.
+	 *
+	 * @param array $rules Rewrite rules.
+	 * @return array
+	 */
+	public static function filter__rewrite_rules_array( $rules ): array {
+		foreach ( $rules as $regex => $query ) {
+			if ( strpos( $regex, 'attachment' ) || strpos( $query, 'attachment' ) ) {
+				unset( $rules[ $regex ] );
+			}
+		}
+
+		return $rules;
+	}
+
+	/**
+	 * Remove the attachment link.
+	 *
+	 * @param string $link Attachment link.
+	 * @return string
+	 */
+	public static function filter__attachment_link( $link ): string {
+		return '';
+	}
+
+	/**
+	 * Ensure attachment pages return 404s.
+	 *
+	 * @param WP_Query $query WP_Query object.
+	 */
+	public static function action__pre_get_posts( $query ) {
+		if ( is_admin() || ! $query->is_main_query() ) {
+			return;
+		}
+
+		if (
+			$query->queried_object instanceof \WP_Post
+			&& 'attachment' === get_post_type( $query->get_queried_object_id() )
+		) {
+			$query->set_404();
+			status_header( 404 );
+		}
+	}
+
+	/**
+	 * Remove attachment link from admin bar.
+	 *
+	 * @param \WP_Admin_Bar $wp_admin_bar Admin bar class.
+	 */
+	public static function action__admin_bar_menu( \WP_Admin_Bar $wp_admin_bar ) {
+		if ( 'attachment' == get_post_type() ) {
+			$wp_admin_bar->remove_node( 'view' );
+		}
+	}
+}

src/alley/wp/alleyvate/load.php

@@ -20,6 +20,7 @@
 function available_features(): array {
 	return [
 		'clean_admin_bar'               => new Features\Clean_Admin_Bar(),
+		'disable_attachment_routing'    => new Features\Disable_Attachment_Routing(),
 		'disable_comments'              => new Features\Disable_Comments(),
 		'disable_dashboard_widgets'     => new Features\Disable_Dashboard_Widgets(),
 		'disable_sticky_posts'          => new Features\Disable_Sticky_Posts(),

tests/alley/wp/alleyvate/features/test-disable-attachment-routing.php

@@ -0,0 +1,70 @@
+<?php
+/**
+ * Class file for Test_Disable_Attachment_Routing
+ *
+ * (c) Alley <[email protected]>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ *
+ * @package wp-alleyvate
+ */
+
+namespace Alley\WP\Alleyvate\Features;
+
+use Alley\WP\Alleyvate\Feature;
+use Mantle\Testkit\Test_Case;
+
+/**
+ * Tests for fully disabling attachment routing.
+ */
+final class Test_Disable_Attachment_Routing extends Test_Case {
+	use \Mantle\Testing\Concerns\Admin_Screen;
+	use \Mantle\Testing\Concerns\Refresh_Database;
+
+	/**
+	 * Feature instance.
+	 *
+	 * @var Feature
+	 */
+	private Feature $feature;
+
+	/**
+	 * Set up.
+	 */
+	protected function setUp(): void {
+		parent::setUp();
+
+		$this->feature = new Disable_Attachment_Routing();
+	}
+
+	/**
+	 * Test that the attachment permalink is empty.
+	 */
+	public function test_attachment_permalink(): void {
+		$attachment_id = $this->factory()->attachment->create();
+
+		$this->assertNotEmpty( get_permalink( $attachment_id ) );
+
+		$this->feature->boot();
+
+		$this->assertEmpty( get_permalink( $attachment_id ) );
+	}
+
+	/**
+	 * Test that the attachment page returns a 404.
+	 */
+	public function test_attachment_page(): void {
+		$attachment_id = $this->factory()->attachment->create();
+		$permalink     = get_permalink( $attachment_id );
+
+		$this
+			->get( $permalink )
+			->assertOk()
+			->assertQueriedObjectId( $attachment_id );
+
+		$this->feature->boot();
+
+		$this->get( $permalink )->assertNotFound();
+	}
+}