<?php

/**
 * Plugin Name:       Mail Queue
 * Plugin URI:        https://www.webdesign-muenchen.de/wordpress-plugin-mail-queue/
 * Description:       Take Control and improve Security of wp_mail(). Queue outgoing emails, log their sendings and get alerted, if your website wants to send more emails than usual.
 * Version:           1.1
 * Requires at least: 5.2
 * Requires PHP:      7.2
 * Author:            WDM
 * Author URI:        https://www.webdesign-muenchen.de
 * License:           GPLv3 or later
 * License URI:       https://www.gnu.org/licenses/gpl-3.0.html
 */

if (!defined('ABSPATH')) { exit; }


/* ***************************************************************
PLUGIN DEFAULT SETTINGS
**************************************************************** */
function wdm_wpma_get_settings() {
    $defaults = array(
        'enabled'        => '0',
        'alert_enabled'  => '0',
        'email'          => get_option('admin_email'),
        'email_amount'   => '10',
        'queue_amount'   => '1',
        'queue_interval' => '5',
        'clear_queue'    => '14',
        'tableName'      => 'mail_queue',
    );
    $args = get_option('wdm_wpma_settings');
    $options = wp_parse_args($args,$defaults);
    $options['queue_interval'] = intval($options['queue_interval']) * 60;
    $options['clear_queue']    = intval($options['clear_queue']) * 24;
    return $options;
}





/* ***************************************************************
Overwrite wp_mail() if Plugin enabled and no Cron is running
**************************************************************** */
$wdm_wpma_mailid = 0;
$wdm_wpma_options = wdm_wpma_get_settings(); // Get Settings

if ($wdm_wpma_options['enabled'] == '1' && !defined( 'DOING_CRON' )) {

    if ( !function_exists( 'wp_mail' ) ) {
        function wp_mail (  $to, $subject, $message, $headers = '', $attachments = [] ) {

            global $wpdb, $wdm_wpma_options;

            // Write email in Queue
            $tableName = $wpdb->prefix.$wdm_wpma_options['tableName'];
            $data = array(
                'timestamp'=> current_time('mysql',false),
                'recipient'=> maybe_serialize($to),
                'subject'=> $subject,
                'message'=> $message,
                'status' => 'queue',
                'attachments' => ''
            );
            if (isset($headers) && $headers) { $data['headers'] = maybe_serialize($headers); }

            // store attachments in /attachments/ Folder, to address them later
            if (isset($attachments) && $attachments && $attachments != '') { 
                
                $subfolder = time().'-'.rand(0,999999);
                $foldercreated = wp_mkdir_p(plugin_dir_path(__FILE__).'attachments/'.$subfolder);
                if (!$foldercreated) {
                    error_log('Could not create Subfolder for Email attachment');
                    $data['info'] = 'Error: Could not store attachments';
                } else {
                    if (!is_array($attachments)) { $attachments = array($attachments); }
                    $newattachments = array();
                    global $wp_filesystem;
                    if ( ! is_a( $wp_filesystem, 'WP_Filesystem_Base') ){
                        include_once(ABSPATH . 'wp-admin/includes/file.php');
                        WP_Filesystem();
                    }
                    foreach($attachments as $item) {
                        $newfile = plugin_dir_path(__FILE__).'attachments/'.$subfolder.'/'.basename($item);
                        $wp_filesystem->copy($item,$newfile);
                        array_push($newattachments,$newfile);
                    }
                    $data['attachments'] = maybe_serialize($newattachments);
                }
            }
            $wpdb->insert($tableName,$data);

            // Fake Submit by returning 'True'
            return true;

        }
    }

}



