wpsy/Twig/WordpressTwigExtension.php line 173

Open in your IDE?
  1. <?php
  3. /**
  4.  * Class TwigExtension
  5.  *
  6.  * Provide a set of methods which can be used in template engine
  7.  *
  8.  */
  10. namespace Metabolism\WordpressBundle\Twig;
  12. use Metabolism\WordpressBundle\Entity\Image;
  13. use Metabolism\WordpressBundle\Factory\Factory;
  14. use Metabolism\WordpressBundle\Factory\PostFactory;
  15. use Metabolism\WordpressBundle\Factory\TermFactory;
  17. use Twig\Extension\AbstractExtension,
  18.     Twig\TwigFilter,
  19.     Twig\TwigFunction;
  21. class WordpressTwigExtension extends AbstractExtension{
  23.     public function getFilters()
  24.     {
  25.         return [
  26.             new TwigFilter( 'handle', 'sanitize_title' ),
  27.             new TwigFilter( 'placeholder', [$this, 'placeholder'] ),
  28.             new TwigFilter( 'more', [$this, 'more'] ),
  29.             new TwigFilter( 'resize', [$this, 'resize'] ),
  30.             new TwigFilter( 'picture', [$this, 'picture'] ),
  31.             new TwigFilter( 'stripshortcodes','strip_shortcodes' ),
  32.             new TwigFilter( 'function', [$this, 'execFunction'] ),
  33.             new TwigFilter( 'excerpt','wp_trim_words' ),
  34.             new TwigFilter( 'sanitize','sanitize_title' ),
  35.             new TwigFilter( 'shortcodes','do_shortcode' ),
  36.             new TwigFilter( 'wpautop','wpautop' ),
  37.             new TwigFilter( 'array',[$this, 'to_array'] ),
  38.         ];
  39.     }
  41.     /**
  42.      * @return array
  43.      */
  44.     public function getFunctions()
  45.     {
  46.         return [
  47.             new TwigFunction( 'pixel', [$this, 'generatePixel'] ),
  48.             new TwigFunction( 'fn', [$this, 'execFunction'] ),
  49.             new TwigFunction( 'function', [$this, 'execFunction'] ),
  50.             new TwigFunction( 'action', [$this, 'doAction'] ),
  51.             new TwigFunction( 'shortcode', 'do_shortcode' ),
  52.             new TwigFunction( 'login_url', 'wp_login_url' ),
  53.             new TwigFunction( 'search_form', 'get_search_form' ),
  54.             new TwigFunction( 'archive_url', 'get_post_type_archive_link' ),
  55.             new TwigFunction( 'attachment_url', 'wp_get_attachment_url' ),
  56.             new TwigFunction( 'post_url', [$this, 'getPermalink'] ),
  57.             new TwigFunction( 'term_url', [$this, 'getTermLink'] ),
  58.             new TwigFunction( 'bloginfo', 'get_bloginfo' ),
  59.             new TwigFunction( 'dynamic_sidebar', function($id){ return $this->getOutput('dynamic_sidebar', [$id]); }, ['is_safe' => array('html')]  ),
  60.             new TwigFunction( 'comment_form', function($post_id, $args=[]){ return $this->getOutput('comment_form', [$args, $post_id]); }, ['is_safe' => array('html')]  ),
  61.             new TwigFunction( 'is_active_sidebar', 'is_active_sidebar' ),
  62.             new TwigFunction( '_e', 'translate' ),
  63.             new TwigFunction( '_x', '_x' ),
  64.             new TwigFunction( '_ex', '_ex' ),
  65.             new TwigFunction( '_nx', '_nx' ),
  66.             new TwigFunction( '_n_noop', '_n_noop' ),
  67.             new TwigFunction( '_nx_noop', '_nx_noop' ),
  68.             new TwigFunction( '_n', '_n' ),
  69.             new TwigFunction( '__', 'translate' ),
  70.             new TwigFunction( 'translate', 'translate' ),
  71.             new TwigFunction( 'translate_nooped_plural', 'translate_nooped_plural' ),
  72.             new TwigFunction( 'wp_head', function(){ return @$this->getOutput('wp_head'); }, ['is_safe' => array('html')]  ),
  73.             new TwigFunction( 'wp_footer', function(){ return @$this->getOutput('wp_footer'); }, ['is_safe' => array('html')]  ),
  74.             new TwigFunction( 'Post', function($id){ return PostFactory::create($id); } ),
  75.             new TwigFunction( 'User', function($id){ return Factory::create($id, 'user'); } ),
  76.             new TwigFunction( 'Term', function($id){ return TermFactory::create($id); } ),
  77.             new TwigFunction( 'Image', function($id){ return Factory::create($id, 'image'); } )
  78.         ];
  79.     }
  82.     public function to_array( $arr ) {
  84.         return (array)$arr;
  85.     }
  88.     public function generatePixel($w = 1, $h = 1) {
  90.         ob_start();
  92.         $img = imagecreatetruecolor($w, $h);
  93.         imagetruecolortopalette($img, false, 1);
  94.         imagesavealpha($img, true);
  95.         $color = imagecolorallocatealpha($img, 0, 0, 0, 127);
  96.         imagefill($img, 0, 0, $color);
  97.         imagepng($img, null, 9);
  98.         imagedestroy($img);
  100.         $imagedata = ob_get_contents();
  101.         ob_end_clean();
  103.        return 'data:image/png;base64,' . base64_encode($imagedata);
  104.     }
  107.     /**
  108.      * Return resized image
  109.      *
  110.      * @param $image
  111.      * @param $width
  112.      * @param int $height
  113.      * @param array $args
  114.      * @return string
  115.      */
  116.     public function resize($image, $width, $height=0, $args=[])
  117.     {
  118.         if( is_string($image) )
  119.             $image = new Image($image);
  120.         elseif( is_array($image) && isset($image['url']) )
  121.             $image = new Image($image['url']);
  123.         if( !$image instanceof Image )
  124.             $image = new Image();
  126.         $args['resize'] = [$width, $height];
  128.         return $image->edit($args);
  129.     }
  132.     /**
  133.      * Return resized picture
  134.      *
  135.      * @param $image
  136.      * @param $width
  137.      * @param int $height
  138.      * @param array $sources
  139.      * @param bool $alt
  140.      * @param string $loading
  141.      * @return string
  142.      */
  143.     public function picture($image, $width, $height=0, $sources=[], $alt=false, $loading='lazy')
  144.     {
  145.         if( is_string($image) ){
  147.             $image = new Image($image);
  148.         }
  149.         elseif( is_array($image) && isset($image['url']) ){
  151.             if( !$alt && isset($image['alt']) )
  152.                 $alt = $image['alt'];
  154.             $image = new Image($image['url']);
  155.         }
  157.         if( !$image instanceof Image )
  158.             $image = new Image();
  160.         return $image->toHTML($width, $height, $sources, $alt, $loading);
  161.     }
  164.     /**
  165.      * Return function echo
  166.      * @param $function
  167.      * @param array $args
  168.      * @return string
  169.      */
  170.     private function getOutput($function, $args=[])
  171.     {
  172.         ob_start();
  173.         call_user_func_array($function, $args);
  174.         $data = ob_get_contents();
  175.         ob_end_clean();
  177.         return $data;
  178.     }
  181.     /**
  182.      * @param object|int|string $term
  183.      * @param string $taxonomy
  184.      * @param bool|string $meta
  185.      * @return false|string
  186.      */
  187.     public function getTermLink( $term, $taxonomy = 'category', $meta=false )
  188.     {
  190.         if( $meta ){
  192.             $args = array(
  193.                 'meta_query' => array(
  194.                     array(
  195.                         'key'       => $meta,
  196.                         'value'     => $term,
  197.                         'compare'   => 'LIKE'
  198.                     )
  199.                 ),
  200.                 'number'  => 1,
  201.                 'taxonomy'  => $taxonomy,
  202.             );
  204.             $terms = get_terms( $args );
  206.             if( count($terms) )
  207.                 $term = $terms[0];
  208.         }
  211.         $link = get_term_link($term, $taxonomy);
  213.         if( !is_string($link) )
  214.             return false;
  216.         return $link;
  217.     }
  220.     /**
  221.      * @param $content
  222.      * @return array
  223.      */
  224.     public function more( $content )
  225.     {
  226.         if ( preg_match( '/<!--more(.*?)?-->/', $content, $matches ) ) {
  227.             if ( has_block( 'more', $content ) ) {
  228.                 // Remove the core/more block delimiters. They will be left over after $content is split up.
  229.                 $content = preg_replace( '/<!-- \/?wp:more(.*?) -->/', '', $content );
  230.             }
  232.             $content = explode( $matches[0], $content, 2 );
  233.         } else {
  234.             $content = array( $content );
  235.         }
  237.         foreach ($content as &$paragraph)
  238.             $paragraph = force_balance_tags($paragraph);
  240.         return $content;
  241.     }
  244.     /**
  245.      * @param $function_name
  246.      * @return mixed
  247.      */
  248.     public function execFunction( $function_name )
  249.     {
  250.         $args = func_get_args();
  252.         array_shift($args);
  254.         if ( is_string($function_name) )
  255.             $function_name = trim($function_name);
  257.         return call_user_func_array($function_name, ($args));
  258.     }
  261.     /**
  262.      * @param $action_name
  263.      * @param mixed ...$args
  264.      * @return void
  265.      */
  266.     public function doAction( $action_name, ...$args )
  267.     {
  268.         do_action_ref_array( $action_name, $args );
  269.     }
  272.     /**
  273.      * @param $page
  274.      * @param bool $by
  275.      * @return false|string
  276.      */
  277.     public function getPermalink( $page, $by=false )
  278.     {
  279.         switch ( $by ){
  281.             case 'state':
  283.                 $page = get_page_by_state($page);
  284.                 break;
  286.             case 'path':
  288.                 $page = get_page_by_path($page);
  289.                 break;
  291.             case 'title':
  293.                 $page = get_page_by_title($page);
  294.                 break;
  295.         }
  297.         if( $page ){
  299.             $link = get_permalink($page);
  301.             if( !is_string($link) )
  302.                 return false;
  304.             return $link;
  305.         }
  306.         else
  307.             return false;
  308.     }
  311.     /**
  312.      * @param $image
  313.      * @param bool $params
  314.      * @return Image
  315.      */
  316.     public function placeholder($image, $params=false)
  317.     {
  318.         if(!$image instanceof Image)
  319.             return new Image();
  321.         return $image;
  322.     }
  323. }