| @@ -522,8 +522,12 @@ | ||
| 522 | 522 | |
| 523 | 523 | require_once( ABSPATH . 'wp-admin/includes/file.php' ); |
| 524 | 524 | WP_Filesystem(); |
| 525 | 525 | global $wp_filesystem; |
| 526 | + if ( ! is_a( $wp_filesystem, 'WP_Filesystem_Base' ) ) { | |
| 527 | + $creds = request_filesystem_credentials( site_url() ); | |
| 528 | + wp_filesystem( $creds ); | |
| 529 | + } | |
| 526 | 530 | |
| 527 | 531 | $multisite_arg = '/'; |
| 528 | 532 | if ( is_multisite() && ! is_main_site() ) { |
| 529 | 533 | $multisite_arg = '/sites/' . get_current_blog_id() . '/'; |
| @@ -733,13 +737,22 @@ | ||
| 733 | 737 | /** |
| 734 | 738 | * Gets the features for the provided license type. |
| 735 | 739 | */ |
| 736 | 740 | public static final function get_features_for_license( $plan ) { |
| 741 | + $is_new_personal = apply_filters( 'visualizer_is_new_personal', false ); | |
| 737 | 742 | switch ( $plan ) { |
| 738 | 743 | case 1: |
| 739 | - return array( 'import-wp', 'db-query', 'import-wc-report' ); | |
| 744 | + $features = array( 'import-wp', 'import-wc-report', 'import-file', 'import-url' ); | |
| 745 | + if ( ! $is_new_personal ) { | |
| 746 | + $features[] = 'db-query'; | |
| 747 | + } | |
| 748 | + return $features; | |
| 740 | 749 | case 2: |
| 741 | - return array( 'schedule-chart', 'chart-permissions' ); | |
| 750 | + $features = array( 'schedule-chart', 'chart-permissions', 'import-chart', 'data-filter-configuration', 'frontend-actions' ); | |
| 751 | + if ( $is_new_personal ) { | |
| 752 | + $features[] = 'db-query'; | |
| 753 | + } | |
| 754 | + return $features; | |
| 742 | 755 | } |
| 743 | 756 | } |
| 744 | 757 | |
| 745 | 758 | /** |
