Merge pull request #4 from tarosky/feature/sitemap-query-filter

Add filter for attachment query.

Author: fumikito

.github/workflows/wordpress.yml

@@ -15,7 +15,7 @@ jobs:
     runs-on: ${{ matrix.operating-system }}
     strategy:
       matrix:
-        operating-system: [ ubuntu-18.04 ]  # OS. ubuntu-18.04 is also available.
+        operating-system: [ ubuntu-latest ]  # OS. ubuntu-18.04 is also available.
         php: [ '7.2', '7.4', '8.1' ] # PHP versions to check.
         wp: [ 'latest', '5.7' ]      # WordPress version to check.
     services:
@@ -53,7 +53,7 @@ jobs:
 
   assets:
     name: Assets Test
-    runs-on: ubuntu-18.04
+    runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@master
 
@@ -72,19 +72,19 @@ jobs:
     name: Deploy WordPress.org
     needs: [ test, assets ]
     if: contains(github.ref, 'tags/')
-    runs-on: ubuntu-18.04
+    runs-on: ubuntu-latest
     steps:
       - name: Checkout code
-        uses: actions/checkout@v2
+        uses: actions/checkout@master
 
       - name: Setup PHP with composer v2
         uses: shivammathur/setup-php@v2
         with:
           php-version: 7.2
-          tools: composer:v2.2
+          tools: composer
 
       - name: Install Node
-        uses: actions/setup-node@v1
+        uses: actions/setup-node@v3
         with:
           node-version: '14'
 

.wp-env.json

@@ -1,6 +1,7 @@
 {
 	"plugins": [ "." ],
 	"themes": [
-		"https://downloads.wordpress.org/theme/twentytwenty.latest-stable.zip"
+		"https://downloads.wordpress.org/theme/twentytwenty.latest-stable.zip",
+		"https://downloads.wordpress.org/theme/stacks.latest-stable.zip"
 	]
 }

src/Tarosky/Sitemap/Provider/AttachmentSitemapIndexProvider.php

@@ -28,19 +28,23 @@ public function is_active() {
 	 */
 	protected function get_urls() {
 		global $wpdb;
-		$query = <<<SQL
+		$join_clause  = apply_filters( 'taro_sitemap_attachment_query_join', '' );
+		$where_clause = apply_filters( 'taro_sitemap_attachment_query_where', '' );
+		$query        = <<<SQL
 			SELECT
 			    EXTRACT( YEAR_MONTH from p1.post_date ) as date,
 			    COUNT( p1.ID ) AS total
 			FROM {$wpdb->posts} AS p1
 			LEFT JOIN {$wpdb->posts} AS p2
 			ON p1.post_parent = p2.ID
+			{$join_clause}
 			WHERE p1.post_type = 'attachment'
 			  AND p1.post_mime_type LIKE 'image%'
 			  AND p2.post_status = 'publish'
+			  {$where_clause}
 			GROUP BY EXTRACT( YEAR_MONTH from p1.post_date )
 SQL;
-		$urls  = [];
+		$urls         = [];
 		// Already escaped.
 		// phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
 		foreach ( $wpdb->get_results( $query ) as $row ) {

src/Tarosky/Sitemap/Provider/AttachmentSitemapProvider.php

@@ -31,21 +31,25 @@ public function is_active() {
 	 */
 	protected function get_urls() {
 		global $wpdb;
-		$year     = get_query_var( 'year' );
-		$month    = get_query_var( 'monthnum' );
-		$from     = sprintf( '%04d-%02d-01 00:00:00', $year, $month );
-		$to       = ( new \DateTimeImmutable )->modify( sprintf( 'last day of %04d-%02d', $year, $month ) )->format( 'Y-m-d 23:59:59' );
-		$per_page = $this->option()->posts_per_page;
-		$offset   = ( max( 1, get_query_var( 'paged' ) ) - 1 ) * $per_page;
-		$query    = <<<SQL
+		$year         = get_query_var( 'year' );
+		$month        = get_query_var( 'monthnum' );
+		$from         = sprintf( '%04d-%02d-01 00:00:00', $year, $month );
+		$to           = ( new \DateTimeImmutable )->modify( sprintf( 'last day of %04d-%02d', $year, $month ) )->format( 'Y-m-d 23:59:59' );
+		$per_page     = $this->option()->posts_per_page;
+		$offset       = ( max( 1, get_query_var( 'paged' ) ) - 1 ) * $per_page;
+		$join_clause  = apply_filters( 'taro_sitemap_attachment_query_join', '' );
+		$where_clause = apply_filters( 'taro_sitemap_attachment_query_where', '' );
+		$query        = <<<SQL
 			SELECT p1.* FROM {$wpdb->posts} AS p1
 			LEFT JOIN {$wpdb->posts} AS p2
 			ON p1.post_parent = p2.ID
+			{$join_clause}
 			WHERE p1.post_type = 'attachment'
 			  AND p1.post_status = 'inherit'
 			  AND p1.post_date BETWEEN %s AND %s
 			  AND p1.post_mime_type LIKE ':::image:::'
 			  AND p2.post_status = 'publish'
+			  {$where_clause}
 			ORDER BY p1.post_date DESC
 			LIMIT %d, %d
 SQL;