OwlCyberSecurity - MANAGER

Edit File: class-dwe-admin.php

<?php
namespace DWE_Plugin;

use Elementor;
use Elementor\Core\Files\CSS\Post as Post_CSS;

// Exit if accessed directly.
if ( ! defined( 'ABSPATH' ) ) {
	exit;
}

/**
 * Handles admin logic.
 */
final class Admin {
	/**
	 * Holds the current class object.
	 * 
	 * @since 1.0.0
	 * @var object
	 */
	public static $instance;

/**
	 * Holds the settings page slug.
	 * 
	 * @since 1.0.0
	 * @var string
	 */
	public $settings_page;

/**
	 * Holds the settings page title.
	 * 
	 * @since 1.0.0
	 * @var string
	 */
	public $settings_title;

/**
	 * Holds the settings.
	 * 
	 * @since 1.0.0
	 * @var array
	 */
	public $settings;

/**
     * Holds the user roles.
     *
     * @since 1.0.0
     * @var array
     */
    public $roles;

/**
     * Holds the current user role.
     *
     * @since 1.0.0
     * @var string
     */
    public $current_role;

/**
	 * Initializes the admin settings.
	 *
	 * @since 1.0.0
	 * @return void
	 */
	public function __construct()
	{
		if ( ! is_admin() ) {
			return;
		}

$this->settings_page = 'dwe-settings';
		$this->settings_title = __('Dashboard Welcome for Elementor', 'ibx-dwe');
		$this->settings = $this->get_settings();

add_action( 'admin_menu', array( $this, 'admin_menu' ), 1000 );
		add_action( 'admin_init', array( $this, 'admin_init' ) );
	}

/**
	 * Initializes admin related logic.
	 *
	 * @since 1.0.0
	 * @return void
	 */
	public function admin_init()
	{
		$this->update_settings();

global $pagenow;

if ( 'index.php' != $pagenow ) {
            return;
		}

$settings 	= $this->settings;
		$role		= $this->current_role;

if ( ! empty( $settings ) && isset( $settings[ $role ] ) ) {
			if ( isset( $settings[ $role ]['template'] ) && ! empty( $settings[ $role ]['template'] ) ) {
				remove_action( 'welcome_panel', 'wp_welcome_panel' );
				add_action( 'welcome_panel', array( $this, 'welcome_panel' ) );

add_action( 'admin_head', array( $this, 'print_styles' ) );

// custom fallback for the users who don't have
				// enough capabilities to display welcome panel.
				if ( ! current_user_can( 'edit_theme_options' ) ) {
					add_action( 'admin_notices', array( $this, 'welcome_panel' ) );
				}
			}
		}
	}

/**
	 * Renders welcome panel.
	 *
	 * @since 1.0.0
	 * @return void
	 */
	public function welcome_panel()
	{
		include IBX_DWE_DIR . 'includes/welcome-panel.php';
	}
    
	/**
	 * Print styles
	 *
	 * @since 1.0.6
	 * @return void
	 */
    public function print_styles()
    {
		$settings 	= $this->settings;
		$role		= $this->current_role;
		$dismissible = isset( $settings[ $role ]['dismissible'] ) ? true : false;
    	?>
    	<style type="text/css" id="dwe-dashboard-welcome-css">
			.welcome-panel {
				padding: 0;
				background: none;
				overflow: inherit;
			}
			.welcome-panel:before,
			.welcome-panel:after {
				background: none;
			}
			.welcome-panel .welcome-panel-close {
				z-index: 100;
				<?php if ( ! $dismissible ) { ?>
					display: none;
				<?php } ?>
			}
			.welcome-panel .welcome-panel-content {
				display: none !important;
			}
			#dwe-dashboard-welcome {
				-webkit-font-smoothing: antialiased;
			}
			#dwe-dashboard-welcome .fl-builder-content ul,
			#dwe-dashboard-welcome .fl-builder-content ol {
				list-style: inherit;
			}
			#dwe-dashboard-welcome .fl-builder-content p {
				color: inherit;
				font-size: inherit;
				margin: inherit;
				margin-bottom: 10px;
			}
			#dwe-dashboard-welcome input:focus,
			#dwe-dashboard-welcome textarea:focus,
			#dwe-dashboard-welcome select:focus,
			#dwe-dashboard-welcome button:focus {
				-webkit-box-shadow: none;
				box-shadow: none;
			}
		</style>
    	<?php
    }

/**
	 * Add Dashboard Welcome to admin menu.
	 *
	 * @since 1.0.0
	 * @return void
	 */
	public function admin_menu()
	{	
		global $wp_roles;

$this->roles 		= $wp_roles->get_names();
		$this->current_role = $this->get_user_role();

if ( is_multisite() ) {
			$hide_settings = get_blog_option( 1, 'dwe_hide_from_subsites' );
			
			if ( $hide_settings && get_current_blog_id() != 1 ) {
				return;
			}
		}

if ( current_user_can( 'manage_options' ) ) {

$title = __('Dashboard Welcome Elementor', 'ibx-dwe');
			$cap   = 'manage_options';
			$slug  = $this->settings_page;
			$func  = array( $this, 'render_settings' );

add_submenu_page( 'options-general.php', $title, $title, $cap, $slug, $func );
		}
	}

/**
	 * Renders settings content.
	 *
	 * @since 1.0.0
	 * @return void
	 */
	public function render_settings()
	{
		$title 			= $this->settings_title;
		$form_action 	= $this->get_form_action();
		$roles			= $this->roles;
		$current_role	= $this->current_role;
		$templates		= $this->get_templates();
		$settings		= $this->get_settings();

include IBX_DWE_DIR . 'includes/admin-settings.php';
	}

