Add option to enable/disable save chart image #825

Author: girishpanchal30

classes/Visualizer/Module/Chart.php

@@ -527,8 +527,10 @@ public function renderChartPages() {
 			defined( 'WP_TESTS_DOMAIN' ) ? wp_die() : exit();
 		}
 
+		$_POST['save_chart_image'] = isset( $_POST['save_chart_image'] ) && 'yes' === $_POST['save_chart_image'] ? true : false;
+
 		if ( isset( $_POST['chart-img'] ) && ! empty( $_POST['chart-img'] ) ) {
-			$attachment_id = $this->save_chart_image( $_POST['chart-img'], $chart_id );
+			$attachment_id = $this->save_chart_image( $_POST['chart-img'], $chart_id, $_POST['save_chart_image'] );
 			if ( $attachment_id ) {
 				update_post_meta( $chart_id, Visualizer_Plugin::CF_CHART_IMAGE, $attachment_id );
 			}
@@ -1437,9 +1439,20 @@ public function saveFilter() {
 	 *
 	 * @param string $base64_img Chart image.
 	 * @param int    $chart_id Chart ID.
+	 * @param bool   $save_attachment Save attachment.
 	 * @return attachment ID
 	 */
-	public function save_chart_image( $base64_img, $chart_id ) {
+	public function save_chart_image( $base64_img, $chart_id, $save_attachment = true ) {
+		// Delete old chart image.
+		$old_attachment_id = get_post_meta( $chart_id, Visualizer_Plugin::CF_CHART_IMAGE, true );
+		if ( $old_attachment_id ) {
+			wp_delete_attachment( $old_attachment_id, true );
+		}
+
+		if ( ! $save_attachment ) {
+			return 0;
+		}
+
 		// Upload dir.
 		$upload_dir  = wp_upload_dir();
 		$upload_path = str_replace( '/', DIRECTORY_SEPARATOR, $upload_dir['path'] ) . DIRECTORY_SEPARATOR;
@@ -1451,12 +1464,6 @@ public function save_chart_image( $base64_img, $chart_id ) {
 		$file_type       = 'image/png';
 		$hashed_filename = $filename;
 
-		// Delete old chart image.
-		$old_attachment_id = get_post_meta( $chart_id, Visualizer_Plugin::CF_CHART_IMAGE, true );
-		if ( $old_attachment_id ) {
-			wp_delete_attachment( $old_attachment_id, true );
-		}
-
 		// Save the image in the uploads directory.
 		require_once ABSPATH . '/wp-admin/includes/file.php';
 		\WP_Filesystem();

classes/Visualizer/Render/Sidebar.php

@@ -510,4 +510,26 @@ protected function load_dependent_assets( $libs ) {
 		}
 
 	}
+
+	/**
+	 * Renders save chart as image setting group.
+	 *
+	 * @access protected
+	 */
+	protected function _renderChartImageSettings() {
+			// Default enable if amp is active.
+			$is_amp = function_exists( 'amp_is_enabled' ) && amp_is_enabled();
+			$this->save_chart_image = null === $this->save_chart_image && $is_amp ? true : $this->save_chart_image;
+
+			self::_renderSectionStart( esc_html__( 'Save chart as an image inside Media Library', 'visualizer' ), false );
+				self::_renderCheckboxItem(
+					esc_html__( 'Save inside media library?', 'visualizer' ),
+					'save_chart_image',
+					$this->save_chart_image ? true : false,
+					'yes',
+					esc_html__( 'To enable save the image as an inside media library.', 'visualizer' ),
+					false
+				);
+			self::_renderSectionEnd();
+	}
 }

classes/Visualizer/Render/Sidebar/ChartJS.php

@@ -310,6 +310,9 @@ protected function _renderGeneralSettings() {
 				''
 			);
 			self::_renderSectionEnd();
+
+			self::_renderChartImageSettings();
+
 		self::_renderGroupEnd();
 	}
 

classes/Visualizer/Render/Sidebar/Google.php

@@ -251,6 +251,9 @@ protected function _renderGeneralSettings() {
 				''
 			);
 			self::_renderSectionEnd();
+
+			self::_renderChartImageSettings();
+
 			do_action( 'visualizer_chart_settings', get_class( $this ), $this->_data, 'general', array( 'generic' => true ) );
 
 		self::_renderGroupEnd();

classes/Visualizer/Render/Sidebar/Type/DataTable/DataTable.php

@@ -149,6 +149,9 @@ protected function _renderGeneralSettings() {
 				);
 
 			self::_renderSectionEnd();
+
+			self::_renderChartImageSettings();
+
 		self::_renderGroupEnd();
 	}
 

classes/Visualizer/Render/Sidebar/Type/DataTable/Tabular.php

@@ -147,6 +147,9 @@ protected function _renderGeneralSettings() {
 				);
 
 			self::_renderSectionEnd();
+
+			self::_renderChartImageSettings();
+
 		self::_renderGroupEnd();
 	}
 

classes/Visualizer/Render/Sidebar/Type/GoogleCharts/Gauge.php

@@ -102,6 +102,8 @@ protected function _renderGeneralSettings() {
 			);
 			self::_renderSectionEnd();
 
+			self::_renderChartImageSettings();
+
 		self::_renderGroupEnd();
 	}
 

classes/Visualizer/Render/Sidebar/Type/GoogleCharts/Geo.php

@@ -86,6 +86,9 @@ protected function _renderGeneralSettings() {
 				''
 			);
 			self::_renderSectionEnd();
+
+			self::_renderChartImageSettings();
+
 		self::_renderGroupEnd();
 	}
 

classes/Visualizer/Render/Sidebar/Type/GoogleCharts/Tabular.php

@@ -94,6 +94,9 @@ protected function _renderGeneralSettings() {
 					$this->creator,
 					''
 				);
+
+			self::_renderChartImageSettings();
+
 			self::_renderGroupEnd();
 	}