spreed/sample-commands/wikipedia.php

89 строки
2.5 KiB
PHP

<?php
declare(strict_types=1);
/**
* @copyright Copyright (c) 2019 Joas Schilling <coding@schilljs.com>
*
* @license GNU AGPL version 3 or any later version
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero 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 Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*
*/
if (PHP_SAPI !== 'cli') {
// Only allow access via the console
exit;
}
if ($argc < 2) {
echo 'Missing search term in call to wikipedia.php';
return 1;
}
$searchTerm = $argv[1];
if ($searchTerm === '--help') {
echo '/wiki - A simple command to find wikipedia articles for a term' . "\n";
echo "\n";
echo 'Example: /wiki Nextcloud' . "\n";
$searchTerm = 'Nextcloud';
}
$endpoint = 'https://en.wikipedia.org/w/api.php';
$parameters = [
'action' => 'opensearch',
'format' => 'json',
'formatversion' => 2,
'search' => $searchTerm,
];
$content = file_get_contents($endpoint . '?' . http_build_query($parameters));
$results = json_decode($content, true);
[, $titles, $descriptions, $links] = $results;
$numArticles = count($titles);
if ($numArticles === 0) {
echo 'Wikipedia did not have any results for "' . $searchTerm . '" :(' . "\n";
return;
}
if ($numArticles !== count($descriptions) || $numArticles !== count($links)) {
echo 'Result returned from wikipedia is maleformed';
return 1;
}
$response = 'Wikipedia search results for "' . $searchTerm . '":' . "\n";
$maxArticles = $numArticles > 7 ? 5 : $numArticles;
$length = (int) ((800 - strlen($response)) / $maxArticles);
foreach ($titles as $key => $title) {
if ($key >= $maxArticles) {
break;
}
$link = " - {$links[$key]}\n";
$remainingLength = max(strlen($title),$length - strlen($link));
if ($remainingLength < strlen("* $title - {$descriptions[$key]}")) {
$response .= substr("* $title - {$descriptions[$key]}", 0, $remainingLength) . '…' . $link;
} else {
$response .= "* $title - {$descriptions[$key]}" . $link;
}
}
if ($maxArticles < $numArticles) {
$response .= '* and ' . ($numArticles - $maxArticles) ." more articles found\n";
}
echo ($response);