| @@ -336,9 +336,9 @@ | ||
| 336 | 336 | * |
| 337 | 337 | * @access public |
| 338 | 338 | * @return array |
| 339 | 339 | */ |
| 340 | - public static function get_all_db_tables_column_mapping() { | |
| 340 | + public static function get_all_db_tables_column_mapping( $chart_id, $use_filter = true ) { | |
| 341 | 341 | $mapping = array(); |
| 342 | 342 | $tables = self::get_db_tables(); |
| 343 | 343 | foreach ( $tables as $table ) { |
| 344 | 344 | $cols = self::get_db_table_columns( $table, true ); |
| @@ -343,8 +343,11 @@ | ||
| 343 | 343 | foreach ( $tables as $table ) { |
| 344 | 344 | $cols = self::get_db_table_columns( $table, true ); |
| 345 | 345 | $names = wp_list_pluck( $cols, 'name' ); |
| 346 | 346 | $mapping[ $table ] = $names; |
| 347 | + } | |
| 348 | + if ( $use_filter ) { | |
| 349 | + return apply_filters( 'visualizer_db_tables_column_mapping', $mapping, $chart_id ); | |
| 347 | 350 | } |
| 348 | 351 | return $mapping; |
| 349 | 352 | } |
| 350 | 353 | |
