JezK

Edit File: pdf-protection.php

<?php
// Include WordPress to access functions like is_user_logged_in()
require_once($_SERVER['DOCUMENT_ROOT'] . '/wp-load.php');

// Start session if not already started
if (session_id() == "") {
    session_start();
}

// Get the requested file path
$request_uri = $_SERVER['REQUEST_URI'];

// Get the file name from the URI
$file = basename($request_uri);

// Define the base directory for uploads
$upload_dir = wp_upload_dir()['basedir'];

// Extract year and month from the request URI
preg_match('/wp-content\/uploads\/(\d{4})\/(\d{2})\//', $request_uri, $matches);
$year = $matches[1] ?? null;
$month = $matches[2] ?? null;

// Build the full file path
$file_path = $upload_dir . "/$year/$month/$file";

// List of restricted PDF files
$restricted_pdfs = [
    'rare-metal-20181024_5P-6P.pdf',
    'REIA-Input-for-EU-RawMaterials-Act.pdf',
    'REIA-Innovation-Fund-Inclusion-of-Rare-Earths-web-final-1.pdf',
    'Press-release-3.pdf',
    'LCA-Report.pdf',
    'REIA-Quarterly-Report-for-Q4-2022.pdf',
    'REIA-Monthly-BOD-Report-June-2022.pdf',
    'REIA-Monthly-BOD-Report-February-2022.pdf',
    'REIA-Quarterly-Report-for-Q2-2022.pdf',
    'REIA-Monthly-BOD-Report-May-2022.pdf',
    'REIA-Monthly-BOD-Report-November-2021.pdf',
    'REIA-Bylaws-21-Jun-2023-CURRENT.pdf',
    'REIA-Monthly-BOD-Report-December-2021.pdf',
    'REIA-Articles-English-25-Jun-2019-CURRENT.pdf',
    'White-Paper-PCR-1.pdf',
    'REIA-2023-12-IGM-Minutes.pdf',
    'Minutes-REIA-General-Assembly-2020.pdf',
    'Minutes_2023_AGM.pdf',
    'Minutes-General-Assembly-25061984.pdf',
    'REIA-2022-AGM-Minutes.pdf',
    '2024-REIA-Annual-Conference-Guide.pdf',
    'Minutes-REIA-General-Assembly-2021.pdf',
    // '2024_Q3-REIA-Public74.pdf',
    'Minutes-(Summary)-of-18-December-2024-REIA-Board-Meeting-17012025.pdf',
    // 'Rare-Earth-Market-Report-Q2-2024-Public.pdf',
    // 'REIA-Quarterly-Rare-Earth-Market-Report-2024-Q4.pdf',
    // '2025_Q1-REIA-Public.pdf',
    '2025_Q2-REIA-Public.pdf',
    // '2025_Q2-REIA-Public-1.pdf',
	// 'REIA-REE-Market-Report-2025_Q3-Public.pdf',
	// 'REIA-REE-Market-Report-Q4-2025-Public-1.pdf',
    'REIA-Quarterly-Rare-Earth-Market-Report_Q1-2026-REIA-Members-only.pdf',
    'REIA-REE-Market-Report_Q1-2026-Public.pdf',
    // Add more if needed
];

// Check if file is restricted
if (in_array($file, $restricted_pdfs)) {

// Require login or special session access
    if (!is_user_logged_in() && !isset($_SESSION['form_submit'])) {
        wp_redirect(wp_login_url());
        exit;
    }

// Serve the restricted file
    if (file_exists($file_path)) {
    	require_once('wp-content/plugins/paid-memberships-pro/preheaders/check.php');
        if (check_file($file_path, $_SERVER)) exit;
        header('Content-Type: application/pdf');
        header('Content-Disposition: inline; filename="' . basename($file_path) . '"');
        header('Content-Transfer-Encoding: binary');
        header('Accept-Ranges: bytes');
        @readfile($file_path);
        exit;
    } else {
        header("HTTP/1.0 404 Not Found");
        echo "Restricted file not found!";
        exit;
    }

} else {
    // Serve public PDF file
    if (file_exists($file_path)) {
    	require_once('wp-content/plugins/paid-memberships-pro/preheaders/check.php');
        if (check_file($file_path, $_SERVER)) exit;
        header('Content-Type: application/pdf');
        header('Content-Disposition: inline; filename="' . basename($file_path) . '"');
        header('Content-Transfer-Encoding: binary');
        header('Accept-Ranges: bytes');
        @readfile($file_path);
        exit;
    } else {
        header("HTTP/1.0 404 Not Found");
        echo "Public file not found!";
        exit;
    }
}