File "WPDA_Query_Builder.php"

Full Path: /home/vantageo/public_html/cache/cache/.wp-cli/wp-content/plugins/wp-data-access/WPDataAccess/Query_Builder/WPDA_Query_Builder.php
File size: 27.06 KB
MIME-type: text/x-php
Charset: utf-8

<?php

namespace WPDataAccess\Query_Builder;

use WPDataAccess\Connection\WPDADB;
use WPDataAccess\Data_Dictionary\WPDA_Dictionary_Lists;
use WPDataAccess\WPDA;
class WPDA_Query_Builder {
    const EXPLAIN_COMMANDS = array(
        'select',
        'delete',
        'insert',
        'replace',
        'update'
    );

    const EXPLAIN_PROTECTED = array(
        'delete',
        'insert',
        'replace',
        'update'
    );

    protected $databases = array();

    protected $default_database = '';

    protected $wpnonce = '';

    protected $sql_hints = 'off';

    public function show() {
        $dbs = WPDA_Dictionary_Lists::get_db_schemas();
        foreach ( $dbs as $db ) {
            $this->databases[] = $db['schema_name'];
        }
        $this->default_database = WPDA::get_user_default_scheme();
        $sql_hints = get_option( 'wpda_sql_hints' );
        if ( false !== $sql_hints ) {
            $this->sql_hints = $sql_hints;
        }
        $this->wpnonce = wp_create_nonce( 'wpda-query-builder-' . WPDA::get_current_user_id() );
        $this->html();
        $this->css();
        $this->js();
    }

    protected function html() {
        $title = 'Query Builder';
        ?>
			<div class="wrap">
				<span style="float:right">
					<label style="font-weight:normal"><input type="checkbox" id="wpda_sql_hints" <?php 
        echo ( 'on' === $this->sql_hints ? 'checked' : '' );
        ?>> Enable SQL hints</label>
				</span>
				<h1 class="wp-heading-inline">
					<?php 
        echo esc_attr( $title );
        ?>
				</h1>
				<div id="wpda_query_builder_open" style="display: none">
					<fieldset class="wpda_fieldset" style="position:relative">
						<legend>
							Queries stored for user <?php 
        echo esc_attr( WPDA::get_current_user_login() );
        ?>
						</legend>
						<label>
							Select
						</label>
						<select id="wpda_query_builder_open_select"></select>
						<a id="wpda_query_builder_open_open"
						   href="javascript:void(0)"
						   onclick="tabOpen()"
						   class="wpda_tooltip button button-primary"
						   title="Open query in new tab">
							<i class="fas fa-folder-open wpda_icon_on_button"></i> Open selected</a>
						<a id="wpda_query_builder_open_openall"
						   href="javascript:void(0)"
						   onclick="tabOpenAll()"
						   class="wpda_tooltip button button-secondary"
						   title="Open all stored queries in new tabs">
							<i class="fas fa-folder-open wpda_icon_on_button"></i> Open all</a>
						<a id="wpda_query_builder_open_delete"
						   href="javascript:void(0)"
						   onclick="deleteQuery()"
						   class="wpda_tooltip button button-secondary"
						   title="Delete query">
							<i class="fas fa-trash wpda_icon_on_button"></i> Delete selected</a>
						<a href="javascript:void(0)"
						   onclick="closeQuery()"
						   class="wpda_tooltip button button-secondary"
						   title="Close panel">
							<i class="fas fa-times-circle wpda_icon_on_button"></i> Cancel</a>
					</fieldset>
				</div>
				<div id="wpda_query_builder">
					<nav class="nav-tab-wrapper">
					</nav>
				</div>
			</div>
			<?php 
    }

