1 files changed, 97 insertions, 0 deletions
diff --git a/plugins/jetpack/json-endpoints/jetpack/class.jetpack-json-api-get-database-object-backup-endpoint.php b/plugins/jetpack/json-endpoints/jetpack/class.jetpack-json-api-get-database-object-backup-endpoint.php
new file mode 100644
index 00000000..b7134730
--- /dev/null
+++ b/plugins/jetpack/json-endpoints/jetpack/class.jetpack-json-api-get-database-object-backup-endpoint.php
@@ -0,0 +1,97 @@
+<?php
+
+class Jetpack_JSON_API_Get_Database_Object_Backup_Endpoint extends Jetpack_JSON_API_Endpoint {
+	// /sites/%s/database-object/backup      -> $blog_id
+
+	protected $needed_capabilities = array(); // This endpoint is only accessible using a site token
+	protected $object_type;
+	protected $object_id;
+
+	// Full list of database objects that can be retrieved via this endpoint.
+	protected $object_types = array(
+		'woocommerce_attribute' => array(
+			'table'     => 'woocommerce_attribute_taxonomies',
+			'id_field'  => 'attribute_id',
+		),
+
+		'woocommerce_downloadable_product_permission' => array(
+			'table'    => 'woocommerce_downloadable_product_permissions',
+			'id_field' => 'permission_id',
+		),
+
+		'woocommerce_order_item' => array(
+			'table'     => 'woocommerce_order_items',
+			'id_field'  => 'order_item_id',
+			'meta_type' => 'order_item',
+		),
+
+		'woocommerce_payment_token' => array(
+			'table'     => 'woocommerce_payment_tokens',
+			'id_field'  => 'token_id',
+			'meta_type' => 'payment_token',
+		),
+
+		'woocommerce_tax_rate' => array(
+			'table'          => 'woocommerce_tax_rates',
+			'id_field'       => 'tax_rate_id',
+			'child_table'    => 'woocommerce_tax_rate_locations',
+			'child_id_field' => 'tax_rate_id',
+		),
+
+		'woocommerce_webhook' => array(
+			'table'    => 'wc_webhooks',
+			'id_field' => 'webhook_id',
+		),
+	);
+
+	function validate_input( $object ) {
+		$query_args = $this->query_args();
+
+		if ( empty( $query_args['object_type'] ) || empty( $query_args['object_id'] ) ) {
+			return new WP_Error( 'invalid_args', __( 'You must specify both an object type and id to fetch', 'jetpack' ), 400 );
+		}
+
+		if ( empty( $this->object_types[ $query_args['object_type'] ] ) ) {
+			return new WP_Error( 'invalid_args', __( 'Specified object_type not recognized', 'jetpack' ), 400 );
+		}
+
+		$this->object_type = $this->object_types[ $query_args['object_type'] ];
+		$this->object_id   = $query_args['object_id'];
+
+		return true;
+	}
+
+	protected function result() {
+		global $wpdb;
+
+		$table    = $wpdb->prefix . $this->object_type['table'];
+		$id_field = $this->object_type['id_field'];
+
+		// Fetch the requested object
+		$query  = $wpdb->prepare( 'select * from `' . $table . '` where `' . $id_field . '` = %d', $this->object_id );
+		$object = $wpdb->get_row( $query );
+
+		if ( empty( $object ) ) {
+			return new WP_Error( 'object_not_found', __( 'Object not found', 'jetpack' ), 404 );
+		}
+
+		$result = array( 'object' => $object );
+
+		// Fetch associated metadata (if this object type has any)
+		if ( ! empty( $this->object_type['meta_type'] ) ) {
+			$result['meta'] = get_metadata( $this->object_type['meta_type'], $this->object_id );
+		}
+
+		// If there is a child linked table (eg: woocommerce_tax_rate_locations), fetch linked records
+		if ( ! empty( $this->object_type['child_table'] ) ) {
+			$child_table    = $wpdb->prefix . $this->object_type['child_table'];
+			$child_id_field = $this->object_type['child_id_field'];
+
+			$query = $wpdb->prepare( 'select * from `' . $child_table . '` where `' . $child_id_field . '` = %d', $this->object_id );
+			$result[ 'children' ] = $wpdb->get_results( $query );
+		}
+
+		return $result;
+	}
+
+}