mirror of
https://github.com/WenPai-org/wp-domain-mapping.git
synced 2025-08-03 22:39:47 +08:00
341 lines
7.8 KiB
PHP
341 lines
7.8 KiB
PHP
<?php
|
|
/**
|
|
* Common functions for WP Domain Mapping plugin
|
|
*
|
|
* @package WP Domain Mapping
|
|
*/
|
|
|
|
// Prevent direct access
|
|
if ( ! defined( 'ABSPATH' ) ) {
|
|
exit;
|
|
}
|
|
|
|
/**
|
|
* Ensure URL has a protocol
|
|
*
|
|
* @param string $domain Domain name
|
|
* @return string Domain with protocol
|
|
*/
|
|
function dm_ensure_protocol( $domain ) {
|
|
if ( preg_match( '#^https?://#', $domain ) ) {
|
|
return $domain;
|
|
}
|
|
return 'http://' . $domain;
|
|
}
|
|
|
|
/**
|
|
* Clean domain name (remove protocol and trailing slash)
|
|
*
|
|
* @param string $domain Domain name
|
|
* @return string Cleaned domain
|
|
*/
|
|
function dm_clean_domain( $domain ) {
|
|
// Remove protocol
|
|
$domain = preg_replace( '#^https?://#', '', $domain );
|
|
|
|
// Remove trailing slash
|
|
$domain = rtrim( $domain, '/' );
|
|
|
|
// Convert IDN to ASCII (Punycode)
|
|
if ( function_exists( 'idn_to_ascii' ) && preg_match( '/[^a-z0-9\-\.]/i', $domain ) ) {
|
|
if (defined('INTL_IDNA_VARIANT_UTS46')) {
|
|
// PHP 7.2+
|
|
$domain = idn_to_ascii( $domain, 0, INTL_IDNA_VARIANT_UTS46 );
|
|
} else {
|
|
// PHP < 7.2
|
|
$domain = idn_to_ascii( $domain );
|
|
}
|
|
}
|
|
|
|
return $domain;
|
|
}
|
|
|
|
/**
|
|
* Validate a domain name
|
|
*
|
|
* @param string $domain The domain
|
|
* @return bool True if valid
|
|
*/
|
|
function dm_validate_domain( $domain ) {
|
|
// Basic validation
|
|
return (bool) preg_match( '/^[a-z0-9]+([\-\.]{1}[a-z0-9]+)*\.[a-z]{2,}$/i', $domain );
|
|
}
|
|
|
|
/**
|
|
* Display IDN warning message
|
|
*
|
|
* @return string Warning message
|
|
*/
|
|
function dm_idn_warning() {
|
|
return sprintf(
|
|
/* translators: %s: URL to punycode converter */
|
|
wp_kses(
|
|
__( 'International Domain Names should be in <a href="%s" target="_blank">punycode</a> format.', 'wp-domain-mapping' ),
|
|
array( 'a' => array( 'href' => array(), 'target' => array() ) )
|
|
),
|
|
'https://www.punycoder.com/'
|
|
);
|
|
}
|
|
|
|
/**
|
|
* Check if user is a site admin
|
|
*
|
|
* @return bool True if user is a site admin
|
|
*/
|
|
function dm_is_site_admin() {
|
|
return current_user_can( 'manage_network' );
|
|
}
|
|
|
|
/**
|
|
* Get domain mapping table names
|
|
*
|
|
* @return array Array of table names
|
|
*/
|
|
function dm_get_table_names() {
|
|
global $wpdb;
|
|
|
|
return array(
|
|
'domains' => $wpdb->base_prefix . WP_DOMAIN_MAPPING_TABLE_DOMAINS,
|
|
'logins' => $wpdb->base_prefix . WP_DOMAIN_MAPPING_TABLE_LOGINS,
|
|
'logs' => $wpdb->base_prefix . WP_DOMAIN_MAPPING_TABLE_LOGS,
|
|
);
|
|
}
|
|
|
|
/**
|
|
* Log domain mapping action
|
|
*
|
|
* @param string $action Action type
|
|
* @param string $domain Domain name
|
|
* @param int $blog_id Blog ID
|
|
* @param int $user_id User ID (optional)
|
|
*/
|
|
function dm_log_action( $action, $domain, $blog_id, $user_id = null ) {
|
|
global $wpdb;
|
|
|
|
if ( null === $user_id ) {
|
|
$user_id = get_current_user_id();
|
|
}
|
|
|
|
$tables = dm_get_table_names();
|
|
|
|
$wpdb->insert(
|
|
$tables['logs'],
|
|
array(
|
|
'user_id' => $user_id,
|
|
'action' => $action,
|
|
'domain' => $domain,
|
|
'blog_id' => $blog_id
|
|
),
|
|
array( '%d', '%s', '%s', '%d' )
|
|
);
|
|
}
|
|
|
|
/**
|
|
* Get domain by name
|
|
*
|
|
* @param string $domain Domain name
|
|
* @return object|null Domain object or null
|
|
*/
|
|
function dm_get_domain_by_name( $domain ) {
|
|
global $wpdb;
|
|
|
|
$tables = dm_get_table_names();
|
|
|
|
return $wpdb->get_row( $wpdb->prepare(
|
|
"SELECT * FROM {$tables['domains']} WHERE domain = %s",
|
|
$domain
|
|
));
|
|
}
|
|
|
|
/**
|
|
* Get domains by blog ID
|
|
*
|
|
* @param int $blog_id Blog ID
|
|
* @return array Array of domain objects
|
|
*/
|
|
function dm_get_domains_by_blog_id( $blog_id ) {
|
|
global $wpdb;
|
|
|
|
$tables = dm_get_table_names();
|
|
|
|
return $wpdb->get_results( $wpdb->prepare(
|
|
"SELECT * FROM {$tables['domains']} WHERE blog_id = %d ORDER BY active DESC, domain ASC",
|
|
$blog_id
|
|
));
|
|
}
|
|
|
|
/**
|
|
* Add a new domain mapping
|
|
*
|
|
* @param int $blog_id Blog ID
|
|
* @param string $domain Domain name
|
|
* @param int $active Whether domain is primary (1) or not (0)
|
|
* @return bool|int False on failure, insert ID on success
|
|
*/
|
|
function dm_add_domain( $blog_id, $domain, $active = 0 ) {
|
|
global $wpdb;
|
|
|
|
$tables = dm_get_table_names();
|
|
$domain = dm_clean_domain( $domain );
|
|
|
|
// Validate domain
|
|
if ( ! dm_validate_domain( $domain ) ) {
|
|
return false;
|
|
}
|
|
|
|
// Check if domain already exists
|
|
if ( dm_get_domain_by_name( $domain ) ) {
|
|
return false;
|
|
}
|
|
|
|
// If setting as primary, reset other domains
|
|
if ( $active ) {
|
|
$wpdb->update(
|
|
$tables['domains'],
|
|
array( 'active' => 0 ),
|
|
array( 'blog_id' => $blog_id ),
|
|
array( '%d' ),
|
|
array( '%d' )
|
|
);
|
|
}
|
|
|
|
// Insert new domain
|
|
$result = $wpdb->insert(
|
|
$tables['domains'],
|
|
array(
|
|
'blog_id' => $blog_id,
|
|
'domain' => $domain,
|
|
'active' => $active
|
|
),
|
|
array( '%d', '%s', '%d' )
|
|
);
|
|
|
|
if ( $result ) {
|
|
dm_log_action( 'add', $domain, $blog_id );
|
|
return $wpdb->insert_id;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
/**
|
|
* Update domain mapping
|
|
*
|
|
* @param string $domain Domain name
|
|
* @param int $blog_id Blog ID
|
|
* @param int $active Whether domain is primary (1) or not (0)
|
|
* @return bool True on success, false on failure
|
|
*/
|
|
function dm_update_domain( $domain, $blog_id, $active ) {
|
|
global $wpdb;
|
|
|
|
$tables = dm_get_table_names();
|
|
|
|
// If setting as primary, reset other domains
|
|
if ( $active ) {
|
|
$wpdb->update(
|
|
$tables['domains'],
|
|
array( 'active' => 0 ),
|
|
array( 'blog_id' => $blog_id ),
|
|
array( '%d' ),
|
|
array( '%d' )
|
|
);
|
|
}
|
|
|
|
// Update domain
|
|
$result = $wpdb->update(
|
|
$tables['domains'],
|
|
array( 'active' => $active ),
|
|
array( 'domain' => $domain ),
|
|
array( '%d' ),
|
|
array( '%s' )
|
|
);
|
|
|
|
if ( $result !== false ) {
|
|
dm_log_action( 'edit', $domain, $blog_id );
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
/**
|
|
* Delete domain mapping
|
|
*
|
|
* @param string $domain Domain name
|
|
* @return bool True on success, false on failure
|
|
*/
|
|
function dm_delete_domain( $domain ) {
|
|
global $wpdb;
|
|
|
|
$tables = dm_get_table_names();
|
|
|
|
// Get domain info for logging
|
|
$domain_info = dm_get_domain_by_name( $domain );
|
|
|
|
if ( ! $domain_info ) {
|
|
return false;
|
|
}
|
|
|
|
// Delete domain
|
|
$result = $wpdb->delete(
|
|
$tables['domains'],
|
|
array( 'domain' => $domain ),
|
|
array( '%s' )
|
|
);
|
|
|
|
if ( $result ) {
|
|
dm_log_action( 'delete', $domain, $domain_info->blog_id );
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
/**
|
|
* Save health check result
|
|
*
|
|
* @param string $domain Domain name
|
|
* @param array $result Health check result
|
|
*/
|
|
function dm_save_health_result( $domain, $result ) {
|
|
$health_results = get_site_option( 'dm_domain_health_results', array() );
|
|
$domain_key = md5( $domain );
|
|
|
|
$health_results[$domain_key] = $result;
|
|
update_site_option( 'dm_domain_health_results', $health_results );
|
|
}
|
|
|
|
/**
|
|
* Get health check result
|
|
*
|
|
* @param string $domain Domain name
|
|
* @return array|null Health check result or null
|
|
*/
|
|
function dm_get_health_result( $domain ) {
|
|
$health_results = get_site_option( 'dm_domain_health_results', array() );
|
|
$domain_key = md5( $domain );
|
|
|
|
return isset( $health_results[$domain_key] ) ? $health_results[$domain_key] : null;
|
|
}
|
|
|
|
/**
|
|
* Format action name for display
|
|
*
|
|
* @param string $action Action name
|
|
* @return string Formatted action name
|
|
*/
|
|
function dm_format_action_name( $action ) {
|
|
switch ( $action ) {
|
|
case 'add':
|
|
return __( 'Added', 'wp-domain-mapping' );
|
|
case 'edit':
|
|
return __( 'Updated', 'wp-domain-mapping' );
|
|
case 'delete':
|
|
return __( 'Deleted', 'wp-domain-mapping' );
|
|
case 'import':
|
|
return __( 'Imported', 'wp-domain-mapping' );
|
|
default:
|
|
return ucfirst( $action );
|
|
}
|
|
}
|