/**
	 * Renders Elementor template in welcome panel.
	 *
	 * @since 1.0.0
	 * @return void
	 */
	public function render_template()
	{
		$settings 	= $this->settings;
		$role		= $this->current_role;

if ( ! empty( $settings ) && isset( $settings[ $role ] ) ) {
			if ( isset( $settings[ $role ]['template'] ) && ! empty( $settings[ $role ]['template'] ) ) {
				$template_id = $settings[ $role ]['template'];
				$site_id 	 = isset( $settings[ $role ]['site'] ) ? $settings[ $role ]['site'] : '';
				$dismissible = isset( $settings[ $role ]['dismissible'] ) ? true : false;
				$is_multisite = is_multisite();

$elementor = Elementor\Plugin::$instance;

echo '<style>';
				$css = file_get_contents( IBX_DWE_DIR . 'assets/css/dashboard.css' );

if ( ! $dismissible ) {
					$css .= '.welcome-panel .welcome-panel-close { display: none; }';
				}

$css = str_replace( array("\r\n", "\n", "\r\t", "\t", "\r"), '', $css );
				$css = preg_replace('/\s+/', ' ', $css);

echo $css;
				echo '</style>';

$elementor->frontend->register_styles();
				$elementor->frontend->enqueue_styles();

if ( ! empty( $site_id ) && $is_multisite ) {
					switch_to_blog( $site_id );
				}

echo $elementor->frontend->get_builder_content( $template_id, true );

if ( ! empty( $site_id ) && $is_multisite ) {
					restore_current_blog();
				}

$elementor->frontend->register_scripts();
				$elementor->frontend->enqueue_scripts();
			}
		}
	}

/**
	 * Get setting form action attribute.
	 *
	 * @since 1.0.0
	 * @return string
	 */
	private function get_form_action()
	{
		return admin_url( '/admin.php?page=' . $this->settings_page );
	}

/**
	 * Get Elementor saved templates.
	 *
	 * @since 1.0.0
	 * @return array
	 */
	private function get_templates()
	{
		$args = array(
            'post_type'         => 'elementor_library',
			'posts_per_page'    => '-1',
			'post_status'		=> 'publish'
		);

$templates = get_posts( $args );

// Multisite support.
		if ( is_multisite() ) {

$blog_id = get_current_blog_id();

if ( $blog_id != 1 ) {
                switch_to_blog(1);

// Get posts from main site.
                $main_posts = get_posts( $args );

// Loop through each main site post
                // and add site_id to post object.
                foreach ( $main_posts as $main_post ) {
                    $main_post->site_id = 1;
                }

$templates = array_merge( $templates, $main_posts );

restore_current_blog();
            }
            else {
                // Loop through each main site post
                // and add site_id to post object.
                foreach ( $templates as $template ) {
                    $template->site_id = 1;
                }
            }
        }
		
		$data = array();

if ( ! empty( $templates ) && ! is_wp_error( $templates ) ){
            foreach ( $templates as $post ) {
                $data[ $post->ID ] = array(
					'title'	=> $post->post_title,
					'site'	=> isset( $post->site_id ) ? $post->site_id : null
				);
            }
		}
		
        return $data;
	}

/**
	 * Get user roles.
	 *
	 * @since 1.0.0
	 * @return mixed
	 */
	private function get_user_role()
	{
		// Get current user role in multisite network using WP_User_Query.
        if ( is_multisite() ) {
			$user_query = new \WP_User_Query( array( 'blog_id' => 1 , 'include' => array( get_current_user_id() ) ) );
			
            if ( ! empty( $user_query->results ) ) {
				$roles = $user_query->results[0]->roles;
				
                if ( is_array( $roles ) && count( $roles ) ) {
                    return $roles[0];
                }
            }
        }

$user   = wp_get_current_user();
        $roles  = $user->roles;
        $roles  = array_shift( $roles );

return $roles;
	}

/**
	 * Get setting form database.
	 *
	 * @since 1.0.0
	 * @return array
	 */
	public function get_settings()
	{
		$key = '_dwe_templates';

if ( is_multisite() ) {
			$settings = get_option( $key, false );
			$settings = ! $settings ? get_blog_option( 1, $key ) : $settings;
		} else {
			$settings = get_option( $key, array() );
		}

if ( ! is_array( $settings ) ) {
			$settings = array();
		}

$this->settings = $settings;

return $settings;
	}

/**
	 * Update settings in database
	 *
	 * @since 1.0.0
	 */
	public function update_settings()
	{
		if ( ! isset( $_POST['dwe_settings_nonce'] ) || ! wp_verify_nonce( $_POST['dwe_settings_nonce'], 'dwe_settings' ) ) {
			return;
		}

if ( ! isset( $_POST['dwe_templates'] ) ) {
			return;
		}

$data = array();

foreach ( $_POST['dwe_templates'] as $user_role => $template ) {
			$data[ $user_role ] = array_map( 'sanitize_text_field', wp_unslash( $template ) );
		}

update_option( '_dwe_templates', $data );

if ( isset( $_POST['dwe_hide_from_subsites'] ) ) {
			update_option( 'dwe_hide_from_subsites', true );
		} else {
			update_option( 'dwe_hide_from_subsites', false );
		}
	}

/**
	 * Delete setting form database.
	 *
	 * @since 1.0.0
	 */
	public function delete_settings()
	{
		delete_option( '_dwe_templates' );
	}

/**
	 * Get class instance.
	 *
	 * @since 1.0.0
	 * @return object
	 */
	public static function get_instance()
	{
		if ( ! isset( self::$instance ) && ! ( self::$instance instanceof DWE_Plugin\Admin ) ) {
			self::$instance = new Admin();
		}

return self::$instance;
	}
}