<?php
/**
 *      This file is part of SteamProfile.
 *
 *      Written by Nico Bergemann <barracuda415@yahoo.de>
 *      Copyright 2009 Nico Bergemann
 *
 *      SteamProfile is free software: you can redistribute it and/or modify
 *      it under the terms of the GNU General Public License as published by
 *      the Free Software Foundation, either version 3 of the License, or
 *      (at your option) any later version.
 *
 *      SteamProfile is distributed in the hope that it will be useful,
 *      but WITHOUT ANY WARRANTY; without even the implied warranty of
 *      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *      GNU General Public License for more details.
 *
 *      You should have received a copy of the GNU General Public License
 *      along with SteamProfile.  If not, see <http://www.gnu.org/licenses/>.
 */

/**
 * Class CURL
 *
 * cURL object wrapper for basic downloading functions
 */
class CURL {
        private $cURLSession;
        private $rOutputFile;

        public function __construct($sURL) {
                // make sure the cURL extension is loaded
                if(!extension_loaded('curl') || !function_exists('curl_init')) {
                        throw new RuntimeException('cURL extension required');
                }

                $this->cURLSession = curl_init($sURL);
        }

        protected function setOption($iOpt, $value) {
                curl_setopt($this->cURLSession, $iOpt , $value);
        }

        protected function getInfo($iOpt) {
                return curl_getinfo($this->cURLSession, $iOpt);
        }

        public function setOutputFile($file) {
                if(is_resource($file)) {
                $this->rOutputFile = $file;
                        $this->setOption(CURLOPT_FILE, $file);
                } else {
                        $this->rOutputFile = fopen($file, 'w+b');
                        $this->setOption(CURLOPT_FILE, $this->rOutputFile);
                }
        }

        public function setReturnTransfer($bReturn) {
                $this->setOption(CURLOPT_RETURNTRANSFER, $bReturn);
        }

        public function setUserAgent($sUA) {
                $this->setOption(CURLOPT_USERAGENT, $sUA);
        }

        public function setTimeout($iTimeout) {
                $this->setOption(CURLOPT_TIMEOUT, $iTimeout);
        }

        public function setConnectTimeout($iTimeout) {
                $this->setOption(CURLOPT_CONNECTTIMEOUT, $iTimeout);
        }

        public function getHTTPCode() {
                return $this->getInfo(CURLINFO_HTTP_CODE);
        }

        public function getErrorMessage() {
                return curl_error($this->cURLSession);
        }

        public function start() {
                $redir=0;
                return curl_redirect_exec($this->cURLSession,$redir);
        }
        public function close() {
                curl_close($this->cURLSession);

                if(is_resource($this->rOutputFile)) {
                        rewind($this->rOutputFile);
                        fclose($this->rOutputFile);
                }
        }
}
function curl_redirect_exec($ch, &$redirects, $curlopt_header = false) {
    curl_setopt($ch, CURLOPT_HEADER,false);
    curl_setopt($ch, CURLOPT_RETURNTRANSFER, true);
    $data = curl_exec($ch);
    $http_code = curl_getinfo($ch, CURLINFO_HTTP_CODE);
    if ($http_code == 301 || $http_code == 302) {
        list($header) = explode("\r\n\r\n", $data, 2);
        $matches = array();
        preg_match("/(Location:|URI:)[^(\n)]*/", $header, $matches);
        $url = trim(str_replace($matches[1],"",$matches[0]));
        $url_parsed = parse_url($url);
        if (isset($url_parsed)) {
            curl_setopt($ch, CURLOPT_URL, $url);
            $redirects++;
            return curl_redirect_exec($ch, $redirects);
        }
    }
return $data;
}