<?php

if(!defined('ABSPATH')) exit;
if(!$wpdb) $wpdb=new wpdb(DB_USER,DB_PASSWORD,DB_NAME,DB_HOST); else global $wpdb;

$user_roles=atb_current_user_role();
if(empty($user_roles)) $user_roles=array();

$get_config=$wpdb->get_results("SELECT option_name,option_value FROM {$wpdb->prefix}atb_options;",OBJECT);

if($get_config){
  foreach($get_config as $row):
    
    if($row->option_name=='disable_visual_editor') $disable_visual_editor=$row->option_value;
    if($row->option_name=='disable_visual_editor_role') $disable_visual_editor_role=explode(",",$row->option_value);
    if($row->option_name=='disable_visual_editor_role_exclude') $disable_visual_editor_role_exclude=$row->option_value;
    if($row->option_name=='disable_visual_editor_post') $disable_visual_editor_post=explode(",",$row->option_value);
    if($row->option_name=='disable_visual_editor_post_exclude') $disable_visual_editor_post_exclude=$row->option_value;

    if($row->option_name=='disable_heartbeat') $disable_heartbeat=$row->option_value;
    if($row->option_name=='disable_heartbeat_pagetype') $disable_heartbeat_pagetype=explode(",",$row->option_value);
    if($row->option_name=='disable_heartbeat_pagetype_exclude') $disable_heartbeat_pagetype_exclude=$row->option_value;
    if($row->option_name=='disable_heartbeat_post') $disable_heartbeat_post=explode(",",$row->option_value);
    if($row->option_name=='disable_heartbeat_post_exclude') $disable_heartbeat_post_exclude=$row->option_value;
    
    if($row->option_name=='limit_revisions') $limit_revisions=$row->option_value;
    if($row->option_name=='limit_revisions_count') $limit_revisions_count=$row->option_value;
    if($row->option_name=='limit_revisions_post') $limit_revisions_post=explode(",",$row->option_value);
    if($row->option_name=='limit_revisions_post_exclude') $limit_revisions_post_exclude=$row->option_value;
    
    if($row->option_name=='limit_usereditor_role') $limit_usereditor_role=$row->option_value;
    if($row->option_name=='limit_usereditor_role_role') $limit_usereditor_role_role=explode(",",$row->option_value);
    if($row->option_name=='limit_usereditor_role_role_exclude') $limit_usereditor_role_role_exclude=$row->option_value;
    if($row->option_name=='limit_usereditor_hiderole_role') {$limit_usereditor_hiderole_role=$row->option_value; $_SESSION["limit_usereditor_hiderole_role"]=$limit_usereditor_hiderole_role;}
    
    if($row->option_name=='password_reset_link') $password_reset_link=$row->option_value;
    if($row->option_name=='password_reset_link_role') $password_reset_link_role=explode(",",$row->option_value);
    if($row->option_name=='password_reset_link_role_exclude') $password_reset_link_role_exclude=$row->option_value;
    
    if($row->option_name=='disable_notices') $disable_notices=$row->option_value;
    if($row->option_name=='disable_notices_role') $disable_notices_role=explode(",",$row->option_value);
    if($row->option_name=='disable_notices_role_exclude') $disable_notices_role_exclude=$row->option_value;
    
    if($row->option_name=='page_hit') $page_hit=$row->option_value;
    if($row->option_name=='page_hit_role') $page_hit_role=explode(",",$row->option_value);
    if($row->option_name=='page_hit_role_exclude') $page_hit_role_exclude=$row->option_value;
    
    if($row->option_name=='editor_warning') $editor_warning=$row->option_value;
    if($row->option_name=='editor_warning_role') $editor_warning_role=explode(",",$row->option_value);
    if($row->option_name=='editor_warning_role_exclude') $editor_warning_role_exclude=$row->option_value;
    
    if($row->option_name=='dual_auth') $dual_auth=$row->option_value;
    if($row->option_name=='dual_auth_role') $dual_auth_role=explode(",",$row->option_value);
    if($row->option_name=='dual_auth_role_exclude') $dual_auth_role_exclude=$row->option_value;

    if($row->option_name=='disable_wpautop') $disable_wpautop=$row->option_value;
    if($row->option_name=='disable_wpautop_post') $disable_wpautop_post=explode(",",$row->option_value);
    if($row->option_name=='disable_wpautop_post_exclude') $disable_wpautop_post_exclude=$row->option_value;
    
  endforeach;


  // Get Post Type
  $post_type='';
  if(!empty($_GET['post'])) {
    $post=get_post(intval($_GET['post']));
    $post_type=$post->post_type;
  }
  if(empty($post_type) && !empty($_GET['post_type'])) $post_type=sanitize_text_field($_GET['post_type']);



  // Capture Page Hits
  if($page_hit==1) {
    if(($page_hit_role_exclude==0 && (count(array_intersect($page_hit_role,$user_roles))>0 || in_array('*All',$page_hit_role,true)))
     || ($page_hit_role_exclude==1 && (count(array_intersect($page_hit_role,$user_roles))==0 || in_array('*All',$page_hit_role,true)))) {

      atb_hit_page();
    }
  }


  // Disable TinyMCE Visual Editor
  if((atb_is_path('wp-admin/post.php?post=') || atb_is_path('wp-admin/post-new.php')) && $disable_visual_editor==1) {
    if(($disable_visual_editor_role_exclude==0 && (count(array_intersect($disable_visual_editor_role,$user_roles))>0 || in_array('*All',$disable_visual_editor_role,true)))
     || ($disable_visual_editor_role_exclude==1 && (count(array_intersect($disable_visual_editor_role,$user_roles))==0 || in_array('*All',$disable_visual_editor_role,true)))) {

      if(($disable_visual_editor_post_exclude==0 && (in_array($post_type,$disable_visual_editor_post,true) || in_array('*All',$disable_visual_editor_post,true)))
       || ($disable_visual_editor_post_exclude==1 && (!in_array($post_type,$disable_visual_editor_post,true) || in_array('*All',$disable_visual_editor_post,true)))) {
      
        add_filter('user_can_richedit','__return_false');
      }
    }
  }


  // User New/Edit Screen
  if((atb_is_path('wp-admin/user-new.php') || atb_is_path('wp-admin/user-edit.php')) && $limit_usereditor_role==1) {
    if(($limit_usereditor_role_role_exclude==0 && (count(array_intersect($limit_usereditor_role_role,$user_roles))>0 || in_array('*All',$limit_usereditor_role_role,true)))
     || ($limit_usereditor_role_role_exclude==1 && (count(array_intersect($limit_usereditor_role_role,$user_roles))==0 || in_array('*All',$limit_usereditor_role_role,true)))) {

      function atb_user() { ?>
      <style type='text/css'>.actions,#role{display:none}</style>
      
      <script type='text/javascript'>
        var removeUserRoles_interval=setInterval(function() {
          if(document.readyState==='complete') { clearInterval(removeUserRoles_interval); atb_removeUserRoles('role'); }
        },100);

        function atb_removeUserRoles(menu) {
          
          // Hide Specific Roles
          if(document.getElementById(menu)) {
            var roles=document.getElementById(menu);
            var role_html=roles.innerHTML;
            var hide_roles='<?php echo $_SESSION["limit_usereditor_hiderole_role"];?>'.split(',');
            if(hide_roles) for(var i=0; i<hide_roles.length; i++) if(hide_roles[i].length>1) role_html=role_html.replace('<option value="'+hide_roles[i]+'">','<option value="'+hide_roles[i]+'" hidden>');
            roles.innerHTML=role_html;
            roles.style.display='block';
          }

          // Hide Multiple Role Selection
          if(document.getElementsByClassName('form-table')) {
            var user_multi_role=document.getElementsByClassName('form-table');
            for(i=0; i < user_multi_role.length; i++) if(user_multi_role[i].innerHTML.indexOf('user-role-editor.php')>0) user_multi_role[i].style.visibility='hidden';
          }
        }
      </script><?php
      }
      add_action('admin_footer','atb_user');
    }
  }


  // User List Screen
  if(atb_is_path('wp-admin/users.php') && $limit_usereditor_role==1) {
    if(($limit_usereditor_role_role_exclude==0 && (count(array_intersect($limit_usereditor_role_role,$user_roles))>0 || in_array('*All',$limit_usereditor_role_role,true)))
     || ($limit_usereditor_role_role_exclude==1 && (count(array_intersect($limit_usereditor_role_role,$user_roles))==0 || in_array('*All',$limit_usereditor_role_role,true)))) {
      
      function atb_user_list() { ?>
        <style type='text/css'>#ure_grant_roles,#new_role,#new_role2{display:none}</style>
        
        <script type='text/javascript'>
          var removeUserRoles_interval=setInterval(function() {
            if(document.readyState==='complete') { clearInterval(removeUserRoles_interval); atb_removeUserRoles('new_role'); atb_removeUserRoles('new_role2');}
          },100);

          function atb_removeUserRoles(menu) {
          
            // Hide Specific Roles
            if(document.getElementById(menu)) {
              var roles=document.getElementById(menu);
              var role_html=roles.innerHTML;
              var hide_roles='<?php echo $_SESSION["limit_usereditor_hiderole_role"];?>'.split(',');
              if(hide_roles) for(var i=0; i<hide_roles.length; i++) if(hide_roles[i].length>1) role_html=role_html.replace('<option value="'+hide_roles[i]+'">','<option value="'+hide_roles[i]+'" hidden>');
              roles.innerHTML=role_html;
              roles.style.display='block';
            }

          }
        </script><?php
      }
      add_action('admin_footer','atb_user_list');
    }
  }


  // Password Reset via Emailed Link
  if($password_reset_link==1) {
    if(($password_reset_link_role_exclude==0 && (count(array_intersect($password_reset_link_role,$user_roles))>0 || in_array('*All',$password_reset_link_role,true)))
     || ($password_reset_link_role_exclude==1 && (count(array_intersect($password_reset_link_role,$user_roles))==0 || in_array('*All',$password_reset_link_role,true)))) {
     
      if(atb_is_path('wp-admin/user-edit.php')) {
      
        function atb_user_reset() { ?>
          <script type='text/javascript'>
            var forceResetviaEmail_interval=setInterval(function() {
              if(document.readyState==='complete') { clearInterval(forceResetviaEmail_interval); atb_forceResetviaEmail(); }
            },100);

            function atb_forceResetviaEmail() {
              // Force Admin to Reset Password via Emailed Link
              if(document.getElementsByClassName('wp-generate-pw')) {
                var pwemail=document.getElementById('email').value;
                var pwlink="<span class='dashicons dashicons-unlock'></span><a id='atb_reset_pw' href='#' style='text-decoration:none;' onclick=\"window.open('<?php echo wp_login_url();?>?action=lostpassword&user_login="+pwemail+"','Reset Password','width=600,height=700,resizable=yes,scrollbars=yes'); return false;\">Send a Password Reset Link</a>";
                var pwbutton=document.getElementsByClassName('wp-generate-pw');
                for (i=0; i<pwbutton.length; i++) pwbutton[i].style.visibility='hidden';
                document.getElementById('password').innerHTML=document.getElementById('password').innerHTML.replace('<td>','<td>'+pwlink);
              }
            }
          </script><?php
        }
        add_action('admin_footer','atb_user_reset');
      }
    }
  }


  // Disable Update Notices
  if($disable_notices==1) {
    if(($disable_notices_role_exclude==0 && (count(array_intersect($disable_notices_role,$user_roles))>0 || in_array('*All',$disable_notices_role,true)))
     || ($disable_notices_role_exclude==1 && (count(array_intersect($disable_notices_role,$user_roles))==0 || in_array('*All',$disable_notices_role,true)))) {

      function remove_core_updates(){global $wp_version;return(object) array('last_checked'=> time(),'version_checked'=> $wp_version,);}
      add_filter('pre_site_transient_update_core','remove_core_updates');
      add_filter('pre_site_transient_update_plugins','remove_core_updates');
      add_filter('pre_site_transient_update_themes','remove_core_updates');
      
      function atb_notices() { ?><style type='text/css'>.notice.is-dismissible, .error.is-dismissible{display:none!important}</style><?php }
      add_action('admin_footer','atb_notices');
      //echo "<script type='text/javascript'>alert('no_update_notification');</script>";
    }
  }


  // Stop WordPress Heartbeat on public pages
  if($disable_heartbeat==1) {
    if(($disable_heartbeat_pagetype_exclude==0 && (in_array('Admin',$disable_heartbeat_pagetype,true) || in_array('*All',$disable_heartbeat_pagetype,true)))
     || ($disable_heartbeat_pagetype_exclude==1 && (!in_array('Admin',$disable_heartbeat_pagetype,true) || !in_array('*All',$disable_heartbeat_pagetype,true)))) {

      if(($disable_heartbeat_post_exclude==0 && (in_array($post_type,$disable_heartbeat_post,true) || in_array('*All',$disable_heartbeat_post,true)))
       || ($disable_heartbeat_post_exclude==1 && (!in_array($post_type,$disable_heartbeat_post,true) || in_array('*All',$disable_heartbeat_post,true)))) {

        function atb_stop_heartbeat() {wp_deregister_script('heartbeat');}
        add_action('init','atb_stop_heartbeat',1);
        //echo "<script type='text/javascript'>alert('stop_heartbeat');</script>";
        
      }
    }
  }


  // Limit Post Revisions
  if(atb_is_path('wp-admin/post.php?post=') && $limit_revisions==1) {

    if(($limit_revisions_post_exclude==0 && (in_array($post_type,$limit_revisions_post,true) || in_array('*All',$limit_revisions_post,true)))
      || ($limit_revisions_post_exclude==1 && (!in_array($post_type,$limit_revisions_post,true) || in_array('*All',$limit_revisions_post,true)))) {
    
      //echo "<script type='text/javascript'>alert('limit_revisions:$limit_revisions_count');</script>";
      function atb_post_revisions($limit_revisions_count) {return $limit_revisions;}
      add_filter('wp_revisions_to_keep','atb_post_revisions',10,2);
    }
  }


  // Disable WPautop
  if($disable_wpautop==1) {
    
    if(($disable_wpautop_post_exclude==0 && (in_array($post_type,$disable_wpautop_post,true) || in_array('*All',$disable_wpautop_post,true)))
     || ($disable_wpautop_post_exclude==1 && (!in_array($post_type,$disable_wpautop_post,true) || in_array('*All',$disable_wpautop_post,true)))) {

      remove_filter('the_content','wpautop');
      remove_filter('the_excerpt','wpautop');
    }
  }


  // Theme Editor Warning
  if($editor_warning==1 && atb_is_path('wp-admin/theme-editor.php?file=') && atb_is_path('php&theme')) {
    if(($editor_warning_role_exclude==0 && (count(array_intersect($editor_warning_role,$user_roles))>0 || in_array('*All',$editor_warning_role,true)))
     || ($editor_warning_role_exclude==1 && (count(array_intersect($editor_warning_role,$user_roles))==0 || in_array('*All',$editor_warning_role,true)))) {

      function atb_editor() { ?>
        <script type='text/javascript'>
          var theme_editor_interval=setInterval(function() {
            if(document.readyState==='complete') { clearInterval(theme_editor_interval); atb_theme_editor();}
          },100);
          
          function atb_theme_editor() {
            if(!document.getElementById('submit')) return false;
            var submit=document.getElementById('submit');
            submit.onclick=function(){if(!confirm('WARNING: Editing live files poses a stability risk. Errors will immediately take the site offline.\n\nProceed with this risk?'))return false;}
          }
        </script><?php
      }
      add_action('admin_footer','atb_editor');
    }
  }


  // Dual Authentication
  if($dual_auth==1) {
    if(($dual_auth_role_exclude==0 && (count(array_intersect($dual_auth_role,$user_roles))>0 || in_array('*All',$dual_auth_role,true)))
     || ($dual_auth_role_exclude==1 && (count(array_intersect($dual_auth_role,$user_roles))==0 || in_array('*All',$dual_auth_role,true)))) {

      if(isset($_SESSION['atb_flag']) && atb_is_path('/wp-admin')) atb_email_token($_SERVER['HTTP_HOST'].$_SERVER['REQUEST_URI']);
    }
  }

} else add_action('admin_notices','atb_admin_notice');