X-Git-Url: https://git.ao2.it/tweeper.git/blobdiff_plain/8e7df9a925ce30ebdd2bf6441c9dd639dd0ffddd..6525c19868a0511abaaac9d2ba452ba640899209:/tweeper.php?ds=sidebyside diff --git a/tweeper.php b/tweeper.php index ef55022..bcc85fe 100644 --- a/tweeper.php +++ b/tweeper.php @@ -1,212 +1,173 @@ - * + +/** + * @file + * Tweeper - a Twitter to RSS web scraper. + * + * Copyright (C) 2013-2020 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'); - -class Tweeper { - - private static $USER_AGENT = "Mozilla/5.0 (Windows NT 6.1; rv:22.0) Gecko/20130405 Firefox/22.0"; +require_once 'autoload.php'; - public function __construct($stylesheet, $generate_enclosure = FALSE) { - $stylesheet_contents = $this->get_contents($stylesheet); - - $xslDoc = new DOMDocument(); - $xslDoc->loadXML($stylesheet_contents); - - $this->xsltProcessor = new XSLTProcessor(); - $this->xsltProcessor->registerPHPFunctions(); - $this->xsltProcessor->setParameter('', 'generateEnclosure', $generate_enclosure); - $this->xsltProcessor->importStylesheet($xslDoc); - } - - public static function epoch_to_gmdate($timestamp) - { - return gmdate('D, d M Y H:i:s', $timestamp) . ' GMT'; - } - - public static function str_to_gmdate($date) - { - $timestamp = strtotime($date); - return Tweeper::epoch_to_gmdate($timestamp); - } - - private static 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: {" + Tweeper::$USER_AGENT + "}\r\n" - )) - ) - ); - - $context = stream_context_create($opts); - $contents = file_get_contents($uri, false, $context); - return $contents; - } +use Tweeper\Tweeper; - private static function get_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 => Tweeper::$USER_AGENT, - )); - curl_exec($ch); - $url_info = curl_getinfo($ch);; - curl_close($ch); - - return $url_info; - } - - public static function generate_enclosure($url) - { - $supported_content_types = array( - "application/ogg", - "audio/aac", - "audio/mp4", - "audio/mpeg", - "audio/vorbis", - "audio/wav", - "audio/webm", - "audio/x-midi", - "image/gif", - "image/jpeg", - "video/avi", - "video/mp4", - "video/mpeg", - "video/ogg", - ); - - $url_info = Tweeper::get_info($url); - - $supported = in_array($url_info['content_type'], $supported_content_types); - if (!$supported) { - error_log("Unsupported enclosure content type \"" . $url_info['content_type'] . "\" for URL: " . $url_info['url']); - return ''; - } - - $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']); +date_default_timezone_set('UTC'); - $dom->appendChild($enc); +/** + * Check if the script is being run from the command line. + */ +function is_cli() { + return (php_sapi_name() === "cli"); +} - return $dom->saveXML($enc); +/** + * Show the script usage. + */ +function usage($argv) { + if (is_cli()) { + $usage = "{$argv[0]} [-e|-m <0|1>|-u <0|1>|-v <0|1>|-h|--help] \n"; } - - public function tweep($uri) { - $html = Tweeper::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>&show_usernames=<0|1>&show_multimedia=<0|1>&verbose_output=<0|1>"); } -} - -function usage($argv) -{ - if (php_sapi_name() != 'cli') - $usage = htmlentities("{$_SERVER['SCRIPT_NAME']}?src_url=&generate_enclosure=<0|1>"); - else - $usage = "{$argv[0]} [-e|-h|--help] \n"; return "usage: $usage"; } -function parse_options_cli($argv, $argc) -{ - $options = array( - 'generate_enclosure' => FALSE - ); - - if ($argc < 2) +/** + * Parse command line options. + */ +function parse_options_cli($argv, $argc) { + $options = [ + 'generate_enclosure' => FALSE, + 'show_usernames' => TRUE, + 'show_multimedia' => TRUE, + 'verbose_output' => TRUE, + ]; + + if ($argc < 2) { return $options; + } - $cli_options = getopt("eh", array("help")); + $cli_options = getopt("em:u:v:h", ["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: - die(usage($argv)); + case 'e': + $options['generate_enclosure'] = TRUE; + break; + + case 'm': + $ret = filter_var($val, FILTER_VALIDATE_BOOLEAN, FILTER_NULL_ON_FAILURE); + if (NULL === $ret) { + fwrite(STDERR, "Invalid argument for the -m option.\n"); + fwrite(STDERR, usage($argv)); + exit(1); + } + $options['show_multimedia'] = $val; + break; + + case 'u': + $ret = filter_var($val, FILTER_VALIDATE_BOOLEAN, FILTER_NULL_ON_FAILURE); + if (NULL === $ret) { + fwrite(STDERR, "Invalid argument for the -u option.\n"); + fwrite(STDERR, usage($argv)); + exit(1); + } + $options['show_usernames'] = $val; + break; + + case 'v': + $ret = filter_var($val, FILTER_VALIDATE_BOOLEAN, FILTER_NULL_ON_FAILURE); + if (NULL === $ret) { + fwrite(STDERR, "Invalid argument for the -v option.\n"); + fwrite(STDERR, usage($argv)); + exit(1); + } + $options['verbose_output'] = $val; + break; + + case 'h': + case 'help': + echo usage($argv); + exit(0); + + default: + fwrite(STDERR, usage($argv)); + exit(1); } } - $options['src_url'] = $argv[count($cli_options) + 1]; + // For now assume that the URL is the lest argument, in the future we could + // switch to PHP >= 7.1 and use the $optind argument of getopt(). + $options['src_url'] = array_pop($argv); return $options; } -function parse_options_query_string() -{ - $options = array( - 'generate_enclosure' => FALSE - ); - - if (isset($_GET['src_url'])) +/** + * Parse options passed from a query string. + */ +function parse_options_query_string() { + $options = [ + 'generate_enclosure' => FALSE, + 'show_usernames' => TRUE, + 'show_multimedia' => TRUE, + 'verbose_output' => TRUE, + ]; + + if (isset($_GET['src_url'])) { $options['src_url'] = $_GET['src_url']; + } - if (isset($_GET['generate_enclosure'])) + if (isset($_GET['generate_enclosure'])) { $options['generate_enclosure'] = $_GET['generate_enclosure'] == 1; + } - return $options; -} + if (isset($_GET['show_multimedia'])) { + $options['show_multimedia'] = $_GET['show_multimedia'] != 0; + } + if (isset($_GET['show_usernames'])) { + $options['show_usernames'] = $_GET['show_usernames'] != 0; + } -if (php_sapi_name() != 'cli') - $options = parse_options_query_string(); -else - $options = parse_options_cli($argv, $argc); + if (isset($_GET['verbose_output'])) { + $options['verbose_output'] = $_GET['verbose_output'] != 0; + } -if (!isset($options['src_url'])) - die(usage($argv)); + return $options; +} -$url = parse_url($options['src_url']); -if (FALSE === $url || empty($url["host"])) - die("Invalid url: ${options['src_url']}\n"); +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'); +} -$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, $options['generate_enclosure']); -echo $tweeper->tweep($options['src_url']); +$tweeper = new Tweeper($options['generate_enclosure'], $options['show_usernames'], $options['show_multimedia'], $options['verbose_output']); +$output = $tweeper->tweep($options['src_url']); +if (is_null($output)) { + exit(1); +} +echo $output;