qdr.app.woo.plugins/qdr-temporary-shared-storage/public/class-qdr-tss-download.php

<?php
/**
 * Download functionality of the plugin.
 *
 * @package    QDR_Temporary_Shared_Storage
 * @since      1.0.0
 */
class QDR_TSS_Download {

    /**
     * The ID of this plugin.
     *
     * @since    1.0.0
     * @access   private
     * @var      string    $plugin_name    The ID of this plugin.
     */
    private $plugin_name;

    /**
     * The version of this plugin.
     *
     * @since    1.0.0
     * @access   private
     * @var      string    $version    The current version of this plugin.
     */
    private $version;

    /**
     * The database manager instance.
     *
     * @since    1.0.0
     * @access   private
     * @var      QDR_TSS_DB_Manager    $db_manager    The database manager instance.
     */
    private $db_manager;

    /**
     * The file manager instance.
     *
     * @since    1.0.0
     * @access   private
     * @var      QDR_TSS_File_Manager    $file_manager    The file manager instance.
     */
    private $file_manager;

    /**
     * Initialize the class and set its properties.
     *
     * @since    1.0.0
     * @param    string    $plugin_name    The name of this plugin.
     * @param    string    $version        The version of this plugin.
     */
    public function __construct($plugin_name, $version) {
        $this->plugin_name = $plugin_name;
        $this->version = $version;
        $this->db_manager = QDR_TSS_DB_Manager::instance();
        $this->file_manager = QDR_TSS_File_Manager::instance();
    }

    /**
     * Process download request.
     *
     * @since    1.0.0
     * @param    array    $atts    Shortcode attributes.
     * @return   string            Download form or file download.
     */
    public function process_download_request($atts) {
        // Extract attributes
        $atts = shortcode_atts(array(
            'id' => null,
        ), $atts, 'qdr_tss_download');
        
        // Get media ID from shortcode attribute or URL parameter
        $media_id = $atts['id'];
        if (!$media_id && isset($_GET['id'])) {
            $media_id = intval($_GET['id']);
        }
        
        // If no media ID, show form to enter media ID, password and reference
        if (!$media_id) {
            return $this->render_download_form();
        }
        
        // Get media information
        $item = $this->db_manager->get_item_by_media_id($media_id);
        
        if (!$item) {
            return '<div class="qdr-tss-error">' . __('File not found.', 'qdr-temporary-shared-storage') . '</div>';
        }
        
        // Check if file is active
        $current_time = current_time('mysql');
        if ($current_time < $item->active_from) {
            return '<div class="qdr-tss-error">' . __('This file is not yet available for download.', 'qdr-temporary-shared-storage') . '</div>';
        }
        
        if ($current_time > $item->active_to) {
            return '<div class="qdr-tss-error">' . __('This file has expired and is no longer available for download.', 'qdr-temporary-shared-storage') . '</div>';
        }
        
        // Check if form is submitted
        if (isset($_POST['qdr_tss_download_submit'])) {
            // Verify nonce
            if (!isset($_POST['qdr_tss_nonce']) || !wp_verify_nonce($_POST['qdr_tss_nonce'], 'qdr_tss_download')) {
                return '<div class="qdr-tss-error">' . __('Security check failed.', 'qdr-temporary-shared-storage') . '</div>';
            }
            
            // Verify password
            $password = isset($_POST['qdr_tss_password']) ? $_POST['qdr_tss_password'] : '';
            if (!wp_check_password($password, $item->password)) {
                return '<div class="qdr-tss-error">' . __('Invalid password.', 'qdr-temporary-shared-storage') . '</div>' . $this->render_download_form($media_id);
            }
            
            // Verify reference if set
            if (!empty($item->reference)) {
                $reference = isset($_POST['qdr_tss_reference']) ? $_POST['qdr_tss_reference'] : '';
                if ($reference !== $item->reference) {
                    return '<div class="qdr-tss-error">' . __('Invalid reference.', 'qdr-temporary-shared-storage') . '</div>' . $this->render_download_form($media_id);
                }
            }
            
            // Update download count and last download time
            $this->db_manager->increment_download_count($item->id);
            
            // Initiate download
            $file_path = QDR_TSS_UPLOADS_DIR . $item->file_path;
            
            if (file_exists($file_path)) {
                // Set headers for download
                header('Content-Description: File Transfer');
                header('Content-Type: application/octet-stream');
                header('Content-Disposition: attachment; filename="' . $item->original_file_name . '"');
                header('Expires: 0');
                header('Cache-Control: must-revalidate');
                header('Pragma: public');
                header('Content-Length: ' . filesize($file_path));
                ob_clean();
                flush();
                readfile($file_path);
                exit;
            } else {
                return '<div class="qdr-tss-error">' . __('File not found on server.', 'qdr-temporary-shared-storage') . '</div>';
            }
        }
        
        // Show download form with media information
        $output = '<div class="qdr-tss-download-info">';
        $output .= '<h2>' . __('File Information', 'qdr-temporary-shared-storage') . '</h2>';
        $output .= '<p><strong>' . __('File Name:', 'qdr-temporary-shared-storage') . '</strong> ' . esc_html($item->original_file_name) . '</p>';
        
        if (!empty($item->description)) {
            $output .= '<p><strong>' . __('Description:', 'qdr-temporary-shared-storage') . '</strong> ' . esc_html($item->description) . '</p>';
        }
        
        if (!empty($item->message)) {
            $output .= '<p><strong>' . __('Message:', 'qdr-temporary-shared-storage') . '</strong> ' . esc_html($item->message) . '</p>';
        }
        
        $output .= '</div>';
        $output .= $this->render_download_form($media_id);
        
        return $output;
    }

    /**
     * Render download form.
     *
     * @since    1.0.0
     * @param    int       $media_id    Optional. Media ID.
     * @return   string                 Download form HTML.
     */
    private function render_download_form($media_id = null) {
        $form = '<div class="qdr-tss-download-form">';
        $form .= '<form method="post" action="">';
        
        // Add nonce for security
        $form .= wp_nonce_field('qdr_tss_download', 'qdr_tss_nonce', true, false);
        
        if (!$media_id) {
            $form .= '<div class="qdr-tss-form-group">';
            $form .= '<label for="qdr_tss_media_id">' . __('Media ID', 'qdr-temporary-shared-storage') . '</label>';
            $form .= '<input type="text" name="id" id="qdr_tss_media_id" required>';
            $form .= '</div>';
        } else {
            $form .= '<input type="hidden" name="id" value="' . esc_attr($media_id) . '">';
        }
        
        $form .= '<div class="qdr-tss-form-group">';
        $form .= '<label for="qdr_tss_password">' . __('Password', 'qdr-temporary-shared-storage') . '</label>';
        $form .= '<input type="password" name="qdr_tss_password" id="qdr_tss_password" required>';
        $form .= '</div>';
        
        $form .= '<div class="qdr-tss-form-group">';
        $form .= '<label for="qdr_tss_reference">' . __('Reference (if required)', 'qdr-temporary-shared-storage') . '</label>';
        $form .= '<input type="text" name="qdr_tss_reference" id="qdr_tss_reference">';
        $form .= '</div>';
        
        $form .= '<div class="qdr-tss-form-group">';
        $form .= '<button type="submit" name="qdr_tss_download_submit" class="qdr-tss-button">' . __('Download File', 'qdr-temporary-shared-storage') . '</button>';
        $form .= '</div>';
        
        $form .= '</form>';
        $form .= '</div>';
        
        return $form;
    }
}