← All changes | classes/Visualizer/Module/Setup.php +218 -31 3.1.2 → 3.7.9 View file →

		@@ -44,15 +44,20 @@
44	44
45	45	register_activation_hook( VISUALIZER_BASEFILE, array( $this, 'activate' ) );
46	46	register_deactivation_hook( VISUALIZER_BASEFILE, array( $this, 'deactivate' ) );
47	47	$this->_addAction( 'visualizer_schedule_refresh_db', 'refreshDbChart' );
48		- $this->_addFilter( 'visualizer_schedule_refresh_chart', 'refresh_db_for_chart', null, 10, 3 );
	48	+ $this->_addFilter( 'visualizer_schedule_refresh_chart', 'refresh_db_for_chart', 10, 3 );
49	49
	50	+ $this->_addAction( 'admin_init', 'adminInit' );
50	51	$this->_addAction( 'init', 'setupCustomPostTypes' );
51	52	$this->_addAction( 'plugins_loaded', 'loadTextDomain' );
52	53	$this->_addFilter( 'visualizer_logger_data', 'getLoggerData' );
53		- $this->_addFilter( 'visualizer_get_chart_counts', 'getChartCountsByTypeAndMeta' );
	54	+ $this->_addFilter( 'visualizer_get_chart_counts', 'getUsage', 10, 2 );
54	55
	56	+ $this->_addAction( 'init', 'checkIsExistingUser' );
	57	+
	58	+ // only for testing
	59	+ // $this->_addAction( 'admin_init', 'getUsage' );