    protected function css() {
        // Overwrite default style
        ?>
			<style>
				.ui-corner-all, .ui-corner-top, .ui-corner-right, .ui-corner-tr, .ui-corner-left, .ui-corner-tl {
					border-radius: 0 !important;
				}
				.ui-widget.ui-widget-content {
					border: none;
				}
				.wpda_query_builder_tabs .ui-widget-content {
					background: none;
					color: inherit;
				}
				/*.ui-widget-header {*/
				/*	font-weight: normal;*/
				/*	background: none;*/
				/*	border: none;*/
				/*    border-bottom: 1px solid #c3c4c7;*/
				/*}*/
				.ui-tabs .ui-tabs-nav li.ui-tabs-active .ui-tabs-anchor, .ui-tabs .ui-tabs-nav li.ui-state-disabled .ui-tabs-anchor, .ui-tabs .ui-tabs-nav li.ui-tabs-loading .ui-tabs-anchor {
					cursor: pointer;
				}
				.ui-state-default, .ui-widget-content .ui-state-default, .ui-widget-header .ui-state-default, .ui-button, html .ui-button.ui-state-disabled:hover, html .ui-button.ui-state-disabled:active {
					border: 1px solid #c3c4c7;
					font-weight: bold;
					color: #000 !important;
					background: #dcdcde !important;
				}
				.ui-state-default:hover {
					background: #f0f0f1 !important;
				}
				.ui-state-active, .ui-widget-content .ui-state-active, .ui-widget-header .ui-state-active, a.ui-button:active, .ui-button:active, .ui-button.ui-state-active:hover {
					border: 1px solid #c3c4c7;
					border-bottom: none;
					font-weight: bold;
					color: #000 !important;
					background: #f0f0f1 !important;
				}
				.ui-tabs-anchor {
					color: #50575e !important;
				}
				.ui-tabs-anchor span.dashicons {
					padding-left: 5px;
				}
				.ui-tabs .ui-tabs-panel {
					padding: 0;
					font-family: inherit;
					font-size: inherit;
				}
				.ui-tabs .ui-tabs-nav li {
					margin-left: 4px;
				}
				.ui-tabs .ui-tabs-nav .ui-tabs-anchor {
					line-height: 20px;
				}
			</style>
			<?php 
    }

    protected function js() {
        ?>
				<script type="application/javascript">
					var wpda_default_database = '<?php 
        echo esc_attr( $this->default_database );
        ?>';
					var wpda_databases = '';
					<?php 
        global $wpdb;
        $database_options = '';
        foreach ( $this->databases as $database ) {
            if ( $wpdb->dbname === $database ) {
                $database_printed = "WordPress database ({$database})";
            } else {
                $database_printed = $database;
            }
            $database_options .= '<option value="' . $database . '">' . $database_printed . '</option>';
        }
        echo "wpda_databases = '{$database_options}';";
        // phpcs:ignore WordPress.Security.EscapeOutput
        ?>
					var wpda_home_url = "<?php 
        echo admin_url( 'admin.php' );
        // phpcs:ignore WordPress.Security.EscapeOutput
        ?>";
					var wpda_wpnonce = "<?php 
        echo esc_attr( $this->wpnonce );
        ?>";
					var wpda_loader_url = "<?php 
        echo plugins_url( '../../assets/images/loading.gif', __FILE__ );
        // phpcs:ignore WordPress.Security.EscapeOutput
        ?>";
					var vqbInstalled = false;
					<?php 
        ?>

					function getColumns(schema, table, activeIndex, callback, rebuild = false) {
						const url = location.pathname + '?action=wpda_get_table_widget_info';
						const data = {
							wpdaschema_name: schema,
							table_name: table,
							wpda_wpnonce: '<?php 
        echo esc_attr( wp_create_nonce( 'wpda-getdata-access-' . WPDA::get_current_user_login() ) );
        ?>'
						};
						jQuery.post(
							url,
							data,
							function (data) {
								callback(schema, table, activeIndex, JSON.parse(data), rebuild);
							}
						);
					}

					jQuery(
						function() {
							tabNew();
							jQuery('#wpda_query_builder nav').sortable();
							jQuery('.wpda_tooltip').tooltip();
							jQuery("#wpda_sql_hints").on('change', function() {
								jQuery.ajax({
									method: 'POST',
									url: wpda_home_url + "?action=wpda_query_builder_set_db_hints",
									data: {
										wpda_wpnonce: wpda_wpnonce,
										wpda_sqlhints: jQuery(this).is(":checked") ? 'on' : 'off'
									}
								}).done(
									function (msg) {}
								).fail(
									function (msg) {}
								);

							});
						}
					);
				</script>
			<?php 
    }

