X-Git-Url: https://git.ao2.it/tweeper.git/blobdiff_plain/d38cac35c8e0aed0a223533bb54e46958c693212..b187bb677361d17a468abf749332d081a194b4bd:/tweeper.php?ds=inline diff --git a/tweeper.php b/tweeper.php index 45bc786..ba8b1d7 100644 --- a/tweeper.php +++ b/tweeper.php @@ -1,163 +1,78 @@ - * +/** + * @file + * Tweeper - a Twitter to RSS web scraper. + * + * Copyright (C) 2013-2015 Antonio Ospite + * * This program 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. - * + * * This program 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 this program. If not, see . */ -date_default_timezone_set('UTC'); - -define('USER_AGENT', 'Mozilla/5.0 (Windows NT 6.1; rv:22.0) Gecko/20130405 Firefox/22.0'); +require_once 'src/Tweeper.php'; -function str_to_gmdate($date) -{ - $timestamp = strtotime($date); - return epoch_to_gmdate($timestamp); -} - -function epoch_to_gmdate($timestamp) -{ - return gmdate('D, d M Y H:i:s', $timestamp) . ' GMT'; -} +use Tweeper\Tweeper; -function get_url_info($url) -{ - $ch = curl_init($url); - curl_setopt_array($ch, array( - CURLOPT_HEADER => TRUE, - CURLOPT_NOBODY => TRUE, - CURLOPT_FOLLOWLOCATION => TRUE, // follow http redirects to get the real URL - CURLOPT_RETURNTRANSFER => TRUE, - CURLOPT_SSL_VERIFYHOST => FALSE, - CURLOPT_SSL_VERIFYPEER => FALSE, - CURLOPT_USERAGENT => USER_AGENT, - )); - curl_exec($ch); - $url_info = curl_getinfo($ch);; - curl_close($ch); - - return $url_info; -} +date_default_timezone_set('UTC'); -function generate_enclosure($url) -{ - $url_info = get_url_info($url); - - switch($url_info['content_type']) { - case "video/avi": - case "image/gif": - case "image/jpeg": - case "audio/x-midi": - case "audio/mpeg": - case "video/mpeg": - case "audio/vorbis": - case "application/ogg": - case "audio/webm": - case "audio/aac": - case "audio/mp4": - case "audio/wav": - case "video/mp4": - case "video/ogg": - $dom = new DomDocument(); - $enc = $dom->createElement('enclosure'); - $enc->setAttribute( 'url', $url_info['url']); - $enc->setAttribute( 'length', $url_info['download_content_length']); - $enc->setAttribute( 'type', $url_info['content_type']); - - $dom->appendChild($enc); - - return $dom->saveXML($enc); - default: - return; - } +/** + * Check if the script is being run from the command line. + */ +function is_cli() { + return (php_sapi_name() === "cli"); } - -class Tweeper { - - public function __construct($stylesheet) { - $stylesheet_contents = $this->get_contents($stylesheet); - - $xslDoc = new DOMDocument(); - $xslDoc->loadXML($stylesheet_contents); - - $this->xsltProcessor = new XSLTProcessor(); - $this->xsltProcessor->registerPHPFunctions(); - $this->xsltProcessor->importStylesheet($xslDoc); - } - - private function get_contents($uri) { - # https://www.wjsams.com/c/docs/Wiki/Php-HowToSetUserAgentOnFileGetContents - $opts = array( - 'http' => array( - 'method' => "GET", - 'header' => join(array( - "Accept-language: en\r\n", - "User-Agent: {" + USER_AGENT + "}\r\n" - )) - ) - ); - - $context = stream_context_create($opts); - $contents = file_get_contents($uri, false, $context); - return $contents; +/** + * Show the script usage. + */ +function usage($argv) { + if (is_cli()) { + $usage = "{$argv[0]} [-e|-h|--help] \n"; } - - public function tweep($uri) { - $html = $this->get_contents($uri); - - $xmlDoc = new DOMDocument(); - $xmlDoc->loadHTML($html); - - $output = $this->xsltProcessor->transformToXML($xmlDoc); - - if (FALSE === $output) { - trigger_error('XSL transformation failed.', E_USER_ERROR); - return NULL; - } - return $output; + else { + $usage = htmlentities("{$_SERVER['SCRIPT_NAME']}?src_url=&generate_enclosure=<0|1>"); } -} - -function usage($argv) -{ - if (php_sapi_name() != 'cli') - $usage = htmlentities("{$_SERVER['SCRIPT_NAME']}?src_url="); - else - $usage = "{$argv[0]} [-h|--help] \n"; return "usage: $usage"; } -function parse_options_cli($argv, $argc) -{ - $options = array(); +/** + * Parse command line options. + */ +function parse_options_cli($argv, $argc) { + $options = array( + 'generate_enclosure' => FALSE, + ); - if ($argc < 2) + if ($argc < 2) { return $options; + } - $cli_options = getopt("h", array("help")); + $cli_options = getopt("eh", array("help")); foreach ($cli_options as $opt => $val) { switch ($opt) { - case 'h': - case 'help': - echo usage($argv); - exit(0); - default: - die(usage($argv)); + case 'e': + $options['generate_enclosure'] = TRUE; + break; + + case 'h': + case 'help': + echo usage($argv); + exit(0); + + default: + fwrite(STDERR, usage($argv)); + exit(1); } } @@ -166,32 +81,42 @@ function parse_options_cli($argv, $argc) return $options; } -function parse_options_query_string() -{ - $options = array(); +/** + * Parse options passed from a query string. + */ +function parse_options_query_string() { + $options = array( + 'generate_enclosure' => FALSE, + ); - if (isset($_GET['src_url'])) + if (isset($_GET['src_url'])) { $options['src_url'] = $_GET['src_url']; + } + + if (isset($_GET['generate_enclosure'])) { + $options['generate_enclosure'] = $_GET['generate_enclosure'] == 1; + } return $options; } - -if (php_sapi_name() != 'cli') - $options = parse_options_query_string(); -else +if (is_cli()) { $options = parse_options_cli($argv, $argc); + $error_stream = fopen('php://stderr', 'w'); +} +else { + $options = parse_options_query_string(); + $error_stream = fopen('php://output', 'w'); +} -if (!isset($options['src_url'])) - die(usage($argv)); - -$url = parse_url($options['src_url']); -if (FALSE === $url || empty($url["host"])) - die("Invalid url: ${options['src_url']}\n"); - -$stylesheet = __DIR__ . "/rss_converter_" . $url["host"] . ".xsl"; -if (FALSE === file_exists($stylesheet)) - die("Conversion to RSS not supported: {$url["host"]}\n"); +if (!isset($options['src_url'])) { + fwrite($error_stream, usage(is_cli() ? $argv : NULL)); + exit(1); +} -$tweeper = new Tweeper($stylesheet); -echo $tweeper->tweep($options['src_url']); +$tweeper = new Tweeper($options['generate_enclosure']); +$output = $tweeper->tweep($options['src_url']); +if (is_null($output)) { + exit(1); +} +echo $output;