X-Git-Url: https://git.ao2.it/tweeper.git/blobdiff_plain/cabaa4bbe13d8c0df16ac224c96f07525ec9377c..cdcd3418129fdf2513a2126bff1c2b38973eb9f5:/tweeper.php diff --git a/tweeper.php b/tweeper.php index c0db320..ba8b1d7 100644 --- a/tweeper.php +++ b/tweeper.php @@ -1,166 +1,122 @@ - * +/** + * @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); -} +use Tweeper\Tweeper; -function epoch_to_gmdate($timestamp) -{ - return gmdate('D, d M Y H:i:s', $timestamp) . ' GMT'; -} +date_default_timezone_set('UTC'); -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; +/** + * Check if the script is being run from the command line. + */ +function is_cli() { + return (php_sapi_name() === "cli"); } -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; +/** + * Show the script usage. + */ +function usage($argv) { + if (is_cli()) { + $usage = "{$argv[0]} [-e|-h|--help] \n"; + } + else { + $usage = htmlentities("{$_SERVER['SCRIPT_NAME']}?src_url=&generate_enclosure=<0|1>"); } -} - - -class Tweeper { - public function __construct($stylesheet) { - $stylesheet_contents = $this->get_contents($stylesheet); + return "usage: $usage"; +} - $xslDoc = new DOMDocument(); - $xslDoc->loadXML($stylesheet_contents); +/** + * Parse command line options. + */ +function parse_options_cli($argv, $argc) { + $options = array( + 'generate_enclosure' => FALSE, + ); - $this->xsltProcessor = new XSLTProcessor(); - $this->xsltProcessor->registerPHPFunctions(); - $this->xsltProcessor->importStylesheet($xslDoc); + if ($argc < 2) { + return $options; } - 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; + $cli_options = getopt("eh", array("help")); + foreach ($cli_options as $opt => $val) { + switch ($opt) { + case 'e': + $options['generate_enclosure'] = TRUE; + break; + + case 'h': + case 'help': + echo usage($argv); + exit(0); + + default: + fwrite(STDERR, usage($argv)); + exit(1); + } } - public function tweep($uri) { - $html = $this->get_contents($uri); + $options['src_url'] = $argv[count($cli_options) + 1]; - $xmlDoc = new DOMDocument(); - $xmlDoc->loadHTML($html); + return $options; +} - $output = $this->xsltProcessor->transformToXML($xmlDoc); +/** + * Parse options passed from a query string. + */ +function parse_options_query_string() { + $options = array( + 'generate_enclosure' => FALSE, + ); - if (FALSE === $output) { - trigger_error('XSL transformation failed.', E_USER_ERROR); - return NULL; - } - return $output; + if (isset($_GET['src_url'])) { + $options['src_url'] = $_GET['src_url']; } -} -function usage($argv) -{ - if (php_sapi_name() != 'cli') - $usage = htmlentities("{$_SERVER['SCRIPT_NAME']}?src_url="); - else - $usage = "{$argv[0]} \n"; + if (isset($_GET['generate_enclosure'])) { + $options['generate_enclosure'] = $_GET['generate_enclosure'] == 1; + } - return "usage: $usage"; + return $options; } -if (isset($_GET['src_url'])) { - $src_url = $_GET['src_url']; -} else if (isset($argv[1])) { - if ($argv[1] == "-h" || $argv[1] == "--help") { - echo usage($argv); - return 0; - } - $src_url = $argv[1]; -} else { - die(usage($argv)); +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'); } -$url = parse_url($src_url); -if (FALSE === $url || empty($url["host"])) - die("Invalid url: $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($src_url); +$tweeper = new Tweeper($options['generate_enclosure']); +$output = $tweeper->tweep($options['src_url']); +if (is_null($output)) { + exit(1); +} +echo $output;