| @@ -129,12 +129,16 @@ | ||
| 129 | 129 | * |
| 130 | 130 | * @return string The new html code. |
| 131 | 131 | */ |
| 132 | 132 | public function addProUpsell( $old, $feature = null ) { |
| 133 | - $biz_features = array( 'schedule-chart', 'chart-permissions', 'db-query' ); | |
| 133 | + $pro_features = Visualizer_Module::get_features_for_license( 1 ); | |
| 134 | + $biz_features = Visualizer_Module::get_features_for_license( 2 ); | |
| 134 | 135 | $return = ''; |
| 135 | 136 | $feature = strval( $feature ); |
| 136 | - if ( empty( $feature ) || ( in_array( $feature, $biz_features, true ) && ! apply_filters( 'visualizer_is_business', false ) ) ) { | |
| 137 | + if ( empty( $feature ) || | |
| 138 | + ( in_array( $feature, $biz_features, true ) && ! apply_filters( 'visualizer_is_business', false ) ) || | |
| 139 | + ( in_array( $feature, $pro_features, true ) && ! Visualizer_Module::is_pro() ) | |
| 140 | + ) { | |
| 137 | 141 | $msg = sprintf( __( 'Upgrade to %s to activate this feature!', 'visualizer' ), 'PRO' ); |
| 138 | 142 | if ( Visualizer_Module::is_pro() && in_array( $feature, $biz_features, true ) ) { |
| 139 | 143 | $msg = sprintf( __( 'Upgrade your license to at least the %s version to activate this feature!', 'visualizer' ), 'DEVELOPER' ); |
| 140 | 144 | } |
