can download from library too

Author: contactashish13

classes/Visualizer/Render.php

@@ -190,4 +190,31 @@ protected function get_chart_type( $with_library = false ) {
 		$array = explode( '_', $lib_type );
 		return end( $array );
 	}
+
+	/**
+	 * Determines if the type of chart can have a particular action.
+	 *
+	 * @since ?
+	 *
+	 * @access protected
+	 * @return bool
+	 */
+	protected function can_chart_have_action( $action, $chart_id = null ) {
+		$type = null;
+		if ( ! $chart_id ) {
+			$type = $this->get_chart_type( false );
+		} else {
+			$type = get_post_meta( $chart_id, Visualizer_Plugin::CF_CHART_TYPE, true );
+			$type = ucwords( $type );
+		}
+
+		switch ( $action ) {
+			case 'image':
+				return ! in_array( $type, array( 'Gauge', 'Tabular', 'DataTable', 'Table' ), true );
+		}
+
+		return true;
+	}
+
+
 }

classes/Visualizer/Render/Library.php

@@ -336,6 +336,9 @@ private function _renderChartBox( $placeholder_id, $chart_id ) {
 		echo '<a class="visualizer-chart-action visualizer-chart-clone" href="', $clone_url, '" title="', esc_attr__( 'Clone', 'visualizer' ), '"></a>';
 		echo '<a class="visualizer-chart-action visualizer-chart-edit" href="javascript:;" title="', esc_attr__( 'Edit', 'visualizer' ), '" data-chart="', $chart_id, '"></a>';
 		echo '<a class="visualizer-chart-action visualizer-chart-export" href="javascript:;" title="', esc_attr__( 'Export', 'visualizer' ), '" data-chart="', $export_link, '"></a>';
+		if ( $this->can_chart_have_action( 'image', $chart_id ) ) {
+			echo '<a class="visualizer-chart-action visualizer-chart-image" href="javascript:;" title="', esc_attr__( 'Download as image', 'visualizer' ), '" data-chart="visualizer-', $chart_id, '" data-chart-title="', $title, '"></a>';
+		}
 		echo '<span class="visualizer-chart-shortcode" title="', esc_attr__( 'Click to select', 'visualizer' ), '">';
 		echo '&nbsp;[visualizer id=&quot;', $chart_id, '&quot;]&nbsp;';
 		echo '</span>';

classes/Visualizer/Render/Sidebar.php

@@ -262,8 +262,7 @@ protected function _renderActionSettings() {
 			);
 
 			// not all charts support downloading as an image.
-			$class = $this->get_chart_type( false );
-			$disabled   = in_array( $class, array( 'Gauge', 'Tabular', 'DataTable', 'Table' ), true );
+			$disabled   = ! $this->can_chart_have_action( 'image' );
 			self::_renderCheckboxItem(
 				esc_html__( 'Download Image', 'visualizer' ),
 				'actions[]',

css/library.css

@@ -107,6 +107,10 @@
 	background-position: -46px -64px;
 }
 
+.visualizer-chart-image {
+	background-position: -206px -129px;
+}
+
 .visualizer-nochart-clone,
 .visualizer-nochart-delete,
 .visualizer-nochart-edit {

js/library.js

@@ -139,6 +139,11 @@
             return false;
         });
 
+        $(".visualizer-chart-image").on("click", function () {
+            $('body').trigger('visualizer:action:specificchart', {action: 'image', id: $(this).attr("data-chart"), data: null, dataObj: {name: $(this).attr("data-chart-title")}});
+            return false;
+        });
+
         // if vaction=addnew is found as a GET request parameter, show the modal.
         if(location.href.indexOf('vaction=addnew') !== -1){
             $('.add-new-chart').trigger('click');

js/render-chartjs.js

@@ -406,6 +406,7 @@
     });
 
     // front end actions
+    // 'image' is also called from the library
     $('body').on('visualizer:action:specificchart', function(event, v){
         var id = v.id;
         if(typeof rendered_charts[id] === 'undefined'){

js/render-google.js

@@ -432,6 +432,7 @@ var __visualizer_chart_images   = [];
     });
 
     // front end actions
+    // 'image' is also called from the library
     $('body').on('visualizer:action:specificchart', function(event, v){
         var id = v.id;
         if(typeof rendered_charts[id] === 'undefined'){