// show wp_mail() errors
function wdm_wpma_mail_failed( $wp_error ) {
    global $wpdb,$wdm_wpma_options,$wdm_wpma_mailid;
    if (isset($wdm_wpma_mailid) && $wdm_wpma_mailid != 0) {
        $tableName = $wpdb->prefix.$wdm_wpma_options['tableName'];
        $wpMailFailedError = isset( $wp_error->errors ) && isset( $wp_error->errors['wp_mail_failed'][0] ) ? implode( '; ', $wp_error->errors['wp_mail_failed'] ) : '<em>Unknown</em>';
        $wpdb->update($tableName,array('timestamp'=>current_time('mysql',false),'status'=>'error', 'info'=>$wpMailFailedError),array('id'=>intval($wdm_wpma_mailid)),array('%s', '%s', '%s'),'%d');
    }
    return error_log(print_r($wp_error, true));
} 
add_action('wp_mail_failed','wdm_wpma_mail_failed',10,1);








/* ***************************************************************
CRON
**************************************************************** */
function wdm_wpma_search_mail_from_queue() {
    
    global $wpdb,$wdm_wpma_options,$wdm_wpma_mailid;
    if ($wdm_wpma_options['enabled'] != '1') { return; }
    $tableName = $wpdb->prefix.$wdm_wpma_options['tableName'];

    // Mails in Queue?
    $mailjobs     = $wpdb->get_results("SELECT * FROM `$tableName` WHERE `status` = 'queue' ORDER BY `id`",'ARRAY_A');
    $mailsInQueue = is_array($mailjobs) ? count($mailjobs) : 0;

    // Alert Admin, if too many mails in the Queue.
    if ($wdm_wpma_options['alert_enabled'] == '1' && $mailsInQueue > intval($wdm_wpma_options['email_amount'])) {

        // Last alerts older than 6 hours?
        $alerts = $wpdb->get_results("SELECT * FROM `$tableName` WHERE `status` = 'alert' AND `timestamp` > NOW() - INTERVAL 6 HOUR",'ARRAY_A');

        // If no alerts, then send one
       if (!$alerts) {
            $alertMessage = 'Hi,';
            $alertMessage .= "\n\n";
            $alertMessage .= 'this is an important message from your WordPress website '.esc_url(get_option('siteurl')).'.';
            $alertMessage .= "\n";
            $alertMessage .= "\n".'The Mail Queue Plugin has detected that your website tries to send more emails than expected.';
            $alertMessage .= "\n".'Please take a close look at the email queue, because it contains more messages than the specified limit.';
            $alertMessage .= "\n";
            $alertMessage .= "\n".'In case this is the usual amount of emails, you can adjust the threshold for alerts in the settings of your Mail Queue Plugin.';
            $alertMessage .= "\n\n";
            $alertMessage .= "-- ";
            $alertMessage .= "\n";
            $alertMessage .= admin_url();
            $alertSubject = '🔴 WordPress Mail Queue Alert - '.esc_html(get_option('blogname'));
            $data = array(
                'timestamp'=> current_time('mysql',false),
                'recipient'=> sanitize_email($wdm_wpma_options['email']),
                'subject'  => $alertSubject,
                'message'  => $alertMessage,
                'status'   => 'alert',
                'info'     => json_encode([
                    'in_queue'       => strval( $mailsInQueue ),
                    'email_amount'   => intval($wdm_wpma_options['email_amount']),
                    'queue_amount'   => intval($wdm_wpma_options['queue_amount']),
                    'queue_interval' => intval($wdm_wpma_options['queue_interval']),
                ]),
            );
            $wpdb->insert($tableName,$data);
            wp_mail($wdm_wpma_options['email'],$alertSubject,$alertMessage);
        }

    }

    // Send Mails in Queue
    if ($mailsInQueue > 0) {
        $results = array_slice($mailjobs,0,intval($wdm_wpma_options['queue_amount']));
        if ($results && count($results) > 0) {
            foreach($results as $item) {
                if ($item['recipient'] && $item['recipient'] != '') { $to = maybe_unserialize($item['recipient']); } else { $to = $wdm_wpma_options['email']; $item['subject'] = 'ERROR // '.$item['subject']; }
                if ($item['headers'] && $item['headers'] != '') { $headers = maybe_unserialize($item['headers']); } else { $headers = ''; }
                if ($item['attachments'] && $item['attachments'] != '') { $attachments = maybe_unserialize($item['attachments']); } else { $attachments = ''; }
                $wdm_wpma_mailid = $item['id'];       
                $sendstatus = wp_mail($to,$item['subject'],$item['message'],$headers,$attachments); // Finally sends the email for real
                if ($sendstatus) {
                    $wpdb->update($tableName,array('timestamp'=>current_time('mysql',false),'status'=>'sent'),array('id'=>$item['id']),'%s','%d');
                }
                if (is_array($attachments)) {
                    global $wp_filesystem;
                    if ( ! is_a( $wp_filesystem, 'WP_Filesystem_Base') ){
                        include_once(ABSPATH . 'wp-admin/includes/file.php');
                        WP_Filesystem();
                    }
                    $attachmentfolder = pathinfo($attachments[0]);
                    $wp_filesystem->delete($attachmentfolder['dirname'],true,'d');
                }
            }
        }
    }

    // Delete old logs
    $wpdb->query("DELETE FROM `$tableName` WHERE `status` != 'queue' AND `timestamp` < NOW() - INTERVAL ".esc_sql($wdm_wpma_options['clear_queue'])." HOUR");

}
add_action('wp_mail_queue_hook','wdm_wpma_search_mail_from_queue');