    public function execute() {
        $response = array(
            'tabs'   => array(),
            'status' => null,
        );
        WPDA::sent_header( 'application/json' );
        ob_start();
        // Only for admins
        if ( current_user_can( 'manage_options' ) && isset( 
            $_POST['wpda_wpnonce'],
            $_POST['wpda_schemaname'],
            $_POST['wpda_sqlquery'],
            $_POST['wpda_sqllimit'],
            $_POST['wpda_protect']
         ) ) {
            $wpda_wpnonce = sanitize_text_field( wp_unslash( $_POST['wpda_wpnonce'] ) );
            // input var okay.
            $wpda_schemaname = sanitize_text_field( wp_unslash( $_POST['wpda_schemaname'] ) );
            // input var okay.
            $wpda_sqlquery = html_entity_decode( wp_unslash( $_POST['wpda_sqlquery'] ), ENT_QUOTES );
            // input var okay.
            $wpda_sqllimit = sanitize_text_field( wp_unslash( $_POST['wpda_sqllimit'] ) );
            // input var okay.
            $wpda_protect = sanitize_text_field( wp_unslash( $_POST['wpda_protect'] ) );
            // input var okay.
            // Check query
            if ( $this->check_query( $wpda_protect, $wpda_schemaname, $wpda_sqlquery ) ) {
                if ( wp_verify_nonce( $wpda_wpnonce, 'wpda-query-builder-' . WPDA::get_current_user_id() ) ) {
                    // Execute query
                    $wpdadb = WPDADB::get_db_connection( $wpda_schemaname );
                    if ( null !== $wpdadb ) {
                        $wpdadb->suppress_errors( true );
                    }
                    $sqllines = explode( "\n", $wpda_sqlquery );
                    //phpcs:ignore - 8.1 proof
                    $sqlcmds = array();
                    $start_i = 0;
                    for ($i = 0; $i < count( $sqllines ); $i++) {
                        //phpcs:ignore - 8.1 proof
                        if ( '/' === trim( $sqllines[$i] ) ) {
                            $sql = '';
                            for ($j = $start_i; $j < $i; $j++) {
                                if ( '' !== trim( $sqllines[$j] ) ) {
                                    $sql .= "{$sqllines[$j]} ";
                                }
                            }
                            $sqlcmds[] = $sql;
                            $start_i = $i + 1;
                        }
                    }
                    if ( count( $sqlcmds ) > 0 ) {
                        //phpcs:ignore - 8.1 proof
                        $tabs = array();
                        $vars = array();
                        $tmps = array();
                        // Process multiple SQL commands
                        for ($i = 0; $i < count( $sqlcmds ); $i++) {
                            //phpcs:ignore - 8.1 proof
                            if ( '' !== $wpda_sqllimit && 'select' === strtolower( substr( $sqlcmds[$i], 0, 6 ) ) ) {
                                $sqlcmds[$i] .= " limit {$wpda_sqllimit} ";
                            }
                            $columns = WPDA::get_columns_from_query( $wpda_schemaname, $sqlcmds[$i] );
                            // Need to reconnect when switching from local to remote and vice versa
                            $reconnected = false;
                            if ( 'use' === substr( strtolower( trim( $sqlcmds[$i] ) ), 0, 3 ) ) {
                                $use_cmd = explode( ' ', trim( $sqlcmds[$i] ) );
                                //phpcs:ignore - 8.1 proof
                                if ( 2 === count( $use_cmd ) && (strtolower( trim( $use_cmd[1] ) ) !== strtolower( trim( $wpda_schemaname ) ) && 'rdb:' === substr( strtolower( trim( $use_cmd[1] ) ), 0, 4 ) || 'rdb:' === substr( strtolower( trim( $wpda_schemaname ) ), 0, 4 )) ) {
                                    $wpda_schemaname = $use_cmd[1];
                                    $wpdadb = WPDADB::get_db_connection( $wpda_schemaname );
                                    if ( null !== $wpdadb ) {
                                        $wpdadb->suppress_errors( true );
                                    }
                                    $reconnected = true;
                                }
                            }
                            $exequery = true;
                            $var_name = null;
                            $wpdadb_saved = null;
                            switch ( strtolower( substr( trim( $sqlcmds[$i] ), 0, 7 ) ) ) {
                                case 'wpdavar':
                                    $use_cmd = explode( ' ', trim( $sqlcmds[$i] ) );
                                    //phpcs:ignore - 8.1 proof
                                    $var_name = ( isset( $use_cmd[1] ) ? $use_cmd[1] : null );
                                    if ( $var_name !== null ) {
                                        $vars[$var_name] = count( $tabs );
                                        $tmps[$var_name] = $this->get_create_table( $wpdadb, count( $tabs ) );
                                        $exequery = false;
                                    }
                                    break;
                                case 'wpdatmp':
                                    $use_cmd = explode( ' ', trim( $sqlcmds[$i] ) );
                                    $var_name = ( isset( $use_cmd[1] ) ? $use_cmd[1] : null );
                                    if ( $var_name !== null ) {
                                        $wpdadb_saved = $this->create_tmp_table( $wpdadb, $tmps[$var_name], $vars[$var_name] );
                                        $exequery = false;
                                    }
                                    $var_name = null;
                                    // reset
                                    break;
                            }
                            if ( !$reconnected && $exequery && null !== $wpdadb ) {
                                $wpdadb->query( $sqlcmds[$i] );
                            }
                            if ( null === $var_name ) {
                                $status = ( null !== $wpdadb_saved ? $wpdadb_saved : (( null !== $wpdadb ? clone $wpdadb : null )) );
                                $tabs[] = array(
                                    'cmd'     => $sqlcmds[$i],
                                    'status'  => $status,
                                    'columns' => $columns,
                                );
                            } else {
                                $tabs[] = array(
                                    'cmd'     => $sqlcmds[$i],
                                    'status'  => ( isset( $vars[$var_name]['status'] ) ? $vars[$var_name]['status'] : null ),
                                    'wpdavar' => ( isset( $tmps[$var_name] ) ? $tmps[$var_name] : null ),
                                    'columns' => null,
                                );
                            }
                            if ( null === $wpdadb || '' !== $wpdadb->last_error ) {
                                break;
                            }
                        }
                        $response['tabs'] = $tabs;
                    } else {
                        $response['columns'] = WPDA::get_columns_from_query( $wpda_schemaname, $wpda_sqlquery );
                        // Process single SQL command
                        if ( '' !== $wpda_sqllimit && 'select' === strtolower( substr( $wpda_sqlquery, 0, 6 ) ) ) {
                            $wpda_sqlquery .= " limit {$wpda_sqllimit} ";
                        }
                        if ( null !== $wpdadb ) {
                            $wpdadb->query( $wpda_sqlquery );
                        }
                        $response['cmd'] = $wpda_sqlquery;
                        $response['status'] = $wpdadb;
                    }
                } else {
                    $response['status'] = '<strong>WP Data Access error:</strong> Token expired, please refresh page';
                }
            } else {
                $response['status'] = '<strong>WP Data Access error:</strong> Query not allowed - WordPress tables are protected';
            }
        } else {
            $response['status'] = '<strong>WP Data Access error:</strong> Query execution failed';
        }
        ob_end_clean();
        echo json_encode( $response );
    }

