<?php

/**

 * Smarty plugin

 * @package Smarty

 * @subpackage plugins

 */

/**

 * Smarty {fetch} plugin

 *

 * Type:     function<br>

 * Name:     fetch<br>

 * Purpose:  fetch file, web or ftp data and display results

 * @link http://smarty.php.net/manual/en/language.function.fetch.php {fetch}

 *       (Smarty online manual)

 * @author Monte Ohrt <monte at ohrt dot com>

 * @param array

 * @param Smarty

 * @return string|null if the assign parameter is passed, Smarty assigns the

 *                     result to a template variable

 */

function smarty_function_fetch($params, &$smarty)

{

    if (empty($params['file'])) {

        $smarty->_trigger_fatal_error("[plugin] parameter 'file' cannot be empty");

        return;

    }

    $content = '';

    if ($smarty->security && !preg_match('!^(http|ftp)://!i', $params['file'])) {

        $_params = array('resource_type' => 'file', 'resource_name' => $params['file']);

        require_once(SMARTY_CORE_DIR . 'core.is_secure.php');

        if(!smarty_core_is_secure($_params, $smarty)) {

            $smarty->_trigger_fatal_error('[plugin] (secure mode) fetch \'' . $params['file'] . '\' is not allowed');

            return;

        }

        // fetch the file

        if($fp = @fopen($params['file'],'r')) {

            while(!feof($fp)) {

                $content .= fgets ($fp,4096);

            }

            fclose($fp);

        } else {

            $smarty->_trigger_fatal_error('[plugin] fetch cannot read file \'' . $params['file'] . '\'');

            return;

        }

    } else {

        // not a local file

        if(preg_match('!^http://!i',$params['file'])) {

            // http fetch

            if($uri_parts = parse_url($params['file'])) {

                // set defaults

                $host = $server_name = $uri_parts['host'];

                $timeout = 30;

                $accept = "image/gif, image/x-xbitmap, image/jpeg, image/pjpeg, */*";

                $agent = "Smarty Template Engine ".$smarty->_version;

                $referer = "";

                $uri = !empty($uri_parts['path']) ? $uri_parts['path'] : '/';

                $uri .= !empty($uri_parts['query']) ? '?' . $uri_parts['query'] : '';

                $_is_proxy = false;

                if(empty($uri_parts['port'])) {

                    $port = 80;

                } else {

                    $port = $uri_parts['port'];

                }

                if(!empty($uri_parts['user'])) {

                    $user = $uri_parts['user'];

                }

                if(!empty($uri_parts['pass'])) {

                    $pass = $uri_parts['pass'];

                }

                // loop through parameters, setup headers

                foreach($params as $param_key => $param_value) {

                    switch($param_key) {

                        case "file":

                        case "assign":

                        case "assign_headers":

                            break;

                        case "user":

                            if(!empty($param_value)) {

                                $user = $param_value;

                            }

                            break;

                        case "pass":

                            if(!empty($param_value)) {

                                $pass = $param_value;

                            }

                            break;

                        case "accept":

                            if(!empty($param_value)) {

                                $accept = $param_value;

                            }

                            break;

                        case "header":

                            if(!empty($param_value)) {

                                if(!preg_match('![\w\d-]+: .+!',$param_value)) {

                                    $smarty->_trigger_fatal_error("[plugin] invalid header format '".$param_value."'");

                                    return;

                                } else {

                                    $extra_headers[] = $param_value;

                                }

                            }

                            break;

                        case "proxy_host":

                            if(!empty($param_value)) {

                                $proxy_host = $param_value;

                            }

                            break;

                        case "proxy_port":

                            if(!preg_match('!\D!', $param_value)) {

                                $proxy_port = (int) $param_value;

                            } else {

                                $smarty->_trigger_fatal_error("[plugin] invalid value for attribute '".$param_key."'");

                                return;

                            }

                            break;

                        case "agent":

                            if(!empty($param_value)) {

                                $agent = $param_value;

                            }

                            break;

                        case "referer":

                            if(!empty($param_value)) {

                                $referer = $param_value;

                            }

                            break;

                        case "timeout":

                            if(!preg_match('!\D!', $param_value)) {

                                $timeout = (int) $param_value;

                            } else {

                                $smarty->_trigger_fatal_error("[plugin] invalid value for attribute '".$param_key."'");

                                return;

                            }

                            break;

                        default:

                            $smarty->_trigger_fatal_error("[plugin] unrecognized attribute '".$param_key."'");

                            return;

                    }

                }

                if(!empty($proxy_host) && !empty($proxy_port)) {

                    $_is_proxy = true;

                    $fp = fsockopen($proxy_host,$proxy_port,$errno,$errstr,$timeout);

                } else {

                    $fp = fsockopen($server_name,$port,$errno,$errstr,$timeout);

                }

                if(!$fp) {

                    $smarty->_trigger_fatal_error("[plugin] unable to fetch: $errstr ($errno)");

                    return;

                } else {

                    if($_is_proxy) {

                        fputs($fp, 'GET ' . $params['file'] . " HTTP/1.0\r\n");

                    } else {

                        fputs($fp, "GET $uri HTTP/1.0\r\n");

                    }

                    if(!empty($host)) {

                        fputs($fp, "Host: $host\r\n");

                    }

                    if(!empty($accept)) {

                        fputs($fp, "Accept: $accept\r\n");

                    }

                    if(!empty($agent)) {

                        fputs($fp, "User-Agent: $agent\r\n");

                    }

                    if(!empty($referer)) {

                        fputs($fp, "Referer: $referer\r\n");

                    }

                    if(isset($extra_headers) && is_array($extra_headers)) {

                        foreach($extra_headers as $curr_header) {

                            fputs($fp, $curr_header."\r\n");

                        }

                    }

                    if(!empty($user) && !empty($pass)) {

                        fputs($fp, "Authorization: BASIC ".base64_encode("$user:$pass")."\r\n");

                    }

                    fputs($fp, "\r\n");

                    while(!feof($fp)) {

                        $content .= fgets($fp,4096);

                    }

                    fclose($fp);

                    $csplit = split("\r\n\r\n",$content,2);

                    $content = $csplit[1];

                    if(!empty($params['assign_headers'])) {

                        $smarty->assign($params['assign_headers'],split("\r\n",$csplit[0]));

                    }

                }

            } else {

                $smarty->_trigger_fatal_error("[plugin] unable to parse URL, check syntax");

                return;

            }

        } else {

            // ftp fetch

            if($fp = @fopen($params['file'],'r')) {

                while(!feof($fp)) {

                    $content .= fgets ($fp,4096);

                }

                fclose($fp);

            } else {

                $smarty->_trigger_fatal_error('[plugin] fetch cannot read file \'' . $params['file'] .'\'');

                return;

            }

        }

    }

    if (!empty($params['assign'])) {

        $smarty->assign($params['assign'],$content);

    } else {

        return $content;

    }

}

/* vim: set expandtab: */

?>