Export data? #41

Author: contactashish13

classes/Visualizer/Module/Chart.php

@@ -61,6 +61,7 @@ public function __construct( Visualizer_Plugin $plugin ) {
 		$this->_addAjaxAction( Visualizer_Plugin::ACTION_CLONE_CHART, 'cloneChart' );
 
         // Added by Ash/Upwork
+		$this->_addAjaxAction( Visualizer_Plugin::ACTION_EXPORT_DATA, 'exportData' );
         if( defined( 'Visualizer_Pro' ) ){
             global $Visualizer_Pro;
             list($action, $name, $class) = $Visualizer_Pro->_getAjaxAction($this);
@@ -597,4 +598,74 @@ public function cloneChart() {
 		exit;
 	}
 
+	/**
+	 * Exports the chart data
+	 *
+	 * @since 1.0.0
+	 *
+	 * @access public
+	 */
+	public function exportData() {
+		$chart_id = $success = false;
+		$capable = current_user_can( 'edit_posts' );
+		if ( $capable ) {
+			$chart_id = filter_input( INPUT_GET, 'chart', FILTER_VALIDATE_INT, array( 'options' => array( 'min_range' => 1 ) ) );
+			if ( $chart_id ) {
+				$chart = get_post( $chart_id );
+				$success = $chart && $chart->post_type == Visualizer_Plugin::CPT_VISUALIZER;
+			}
+		}
+
+		if ( $success ) {
+            $settings   = get_post_meta($chart_id, Visualizer_Plugin::CF_SETTINGS, true);
+            $filename   = $settings["title"];
+            if (empty($filename)) {
+                $filename   = "export.csv";
+            } else {
+                $filename   .= ".csv";
+            }
+            $rows       = array();
+            $series     = get_post_meta($chart_id, Visualizer_Plugin::CF_SERIES, true); 
+            $data       = unserialize($chart->post_content);
+            if (!empty($series)) {
+                $row    = array();
+                foreach ($series as $array) {
+                    $row[]  = $array["label"];
+                }
+                $rows[]     = $row;
+
+                $row    = array();
+                foreach ($series as $array) {
+                    $row[]  = $array["type"];
+                }
+                $rows[]     = $row;
+            }
+
+            if (!empty($data)) {
+                foreach ($data as $array) {
+                    $rows[] = $array;
+                }
+            }
+
+            $fp             = tmpfile();
+            foreach ($rows as $row) {
+                fputcsv($fp, $row);
+            }
+            rewind($fp);
+
+            $csv            = "";
+            while (($array = fgetcsv($fp)) !== FALSE) {
+                if (strlen($csv) > 0) $csv .= PHP_EOL;
+                $csv        .= implode(",", $array);
+            }
+            fclose($fp);
+
+            echo wp_send_json_success(array(
+                "csv"       => $csv,
+                "name"      => $filename
+            ));
+		}
+
+		exit;
+	}
 }

classes/Visualizer/Plugin.php

@@ -49,6 +49,8 @@ class Visualizer_Plugin {
 	const ACTION_CLONE_CHART  = 'visualizer-clone-chart';
 	const ACTION_DELETE_CHART = 'visualizer-delete-chart';
 	const ACTION_UPLOAD_DATA  = 'visualizer-upload-data';
+    // Added by Ash/Upwork
+	const ACTION_EXPORT_DATA  = 'visualizer-export-data';
 
 	// custom filters
 	const FILTER_CHART_WRAPPER_CLASS = 'visualizer-chart-wrapper-class';

classes/Visualizer/Render/Page/Data.php

@@ -108,11 +108,16 @@ protected function _renderSidebarContent() {
                     <a href="<?php echo Visualizer_Plugin::PRO_TEASER_URL;?>" title="<?php echo Visualizer_Plugin::PRO_TEASER_TITLE;?>" class="check-pro-btn" target="_new">
                         <input type="button" class="button preview preview-btn" id="existing-chart-free" value="<?php esc_attr_e( 'Check PRO Version ', Visualizer_Plugin::NAME );?>">
                     </a>
-
-
 <?php
                     }
 
+                    $export_link = add_query_arg( array(
+                        'action' => Visualizer_Plugin::ACTION_EXPORT_DATA,
+                        'chart'  => $this->chart->ID,
+                    ), admin_url( 'admin-ajax.php' ) );
+?>
+                    <input type="button" class="button" id="export-data" value="<?php esc_attr_e( 'Export Data', Visualizer_Plugin::NAME );?>" data-url="<?php echo $export_link;?>">
+<?php
 					echo'<input type="button" name="advanced_button" class="advanced-settings-btn preview-btn" value="'. __( 'Advanced', Visualizer_Plugin::NAME ).' &raquo;">';
                     // Added by Ash/Upwork
 

js/frame.js

@@ -55,6 +55,25 @@
 			$(this).parent().find('.section-description:first').toggle();
 			return false;
 		});
+
+        $("#export-data").on("click", function() {
+            $.ajax({
+                url: $(this).attr("data-url"),
+                method: "get",
+                success: function( data, textStatus, jqXHR ){
+                    var a = document.createElement("a");
+                    document.body.appendChild(a);
+                    a.style = "display: none";
+                    var blob = new Blob([data.data.csv], {type: "application/csv"}),
+                    url = window.URL.createObjectURL(blob);
+                    a.href = url;
+                    a.download = data.data.name;
+                    a.click();
+                    window.URL.revokeObjectURL(url);
+                }
+            });
+        });
+
 	});
 })(jQuery);