Merge pull request #632 from humanmade/backport-564-to-v3-branch

[Backport v3-branch] Added `get_s3_path` function instead of hard coding s3 path.
humanmade · May 4, 2023 · b65f2c7 · b65f2c7
2 parents 1f3a7a2 + fc88fbd
commit b65f2c7
Showing 1 changed file with 14 additions and 6 deletions.
diff --git a/inc/class-plugin.php b/inc/class-plugin.php
@@ -154,6 +154,13 @@ public function register_stream_wrapper() {
 		stream_context_set_option( stream_context_get_default(), 's3', 'seekable', true );
 	}
 
+	/**
+	 * Get the s3:// path for the bucket.
+	 */
+	public function get_s3_path() {
+		return 's3://' . $this->bucket;
+	}
+
 	/**
 	 * Overwrite the default wp_upload_dir.
 	 *
@@ -163,19 +170,20 @@ public function register_stream_wrapper() {
 	public function filter_upload_dir( array $dirs ) : array {
 
 		$this->original_upload_dir = $dirs;
+		$s3_path = $this->get_s3_path();
 
-		$dirs['path']    = str_replace( WP_CONTENT_DIR, 's3://' . $this->bucket, $dirs['path'] );
-		$dirs['basedir'] = str_replace( WP_CONTENT_DIR, 's3://' . $this->bucket, $dirs['basedir'] );
+		$dirs['path']    = str_replace( WP_CONTENT_DIR, $s3_path, $dirs['path'] );
+		$dirs['basedir'] = str_replace( WP_CONTENT_DIR, $s3_path, $dirs['basedir'] );
 
 		if ( ! defined( 'S3_UPLOADS_DISABLE_REPLACE_UPLOAD_URL' ) || ! S3_UPLOADS_DISABLE_REPLACE_UPLOAD_URL ) {
 
 			if ( defined( 'S3_UPLOADS_USE_LOCAL' ) && S3_UPLOADS_USE_LOCAL ) {
-				$dirs['url']     = str_replace( 's3://' . $this->bucket, $dirs['baseurl'] . '/s3/' . $this->bucket, $dirs['path'] );
-				$dirs['baseurl'] = str_replace( 's3://' . $this->bucket, $dirs['baseurl'] . '/s3/' . $this->bucket, $dirs['basedir'] );
+				$dirs['url']     = str_replace( $s3_path, $dirs['baseurl'] . '/s3/' . $this->bucket, $dirs['path'] );
+				$dirs['baseurl'] = str_replace( $s3_path, $dirs['baseurl'] . '/s3/' . $this->bucket, $dirs['basedir'] );
 
 			} else {
-				$dirs['url']     = str_replace( 's3://' . $this->bucket, $this->get_s3_url(), $dirs['path'] );
-				$dirs['baseurl'] = str_replace( 's3://' . $this->bucket, $this->get_s3_url(), $dirs['basedir'] );
+				$dirs['url']     = str_replace( $s3_path, $this->get_s3_url(), $dirs['path'] );
+				$dirs['baseurl'] = str_replace( $s3_path, $this->get_s3_url(), $dirs['basedir'] );
 			}
 		}