Changeset 52286 for trunk/src/wp-includes/block-template-utils.php

Timestamp:

11/30/2021 05:30:22 PM (3 years ago)

Author:

spacedmonkey

Message:

Site Editor: Add site export REST API endpoint.

Add a REST API to export site templates and template part as html files. When the REST API is requested, it responds by downloading a single ZIP file and exits early, without completing full request. To create the exported zip, the ZipArchive class is required. If this class is not present then the export will gracefully fail, returning a WP_Error object and 500 status error code.

Props spacedmonkey, youknowriad, Mamaduka, walbo, peterwilsoncc.
Fixes #54448 .

File:

• trunk/src/wp-includes/block-template-utils.php (modified) (2 diffs)

trunk/src/wp-includes/block-template-utils.php

@@ -456 +456 @@
 
 /**
+ * Parses a block template and removes the theme attribute from each template part.
+ *
+ * @access private
+ * @since 5.9.0
+ *
+ * @param string $template_content Serialized block template content.
+ * @return string Updated block template content.
+ */
+function _remove_theme_attribute_in_block_template_content( $template_content ) {
+    $has_updated_content = false;
+    $new_content         = '';
+    $template_blocks     = parse_blocks( $template_content );
+
+    $blocks = _flatten_blocks( $template_blocks );
+    foreach ( $blocks as $key => $block ) {
+        if ( 'core/template-part' === $block['blockName'] && isset( $block['attrs']['theme'] ) ) {
+            unset( $blocks[ $key ]['attrs']['theme'] );
+            $has_updated_content = true;
+        }
+    }
+
+    if ( ! $has_updated_content ) {
+        return $template_content;
+    }
+
+    foreach ( $template_blocks as $block ) {
+        $new_content .= serialize_block( $block );
+    }
+
+    return $new_content;
+}
+
+/**
  * Build a unified template object based on a theme file.
  *
@@ -864 +897 @@
     block_template_part( 'footer' );
 }
+
+/**
+ * Creates an export of the current templates and
+ * template parts from the site editor at the
+ * specified path in a ZIP file.
+ *
+ * @since 5.9.0
+ *
+ * @return WP_Error|string Path of the ZIP file or error on failure.
+ */
+function wp_generate_block_templates_export_file() {
+    if ( ! class_exists( 'ZipArchive' ) ) {
+        return new WP_Error( __( 'Zip Export not supported.' ) );
+    }
+
+    $obscura  = wp_generate_password( 12, false, false );
+    $filename = get_temp_dir() . 'edit-site-export-' . $obscura . '.zip';
+
+    $zip = new ZipArchive();
+    if ( true !== $zip->open( $filename, ZipArchive::CREATE ) ) {
+        return new WP_Error( __( 'Unable to open export file (archive) for writing.' ) );
+    }
+
+    $zip->addEmptyDir( 'theme' );
+    $zip->addEmptyDir( 'theme/templates' );
+    $zip->addEmptyDir( 'theme/parts' );
+
+    // Load templates into the zip file.
+    $templates = get_block_templates();
+    foreach ( $templates as $template ) {
+        $template->content = _remove_theme_attribute_in_block_template_content( $template->content );
+
+        $zip->addFromString(
+            'theme/templates/' . $template->slug . '.html',
+            $template->content
+        );
+    }
+
+    // Load template parts into the zip file.
+    $template_parts = get_block_templates( array(), 'wp_template_part' );
+    foreach ( $template_parts as $template_part ) {
+        $zip->addFromString(
+            'theme/parts/' . $template_part->slug . '.html',
+            $template_part->content
+        );
+    }
+
+    // Save changes to the zip file.
+    $zip->close();
+
+    return $filename;
+}