    private function create_tmp_table( $wpdadb, $sql, $data ) {
        // Create temporary table
        $wpdadb->query( str_ireplace( 'CREATE TABLE', 'CREATE TEMPORARY TABLE', $sql['create_table'] ) );
        $wpdadb_created = clone $wpdadb;
        // Copy data to temporary table
        foreach ( $data['status']->last_result as $resultset ) {
            $column_values = array();
            foreach ( $resultset as $key => $val ) {
                $column_values[$key] = $val;
            }
            $wpdadb->insert( $sql['table_name'], $column_values );
        }
        return $wpdadb_created;
    }

    private function get_create_table( $wpdadb, $query ) {
        $wpdadb->query( "set sql_quote_show_create = 'ON'" );
        $explain = $wpdadb->get_results( "explain {$query['status']->last_query}", 'ARRAY_A' );
        if ( '' !== $wpdadb->last_error ) {
            return null;
        }
        $table_name = $explain[0]['table'];
        $create_table = $wpdadb->get_results( "show create table `{$table_name}`", 'ARRAY_N' );
        if ( '' !== $wpdadb->last_error ) {
            return null;
        }
        $sql = str_replace( "\n", '', $create_table[0][1] );
        $pos = stripos( $sql, ') ENGINE=' );
        return ( false === $pos ? array(
            'table_name'   => $table_name,
            'create_table' => $sql,
        ) : array(
            'table_name'   => $table_name,
            'create_table' => substr( $sql, 0, $pos + 1 ),
        ) );
    }

