Mr.X


<?php
/**
 * REST API: WP_REST_Edit_Site_Export_Controller class
 *
 * @package    WordPress
 * @subpackage REST_API
 */

/**
 * Controller which provides REST endpoint for exporting current templates
 * and template parts.
 *
 * @since 5.9.0
 *
 * @see WP_REST_Controller
 */
class WP_REST_Edit_Site_Export_Controller extends WP_REST_Controller {

    /**
     * Constructor.
     *
     * @since 5.9.0
     */
    public function __construct() {
        $this->namespace = 'wp-block-editor/v1';
        $this->rest_base = 'export';
    }

    /**
     * Registers the site export route.
     *
     * @since 5.9.0
     */
    public function register_routes() {
        register_rest_route(
            $this->namespace,
            '/' . $this->rest_base,
            array(
                array(
                    'methods'             => WP_REST_Server::READABLE,
                    'callback'            => array( $this, 'export' ),
                    'permission_callback' => array( $this, 'permissions_check' ),
                ),
            )
        );
    }

    /**
     * Checks whether a given request has permission to export.
     *
     * @since 5.9.0
     *
     * @return WP_Error|true True if the request has access, or WP_Error object.
     */
    public function permissions_check() {
        if ( ! current_user_can( 'edit_theme_options' ) ) {
            return new WP_Error(
                'rest_cannot_export_templates',
                __( 'Sorry, you are not allowed to export templates and template parts.' ),
                array( 'status' => rest_authorization_required_code() )
            );
        }

        return true;
    }

    /**
     * Output a ZIP file with an export of the current templates
     * and template parts from the site editor, and close the connection.
     *
     * @since 5.9.0
     *
     * @return WP_Error|void
     */
    public function export() {
        // Generate the export file.
        $filename = wp_generate_block_templates_export_file();

        if ( is_wp_error( $filename ) ) {
            $filename->add_data( array( 'status' => 500 ) );

            return $filename;
        }

        header( 'Content-Type: application/zip' );
        header( 'Content-Disposition: attachment; filename=edit-site-export.zip' );
        header( 'Content-Length: ' . filesize( $filename ) );
        flush();
        readfile( $filename );
        unlink( $filename );
        exit;
    }
}