<?php
namespace um\core;

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

if ( ! class_exists( 'Shortcodes' ) ) {
    class Shortcodes {

        function __construct() {

            $this->message_mode = false;
            $this->custom_message = '';

            $this->loop = array();

            add_shortcode( 'ultimatemember', array( &$this, 'ultimatemember' ) );

            add_shortcode( 'um_loggedin', array( &$this, 'um_loggedin' ) );
            add_shortcode( 'um_loggedout', array( &$this, 'um_loggedout' ) );
            add_shortcode( 'um_show_content', array( &$this, 'um_shortcode_show_content_for_role' ) );
            add_shortcode( 'ultimatemember_searchform', array( &$this, 'ultimatemember_searchform' ) );


            add_filter( 'body_class', array( &$this, 'body_class' ), 0 );

	        add_filter( 'um_shortcode_args_filter', array( &$this, 'display_logout_form' ), 99 );
	        add_filter( 'um_shortcode_args_filter', array( &$this, 'parse_shortcode_args' ), 99 );

	        $base_uri = apply_filters( 'um_emoji_base_uri', 'https://s.w.org/images/core/emoji/' );

	        $this->emoji[':)'] = $base_uri . '72x72/1f604.png';
	        $this->emoji[':smiley:'] = $base_uri . '72x72/1f603.png';
            $this->emoji[':D'] = $base_uri . '72x72/1f600.png';
            $this->emoji[':$'] = $base_uri . '72x72/1f60a.png';
            $this->emoji[':relaxed:'] = $base_uri . '72x72/263a.png';
            $this->emoji[';)'] = $base_uri . '72x72/1f609.png';
            $this->emoji[':heart_eyes:'] = $base_uri . '72x72/1f60d.png';
            $this->emoji[':kissing_heart:'] = $base_uri . '72x72/1f618.png';
            $this->emoji[':kissing_closed_eyes:'] = $base_uri . '72x72/1f61a.png';
            $this->emoji[':kissing:'] = $base_uri . '72x72/1f617.png';
            $this->emoji[':kissing_smiling_eyes:'] = $base_uri . '72x72/1f619.png';
            $this->emoji[';P'] = $base_uri . '72x72/1f61c.png';
            $this->emoji[':P'] = $base_uri . '72x72/1f61b.png';
            $this->emoji[':stuck_out_tongue_closed_eyes:'] = $base_uri . '72x72/1f61d.png';
            $this->emoji[':flushed:'] = $base_uri . '72x72/1f633.png';
            $this->emoji[':grin:'] = $base_uri . '72x72/1f601.png';
            $this->emoji[':pensive:'] = $base_uri . '72x72/1f614.png';
            $this->emoji[':relieved:'] = $base_uri . '72x72/1f60c.png';
            $this->emoji[':unamused'] = $base_uri . '72x72/1f612.png';
            $this->emoji[':('] = $base_uri . '72x72/1f61e.png';
            $this->emoji[':persevere:'] = $base_uri . '72x72/1f623.png';
            $this->emoji[":'("] = $base_uri . '72x72/1f622.png';
            $this->emoji[':joy:'] = $base_uri . '72x72/1f602.png';
            $this->emoji[':sob:'] = $base_uri . '72x72/1f62d.png';
            $this->emoji[':sleepy:'] = $base_uri . '72x72/1f62a.png';
            $this->emoji[':disappointed_relieved:'] = $base_uri . '72x72/1f625.png';
            $this->emoji[':cold_sweat:'] = $base_uri . '72x72/1f630.png';
            $this->emoji[':sweat_smile:'] = $base_uri . '72x72/1f605.png';
            $this->emoji[':sweat:'] = $base_uri . '72x72/1f613.png';
            $this->emoji[':weary:'] = $base_uri . '72x72/1f629.png';
            $this->emoji[':tired_face:'] = $base_uri . '72x72/1f62b.png';
            $this->emoji[':fearful:'] = $base_uri . '72x72/1f628.png';
            $this->emoji[':scream:'] = $base_uri . '72x72/1f631.png';
            $this->emoji[':angry:'] = $base_uri . '72x72/1f620.png';
            $this->emoji[':rage:'] = $base_uri . '72x72/1f621.png';
            $this->emoji[':triumph'] = $base_uri . '72x72/1f624.png';
            $this->emoji[':confounded:'] = $base_uri . '72x72/1f616.png';
            $this->emoji[':laughing:'] = $base_uri . '72x72/1f606.png';
            $this->emoji[':yum:'] = $base_uri . '72x72/1f60b.png';
            $this->emoji[':mask:'] = $base_uri . '72x72/1f637.png';
            $this->emoji[':cool:'] = $base_uri . '72x72/1f60e.png';
            $this->emoji[':sleeping:'] = $base_uri . '72x72/1f634.png';
            $this->emoji[':dizzy_face:'] = $base_uri . '72x72/1f635.png';
            $this->emoji[':astonished:'] = $base_uri . '72x72/1f632.png';
            $this->emoji[':worried:'] = $base_uri . '72x72/1f61f.png';
            $this->emoji[':frowning:'] = $base_uri . '72x72/1f626.png';
            $this->emoji[':anguished:'] = $base_uri . '72x72/1f627.png';
            $this->emoji[':smiling_imp:'] = $base_uri . '72x72/1f608.png';
            $this->emoji[':imp:'] = $base_uri . '72x72/1f47f.png';
            $this->emoji[':open_mouth:'] = $base_uri . '72x72/1f62e.png';
            $this->emoji[':grimacing:'] = $base_uri . '72x72/1f62c.png';
            $this->emoji[':neutral_face:'] = $base_uri . '72x72/1f610.png';
            $this->emoji[':confused:'] = $base_uri . '72x72/1f615.png';
            $this->emoji[':hushed:'] = $base_uri . '72x72/1f62f.png';
            $this->emoji[':no_mouth:'] = $base_uri . '72x72/1f636.png';
            $this->emoji[':innocent:'] = $base_uri . '72x72/1f607.png';
            $this->emoji[':smirk:'] = $base_uri . '72x72/1f60f.png';
            $this->emoji[':expressionless:'] = $base_uri . '72x72/1f611.png';

        }


	    /**
	     * Conditional logout form
	     *
	     * @param array $args
	     *
	     * @return array
	     */
	    function display_logout_form( $args ) {
		    if ( is_user_logged_in() && isset( $args['mode'] ) && $args['mode'] == 'login' ) {

			    if ( get_current_user_id() != um_user( 'ID' ) ) {
				    um_fetch_user( get_current_user_id() );
			    }

			    $args['template'] = 'logout';
		    }

		    return $args;
	    }


	    /**
	     * Filter shortcode args
	     *
	     * @param array $args
	     *
	     * @return array
	     */
	    function parse_shortcode_args( $args ) {
		    if ( $this->message_mode == true ) {

			    if ( ! empty( $_REQUEST['um_role'] ) ) {
				    $args['template'] = 'message';
				    $roleID = esc_attr( $_REQUEST['um_role'] );
				    $role = UM()->roles()->role_data( $roleID );

				    if ( ! empty( $role ) && ! empty( $role["status"] ) ) {
					    $message_key = $role["status"] . '_message';
					    $this->custom_message = ! empty( $role[ $message_key ] ) ? $role[ $message_key ] : '';
				    }
			    }

		    }

		    foreach ( $args as $k => $v ) {
			    $args[ $k ] = maybe_unserialize( $args[ $k ] );
		    }

		    return $args;
	    }


        /***
         ***	@emoji support
         */
        function emotize($content) {
            $content = stripslashes($content);
            foreach ($this->emoji as $code => $val) {
                $regex = str_replace(array('(', ')'), array("\\" . '(', "\\" . ')'), $code);
                $content = preg_replace('/(' . $regex . ')(\s|$)/', '<img src="' . $val . '" alt="' . $code . '" title="' . $code . '" class="emoji" />$2', $content);
            }
            return $content;
        }

        /***
         ***	@extend body classes
         */
        function body_class($classes) {
            $array = UM()->config()->permalinks;
            if (!$array) {
                return $classes;
            }

            foreach ($array as $slug => $info) {
                if (um_is_core_page($slug)) {

                    $classes[] = 'um-page-' . $slug;

                    if (is_user_logged_in()) {
                        $classes[] = 'um-page-loggedin';
                    } else {
                        $classes[] = 'um-page-loggedout';
                    }

                }
            }

            if( um_is_core_page('user') && um_is_user_himself() ){
                $classes[] = 'um-own-profile';
            }

            return $classes;
        }

        /***
         ***	@Retrieve core login form
         */
        function core_login_form() {
            $forms = get_posts(array('post_type' => 'um_form', 'posts_per_page' => 1, 'meta_key' => '_um_core', 'meta_value' => 'login'));
            $form_id = isset( $forms[0]->ID ) ? $forms[0]->ID: 0;

            return $form_id;
        }


	    /**
	     * Load a compatible template
	     *
	     * @param $tpl
	     */
        function load_template( $tpl ) {
            $loop = ( $this->loop ) ? $this->loop : array();

            if ( isset( $this->set_args ) && is_array( $this->set_args ) ) {
                $args = $this->set_args;
                
                unset( $args['file'] );
                unset( $args['theme_file'] );
                unset( $args['tpl'] );

                extract( $args );
            }

            $file = um_path . "templates/{$tpl}.php";
            $theme_file = get_stylesheet_directory() . "/ultimate-member/templates/{$tpl}.php";


            if ( file_exists( $theme_file ) ) {
                $file = $theme_file;
            }

            if ( file_exists( $file ) ) {

                include $file;
            }
        }


        /***
         ***	@Add class based on shortcode
         */
        function get_class($mode, $args = array()) {

            $classes = 'um-' . $mode;

            if (is_admin()) {
                $classes .= ' um-in-admin';
            }

            if (isset(UM()->form()->errors) && UM()->form()->errors) {
                $classes .= ' um-err';
            }

            if (UM()->fields()->editing == true) {
                $classes .= ' um-editing';
            }

            if (UM()->fields()->viewing == true) {
                $classes .= ' um-viewing';
            }

            if (isset($args['template']) && $args['template'] != $args['mode']) {
                $classes .= ' um-' . $args['template'];
            }

            $classes = apply_filters('um_form_official_classes__hook', $classes);
            return $classes;
        }

	    /**
	     * Logged-in only content
	     *
	     * @param array $args
	     * @param string $content
	     *
	     * @return string
	     */
        function um_loggedin( $args = array(), $content = "" ) {
            ob_start();

            $defaults = array(
                'lock_text' => __( 'This content has been restricted to logged in users only. Please <a href="{login_referrer}">login</a> to view this content.', 'ultimate-member' ),
                'show_lock' => 'yes',
            );

            $args = wp_parse_args( $args, $defaults );

            $args['lock_text'] = $this->convert_locker_tags( $args['lock_text'] );

            if ( ! is_user_logged_in() ) {
                if ( $args['show_lock'] == 'no' ) {
                    echo '';
                } else {
                    $this->set_args = $args;
                    $this->load_template( 'login-to-view' );
                }
            } else {
                echo do_shortcode( $this->convert_locker_tags( wpautop( $content ) ) );
            }

            $output = ob_get_clean();
            return $output;
        }

        /***
         ***	@Logged-out only content
         */
        function um_loggedout($args = array(), $content = "") {
            ob_start();

            // Hide for logged in users
            if (is_user_logged_in()) {
                echo '';
            } else {
                echo do_shortcode(wpautop($content));
            }

            $output = ob_get_contents();
            ob_end_clean();
            return $output;
        }

        /***
         ***	@Shortcode
         */
        function ultimatemember($args = array()) {
            return $this->load($args);
        }

        /***
         ***	@Load a module with global function
         */
        function load($args) {
            ob_start();

            $defaults = array();
            $args = wp_parse_args($args, $defaults);

            // when to not continue
            $this->form_id = (isset($args['form_id'])) ? $args['form_id'] : null;
            if (!$this->form_id) {
                return;
            }

            $this->form_status = get_post_status($this->form_id);
            if ($this->form_status != 'publish') {
                return;
            }

            // get data into one global array
            $post_data = UM()->query()->post_data($this->form_id);

            $args = apply_filters('um_pre_args_setup', $post_data);

            if (!isset($args['template'])) {
                $args['template'] = '';
            }

            if (isset($post_data['template']) && $post_data['template'] != $args['template']) {
                $args['template'] = $post_data['template'];
            }

            if (!$this->template_exists($args['template'])) {
                $args['template'] = $post_data['mode'];
            }

            if (!isset($post_data['template'])) {
                $post_data['template'] = $post_data['mode'];
            }

	        if( 'directory' != $args['mode'] ) {

		        $args = array_merge( $post_data, $args );

		        if (empty( $args['use_custom_settings'] )) {
			        $args = array_merge( $args, $this->get_css_args( $args ) );
		        } else {
			        $args = array_merge( $this->get_css_args( $args ), $args );
		        }
	        }
            // filter for arguments
            $args = apply_filters('um_shortcode_args_filter', $args);

            extract($args, EXTR_SKIP);

	        if ( 'register' == $mode && is_user_logged_in() ) {
		        return __( 'You are already registered', 'ultimate-member' );
	        }

            // for profiles only
            if ( $mode == 'profile' && um_profile_id() ) {
	            $current_user_roles = UM()->roles()->get_all_user_roles( um_profile_id() );
	            if ( ! empty( $args['role'] ) && ! in_array( $args['role'], $current_user_roles ) ) {
		            return '';
	            }
            }

            // start loading the template here
            do_action("um_pre_{$mode}_shortcode", $args);

            do_action("um_before_form_is_loaded", $args);

            do_action("um_before_{$mode}_form_is_loaded", $args);

            $this->template_load( $template, $args );

            $this->dynamic_css( $args );

            if (um_get_requested_user() || $mode == 'logout') {
                um_reset_user();
            }

            do_action('um_after_everything_output');

            $output = ob_get_contents();
            ob_end_clean();
            return $output;
        }


        /**
         * Get dynamic CSS args
         *
         * @param $args
         * @return array
         */
        function get_css_args( $args ) {
            $arr = um_styling_defaults( $args['mode'] );
            $arr = array_merge( $arr, array( 'form_id' => $args['form_id'], 'mode' => $args['mode'] ) );
            return $arr;
        }


	    /**
	     * Load dynamic css
	     *
	     * @param array $args
	     *
	     * @return string
	     */
        function dynamic_css( $args = array() ) {
	        /*
	         * for fix the issue #306
	         */
			$disable_css = apply_filters( 'um_disable_dynamic_global_css', false );
	        if ( $disable_css )
	        	return '';

	        /**
	         * @var $mode
	         */
            extract( $args );

            include_once um_path . 'assets/dynamic_css/dynamic_global.php';

            if ( isset( $mode ) && in_array( $mode, array( 'profile', 'directory' ) ) ) {
                $file = um_path . 'assets/dynamic_css/dynamic_' . $mode . '.php';

                if ( file_exists( $file ) )
                    include_once $file;
            }

	        return '';
        }


	    /**
	     * Loads a template file
	     *
	     * @param $template
	     * @param array $args
	     */
        function template_load( $template, $args = array() ) {
            if ( is_array( $args ) ) {
                $this->set_args = $args;
            }
            $this->load_template( $template );
        }


        /***
         ***	@Checks if a template file exists
         */
        function template_exists($template) {

            $file = um_path . 'templates/' . $template . '.php';
            $theme_file = get_stylesheet_directory() . '/ultimate-member/templates/' . $template . '.php';

            if (file_exists($theme_file) || file_exists($file)) {
                return true;
            }

            return false;
        }

        /***
         ***	@Get File Name without path and extension
         */
        function get_template_name($file) {
            $file = basename($file);
            $file = preg_replace('/\\.[^.\\s]{3,4}$/', '', $file);
            return $file;
        }

        /***
         ***	@Get Templates
         */
        function get_templates($excluded = null) {

            if ($excluded) {
                $array[$excluded] = __('Default Template', 'ultimate-member');
            }

            $paths[] = glob(um_path . 'templates/' . '*.php');

            if (file_exists(get_stylesheet_directory() . '/ultimate-member/templates/')) {
                $paths[] = glob(get_stylesheet_directory() . '/ultimate-member/templates/' . '*.php');
            }

            if( isset( $paths ) && ! empty( $paths ) ){

                foreach ($paths as $k => $files) {

                    if( isset( $files ) && ! empty( $files ) ){

                        foreach ($files as $file) {

                            $clean_filename = $this->get_template_name( $file );

                            if ( 0 === strpos( $clean_filename, $excluded ) ) {

                                $source = file_get_contents( $file );
                                $tokens = token_get_all( $source );
                                $comment = array(
                                    T_COMMENT, // All comments since PHP5
                                    T_DOC_COMMENT, // PHPDoc comments
                                );
                                foreach ( $tokens as $token ) {
                                    if ( in_array( $token[0], $comment ) && strstr( $token[1], '/* Template:' ) && $clean_filename != $excluded ) {
                                        $txt = $token[1];
                                        $txt = str_replace('/* Template: ', '', $txt );
                                        $txt = str_replace(' */', '', $txt );
                                        $array[ $clean_filename ] = $txt;
                                    }
                                }

                            }

                        }

                    }

                }

            }

            return $array;

        }


		/**
		* Get Shortcode for given form ID
		*
		* @param $post_id
		*
		* @return string
		*/
		function get_shortcode( $post_id ) {
			$shortcode = '[ultimatemember form_id=' . $post_id . ']';
			return $shortcode;
		}


		/**
		* Convert access lock tags
		*
		* @param $str
		*
		* @return mixed|string
		*/
		function convert_locker_tags( $str ) {
			return um_convert_tags( $str, array(), false );
		}


		/**
		* Convert user tags in a string
		*
		* @param $str
		*
		* @return mixed
		*/
		function convert_user_tags( $str ) {

			$pattern_array = array(
				'{first_name}',
				'{last_name}',
				'{display_name}',
				'{user_avatar_small}',
				'{username}',
			);

			$pattern_array = apply_filters( 'um_allowed_user_tags_patterns', $pattern_array );

			//$matches = false;
			foreach ( $pattern_array as $pattern ) {
				$value = '';

				if ( preg_match( $pattern, $str ) ) {

					$usermeta = str_replace( '{', '', $pattern );
					$usermeta = str_replace( '}', '', $usermeta );

					if ( $usermeta == 'user_avatar_small' ) {
						$value = get_avatar( um_user( 'ID' ), 40 );
					} elseif ( um_user( $usermeta ) ) {
						$value = um_user( $usermeta );
					}

					if ( $usermeta == 'username' ) {
						$value = um_user( 'user_login' );
					}

					$value = apply_filters( "um_profile_tag_hook__{$usermeta}", $value, um_user( 'ID' ) );

					if ( $value ) {
						$str = preg_replace( '/' . $pattern . '/', $value, $str );
					}

				}

			}

			return $str;
		}


	    /**
	     * Shortcode: Show custom content to specific role
	     *
	     * Show content to specific roles
	     * [um_show_content roles='member'] <!-- insert content here -->  [/um_show_content]
	     * You can add multiple target roles, just use ',' e.g.  [um_show_content roles='member,candidates,pets']
	     *
	     * Hide content from specific roles
	     * [um_show_content not='contributors'] <!-- insert content here -->  [/um_show_content]
	     * You can add multiple target roles, just use ',' e.g.  [um_show_content roles='member,candidates,pets']
	     *
	     * @param  array $atts
	     * @param  string $content
	     * @return string
	     */
	    function um_shortcode_show_content_for_role( $atts = array() , $content = '' ) {
		    global $user_ID;

		    if ( ! is_user_logged_in() ) {
			    return;
		    }

		    $a = shortcode_atts( array(
			    'roles' => '',
			    'not' => '',
			    'is_profile' => false,
		    ), $atts );

		    if ( $a['is_profile'] ) {
			    um_fetch_user( um_profile_id() );
		    } else {
			    um_fetch_user( $user_ID );
		    }

		    $current_user_roles = um_user( 'roles' );

		    if ( ! empty( $a['not'] ) && ! empty( $a['roles'] ) ) {
			    return do_shortcode( $content );
		    }

		    if ( ! empty( $a['not'] ) ) {
			    $not_in_roles = explode( ",", $a['not'] );

			    if ( is_array( $not_in_roles ) && count( array_intersect( $current_user_roles, $not_in_roles ) ) <= 0 ) {
				    return do_shortcode( $content );
			    }
		    } else {
			    $roles = explode( ",", $a['roles'] );

			    if ( is_array( $roles ) && count( array_intersect( $current_user_roles, $roles ) ) > 0 ) {
				    return do_shortcode( $content );
			    }
		    }

		    return '';
	    }


	    /**
	     * @param array $args
	     * @param string $content
	     *
	     * @return string
	     */
        public function ultimatemember_searchform( $args = array(), $content = "" ) {
            // turn off buffer
            ob_start();

            // load template
            $this->load_template( 'searchform' );

            // get the buffer
            $template = ob_get_contents();

            // clear the buffer
            ob_end_clean();

            return $template;
        }

    }
}