    private function check_query( $wpda_protect, $wpda_schemaname, $wpda_sqlquery ) {
        $sql_parts = explode( ' ', trim( $wpda_sqlquery ) );
        //phpcs:ignore - 8.1 proof
        if ( isset( $sql_parts[0] ) && isset( $sql_parts[2] ) && WPDA::is_wp_table( $sql_parts[2] ) && ('drop' === strtolower( $sql_parts[0] ) || 'alter' === strtolower( $sql_parts[0] ) || 'rename' === strtolower( $sql_parts[0] ) || 'truncate' === strtolower( $sql_parts[0] )) ) {
            return false;
        }
        if ( isset( $sql_parts[0] ) && isset( $sql_parts[1] ) && WPDA::is_wp_table( $sql_parts[1] ) && 'truncate' === strtolower( $sql_parts[0] ) ) {
            return false;
        }
        if ( 'false' !== $wpda_protect && isset( $sql_parts[0] ) && in_array( strtolower( $sql_parts[0] ), self::EXPLAIN_COMMANDS ) ) {
            $wpdadb = WPDADB::get_db_connection( $wpda_schemaname );
            $suppress = $wpdadb->suppress_errors( true );
            $explain = $wpdadb->get_results( "explain {$wpda_sqlquery}", 'ARRAY_A' );
            $wpdadb->suppress_errors( $suppress );
            foreach ( $explain as $check ) {
                if ( in_array( strtolower( $check['select_type'] ), self::EXPLAIN_PROTECTED ) && WPDA::is_wp_table( $check['table'] ) ) {
                    return false;
                }
            }
        }
        return true;
    }

