Importer: Add support for video and audio files (#1585)

Author: obenland

.github/changelog/1585-from-description

@@ -0,0 +1,4 @@
+Significance: minor
+Type: added
+
+Support for all media types in the Mastodon Importer.

includes/wp-admin/import/class-mastodon.php

@@ -368,12 +368,18 @@ function ( $tag ) {
 						'tmp_name' => $file_path,
 					);
 
+					$meta = array();
+					if ( 'image' === strtok( $attachment->{'mediaType'}, '/' ) && ! empty( $attachment->name ) ) {
+						$meta = array( '_wp_attachment_image_alt' => $attachment->name );
+					}
+
 					$attachment_data = array(
 						'post_mime_type' => $attachment->{'mediaType'},
 						'post_title'     => $attachment->name ?? '',
 						'post_content'   => $attachment->name ?? '',
 						'post_status'    => 'inherit',
 						'post_author'    => self::$author,
+						'meta_input'     => $meta,
 					);
 
 					$attachment_id = \media_handle_sideload( $file_array, $post_id, '', $attachment_data );
@@ -387,36 +393,33 @@ function ( $tag ) {
 
 				// If we have attachments, add them to the post content.
 				if ( ! empty( $attachment_ids ) ) {
+					$type = strtok( \get_post_mime_type( $attachment_ids[0] ), '/' );
+
 					if ( site_supports_blocks() ) {
-						// Block editor: Use gallery block.
-						$gallery  = '<!-- wp:gallery {"ids":[' . \implode( ',', $attachment_ids ) . '],"linkTo":"none"} -->' . "\n";
-						$gallery .= '<figure class="wp-block-gallery has-nested-images columns-default is-cropped">';
-
-						foreach ( $attachment_ids as $id ) {
-							$image_src = \wp_get_attachment_image_src( $id, 'large' );
-							if ( ! $image_src ) {
-								continue;
-							}
-
-							$caption  = \get_post_field( 'post_content', $id );
-							$gallery .= "\n<!-- wp:image {\"id\":{$id},\"sizeSlug\":\"large\",\"linkDestination\":\"none\"} -->\n";
-							$gallery .= '<figure class="wp-block-image size-large">';
-							$gallery .= '<img src="' . \esc_url( $image_src[0] ) . '" alt="' . \esc_attr( $caption ) . '" class="' . \esc_attr( 'wp-image-' . $id ) . '"/>';
-							$gallery .= '</figure>';
-							$gallery .= "\n<!-- /wp:image -->\n";
+						if ( 1 === \count( $attachment_ids ) && ( 'video' === $type || 'audio' === $type ) ) {
+							$media = sprintf(
+								'<!-- wp:%1$s {"id":"%2$s"} --><figure class="wp-block-%1$s"><%1$s controls src="%3$s"></%1$s></figure><!-- /wp:%1$s -->',
+								\esc_attr( $type ),
+								\esc_attr( $attachment_ids[0] ),
+								\esc_url( \wp_get_attachment_url( $attachment_ids[0] ) )
+							);
+						} else {
+							$media = self::get_gallery_block( $attachment_ids );
+						}
+					} else { // phpcs:ignore Universal.ControlStructures.DisallowLonelyIf.Found
+						// Classic editor: Use shortcodes.
+						if ( 1 === \count( $attachment_ids ) && ( 'video' === $type || 'audio' === $type ) ) {
+							// Block editor: Use video block.
+							$media = sprintf( '[%1$s src="%2$s"]', \esc_attr( $type ), \esc_url( \wp_get_attachment_url( $attachment_ids[0] ) ) );
+						} else {
+							$media = '[gallery ids="' . \implode( ',', $attachment_ids ) . '" link="none"]';
 						}
-
-						$gallery .= "</figure>\n";
-						$gallery .= '<!-- /wp:gallery -->';
-					} else {
-						// Classic editor: Use gallery shortcode.
-						$gallery = '[gallery ids="' . \implode( ',', $attachment_ids ) . '" link="none"]';
 					}
 
 					\wp_update_post(
 						array(
 							'ID'           => $post_id,
-							'post_content' => $post_data['post_content'] . "\n\n" . $gallery,
+							'post_content' => $post_data['post_content'] . "\n\n" . $media,
 						)
 					);
 				}
@@ -487,4 +490,35 @@ public static function greet() {
 		\wp_import_upload_form( 'admin.php?import=mastodon&amp;step=1' );
 		echo '</div>';
 	}
+
+	/**
+	 * Get gallery block.
+	 *
+	 * @param array $attachment_ids The attachment IDs to use.
+	 * @return string The gallery block markup.
+	 */
+	private static function get_gallery_block( $attachment_ids ) {
+		// Block editor: Use gallery block.
+		$gallery  = '<!-- wp:gallery {"ids":[' . \implode( ',', $attachment_ids ) . '],"linkTo":"none"} -->' . "\n";
+		$gallery .= '<figure class="wp-block-gallery has-nested-images columns-default is-cropped">';
+
+		foreach ( $attachment_ids as $id ) {
+			$image_src = \wp_get_attachment_image_src( $id, 'large' );
+			if ( ! $image_src ) {
+				continue;
+			}
+
+			$caption  = \get_post_field( 'post_content', $id );
+			$gallery .= "\n<!-- wp:image {\"id\":{$id},\"sizeSlug\":\"large\",\"linkDestination\":\"none\"} -->\n";
+			$gallery .= '<figure class="wp-block-image size-large">';
+			$gallery .= '<img src="' . \esc_url( $image_src[0] ) . '" alt="' . \esc_attr( $caption ) . '" class="' . \esc_attr( 'wp-image-' . $id ) . '"/>';
+			$gallery .= '</figure>';
+			$gallery .= "\n<!-- /wp:image -->\n";
+		}
+
+		$gallery .= "</figure>\n";
+		$gallery .= '<!-- /wp:gallery -->';
+
+		return $gallery;
+	}
 }