Viên ngọc phương Đông

<?php

namespace DeliciousBrains\WPMDB\Pro;

use DeliciousBrains\WPMDB\Common\Error\ErrorLog;
use DeliciousBrains\WPMDB\Common\Filesystem\Filesystem;
use DeliciousBrains\WPMDB\Common\FormData\FormData;
use DeliciousBrains\WPMDB\Common\Http\Helper;
use DeliciousBrains\WPMDB\Common\Http\Http;
use DeliciousBrains\WPMDB\Common\Http\WPMDBRestAPIServer;
use DeliciousBrains\WPMDB\Common\MigrationPersistence\Persistence;
use DeliciousBrains\WPMDB\Common\MigrationState\MigrationStateManager;
use DeliciousBrains\WPMDB\Common\MigrationState\StateDataContainer;
use DeliciousBrains\WPMDB\Common\Properties\DynamicProperties;
use DeliciousBrains\WPMDB\Common\Properties\Properties;
use DeliciousBrains\WPMDB\Common\Settings\Settings;
use DeliciousBrains\WPMDB\Pro\UI\Template;
use DeliciousBrains\WPMDB\WPMDBDI;

class UsageTracking
{

/**
     * @var StateDataContainer
     */
    private $state_container;
    /**
     * @var Settings
     */
    private $settings;
    /**
     * @var Properties
     */
    private $props;
    /**
     * @var Filesystem
     */
    private $filesystem;
    /**
     * @var ErrorLog
     */
    private $error_log;
    /**
     * @var Template
     */
    private $template;

private $dynamic_props;
    /**
     * @var FormData
     */
    private $form_data;
    /**
     * @var MigrationStateManager
     */
    private $migration_state_manager;
    /**
     * @var Http
     */
    private $http;

private $api_url;
    /**
     * @var Helper
     */
    private $http_helper;
    /**
     * @var WPMDBRestAPIServer
     */
    private $rest_API_server;

public function __construct(
        Settings $settings,
        Filesystem $filesystem,
        ErrorLog $error_log,
        Template $template,
        FormData $form_data,
        StateDataContainer $state_data_container,
        Properties $properties,
        MigrationStateManager $migration_state_manager,
        Http $http,
        Helper $http_helper,
        WPMDBRestAPIServer $rest_API_server
    ) {
        $this->state_container         = $state_data_container;
        $this->settings                = $settings->get_settings();
        $this->props                   = $properties;
        $this->filesystem              = $filesystem;
        $this->error_log               = $error_log;
        $this->template                = $template;
        $this->dynamic_props           = DynamicProperties::getInstance();
        $this->form_data               = $form_data;
        $this->migration_state_manager = $migration_state_manager;
        $this->http                    = $http;
        $this->http_helper             = $http_helper;

$this->api_url         = apply_filters('wpmdb_logging_endpoint_url', 'https://api2.deliciousbrains.com');
        $this->rest_API_server = $rest_API_server;
    }

/**
     * Adds/updates the `wpmdb_usage` option with most recent 'qualified' plugin use,
     * stores time as well as the action (push/pull/export/find-replace)
     *
     * @param string $action
     */
    public static function log_usage($action = '')
    {
        update_site_option('wpmdb_usage', array('action' => $action, 'time' => time()));
    }

/**
     * Gets just the timestamp of the latest usage to send with the API requests
     *
     * @return int
     */
    public function get_last_usage_time()
    {
        $option = get_site_option('wpmdb_usage');

return ($option && $option['time']) ? $option['time'] : 0;
    }

public function register()
    {
        add_action('wpmdb_additional_settings_advanced', array($this, 'template_toggle_usage_tracking'));

// REST endpoints
        add_action('rest_api_init', [$this, 'register_rest_routes']);
        add_action('wp_ajax_nopriv_wpmdb_track_migration_complete', [$this, 'log_migration_complete']);
        add_action('wp_ajax_wpmdb_track_migration_complete', [$this, 'log_migration_complete']);
        add_action('wpmdb_cancellation', [$this, 'log_migration_cancellation'], 50);
        add_action('wpmdb_error_migration', [$this, 'log_migration_error'], 10, 1);
        add_filter('wpmdb_notification_strings', [$this, 'template_notice_enable_usage_tracking']);
        add_action('wpmdb_after_finalize_migration', [$this, 'log_migration_complete']);
    }

public function register_rest_routes()
    {
        $this->rest_API_server->registerRestRoute(
            '/log-migration',
            [
                'methods'  => 'POST',
                'callback' => [$this, 'log_migration_event'],
            ]
        );

}

/**
     * Send migration update to usage DB
     *
     * @param  string $status complete|error|cancelled
     * @param array $data
     * @throws error
     **/
    public function send_migration_update($status = 'complete', $data = [])
    {
        if ('complete' === $status && isset($_POST['intent']) && in_array($_POST['intent'], ['savefile', 'backup_local'])) {
            $this->http->check_ajax_referer('flush');
        }

$key_rules = array(
            'complete'     => 'bool',
            'migration_id' => 'string',
        );

$state_data = $this->migration_state_manager->set_post_data($key_rules);

$existing_state_data = Persistence::getStateData();
        $state_data          += $existing_state_data;
        //if not logged in
        $filtered_post = $this->http_helper->filter_post_elements(
            $state_data,
            array(
                'action',
                'migration_state_id',
            )
        );

$settings = $this->settings;

if (true !== $settings['allow_tracking']) {
            return false;
        }

$migration_guid = isset($state_data['migration_id']) ? $state_data['migration_id'] : '';
        if ($migration_guid === '') {
            $form_data = json_decode($state_data['form_data']);
            $migration_guid = $form_data->current_migration->migration_id;
        }

$log_data = [
            'migration_complete_time' => time(),
            'migration_guid'          => $migration_guid,
            'migration_status'        => $status,
            'last_stage'              => isset($state_data['stage']) ? $state_data['stage'] : null,
            'error_text'              => isset($data['error_text']) ? $data['error_text'] : null
        ];

$remote_post_args = array(
            'timeout'            => 60,
            'method'             => 'POST',
            'headers'            => array('Content-Type' => 'application/json'),
            'body'               => json_encode($log_data),
            'reject_unsafe_urls' => false,
        );

$api_url = $this->api_url . '/complete';

$result = wp_remote_post($api_url, $remote_post_args);

if (is_wp_error($result) || $result['response']['code'] >= 400) {
            if (defined('WP_DEBUG') && WP_DEBUG) {
                error_log('Error logging migration event');
                error_log(print_r($result, true));
            }
            $this->error_log->log_error('Error logging Migration event', $result);
            $this->http->end_ajax(json_encode($result));
        }

$this->http->end_ajax($result['body']);
    }

/**
     * Log migration cacellation
     *
     * Fires on 'wpmdb_track_migration_complete' hook
     *
     **/
    public function log_migration_complete()
    {
        $this->send_migration_update('complete');
    }

/**
     * Log migration cacellation
     *
     * Fires on 'wpmdb_cancellation' hook
     *
     **/
    public function log_migration_cancellation()
    {
        $this->send_migration_update('cancelled');
    }

/**
     * Logs migration error
     *
     * Fires on 'wpmdb_error_migration' hook
     *
     **/
    public function log_migration_error($error_text)
    {
        $this->send_migration_update('error', ['error_text' => $error_text]);
    }

/**
     * Log Migration Event
     *
     * Callback for log-migration endpoint called at start of Migration
     *
     * @return void
     */
    public function log_migration_event()
    {
        $_POST       = $this->http_helper->convert_json_body_to_post();
        $license     = WPMDBDI::getInstance()->get(License::class);

$license_key = $license->get_licence_key();
        if (empty($license_key)) {
            return;
        }

$key_rules = array(
            'complete'     => 'bool',
            'migration_id' => 'string',
        );

$state_data = $this->migration_state_manager->set_post_data($key_rules);

$existing_state_data = Persistence::getStateData();
        $state_data          += $existing_state_data;

$settings = $this->settings;
        do_action('wpmdb_log_migration_event', $state_data);
        if (true !== $settings['allow_tracking']) {
            return false;
        }

$api_url  = $this->api_url . '/event';
        $cookie   = false === Persistence::getRemoteWPECookie() ? 0 : 1;
        $log_data = array(
            'local_timestamp'                        => time(),
            'licence_key'                            => $license_key,
            'cli'                                    => $this->dynamic_props->doing_cli_migration,
            'setting-compatibility_plugin_installed' => $this->filesystem->file_exists($this->props->mu_plugin_dest),
            'remote_cookie'                          => $cookie,
            'local_platform'                      => $state_data['site_details']['local']['platform'],
            'remote_platform'                     => $state_data['site_details']['remote']['platform'],
        );

// ***+=== @TODO - revisit usage of parse_migration_form_data
        foreach ($this->form_data->parse_and_save_migration_form_data($state_data['form_data']) as $key => $val) {
            if ('connection_info' === $key) {
                continue;
            }
            $log_data['profile-' . $key] = $val;
        }

foreach ($settings as $key => $val) {
            if ('profiles' === $key || 'key' === $key) {
                continue;
            }
            $log_data['setting-' . $key] = $val;
        }

foreach ($GLOBALS['wpmdb_meta'] as $plugin => $arr) {
            $log_data[$plugin . '-active']  = true;
            $log_data[$plugin . '-version'] = $arr['version'];
        }

foreach ($state_data['site_details'] as $site => $info) {
            $log_data[$site . '-site_url'] = $info['site_url'];
            $log_data[$site . '-home_url'] = $info['home_url'];
            $log_data[$site . '-prefix']   = $info['prefix'];

$log_data[$site . '-is_multisite'] = $info['is_multisite'];

if (isset($info['subsites']) && is_array($info['subsites'])) {
                $log_data[$site . '-subsite_count'] = count($info['subsites']);
            }

$log_data[$site . '-is_subdomain_install'] = $info['is_subdomain_install'];
        }

$diagnostic_log = [];

foreach ($this->error_log->get_diagnostic_info() as $group_name => $data) {
            foreach ($data as $key => $val) {
                if (0 === $key) {
                    continue 1;
                }
                $key_name = $group_name;
                if (is_string($key)) {
                    $key_name .= "-{$key}";
                }
                $diagnostic_log[$key_name] = $val;
            }
        }

$log_data['diagnostic_log'] = $diagnostic_log;

foreach ($log_data as $key => $val) {
            if (strpos($key, 'count') !== false || is_array($val)) {
                continue;
            }
            if ('1' === $val) {
                $log_data[$key] = true;
                continue;
            }
            if ('0' === $val) {
                $log_data[$key] = false;
                continue;
            }
            if ('true' === $val) {
                $log_data[$key] = true;
                continue;
            }
            if ('false' === $val) {
                $log_data[$key] = false;
                continue;
            }
        }

$log_data['migration_guid'] = $state_data['migration_id'];

$result = wp_remote_post($api_url, $remote_post_args);
        if (is_wp_error($result) || $result['response']['code'] >= 400) {
            if (defined('WP_DEBUG') && WP_DEBUG) {
                error_log('Error logging migration event');
                error_log(print_r($result, true));
            }
            $this->error_log->log_error('Error logging Migration event', $result);
            $this->http->end_ajax(json_encode($result));
        }

$this->http->end_ajax($result['body']);
    }

public function template_notice_enable_usage_tracking($notifications)
    {
        if (!is_bool($this->settings['allow_tracking'])) {
            $notifications['notice-enable-usage-tracking'] = [
                'message'     => $this->template->template_to_string('notice-enable-usage-tracking', 'pro'),
                'link'        => false,
                'id'          => 'notice-enable-usage-tracking',
                'custom_link' => 'usage_tracking',
            ];
        }

return $notifications;
    }

}