X-Git-Url: https://git.ao2.it/tweeper.git/blobdiff_plain/110456377908327064a91a48e1f4c2c810b848fa..77e6cd315b5c270104a1116102d0296cf3e6495b:/tweeper.php diff --git a/tweeper.php b/tweeper.php index 11e062b..2d5017e 100644 --- a/tweeper.php +++ b/tweeper.php @@ -1,188 +1,157 @@ - * + +/** + * @file + * Tweeper - a Twitter to RSS web scraper. + * + * Copyright (C) 2013-2018 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 'autoload.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); +/** + * Show the script usage. + */ +function usage($argv) { + if (is_cli()) { + $usage = "{$argv[0]} [-e|-m <0|1>|-u <0|1>|-h|--help] \n"; } - - 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; + else { + $usage = htmlentities("{$_SERVER['SCRIPT_NAME']}?src_url=&generate_enclosure=<0|1>&show_usernames=<0|1>&show_multimedia=<0|1>"); } - public function tweep($uri) { - $html = $this->get_contents($uri); - - $xmlDoc = new DOMDocument(); - $xmlDoc->loadHTML($html); + return "usage: $usage"; +} - $output = $this->xsltProcessor->transformToXML($xmlDoc); +/** + * Parse command line options. + */ +function parse_options_cli($argv, $argc) { + $options = array( + 'generate_enclosure' => FALSE, + 'show_usernames' => TRUE, + 'show_multimedia' => TRUE, + ); + + if ($argc < 2) { + return $options; + } - if (FALSE === $output) { - trigger_error('XSL transformation failed.', E_USER_ERROR); - return NULL; + $cli_options = getopt("em:u:h", array("help")); + foreach ($cli_options as $opt => $val) { + switch ($opt) { + 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 'h': + case 'help': + echo usage($argv); + exit(0); + + default: + fwrite(STDERR, usage($argv)); + exit(1); } - return $output; } -} -function usage($argv) -{ - if (php_sapi_name() != 'cli') - $usage = htmlentities("{$_SERVER['SCRIPT_NAME']}?src_url="); - else - $usage = "{$argv[0]} \n"; + // 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 "usage: $usage"; + return $options; } -function parse_options_cli($argv, $argc) -{ - $options = array(); - - if ($argc < 2) - return $options; - - if ($argv[1] == "-h" || $argv[1] == "--help") - echo usage($argv); - else - $options['src_url'] = $argv[1]; +/** + * Parse options passed from a query string. + */ +function parse_options_query_string() { + $options = array( + 'generate_enclosure' => FALSE, + 'show_usernames' => TRUE, + 'show_multimedia' => TRUE, + ); + + if (isset($_GET['src_url'])) { + $options['src_url'] = $_GET['src_url']; + } - return $options; -} + if (isset($_GET['generate_enclosure'])) { + $options['generate_enclosure'] = $_GET['generate_enclosure'] == 1; + } -function parse_options_query_string() -{ - $options = array(); + if (isset($_GET['show_multimedia'])) { + $options['show_multimedia'] = $_GET['show_multimedia'] != 0; + } - if (isset($_GET['src_url'])) - $options['src_url'] = $_GET['src_url']; + if (isset($_GET['show_usernames'])) { + $options['show_usernames'] = $_GET['show_usernames'] != 0; + } 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'], $options['show_usernames'], $options['show_multimedia']); +$output = $tweeper->tweep($options['src_url']); +if (is_null($output)) { + exit(1); +} +echo $output;