tracking_exporter.php

<?php

namespace Podlove\Modules\ImportExport\Export;

class TrackingExporter
{
    public static function init()
    {
        add_action('wp_ajax_podlove-export-tracking', [__CLASS__, 'export_tracking']);
        add_action('wp_ajax_podlove-export-tracking-status', [__CLASS__, 'export_tracking_status']);

        self::init_download();
    }

    public static function init_download()
    {
        if (!is_admin()) {
            return;
        }

        if (isset($_GET['podlove_export_tracking']) && $_GET['podlove_export_tracking']) {
            if (!current_user_can('administrator')) {
                exit;
            }

            if (!wp_verify_nonce($_REQUEST['_podlove_nonce'], 'podlove_export_tracking_download')) {
                exit;
            }

            delete_transient('podlove_tracking_export_finished');

            header('Content-Type: application/octet-stream');
            header('Content-Description: File Transfer');
            header('Content-Disposition: attachment; filename='.TrackingExporter::getDownloadFileName());
            header('Cache-control: private');
            header('Expires: -1');

            readfile(TrackingExporter::get_tracking_export_file_path());
            exit;
        }
    }

    public static function get_tracking_export_file_path()
    {
        $upload_dir = wp_upload_dir();

        return $upload_dir['basedir'].DIRECTORY_SEPARATOR.'tracking.tmp';
    }

    public static function export_tracking()
    {
        global $wpdb;

        if (!current_user_can('administrator')) {
            exit;
        }

        if (!wp_verify_nonce($_REQUEST['_podlove_nonce'], 'podlove_export_tracking')) {
            exit;
        }

        // only one export at a time
        if (get_option('podlove_tracking_export_all') !== false) {
            exit;
        }

        update_option('podlove_tracking_export_all', $wpdb->get_var('SELECT COUNT(*) FROM '.\Podlove\Model\DownloadIntent::table_name()));
        update_option('podlove_tracking_export_progress', 0);

        $rowsPerQuery = 1000;
        $lastId = 0;
        $page = 0;

        $fp = gzopen(self::get_tracking_export_file_path(), 'w');

        do {
            // Keeping track of the $lastId is (roughly) a bajillion times faster than paging via LIMIT.
            $sql = '
				SELECT
					id,
					user_agent_id,
					media_file_id,
					request_id,
					accessed_at,
					source,
					context,
					geo_area_id,
					lat,
					lng,
					httprange
				FROM
					'.\Podlove\Model\DownloadIntent::table_name().'
					WHERE id > '.(int) $lastId."
				LIMIT 0, {$rowsPerQuery}";
            $rows = $wpdb->get_results($sql, ARRAY_A);
            foreach ($rows as $row) {
                gzwrite($fp, implode(',', $row)."\n");
            }

            $lastId = $row['id'];
            ++$page;

            update_option('podlove_tracking_export_progress', $page * $rowsPerQuery);
        } while (count($rows) > 0);

        gzclose($fp);

        set_transient('podlove_tracking_export_finished', true, MINUTE_IN_SECONDS * 3);
        delete_option('podlove_tracking_export_all');
        delete_option('podlove_tracking_export_progress');
        exit;
    }

    public static function export_tracking_status()
    {
        echo json_encode([
            'all' => get_option('podlove_tracking_export_all'),
            'progress' => get_option('podlove_tracking_export_progress'),
            'finished' => (bool) get_transient('podlove_tracking_export_finished'),
        ]);
        exit;
    }

    private static function getDownloadFileName()
    {
        $sitename = sanitize_key(get_bloginfo('name'));

        if (!empty($sitename)) {
            $sitename .= '.';
        }

        return $sitename.'tracking.'.date('Y-m-d').'.csv.gz';
    }
}