<?php
// phpcs:ignore Generic.Commenting.DocComment.MissingShort
/** @noinspection PhpIllegalPsrClassPathInspection */
namespace WPForms\Helpers;
// phpcs:ignore WPForms.PHP.UseStatement.UnusedUseStatement
use WPForms_DB;
use WPForms_Lite;
use WPForms_Pro;
/**
* DB helpers.
*
* @since 1.8.7
*/
class DB {
/**
* Existing tables transient name.
*
* @since 1.8.7
* @since 1.9.0 Changed from 'wpforms_existing_tables' to 'existing_tables'
*
* @var string
*/
const EXISTING_TABLES_TRANSIENT_NAME = 'existing_tables';
/**
* Existing tables transient expiration, sec.
*
* @since 1.8.7
*
* @var int
* @noinspection SummerTimeUnsafeTimeManipulationInspection
*/
const EXISTING_TABLES_TRANSIENT_EXPIRATION = WEEK_IN_SECONDS; // A week.
/**
* Existing tables.
*
* @since 1.8.7
*
* @var array
*/
private static $existing_tables = [];
/**
* Get the list of existing tables and cache the result.
*
* @since 1.8.7
*
* @param string $table_name Table name. Can have SQL wildcard.
*
* @return array List of table names.
*/
public static function get_existing_tables( string $table_name ): array {
global $wpdb;
/**
* Filters existence of a table before a request to the database is executed.
*
* @since 1.8.7
*
* @param array $tables Existing tables with given table name.
* @param string $table_name Table name.
*/
$tables = (array) apply_filters( 'wpforms_helpers_db_pre_get_existing_tables', [], $table_name );
if ( $tables ) {
return $tables;
}
$tables = self::get_existing_tables_cache( $table_name );
if ( $tables ) {
return $tables;
}
// phpcs:disable WordPress.DB.DirectDatabaseQuery.DirectQuery, WordPress.DB.DirectDatabaseQuery.NoCaching
$tables = $wpdb->get_results(
$wpdb->prepare( 'SHOW TABLES LIKE %s', $table_name ),
'ARRAY_N'
);
// phpcs:enable WordPress.DB.DirectDatabaseQuery.DirectQuery, WordPress.DB.DirectDatabaseQuery.NoCaching
$tables = ! empty( $tables ) ? wp_list_pluck( $tables, 0 ) : [];
self::set_existing_tables_cache( $tables, $table_name );
return self::$existing_tables[ $table_name ] ?? [];
}
/**
* Get the list of all existing custom tables starting with `wpforms_*` and cache the result.
*
* @since 1.8.7
*
* @return array List of table names.
*/
public static function get_existing_custom_tables(): array {
global $wpdb;
return self::get_existing_tables( "{$wpdb->prefix}wpforms_%" );
}
/**
* Check if the database table exists and cache the result.
*
* @since 1.8.7
*
* @param string $table_name Table name. Can have SQL wildcard.
*
* @return bool
*/
public static function table_exists( string $table_name ): bool {
/**
* Filters existence of a table before a request to the database is executed.
*
* @since 1.8.7
*
* @param integer $exists Table exists.
* @param string $table_name Table name.
*/
if ( apply_filters( 'wpforms_helpers_db_pre_table_exists', false, $table_name ) ) {
return true;
}
foreach ( self::get_existing_tables( $table_name ) as $existing_table ) {
if ( self::wildcard_match( $table_name, $existing_table ) ) {
return true;
}
}
return false;
}
/**
* Get the list of existing tables from cache.
*
* @since 1.8.7
*
* @param string $table_name Table name. Can have SQL wildcard.
*
* @return array List of table names.
*/
private static function get_existing_tables_cache( string $table_name ): array {
$tables = Transient::get( self::EXISTING_TABLES_TRANSIENT_NAME );
self::$existing_tables = $tables ? $tables : [];
return self::$existing_tables[ $table_name ] ?? [];
}
/**
* Set existing tables cache.
*
* @since 1.8.7
*
* @param array $tables Existing tables with given table name.
* @param string $table_name Table name.
*
* @return void
*/
private static function set_existing_tables_cache( array $tables, string $table_name ) {
if ( empty( $tables ) ) {
return;
}
self::$existing_tables[ $table_name ] = $tables;
/**
* Filters existing tables transient expiration time.
*
* @since 1.8.7
*
* @param integer $expiration Expiration time.
*/
$expiration = apply_filters( 'wpforms_helpers_db_existing_tables_transient_expiration', self::EXISTING_TABLES_TRANSIENT_EXPIRATION );
Transient::set( self::EXISTING_TABLES_TRANSIENT_NAME, self::$existing_tables, $expiration );
}
/**
* Flush existing tables cache.
*
* @since 1.9.0
*
* @return void
*/
public static function flush_existing_tables_cache() {
self::$existing_tables = [];
Transient::delete( self::EXISTING_TABLES_TRANSIENT_NAME );
}
/**
* Wildcard match.
* Works as MySQL LIKE match.
*
* @since 1.8.7
*
* @param string $pattern Pattern.
* @param string $subject String to search into.
*
* @return false|int
*/
private static function wildcard_match( string $pattern, string $subject ) {
$regex = str_replace(
[ '%', '_' ], // MySQL wildcard chars.
[ '.*', '.' ], // Regexp chars.
preg_quote( $pattern, '/' )
);
return preg_match( '/^' . $regex . '$/is', $subject );
}
/**
* Check if all custom tables exist.
*
* @since 1.9.0
*
* @return bool True if all custom tables exist. False if any is missing.
*/
public static function custom_tables_exist(): bool {
global $wpdb;
$existing_tables = self::get_existing_custom_tables();
$custom_tables = wpforms()->is_pro() ? WPForms_Pro::CUSTOM_TABLES : WPForms_Lite::CUSTOM_TABLES;
foreach ( $custom_tables as $table_name => $handler_class ) {
if ( ! in_array( $wpdb->prefix . $table_name, $existing_tables, true ) ) {
return false;
}
}
return true;
}
/**
* Create all custom DB tables.
*
* @since 1.9.0
*
* @param bool $flush_cache Clear existing custom tables cache.
*
* @noinspection PhpPossiblePolymorphicInvocationInspection
*/
public static function create_custom_tables( bool $flush_cache = false ) {
global $wpdb;
if ( $flush_cache ) {
self::flush_existing_tables_cache();
}
$existing_tables = self::get_existing_custom_tables();
$custom_tables = wpforms()->is_pro() ? WPForms_Pro::CUSTOM_TABLES : WPForms_Lite::CUSTOM_TABLES;
$created = false;
foreach ( $custom_tables as $table_name => $handler_class ) {
if ( in_array( $wpdb->prefix . $table_name, $existing_tables, true ) ) {
continue;
}
/**
* Child class of WPForms_DB.
*
* @var $handler WPForms_DB
*/
$handler = new $handler_class();
// Create a table.
$handler->create_table();
$created = true;
}
if ( $created ) {
Transient::delete( self::EXISTING_TABLES_TRANSIENT_NAME );
}
}
}