<?php
/**
* Shell - class to work with $wpdb global object
*/
class DbWpf {
/**
* Execute query and return results
*
* @param string $query query to be executed
* @param string $get what must be returned - one value (one), one row (row), one col (col) or all results (all - by default)
* @param const $outputType type of returned data
* @return mixed data from DB
*/
public static $query = '';
public static function get( $query, $get = 'all', $outputType = ARRAY_A ) {
global $wpdb;
$get = strtolower($get);
$res = null;
$query = self::prepareQuery($query);
self::$query = $query;
$wpdb->wpf_prepared_query = $query;
switch ($get) {
case 'one':
$res = $wpdb->get_var($wpdb->wpf_prepared_query);
break;
case 'row':
$res = $wpdb->get_row($wpdb->wpf_prepared_query, $outputType);
break;
case 'col':
$res = $wpdb->get_col($wpdb->wpf_prepared_query);
break;
case 'all':
default:
$res = $wpdb->get_results($wpdb->wpf_prepared_query, $outputType);
break;
}
return empty($wpdb->last_error) ? $res : false;
}
/**
* Execute one query
*
* @return query results
*/
public static function query( $query, $affected = false ) {
global $wpdb;
$wpdb->wpf_prepared_query = self::prepareQuery($query);
return $affected ? $wpdb->query($wpdb->wpf_prepared_query) : ( $wpdb->query($wpdb->wpf_prepared_query) === false ? false : true );
}
/**
* Get last insert ID
*
* @return int last ID
*/
public static function insertID() {
global $wpdb;
return $wpdb->insert_id;
}
/**
* Get number of rows returned by last query
*
* @return int number of rows
*/
public static function numRows() {
global $wpdb;
return $wpdb->num_rows;
}
/**
* Replace prefixes in custom query. Suported next prefixes:
* #__ Worwpfess prefix
* ^__ Store plugin tables prefix (@see WPF_DB_PREF if config.php)
*
* @__ Compared of WP table prefix + Store plugin prefix (@example wp_s_)
* @param string $query query to be executed
*/
public static function prepareQuery( $query ) {
global $wpdb;
return str_replace(
array('#__', '^__', '@__'),
array($wpdb->prefix, WPF_DB_PREF, $wpdb->prefix . WPF_DB_PREF),
$query);
}
public static function getTableName( $name ) {
global $wpdb;
return $wpdb->prefix . WPF_DB_PREF . $name;
}
public static function getError() {
global $wpdb;
return $wpdb->last_error;
}
public static function lastID() {
global $wpdb;
return $wpdb->insert_id;
}
public static function timeToDate( $timestamp = 0 ) {
if ($timestamp) {
if (!is_numeric($timestamp)) {
$timestamp = dateToTimestampWpf($timestamp);
}
return gmdate('Y-m-d', $timestamp);
} else {
return gmdate('Y-m-d');
}
}
public static function dateToTime( $date ) {
if (empty($date)) {
return '';
}
if (strpos($date, WPF_DATE_DL)) {
return dateToTimestampWpf($date);
}
$arr = explode('-', $date);
return dateToTimestampWpf($arr[2] . WPF_DATE_DL . $arr[1] . WPF_DATE_DL . $arr[0]);
}
public static function exist( $table, $column = '', $value = '' ) {
if (empty($column) && empty($value)) { //Check if table exist
$res = self::get('SHOW TABLES LIKE "' . $table . '"', 'one');
} elseif (empty($value)) { //Check if column exist
$res = self::get('SHOW COLUMNS FROM ' . $table . ' LIKE "' . $column . '"', 'one');
} else { //Check if value in column table exist
$res = self::get('SELECT COUNT(*) AS total FROM ' . $table . ' WHERE ' . $column . ' = "' . $value . '"', 'one');
}
return !empty($res);
}
public static function prepareHtml( $d ) {
if (is_array($d)) {
foreach ($d as $i => $el) {
$d[ $i ] = self::prepareHtml( $el );
}
} else {
$d = esc_html($d);
}
return $d;
}
public static function prepareHtmlIn( $d ) {
if (is_array($d)) {
foreach ($d as $i => $el) {
$d[ $i ] = self::prepareHtml( $el );
}
} else {
$d = wp_filter_nohtml_kses($d);
}
return $d;
}
public static function escape( $data ) {
global $wpdb;
return $wpdb->_escape($data);
}
public static function getAutoIncrement( $table ) {
return (int) self::get('SELECT AUTO_INCREMENT
FROM information_schema.tables
WHERE table_name = "' . $table . '"
AND table_schema = DATABASE( );', 'one');
}
public static function setAutoIncrement( $table, $autoIncrement ) {
return self::query('ALTER TABLE `' . $table . '` AUTO_INCREMENT = ' . $autoIncrement . ';');
}
public static function existsTableColumn( $table, $column ) {
return self::get("SELECT 1 FROM INFORMATION_SCHEMA.COLUMNS WHERE table_name='" . $table . "' AND table_schema=DATABASE( ) AND column_name='" . $column . "'", 'one') == 1;
}
}