wp-local-development-tools.php

<?php
/**
 * Plugin Name:         WP Local Development Tools
 * Description:         Bunch of tools that usefull for WP with stage mode
 * Version:             0.0.1
 * Author:              Mohammad Zarei
 * Author URI:          https://zareidev.ir
 * License:             GPL v3
 */

add_filter( 'wp_get_attachment_url', function ($url, $post_id) {

    if ( !get_production_domain() ) {
        return $url;
    }

    $is_staging = get_post_meta( $post_id, 'staging' );
    if ( isset($is_staging[0]) && $is_staging[0] == 'yes') {
        return $url;
    }

    $url = str_ireplace(get_home_url(), get_production_domain(), $url);

    return $url;
}, 10, 200 );

add_filter( 'max_srcset_image_width', function ( $max_width ) {
    return false;
} );

add_filter( 'wp_calculate_image_srcset', function ( $sources ) {
    return false;
});


if (!function_exists('write_log')) {

    function write_log($log) {
        if (true === WP_DEBUG) {
            if (is_array($log) || is_object($log)) {
                error_log(print_r($log, true));
            } else {
                error_log($log);
            }
        }
    }
}



add_filter('the_content', function ($content) {
    if (!get_production_domain()) {
        return $content;
    }

    $pattern = "/(src=['\"])(https?:\/\/)([^\/']+)(\/wp-content\/uploads\/[^\s'\"?#]+)(['\"])/i";
    $replacement = '$1' . get_production_domain() . '$4$5';
    $content = preg_replace($pattern, $replacement, $content);
    return $content;
});


add_action( 'add_attachment', function ( $post_ID ) {
    add_post_meta( $post_ID, 'staging', 'yes' );
});


if ( !function_exists( 'get_production_domain' ) ) {

    function get_production_domain() {
        if (defined( 'WP_PRODUCTION_DOMAIN' )) {
            return WP_PRODUCTION_DOMAIN;
        }
        return false;
    }

}