// Custom Cron Interval
function wdm_wpma_cron_interval( $schedules ) { 
    global $wdm_wpma_options;
    $schedules['wdm_wpma_interval'] = array(
        'interval' => $wdm_wpma_options['queue_interval'],
        'display'  => esc_html__('WP Mail Queue'), );
    return $schedules;
}
add_filter('cron_schedules','wdm_wpma_cron_interval');

// Set or Remove Cron
$next_wpma_cron_timestamp = wp_next_scheduled('wp_mail_queue_hook');
if ($next_wpma_cron_timestamp && $wdm_wpma_options['enabled'] != '1') {
    wp_unschedule_event($next_wpma_cron_timestamp,'wp_mail_queue_hook');
} else if (!$next_wpma_cron_timestamp && $wdm_wpma_options['enabled'] == '1') {
    wp_schedule_event(time(),'wdm_wpma_interval','wp_mail_queue_hook');
}



/* ***************************************************************
Install/Uninstall
**************************************************************** */


/* Delete plugin options and database table */
function wdm_wpma_uninstall () {
    global $wpdb;

    $optionName = 'wdm_wpma_settings';
    delete_option( $optionName );

    $tableName = $wpdb->prefix.'mail_queue';
    $wpdb->query( "DROP TABLE IF EXISTS $tableName" );
}

/* Create MySQL Table on Activation: https://codex.wordpress.org/Creating_Tables_with_Plugins */
function wdm_wpma_installDatabaseTables() {
    global $wpdb;
    
    $tableName = $wpdb->prefix.'mail_queue';
   
    $charset_collate = $wpdb->get_charset_collate();

    $sql = "CREATE TABLE $tableName (
    id mediumint(9) NOT NULL AUTO_INCREMENT,
    timestamp TIMESTAMP NOT NULL,
    status varchar(55) DEFAULT '' NOT NULL,
    recipient varchar(255) DEFAULT '' NOT NULL,
    subject varchar(255) DEFAULT '' NOT NULL,
    message text NOT NULL,
    headers text NOT NULL,
    attachments text NOT NULL,
    info varchar(255) DEFAULT '' NOT NULL,
    PRIMARY KEY  (id)
    ) $charset_collate;";

    require_once( ABSPATH . 'wp-admin/includes/upgrade.php' );
    dbDelta( $sql );

    register_uninstall_hook( __FILE__, 'wdm_wpma_uninstall' );
}
register_activation_hook( __FILE__, 'wdm_wpma_installDatabaseTables' );


/* ***************************************************************
Options Page 
**************************************************************** */
if (is_admin()) {
    require_once( plugin_dir_path( __FILE__ ) . 'mail-queue-options.php' );
}