95 lines
2.1 KiB
PHP
95 lines
2.1 KiB
PHP
|
<?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 true;
|
||
|
|
}
|
||
|
|
|
||
|
|
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() )
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* 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;
|
||
|
|
}
|
||
|
|
|
||
|
|
$theme_name = basename( get_stylesheet() );
|
||
|
|
header( 'Content-Type: application/zip' );
|
||
|
|
header( 'Content-Disposition: attachment; filename=' . $theme_name . '.zip' );
|
||
|
|
header( 'Content-Length: ' . filesize( $filename ) );
|
||
|
|
flush();
|
||
|
|
readfile( $filename );
|
||
|
|
unlink( $filename );
|
||
|
|
exit;
|
||
|
|
}
|
||
|
|
}
|