    public function save() {
        $response = array(
            'status' => null,
        );
        WPDA::sent_header( 'application/json' );
        // Only for admins
        if ( current_user_can( 'manage_options' ) && isset( 
            $_POST['wpda_wpnonce'],
            $_POST['wpda_schemaname'],
            $_POST['wpda_sqlqueryname'],
            $_POST['wpda_sqlqueryname_old'],
            $_POST['wpda_sqlquery']
         ) ) {
            $wpda_wpnonce = sanitize_text_field( wp_unslash( $_POST['wpda_wpnonce'] ) );
            // input var okay.
            $wpda_schemaname = sanitize_text_field( wp_unslash( $_POST['wpda_schemaname'] ) );
            // input var okay.
            $wpda_sqlqueryname = sanitize_text_field( wp_unslash( $_POST['wpda_sqlqueryname'] ) );
            // input var okay.
            $wpda_sqlqueryname_old = sanitize_text_field( wp_unslash( $_POST['wpda_sqlqueryname_old'] ) );
            // input var okay.
            $wpda_sqlquery = sanitize_textarea_field( wp_unslash( $_POST['wpda_sqlquery'] ) );
            // input var okay.
            $wpda_vqb = null;
            if ( wp_verify_nonce( $wpda_wpnonce, 'wpda-query-builder-' . WPDA::get_current_user_id() ) ) {
                // Save query
                if ( false === $this->upd_query(
                    $wpda_schemaname,
                    $wpda_sqlqueryname,
                    $wpda_sqlquery,
                    $wpda_sqlqueryname_old,
                    $wpda_vqb
                ) ) {
                    $response['status'] = 'Could not save query';
                } else {
                    $response['status'] = 'Query saved';
                }
            } else {
                $response['status'] = 'Token expired, please refresh page';
            }
        } else {
            $response['status'] = 'Could not save query';
        }
        echo json_encode( $response );
    }

    public function open() {
        $response = array(
            'data'   => array(),
            'status' => null,
        );
        WPDA::sent_header( 'application/json' );
        // Only for admins
        if ( current_user_can( 'manage_options' ) && isset( $_POST['wpda_wpnonce'], $_POST['wpda_exclude'] ) ) {
            $wpda_wpnonce = sanitize_text_field( wp_unslash( $_POST['wpda_wpnonce'] ) );
            // input var okay.
            $wpda_exclude = sanitize_text_field( wp_unslash( $_POST['wpda_exclude'] ) );
            // input var okay.
            if ( wp_verify_nonce( $wpda_wpnonce, 'wpda-query-builder-' . WPDA::get_current_user_id() ) ) {
                // Send list of available queries
                $wpda_query_builder_data = $this->get_query_list( $wpda_exclude );
                uksort( $wpda_query_builder_data, 'strnatcasecmp' );
                //phpcs:ignore - 8.1 proof
                $response['data'] = $wpda_query_builder_data;
            } else {
                $response['status'] = 'Token expired, please refresh page';
            }
        } else {
            $response['status'] = 'Query list not available';
        }
        echo json_encode( $response );
    }

    public function delete() {
        $response = array(
            'status' => null,
        );
        WPDA::sent_header( 'application/json' );
        // Only for admins
        if ( current_user_can( 'manage_options' ) && isset( $_POST['wpda_wpnonce'], $_POST['wpda_sqlqueryname'] ) ) {
            $wpda_wpnonce = sanitize_text_field( wp_unslash( $_POST['wpda_wpnonce'] ) );
            // input var okay.
            $wpda_sqlqueryname = sanitize_text_field( wp_unslash( $_POST['wpda_sqlqueryname'] ) );
            // input var okay.
            if ( wp_verify_nonce( $wpda_wpnonce, 'wpda-query-builder-' . WPDA::get_current_user_id() ) ) {
                // Save query
                if ( false === $this->del_query( $wpda_sqlqueryname ) ) {
                    $response['status'] = 'Could not delete query';
                } else {
                    $response['status'] = 'Query deleted';
                }
            } else {
                $response['status'] = 'Token expired, please refresh page';
            }
        } else {
            $response['status'] = 'Could not save query';
        }
        echo json_encode( $response );
    }

    protected function get_query_list( $exclude = '' ) {
        $wpda_query_builder_data = get_user_meta( WPDA::get_current_user_id(), 'wpda_query_builder' );
        if ( count( $wpda_query_builder_data ) > 0 ) {
            //phpcs:ignore - 8.1 proof
            $queries = $wpda_query_builder_data[0];
            if ( null !== $exclude && '' !== $exclude ) {
                $exclude_array = explode( ',', $exclude );
                //phpcs:ignore - 8.1 proof
                foreach ( $exclude_array as $exclude_item ) {
                    unset($queries[$exclude_item]);
                }
            }
            return $queries;
        } else {
            return array();
        }
    }