55	60	}
56	61	/**
57	62	* Fetches the SDK logger data.
58	63	*
		@@ -60,36 +65,82 @@
60	65	*
61	66	* @access public
62	67	*/
63	68	public function getLoggerData( $data ) {
64		- return $this->getChartCountsByTypeAndMeta();
	69	+ return $this->getUsage( $data );
65	70	}
66	71
67	72	/**
68		- * Fetches the types of charts created and their counts.
	73	+ * Fetches the usage of charts.
69	74	*
	75	+ * @param array $data The default data that needs to be sent.
70	76	* @param array $meta_keys An array of name vs. meta keys - to return how many charts have these keys.
71		- * @access private
	77	+ *
	78	+ * @access public
72	79	*/
73		- public function getChartCountsByTypeAndMeta( $meta_keys = array() ) {
74		- $charts = array();
75		- $charts['chart_types'] = array();
76		- // the initial query arguments to fetch charts
	80	+ public function getUsage( $data, $meta_keys = array() ) {
	81	+ $charts = array( 'types' => array(), 'sources' => array(), 'library' => array(), 'permissions' => 0, 'manual_config' => 0, 'scheduled' => 0 );
77	82	$query_args = array(
78	83	'post_type' => Visualizer_Plugin::CPT_VISUALIZER,
79	84	'posts_per_page' => 300,
	85	+ 'post_status' => 'publish',
80	86	'fields' => 'ids',
81	87	'no_rows_found' => false,
82	88	'update_post_meta_cache' => false,
83	89	'update_post_term_cache' => false,
	90	+ );
84	91
	92	+ // default permissions.
	93	+ $default_perms = array(
	94	+ 'read' => 'all',
	95	+ 'read-specific' => null,
	96	+ 'edit' => 'roles',
	97	+ 'edit-specific' => array( 'administrator' ),
85	98	);
86	99
	100	+ // collect all schedules chart ids.
	101	+ $scheduled = array_merge(
	102	+ array_keys( get_option( Visualizer_Plugin::CF_CHART_SCHEDULE, array() ) ),
	103	+ array_keys( get_option( Visualizer_Plugin::CF_DB_SCHEDULE, array() ) )
	104	+ );
87	105	$query = new WP_Query( $query_args );
88	106	while ( $query->have_posts() ) {
89	107	$chart_id = $query->next_post();
90	108	$type = get_post_meta( $chart_id, Visualizer_Plugin::CF_CHART_TYPE, true );
91		- $charts['chart_types'][ $type ] = isset( $charts['chart_types'][ $type ] ) ? $charts['chart_types'][ $type ] + 1 : 1;
	109	+ $charts['types'][ $type ] = isset( $charts['types'][ $type ] ) ? $charts['types'][ $type ] + 1 : 1;
	110	+ $source = get_post_meta( $chart_id, Visualizer_Plugin::CF_SOURCE, true );
	111	+ $charts['sources'][ $source ] = isset( $charts['sources'][ $source ] ) ? $charts['sources'][ $source ] + 1 : 1;
	112	+ $lib = get_post_meta( $chart_id, Visualizer_Plugin::CF_CHART_LIBRARY, true );
	113	+ $charts['library'][ $lib ] = isset( $charts['library'][ $lib ] ) ? $charts['library'][ $lib ] + 1 : 1;
	114	+ $settings = get_post_meta( $chart_id, Visualizer_Plugin::CF_SETTINGS, true );
	115	+ if ( array_key_exists( 'manual', $settings ) && ! empty( $settings['manual'] ) ) {
	116	+ $charts['manual_config'] = $charts['manual_config'] + 1;
	117	+ }
	118	+
	119	+ // phpcs:ignore WordPress.PHP.StrictInArray.FoundNonStrictFalse
	120	+ if ( in_array( $chart_id, $scheduled, false ) ) {
	121	+ $charts['scheduled'] = $charts['scheduled'] + 1;
	122	+ }
	123	+
	124	+ if ( Visualizer_Module::is_pro() ) {
	125	+ $permissions = get_post_meta( $chart_id, Visualizer_PRO::CF_PERMISSIONS, true );
	126	+ if ( empty( $permissions ) ) {
	127	+ continue;
	128	+ }
	129	+ $permissions = $permissions['permissions'];
	130	+ $customized = false;
	131	+ foreach ( $default_perms as $key => $val ) {
	132	+ if ( ! is_array( $val ) && ! is_null( $val ) && isset( $permissions[ $key ] ) && $permissions[ $key ] !== $val ) {
	133	+ $customized = true;
	134	+ } elseif ( is_array( $val ) && ! is_null( $val ) && isset( $permissions[ $key ] ) && count( $permissions[ $key ] ) !== count( $val ) ) {
	135	+ $customized = true;
	136	+ }
	137	+ }
	138	+ if ( $customized ) {
	139	+ $charts['permissions'] = $charts['permissions'] + 1;
	140	+ }
	141	+ }
	142	+
92	143	if ( ! empty( $meta_keys ) ) {
93	144	foreach ( $meta_keys as $name => $key ) {
94	145	$data = get_post_meta( $chart_id, $key, true );
95	146	if ( ! empty( $data ) ) {
		@@ -99,8 +150,9 @@
99	150	}
100	151	}
101	152	}
102	153	}
	154	+
103	155	return $charts;
104	156	}
105	157
106	158	/**
		@@ -139,27 +191,81 @@
139	191
140	192	/**
141	193	* Activate the plugin
142	194	*/
143		- public function activate() {
	195	+ public function activate( $network_wide ) {
	196	+ if ( is_multisite() && $network_wide ) {
	197	+ foreach ( get_sites( array( 'fields' => 'ids' ) ) as $blog_id ) {
	198	+ switch_to_blog( $blog_id );
	199	+ $this->activate_on_site();
	200	+ restore_current_blog();
	201	+ }
	202	+ } else {
	203	+ $this->activate_on_site();
	204	+ }
	205	+ }
	206	+
	207	+ /**
	208	+ * Activates the plugin on a particular blog instance (supports multisite and single site).
	209	+ */
	210	+ private function activate_on_site() {
144	211	wp_clear_scheduled_hook( 'visualizer_schedule_refresh_db' );
145		- wp_schedule_event( strtotime( 'midnight' ) - get_option( 'gmt_offset' ) * HOUR_IN_SECONDS, 'hourly', 'visualizer_schedule_refresh_db' );
	212	+ wp_schedule_event( strtotime( 'midnight' ) - get_option( 'gmt_offset' ) * HOUR_IN_SECONDS, apply_filters( 'visualizer_chart_schedule_interval', 'hourly' ), 'visualizer_schedule_refresh_db' );
	213	+ add_option( 'visualizer-activated', true );
146	214	}
147	215
148	216	/**
149	217	* Deactivate the plugin
150	218	*/
151		- public function deactivate() {
	219	+ public function deactivate( $network_wide ) {
	220	+ if ( is_multisite() && $network_wide ) {
	221	+ foreach ( get_sites( array( 'fields' => 'ids' ) ) as $blog_id ) {
	222	+ switch_to_blog( $blog_id );
	223	+ $this->deactivate_on_site();
	224	+ restore_current_blog();
	225	+ }
	226	+ } else {
	227	+ $this->deactivate_on_site();
	228	+ }
	229	+ }
	230	+
	231	+ /**
	232	+ * Deactivates the plugin on a particular blog instance (supports multisite and single site).
	233	+ */
	234	+ private function deactivate_on_site() {
152	235	wp_clear_scheduled_hook( 'visualizer_schedule_refresh_db' );
	236	+ delete_option( 'visualizer-activated', true );
153	237	}
154	238
155	239	/**
156		- * Refresh the specific chart from the db. This is mostly for charts that have 0 refresh time i.e. live data.
	240	+ * Check if plugin has been activated and then redirect to the correct page.
	241	+ */
	242	+ public function adminInit() {
	243	+ if ( defined( 'TI_UNIT_TESTING' ) ) {
	244	+ return;
	245	+ }
	246	+
	247	+ define( 'VISUALIZER_SURVEY', Visualizer_Module::is_pro() ? 'https://forms.gle/7Zo7FuZbvQ8DTvRi6' : 'https://forms.gle/muMtbcyvHn1aTvmJ7' );
	248	+ // fire any upgrades necessary.
	249	+ Visualizer_Module_Upgrade::upgrade();
	250	+
	251	+ if ( get_option( 'visualizer-activated' ) ) {
	252	+ delete_option( 'visualizer-activated' );
	253	+ if ( ! headers_sent() ) {
	254	+ $page_name = Visualizer_Module::numberOfCharts() > 0 ? Visualizer_Plugin::NAME : 'viz-support';
	255	+ wp_redirect( esc_url_raw( add_query_arg( 'page', $page_name, admin_url( 'admin.php' ) ) ) );
	256	+ exit();
	257	+ }
	258	+ }
	259	+ }
	260	+
	261	+
	262	+ /**
	263	+ * Refresh the specific chart from the db.
157	264	*
158	265	* @param WP_Post $chart The chart object.
159	266	* @param int $chart_id The chart id.
160		- * @param bool $force If this is true, then the chart data will be refreshed no matter if the chart requests live data or cached data.
161		- * If false, data will be refreshed only if the chart requests live data.
	267	+ * @param bool $force If this is true, then the chart data will be force refreshed. If false, data will be refreshed only if the chart requests live data.
162	268	*
163	269	* @access public
164	270	*/
165	271	public function refresh_db_for_chart( $chart, $chart_id, $force = false ) {
		@@ -170,38 +276,97 @@
170	276	if ( ! $chart ) {
171	277	$chart = get_post( $chart_id );
172	278	}
173	279
174		- // check if its a live-data chart or a cached-data chart.
175		- if ( ! $force ) {
176		- $hours = get_post_meta( $chart_id, Visualizer_Plugin::CF_DB_SCHEDULE, true );
177		- if ( ! empty( $hours ) ) {
178		- // cached, bail!
179		- return $chart;
180		- }
	280	+ if ( ! $chart ) {
	281	+ return $chart;
181	282	}
182	283
183	284	// check if the source is correct.
184	285	$source = get_post_meta( $chart_id, Visualizer_Plugin::CF_SOURCE, true );
185		- if ( $source !== 'Visualizer_Source_Query_Params' ) {
186		- return $chart;
	286	+ $load_series = false;
	287	+ switch ( $source ) {
	288	+ case 'Visualizer_Source_Query':
	289	+ // check if its a live-data chart or a cached-data chart.
	290	+ if ( ! $force ) {
	291	+ $hours = get_post_meta( $chart_id, Visualizer_Plugin::CF_DB_SCHEDULE, true );
	292	+ if ( ! empty( $hours ) ) {
	293	+ // cached, bail!
	294	+ return $chart;
	295	+ }
	296	+ }
	297	+
	298	+ $params = get_post_meta( $chart_id, Visualizer_Plugin::CF_DB_QUERY, true );
	299	+ $source = new Visualizer_Source_Query( $params, $chart_id );
	300	+ $source->fetch( false );
	301	+ $load_series = true;
	302	+ break;
	303	+ case 'Visualizer_Source_Json':
	304	+ // check if its a live-data chart or a cached-data chart.
	305	+ if ( ! $force ) {
	306	+ $hours = get_post_meta( $chart_id, Visualizer_Plugin::CF_JSON_SCHEDULE, true );
	307	+ if ( ! empty( $hours ) ) {
	308	+ // cached, bail!
	309	+ return $chart;
	310	+ }
	311	+ }
	312	+
	313	+ $url = get_post_meta( $chart_id, Visualizer_Plugin::CF_JSON_URL, true );
	314	+ $root = get_post_meta( $chart_id, Visualizer_Plugin::CF_JSON_ROOT, true );
	315	+ $paging = get_post_meta( $chart_id, Visualizer_Plugin::CF_JSON_PAGING, true );
	316	+ $series = get_post_meta( $chart_id, Visualizer_Plugin::CF_SERIES, true );
	317	+ $source = new Visualizer_Source_Json( array( 'url' => $url, 'root' => $root, 'paging' => $paging ) );
	318	+ $source->refresh( $series );
	319	+ break;
	320	+ case 'Visualizer_Source_Csv_Remote':
	321	+ // check if its a live-data chart or a cached-data chart.
	322	+ if ( ! $force ) {
	323	+ $hours = get_post_meta( $chart_id, Visualizer_Plugin::CF_CHART_SCHEDULE, true );
	324	+ if ( ! empty( $hours ) ) {
	325	+ // cached, bail!
	326	+ return $chart;
	327	+ }
	328	+ }
	329	+ do_action( 'visualizer_schedule_import' );
	330	+ return get_post( $chart_id );
	331	+ default:
	332	+ return $chart;
187	333	}
188	334
189		- $params = get_post_meta( $chart_id, Visualizer_Plugin::CF_DB_QUERY, true );
190		- $source = new Visualizer_Source_Query( $params );
191		- $source->fetch( false );
192	335	$error = $source->get_error();
193	336	if ( empty( $error ) ) {
194		- update_post_meta( $chart_id, Visualizer_Plugin::CF_SERIES, $source->getSeries() );
	337	+ $this->disableRevisionsTemporarily();
	338	+ if ( $load_series ) {
	339	+ update_post_meta( $chart_id, Visualizer_Plugin::CF_SERIES, $source->getSeries() );
	340	+ }
195	341
	342	+ $allow_html = false;
	343	+ $settings = get_post_meta( $chart_id, Visualizer_Plugin::CF_SETTINGS, true );
	344	+ if ( isset( $settings['allowHtml'] ) && intval( $settings['allowHtml'] ) === 1 ) {
	345	+ $allow_html = true;
	346	+ }
	347	+
	348	+ $allow_html = apply_filters( 'visualizer_allow_html_content', $allow_html, $chart_id, $chart );
	349	+
	350	+ if ( $allow_html ) {
	351	+ kses_remove_filters();
	352	+ }
	353	+
196	354	wp_update_post(
197	355	array(
198	356	'ID' => $chart_id,
199		- 'post_content' => $source->getData(),
	357	+ 'post_content' => $source->getData( get_post_meta( $chart_id, Visualizer_Plugin::CF_EDITABLE_TABLE, true ) ),
200	358	)
201	359	);
202	360
	361	+ if ( $allow_html ) {
	362	+ kses_init_filters();
	363	+ }
	364	+
203	365	$chart = get_post( $chart_id );
	366	+ delete_post_meta( $chart_id, Visualizer_Plugin::CF_ERROR );
	367	+ } else {
	368	+ update_post_meta( $chart_id, Visualizer_Plugin::CF_ERROR, sprintf( 'Error while updating chart: %s', $error ) );
204	369	}
205	370
206	371	return $chart;
207	372	}
		@@ -229,12 +394,34 @@
229	394	if ( $time > $now ) {
230	395	continue;
231	396	}
232	397
233		- $this->refresh_db_for_chart( null, $chart_id, false );
	398	+ // if the time is nigh, we force an update.
	399	+ $this->refresh_db_for_chart( null, $chart_id, true );
	400	+ // Clear existing chart cache.
	401	+ $cache_key = Visualizer_Plugin::CF_CHART_CACHE . '_' . $chart_id;
	402	+ if ( get_transient( $cache_key ) ) {
	403	+ delete_transient( $cache_key );
	404	+ }
234	405	$hours = get_post_meta( $chart_id, Visualizer_Plugin::CF_DB_SCHEDULE, true );
235	406	$new_schedules[ $chart_id ] = time() + $hours * HOUR_IN_SECONDS;
236	407	}
237	408	update_option( Visualizer_Plugin::CF_DB_SCHEDULE, $new_schedules );
	409	+ }
	410	+
	411	+ /**
	412	+ * Save flag for existing users.
	413	+ */
	414	+ public function checkIsExistingUser() {
	415	+ $chart_exists = get_option( 'visualizer-new-user', '' );
	416	+ if ( '' === $chart_exists ) {
	417	+ $charts = get_posts(
	418	+ array(
	419	+ 'post_type' => Visualizer_Plugin::CPT_VISUALIZER,
	420	+ 'fields' => 'ids',
	421	+ )
	422	+ );
	423	+ update_option( 'visualizer-new-user', ! empty( $charts ) ? 'no' : 'yes' );
	424	+ }
238	425	}
239	426
240	427	}