JezK

Edit File: Template_Query.php

<?php

namespace Essential_Addons_Elementor\Traits;

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

trait Template_Query {

public $current_widget_name = '';

public function set_widget_name( $name = '' ) {
        $this->current_widget_name = $name;
    }

/**
     * Get only filename
     * @param   string
     * @return  string
     */
    public function get_filename_only( $path ) {
        $filename = \explode( '/', $path );
        return \end( $filename );
    }

/**
     * Retrieves Template name from file header.
     *
     * @array
     */
    private $template_headers = [
        'Template Name' => 'Template Name',
    ];

/**
     * Prepare the directory name from the following widget name.
     *
     * @access private
     *
     *
     * @return  string  $widget_name
     */
    private function process_directory_name()
    {
        if ( empty( $this->current_widget_name ) && \method_exists( $this, 'get_name' ) ) {
            $this->current_widget_name = $this->get_name();
        }
        $widget_name = str_replace('eael-', '', $this->current_widget_name);
        $widget_name = str_replace('-', ' ', $widget_name);
        $widget_name = ucwords($widget_name);
        $widget_name = str_replace(' ', '-', $widget_name);

return $widget_name;
    }

/**
     * Retrieve `Theme Template Directory`
     *
     * @return string templates directory from the active theme.
     */
    private function theme_templates_dir() {
        $current_theme = wp_get_theme();

$dir = sprintf(
            '%s/%s/Template/%s',
            $current_theme->theme_root,
            $current_theme->stylesheet,
            $this->process_directory_name()
        );

if ( is_dir( $dir ) ) {
            $file = scandir( $dir );
            $file = array_pop( $file );

return pathinfo( $file, PATHINFO_EXTENSION ) === 'php' ? $dir : false;
        }

return false;
    }

/**
     * Retrieves the lite plugin template directory path.
     *
     * @return  string  templates directory path of lite version.
     */
    private function get_template_dir() {
        return \sprintf(
            '%sincludes/Template/%s',
            EAEL_PLUGIN_PATH, $this->process_directory_name()
        );
    }

/**
     * Retrieves the pro plugin template directory path.
     *
     * @return  string  templates directory path of pro version.
     */
    private function get_pro_template_dir()
    {
        /**
         * Ask whether Pro is loaded, not where it lives. `eael/is_plugin_active` maps to
         * WP's is_plugin_active(), an exact `folder/file.php` match, so a hard-coded
         * folder name reported Pro as inactive whenever it was installed anywhere else -
         * a repo checkout, a re-download suffixed " 2", a manual rename. Every Pro
         * template preset then vanished with no notice. EA Pro registers
         * `eael/pro_enabled` from its own bootstrap regardless of directory, and
         * EAEL_PRO_PLUGIN_PATH is derived from __FILE__, so both follow the real
         * install location. See issue #895.
         */
        if (!apply_filters('eael/pro_enabled', false)) {
            return false;
        }

if (!defined('EAEL_PRO_PLUGIN_PATH')) {
            return false;
        }

return \sprintf(
            '%sincludes/Template/%s',
            EAEL_PRO_PLUGIN_PATH, $this->process_directory_name()
        );
    }

/**
     * Collecting templates from different sources.
     *
     * @return array
     */
	private function get_template_files() {
		$templates = [];

if ( is_dir( $this->get_template_dir() ) ) {
			$templates['lite'] = scandir( $this->get_template_dir(), 1 );
		}

if ( is_dir( $this->get_pro_template_dir() ) ) {
			$templates['pro'] = scandir( $this->get_pro_template_dir(), 1 );
		}

if ( $this->theme_templates_dir() ) {
			$templates['theme'] = scandir( $this->theme_templates_dir(), 1 );
		}

return $templates;
	}

/**
     * Retrieves template list from template directory.
     *
     * @return array template list.
     */
    protected function get_template_list() {
        $files = [];

if ($this->get_template_files()) {
            foreach ($this->get_template_files() as $key => $handler) {
                foreach ($handler as $handle) {
                    if (strpos($handle, '.php') !== false) {

if ($key === 'lite') {
                            $path = sprintf('%s/%s', $this->get_template_dir(), $handle);
                        } else if ($key === 'pro') {
                            $path = sprintf('%s/%s', $this->get_pro_template_dir(), $handle);
                        } else if ($key === 'theme') {
                            $path = sprintf('%s/%s', $this->theme_templates_dir(), $handle);
                        }

$template_info = get_file_data( $path, $this->template_headers );
                        $template_name = $template_info[ 'Template Name' ];

if ( $template_name ) {
                            $files[ $template_name ] = $path;
                        }
                    }
                }
            }
        }

return $files;
    }

/**
	 *
	 * Retrieves template list from template directory.
	 *
	 * @param false $sort
	 * @return array
	 */
    public function get_template_list_for_dropdown($sort = false)
    {
        $files = [];
        $templates = $this->get_template_files();
        if (!empty( $templates)) {
            foreach ($templates as $key => $handler) {
                foreach ($handler as $handle) {
                    if (strpos($handle, '.php') !== false) {

$path = $this->_get_path($key, $handle);
                        $template_info = get_file_data($path, $this->template_headers);
                        $template_name = $template_info['Template Name'];
                        $template_key = str_replace( ' ', '-', strtolower( $template_name ) );
                        if ( $template_name ) {
                            $files[$template_key] = sprintf("%s (%s)", ucfirst($template_name), ucfirst($key));
                        }
                    }
                }
            }
        }
        if($sort){
        	ksort($files);
        }
        return $files;
    }

public function _get_path($key, $handle)
    {
        $path = '';
        if ($key === 'lite') {
            $path = sprintf('%s/%s', $this->get_template_dir(), $handle);
        } else if ($key === 'pro') {
            $path = sprintf('%s/%s', $this->get_pro_template_dir(), $handle);
        } else if ($key === 'theme') {
            $path = sprintf('%s/%s', $this->theme_templates_dir(), $handle);
        }
        return $path;
    }

/**
     * Preparing template options for frontend select
     *
     * @return array teplate select options.
     */
    private function get_template_options() {
        $files = [];

if ( $this->get_template_list() ) {
            foreach ( $this->get_template_list() as $filename => $path ) {
                $filename = \str_replace( ' ', '-', $filename );

$files[ strtolower( $filename ) ] = $path;
            }
        }

return $files;
    }

/**
     * Adding key value pairs in template options.
     *
     * @return array
     */
    private function template_options() {
        $keys = array_keys( $this->get_template_options() );
        $values = array_keys( $this->get_template_list() );

return array_combine( $keys, $values );
    }

/**
     * Retrieve template
     *
     * @param string $filename
     *
     * @return string include-able full template path.
     */
    public function get_template( $filename ) {

if ( in_array( $filename, array_keys( $this->get_template_options() ) ) ) {
            $file = $this->get_template_options()[ $filename ];
            return $file;
        }

return false;
    }

/**
     * Set default option in frontend select control.
     *
     * @return string first option.
     */
    public function get_default() {
        $dt = array_reverse( $this->template_options() );

return strtolower( array_pop( $dt ) );
    }

/**
	 * Get directory name based on given file name
	 * @param $filename
	 * @return int|string
	 */
    protected function get_temp_dir_name($filename){
    	if(empty($filename)){
    		return 'free';
	    }
	    $template_files = array_reverse($this->get_template_files());
	    foreach ($template_files as $key => $handler) {
		    if(in_array($filename,$handler)){
				return $key;
		    }
	    }
	    return 'free';
    }
}