Template_mailer.php

<?php

class Template_mailer {
    
    /**
     * @var  array
     */
    protected $vars = array();
    
    /**
     * Constructor
     */
    public function __construct()
    {        
        if( ! isset(ee()->TMPL))
        {
            ee()->load->library('template', NULL, 'TMPL');
        }
        
        ee()->load->library('email');
        
        ee()->email->initialize(array(
            'mailtype' => 'html',
        ));
    }
    
    /**
     * Magic method for function calls
     *
     * @param   mixed
     * @param   array
     * @return  mixed|void
     */
    public function __call($name, $args)
    {
        $name = strtolower($name);
        
        $valid_methods = array(
            'from',
            'reply_to',
            'to',
            'cc',
            'bcc',
            'subject',
            'set_alt_message',
            'attach',
            'print_debugger'
        );
        
        // if the method name maps to the email class, call it
        if(in_array($name, $valid_methods))
        {
            $result = call_user_func_array(array(ee()->email, $name), $args);
            
            if($result !== NULL)
            {
                return $result;
            }
        }
    }
    
    /**
     * Set variables for tags in the email body
     *
     * @param  string
     * @param  mixed
     */
    public function set($name, $value = NULL)
    {
        if(is_array($name))
        {
            $this->vars = array_merge($this->vars, $name);
        }
        else
        {
            $this->vars[$name] = $value;
        }
    }
    
    /**
     * Sets the template to use for the message copy
     *
     * @param   string
     * @return  string
     */
    protected function template($template)
    {
        // split the template group and name
        $segments = explode('/', $template);
        
        // use index as name if not specified
        if(count($segments) == 1)
        {
            $segments[1] = 'index';
        }
        
        list($template_group, $template_name) = $segments;
        
        // set data items as global variables
        foreach($this->vars as $key => $val)
        {
            ee()->config->_global_vars['email:'.$key] = $val;
        }
        
        // parse the template
        ee()->TMPL->cache_status = 'NO_CACHE';
        $template = ee()->TMPL->fetch_template($template_group, $template_name, false);
        $body = ee()->TMPL->parse_globals($template);
        $body = ee()->TMPL->parse_variables_row($body, $this->vars);
        ee()->TMPL->parse($body, false);
        $body = ee()->TMPL->final_template;
        
        // cleanup unparsed conditionals and annotations
        $body = preg_replace("/".LD."if\s+.*?".RD.".*?".LD.'\/if'.RD."/s", "", $body);
        $body = preg_replace("/\{!--.*?--\}/s", '', $body);
        
        // unset data items
        foreach($this->vars as $key => $val)
        {
            unset(ee()->config->_global_vars['email:'.$key]);
        }
        
        return $body;
    }
    
    /**
     * Sends the email
     *
     * @param   string|array|null
     * @return  bool
     */
    public function send($template)
    {
        // get template output
        $body = $this->template($template);
        
        // set message
        ee()->email->message($body);
        
        // send the email
        $result = ee()->email->send();
        
        ee()->email->clear();
        
        return $result;
    }
    
}