    protected function upd_query_list( $wpda_query_builder_data ) {
        update_user_meta( WPDA::get_current_user_id(), 'wpda_query_builder', $wpda_query_builder_data );
    }

    protected function get_query( $query_name ) {
        $wpda_query_builder_data = $this->get_query_list();
        if ( is_array( $wpda_query_builder_data ) && isset( $wpda_query_builder_data[$query_name] ) ) {
            return $wpda_query_builder_data[$query_name];
        } else {
            return array();
        }
    }

    protected function upd_query(
        $schema_name,
        $query_name,
        $query_sql,
        $query_name_old,
        $wpda_vqb = null
    ) {
        $wpda_query_builder_data = $this->get_query_list();
        if ( '' !== $query_name_old && $query_name !== $query_name_old ) {
            unset($wpda_query_builder_data[$query_name_old]);
        }
        $wpda_query_builder_data[$query_name] = array(
            'schema_name' => $schema_name,
            'query'       => $query_sql,
            'is_visual'   => null !== $wpda_vqb,
        );
        $this->upd_query_list( $wpda_query_builder_data );
    }

    protected function del_query( $query_name ) {
        $wpda_query_builder_data = $this->get_query_list();
        unset($wpda_query_builder_data[$query_name]);
        $this->upd_query_list( $wpda_query_builder_data );
    }

    public function get_visual_query( $query_name ) {
    }

    protected function upd_visual_query( $query_name, $wpda_vqb ) {
    }

    protected function del_visual_query( $query_name ) {
    }

    protected function rewrite_query_name( $query_name ) {
        return str_replace( ' ', '_', $query_name );
    }

    public function get_db_hints() {
        $response = array(
            'status' => 'OK',
        );
        WPDA::sent_header( 'application/json' );
        if ( current_user_can( 'manage_options' ) && isset( $_REQUEST['wpda_wpnonce'], $_REQUEST['wpda_schemaname'] ) ) {
            $wpda_wpnonce = sanitize_text_field( wp_unslash( $_REQUEST['wpda_wpnonce'] ) );
            // input var okay.
            $wpda_schemaname = sanitize_text_field( wp_unslash( $_REQUEST['wpda_schemaname'] ) );
            // input var okay.
            if ( wp_verify_nonce( $wpda_wpnonce, 'wpda-query-builder-' . WPDA::get_current_user_id() ) ) {
                $tables = WPDA_Dictionary_Lists::get_tables( true, $wpda_schemaname );
                $hints = array();
                foreach ( $tables as $table ) {
                    $columns = WPDA_Dictionary_Lists::get_table_columns( $table['table_name'], $wpda_schemaname );
                    $hint_cols = array();
                    foreach ( $columns as $column ) {
                        $hints[$table['table_name']] = $hint_cols;
                        $hint_cols[] = $column['column_name'];
                    }
                }
                $response['tables'] = $hints;
            } else {
                $response['status'] = 'Token expired, please refresh page';
            }
        } else {
            $response['status'] = 'Invalid arguments';
        }
        echo json_encode( $response );
    }

    public function set_db_hints() {
        $response = array(
            'status' => 'OK',
        );
        WPDA::sent_header( 'application/json' );
        if ( current_user_can( 'manage_options' ) && isset( $_REQUEST['wpda_wpnonce'], $_REQUEST['wpda_sqlhints'] ) ) {
            $wpda_wpnonce = sanitize_text_field( wp_unslash( $_REQUEST['wpda_wpnonce'] ) );
            // input var okay.
            $wpda_sqlhints = sanitize_text_field( wp_unslash( $_REQUEST['wpda_sqlhints'] ) );
            // input var okay.
            if ( wp_verify_nonce( $wpda_wpnonce, 'wpda-query-builder-' . WPDA::get_current_user_id() ) ) {
                update_option( 'wpda_sql_hints', $wpda_sqlhints );
            }
        }
        $response['hint'] = $wpda_sqlhints;
        echo json_encode( $response );
    }

}