redirect to the correct page depending on charts created

Author: contactashish13

classes/Visualizer/Module.php

@@ -600,6 +600,21 @@ protected static function hasChartType( $type ) {
 		return $q->found_posts > 0;
 	}
 
+	/**
+	 * Determines how many charts have been created.
+	 */
+	protected static function numberOfCharts() {
+		$args = array(
+			'post_type'      => Visualizer_Plugin::CPT_VISUALIZER,
+			'fields'        => 'ids',
+			'post_status'   => 'publish',
+			'posts_per_page'    => 300,
+		);
+
+		$q = new WP_Query( $args );
+		return $q->found_posts;
+	}
+
 	/**
 	 * Checks if the PRO version is active.
 	 *

classes/Visualizer/Module/Setup.php

@@ -155,7 +155,11 @@ public function onActivation( $plugin ) {
 		}
 
 		if ( $plugin === VISUALIZER_BASENAME ) {
-			wp_redirect( admin_url( 'admin.php?page=viz-support' ) );
+			if ( Visualizer_Module::numberOfCharts() > 0 ) {
+				wp_redirect( admin_url( 'admin.php?page=' . Visualizer_Plugin::NAME ) );
+			} else {
+				wp_redirect( admin_url( 'admin.php?page=viz-support' ) );
+			}
 			exit();
 		}
 	}

templates/docs.php

@@ -22,6 +22,8 @@
 				<?php if ( ! $is_pro ) { ?>
 				<p>We have many more features and charts, and offer email & chat support if you purchase our <a href="<?php echo Visualizer_Plugin::PRO_TEASER_URL; ?>">Pro Version</a>.</p>
 				<?php } ?>
+
+				<p>Ready to begin? Let's <a href="<?php echo admin_url( 'admin.php?page=' . Visualizer_Plugin::NAME . '&vaction=addnew' ); ?>">create a chart!
 			</div>
 		</div>
 	</div>