punycode 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 )); } /** * Check if domain exists for another blog * UPDATED: More precise checking for domain conflicts * * @param string $domain Domain name * @param int $exclude_blog_id Blog ID to exclude from check (optional) * @return object|null Domain object if exists for another blog, null otherwise */ function dm_domain_exists_for_another_blog( $domain, $exclude_blog_id = 0 ) { global $wpdb; $tables = dm_get_table_names(); $query = "SELECT * FROM {$tables['domains']} WHERE domain = %s"; $params = array( $domain ); if ( $exclude_blog_id > 0 ) { $query .= " AND blog_id != %d"; $params[] = $exclude_blog_id; } return $wpdb->get_row( $wpdb->prepare( $query, $params ) ); } /** * 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 ); dm_clear_domain_cache( $blog_id ); return $wpdb->insert_id; } return false; } /** * Update domain mapping * UPDATED: Support changing domain name and blog_id * * @param string $domain Domain name (current) * @param int $blog_id Blog ID * @param int $active Whether domain is primary (1) or not (0) * @param string $new_domain New domain name (optional) * @return bool True on success, false on failure */ function dm_update_domain( $domain, $blog_id, $active, $new_domain = null ) { global $wpdb; $tables = dm_get_table_names(); // If changing domain name if ( $new_domain && $new_domain !== $domain ) { $new_domain = dm_clean_domain( $new_domain ); // Validate new domain if ( ! dm_validate_domain( $new_domain ) ) { return false; } // Check if new domain exists for another blog $existing = dm_domain_exists_for_another_blog( $new_domain, $blog_id ); if ( $existing ) { 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' ) ); } // Prepare update data $data = array( 'active' => $active, 'blog_id' => $blog_id ); $data_format = array( '%d', '%d' ); if ( $new_domain && $new_domain !== $domain ) { $data['domain'] = $new_domain; $data_format[] = '%s'; } // Update domain $result = $wpdb->update( $tables['domains'], $data, array( 'domain' => $domain ), $data_format, array( '%s' ) ); if ( $result !== false ) { dm_log_action( 'edit', $new_domain ?: $domain, $blog_id ); dm_clear_domain_cache( $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 ); dm_clear_domain_cache( $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; } /** * Check if domain has all health checks passing * * @param string $domain Domain name * @return bool True if all health checks pass */ function dm_is_domain_healthy( $domain ) { $health_result = dm_get_health_result( $domain ); if ( ! $health_result ) { return false; } // Check all three criteria: DNS, SSL, and Accessibility $dns_ok = isset( $health_result['dns_status'] ) && $health_result['dns_status'] === 'success'; $ssl_ok = isset( $health_result['ssl_valid'] ) && $health_result['ssl_valid'] === true; $accessible_ok = isset( $health_result['accessible'] ) && $health_result['accessible'] === true; return $dns_ok && $ssl_ok && $accessible_ok; } /** * 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 ); } } // Cache functions function dm_get_domains_by_blog_id_cached($blog_id) { $cache_key = 'dm_domains_' . $blog_id; $domains = wp_cache_get($cache_key, 'domain_mapping'); if (false === $domains) { $domains = dm_get_domains_by_blog_id($blog_id); wp_cache_set($cache_key, $domains, 'domain_mapping', HOUR_IN_SECONDS); } return $domains; } // Clear cache function function dm_clear_domain_cache($blog_id) { wp_cache_delete('dm_domains_' . $blog_id, 'domain_mapping'); }