mirror of
https://github.com/pixelfed/pixelfed.git
synced 2024-11-18 12:31:27 +00:00
573 lines
18 KiB
PHP
Executable file
573 lines
18 KiB
PHP
Executable file
<?php
|
||
|
||
/**
|
||
* @author Mike Cochrane <mikec@mikenz.geek.nz>
|
||
* @author Nick Pope <nick@nickpope.me.uk>
|
||
* @copyright Copyright © 2010, Mike Cochrane, Nick Pope
|
||
* @license http://www.apache.org/licenses/LICENSE-2.0 Apache License v2.0
|
||
*/
|
||
|
||
namespace App\Util\Lexer;
|
||
|
||
/**
|
||
* Twitter Extractor Class.
|
||
*
|
||
* Parses tweets and extracts URLs, usernames, username/list pairs and
|
||
* hashtags.
|
||
*
|
||
* Originally written by {@link http://github.com/mikenz Mike Cochrane}, this
|
||
* is based on code by {@link http://github.com/mzsanford Matt Sanford} and
|
||
* heavily modified by {@link http://github.com/ngnpope Nick Pope}.
|
||
*
|
||
* @author Mike Cochrane <mikec@mikenz.geek.nz>
|
||
* @author Nick Pope <nick@nickpope.me.uk>
|
||
* @copyright Copyright © 2010, Mike Cochrane, Nick Pope
|
||
* @license http://www.apache.org/licenses/LICENSE-2.0 Apache License v2.0
|
||
*/
|
||
class Extractor extends Regex
|
||
{
|
||
/**
|
||
* @var bool
|
||
*/
|
||
protected $extractURLWithoutProtocol = true;
|
||
|
||
/**
|
||
* Provides fluent method chaining.
|
||
*
|
||
* @param string $tweet The tweet to be converted.
|
||
*
|
||
* @see __construct()
|
||
*
|
||
* @return Extractor
|
||
*/
|
||
public static function create($tweet = null)
|
||
{
|
||
return new self($tweet);
|
||
}
|
||
|
||
/**
|
||
* Reads in a tweet to be parsed and extracts elements from it.
|
||
*
|
||
* Extracts various parts of a tweet including URLs, usernames, hashtags...
|
||
*
|
||
* @param string $tweet The tweet to extract.
|
||
*/
|
||
public function __construct($tweet = null)
|
||
{
|
||
parent::__construct($tweet);
|
||
}
|
||
|
||
/**
|
||
* Extracts all parts of a tweet and returns an associative array containing
|
||
* the extracted elements.
|
||
*
|
||
* @param string $tweet The tweet to extract.
|
||
*
|
||
* @return array The elements in the tweet.
|
||
*/
|
||
public function extract($tweet = null)
|
||
{
|
||
if (is_null($tweet)) {
|
||
$tweet = $this->tweet;
|
||
}
|
||
|
||
return [
|
||
'hashtags' => $this->extractHashtags($tweet),
|
||
'urls' => $this->extractURLs($tweet),
|
||
'mentions' => $this->extractMentionedUsernames($tweet),
|
||
'replyto' => $this->extractRepliedUsernames($tweet),
|
||
'hashtags_with_indices' => $this->extractHashtagsWithIndices($tweet),
|
||
'urls_with_indices' => $this->extractURLsWithIndices($tweet),
|
||
'mentions_with_indices' => $this->extractMentionedUsernamesWithIndices($tweet),
|
||
];
|
||
}
|
||
|
||
/**
|
||
* Extract URLs, @mentions, lists and #hashtag from a given text/tweet.
|
||
*
|
||
* @param string $tweet The tweet to extract.
|
||
*
|
||
* @return array list of extracted entities
|
||
*/
|
||
public function extractEntitiesWithIndices($tweet = null)
|
||
{
|
||
if (is_null($tweet)) {
|
||
$tweet = $this->tweet;
|
||
}
|
||
$entities = [];
|
||
$entities = array_merge($entities, $this->extractURLsWithIndices($tweet));
|
||
$entities = array_merge($entities, $this->extractHashtagsWithIndices($tweet, false));
|
||
$entities = array_merge($entities, $this->extractMentionsOrListsWithIndices($tweet));
|
||
$entities = array_merge($entities, $this->extractCashtagsWithIndices($tweet));
|
||
$entities = $this->removeOverlappingEntities($entities);
|
||
|
||
return $entities;
|
||
}
|
||
|
||
/**
|
||
* Extracts all the hashtags from the tweet.
|
||
*
|
||
* @param string $tweet The tweet to extract.
|
||
*
|
||
* @return array The hashtag elements in the tweet.
|
||
*/
|
||
public function extractHashtags($tweet = null)
|
||
{
|
||
$hashtagsOnly = [];
|
||
$hashtagsWithIndices = $this->extractHashtagsWithIndices($tweet);
|
||
|
||
foreach ($hashtagsWithIndices as $hashtagWithIndex) {
|
||
$hashtagsOnly[] = $hashtagWithIndex['hashtag'];
|
||
}
|
||
|
||
return $hashtagsOnly;
|
||
}
|
||
|
||
/**
|
||
* Extracts all the cashtags from the tweet.
|
||
*
|
||
* @param string $tweet The tweet to extract.
|
||
*
|
||
* @return array The cashtag elements in the tweet.
|
||
*/
|
||
public function extractCashtags($tweet = null)
|
||
{
|
||
$cashtagsOnly = [];
|
||
$cashtagsWithIndices = $this->extractCashtagsWithIndices($tweet);
|
||
|
||
foreach ($cashtagsWithIndices as $cashtagWithIndex) {
|
||
$cashtagsOnly[] = $cashtagWithIndex['cashtag'];
|
||
}
|
||
|
||
return $cashtagsOnly;
|
||
}
|
||
|
||
/**
|
||
* Extracts all the URLs from the tweet.
|
||
*
|
||
* @param string $tweet The tweet to extract.
|
||
*
|
||
* @return array The URL elements in the tweet.
|
||
*/
|
||
public function extractURLs($tweet = null)
|
||
{
|
||
$urlsOnly = [];
|
||
$urlsWithIndices = $this->extractURLsWithIndices($tweet);
|
||
|
||
foreach ($urlsWithIndices as $urlWithIndex) {
|
||
$urlsOnly[] = $urlWithIndex['url'];
|
||
}
|
||
|
||
return $urlsOnly;
|
||
}
|
||
|
||
/**
|
||
* Extract all the usernames from the tweet.
|
||
*
|
||
* A mention is an occurrence of a username anywhere in a tweet.
|
||
*
|
||
* @param string $tweet The tweet to extract.
|
||
*
|
||
* @return array The usernames elements in the tweet.
|
||
*/
|
||
public function extractMentionedScreennames($tweet = null)
|
||
{
|
||
$usernamesOnly = [];
|
||
$mentionsWithIndices = $this->extractMentionsOrListsWithIndices($tweet);
|
||
|
||
foreach ($mentionsWithIndices as $mentionWithIndex) {
|
||
$screen_name = mb_strtolower($mentionWithIndex['screen_name']);
|
||
if (empty($screen_name) or in_array($screen_name, $usernamesOnly)) {
|
||
continue;
|
||
}
|
||
$usernamesOnly[] = $screen_name;
|
||
}
|
||
|
||
return $usernamesOnly;
|
||
}
|
||
|
||
/**
|
||
* Extract all the usernames from the tweet.
|
||
*
|
||
* A mention is an occurrence of a username anywhere in a tweet.
|
||
*
|
||
* @return array The usernames elements in the tweet.
|
||
*
|
||
* @deprecated since version 1.1.0
|
||
*/
|
||
public function extractMentionedUsernames($tweet)
|
||
{
|
||
$this->tweet = $tweet;
|
||
|
||
return $this->extractMentionedScreennames($tweet);
|
||
}
|
||
|
||
/**
|
||
* Extract all the usernames replied to from the tweet.
|
||
*
|
||
* A reply is an occurrence of a username at the beginning of a tweet.
|
||
*
|
||
* @param string $tweet The tweet to extract.
|
||
*
|
||
* @return array The usernames replied to in a tweet.
|
||
*/
|
||
public function extractReplyScreenname($tweet = null)
|
||
{
|
||
if (is_null($tweet)) {
|
||
$tweet = $this->tweet;
|
||
}
|
||
$matched = preg_match(self::$patterns['valid_reply'], $tweet, $matches);
|
||
// Check username ending in
|
||
if ($matched && preg_match(self::$patterns['end_mention_match'], $matches[2])) {
|
||
$matched = false;
|
||
}
|
||
|
||
return $matched ? $matches[1] : null;
|
||
}
|
||
|
||
/**
|
||
* Extract all the usernames replied to from the tweet.
|
||
*
|
||
* A reply is an occurrence of a username at the beginning of a tweet.
|
||
*
|
||
* @return array The usernames replied to in a tweet.
|
||
*
|
||
* @deprecated since version 1.1.0
|
||
*/
|
||
public function extractRepliedUsernames()
|
||
{
|
||
return $this->extractReplyScreenname();
|
||
}
|
||
|
||
/**
|
||
* Extracts all the hashtags and the indices they occur at from the tweet.
|
||
*
|
||
* @param string $tweet The tweet to extract.
|
||
* @param bool $checkUrlOverlap if true, check if extracted hashtags overlap URLs and remove overlapping ones
|
||
*
|
||
* @return array The hashtag elements in the tweet.
|
||
*/
|
||
public function extractHashtagsWithIndices($tweet = null, $checkUrlOverlap = true)
|
||
{
|
||
if (is_null($tweet)) {
|
||
$tweet = $this->tweet;
|
||
}
|
||
|
||
if (!preg_match('/[#＃]/iu', $tweet)) {
|
||
return [];
|
||
}
|
||
|
||
preg_match_all(self::$patterns['valid_hashtag'], $tweet, $matches, PREG_SET_ORDER | PREG_OFFSET_CAPTURE);
|
||
$tags = [];
|
||
|
||
foreach ($matches as $match) {
|
||
list($all, $before, $hash, $hashtag, $outer) = array_pad($match, 3, ['', 0]);
|
||
$start_position = $hash[1] > 0 ? StringUtils::strlen(substr($tweet, 0, $hash[1])) : $hash[1];
|
||
$end_position = $start_position + StringUtils::strlen($hash[0].$hashtag[0]);
|
||
|
||
if (preg_match(self::$patterns['end_hashtag_match'], $outer[0])) {
|
||
continue;
|
||
}
|
||
|
||
$tags[] = [
|
||
'hashtag' => $hashtag[0],
|
||
'indices' => [$start_position, $end_position],
|
||
];
|
||
}
|
||
|
||
if (!$checkUrlOverlap) {
|
||
return $tags;
|
||
}
|
||
|
||
// check url overlap
|
||
$urls = $this->extractURLsWithIndices($tweet);
|
||
$entities = $this->removeOverlappingEntities(array_merge($tags, $urls));
|
||
|
||
$validTags = [];
|
||
foreach ($entities as $entity) {
|
||
if (empty($entity['hashtag'])) {
|
||
continue;
|
||
}
|
||
$validTags[] = $entity;
|
||
}
|
||
|
||
return $validTags;
|
||
}
|
||
|
||
/**
|
||
* Extracts all the cashtags and the indices they occur at from the tweet.
|
||
*
|
||
* @param string $tweet The tweet to extract.
|
||
*
|
||
* @return array The cashtag elements in the tweet.
|
||
*/
|
||
public function extractCashtagsWithIndices($tweet = null)
|
||
{
|
||
if (is_null($tweet)) {
|
||
$tweet = $this->tweet;
|
||
}
|
||
|
||
if (!preg_match('/\$/iu', $tweet)) {
|
||
return [];
|
||
}
|
||
|
||
preg_match_all(self::$patterns['valid_cashtag'], $tweet, $matches, PREG_SET_ORDER | PREG_OFFSET_CAPTURE);
|
||
$tags = [];
|
||
|
||
foreach ($matches as $match) {
|
||
list($all, $before, $dollar, $cash_text, $outer) = array_pad($match, 3, ['', 0]);
|
||
$start_position = $dollar[1] > 0 ? StringUtils::strlen(substr($tweet, 0, $dollar[1])) : $dollar[1];
|
||
$end_position = $start_position + StringUtils::strlen($dollar[0].$cash_text[0]);
|
||
|
||
if (preg_match(self::$patterns['end_hashtag_match'], $outer[0])) {
|
||
continue;
|
||
}
|
||
|
||
$tags[] = [
|
||
'cashtag' => $cash_text[0],
|
||
'indices' => [$start_position, $end_position],
|
||
];
|
||
}
|
||
|
||
return $tags;
|
||
}
|
||
|
||
/**
|
||
* Extracts all the URLs and the indices they occur at from the tweet.
|
||
*
|
||
* @param string $tweet The tweet to extract.
|
||
*
|
||
* @return array The URLs elements in the tweet.
|
||
*/
|
||
public function extractURLsWithIndices($tweet = null)
|
||
{
|
||
if (is_null($tweet)) {
|
||
$tweet = $this->tweet;
|
||
}
|
||
|
||
$needle = $this->extractURLWithoutProtocol() ? '.' : ':';
|
||
if (strpos($tweet, $needle) === false) {
|
||
return [];
|
||
}
|
||
|
||
$urls = [];
|
||
preg_match_all(self::$patterns['valid_url'], $tweet, $matches, PREG_SET_ORDER | PREG_OFFSET_CAPTURE);
|
||
|
||
foreach ($matches as $match) {
|
||
list($all, $before, $url, $protocol, $domain, $port, $path, $query) = array_pad($match, 8, ['']);
|
||
$start_position = $url[1] > 0 ? StringUtils::strlen(substr($tweet, 0, $url[1])) : $url[1];
|
||
$end_position = $start_position + StringUtils::strlen($url[0]);
|
||
|
||
$all = $all[0];
|
||
$before = $before[0];
|
||
$url = $url[0];
|
||
$protocol = $protocol[0];
|
||
$domain = $domain[0];
|
||
$port = $port[0];
|
||
$path = $path[0];
|
||
$query = $query[0];
|
||
|
||
// If protocol is missing and domain contains non-ASCII characters,
|
||
// extract ASCII-only domains.
|
||
if (empty($protocol)) {
|
||
if (!$this->extractURLWithoutProtocol || preg_match(self::$patterns['invalid_url_without_protocol_preceding_chars'], $before)) {
|
||
continue;
|
||
}
|
||
|
||
$last_url = null;
|
||
$ascii_end_position = 0;
|
||
|
||
if (preg_match(self::$patterns['valid_ascii_domain'], $domain, $asciiDomain)) {
|
||
$asciiDomain[0] = preg_replace('/'.preg_quote($domain, '/').'/u', $asciiDomain[0], $url);
|
||
$ascii_start_position = StringUtils::strpos($domain, $asciiDomain[0], $ascii_end_position);
|
||
$ascii_end_position = $ascii_start_position + StringUtils::strlen($asciiDomain[0]);
|
||
$last_url = [
|
||
'url' => $asciiDomain[0],
|
||
'indices' => [$start_position + $ascii_start_position, $start_position + $ascii_end_position],
|
||
];
|
||
if (!empty($path)
|
||
|| preg_match(self::$patterns['valid_special_short_domain'], $asciiDomain[0])
|
||
|| !preg_match(self::$patterns['invalid_short_domain'], $asciiDomain[0])) {
|
||
$urls[] = $last_url;
|
||
}
|
||
}
|
||
|
||
// no ASCII-only domain found. Skip the entire URL
|
||
if (empty($last_url)) {
|
||
continue;
|
||
}
|
||
|
||
// $last_url only contains domain. Need to add path and query if they exist.
|
||
if (!empty($path)) {
|
||
// last_url was not added. Add it to urls here.
|
||
$last_url['url'] = preg_replace('/'.preg_quote($domain, '/').'/u', $last_url['url'], $url);
|
||
$last_url['indices'][1] = $end_position;
|
||
}
|
||
} else {
|
||
// In the case of t.co URLs, don't allow additional path characters
|
||
if (preg_match(self::$patterns['valid_tco_url'], $url, $tcoUrlMatches)) {
|
||
$url = $tcoUrlMatches[0];
|
||
$end_position = $start_position + StringUtils::strlen($url);
|
||
}
|
||
$urls[] = [
|
||
'url' => $url,
|
||
'indices' => [$start_position, $end_position],
|
||
];
|
||
}
|
||
}
|
||
|
||
return $urls;
|
||
}
|
||
|
||
/**
|
||
* Extracts all the usernames and the indices they occur at from the tweet.
|
||
*
|
||
* @param string $tweet The tweet to extract.
|
||
*
|
||
* @return array The username elements in the tweet.
|
||
*/
|
||
public function extractMentionedScreennamesWithIndices($tweet = null)
|
||
{
|
||
if (is_null($tweet)) {
|
||
$tweet = $this->tweet;
|
||
}
|
||
|
||
$usernamesOnly = [];
|
||
$mentions = $this->extractMentionsOrListsWithIndices($tweet);
|
||
foreach ($mentions as $mention) {
|
||
if (isset($mention['list_slug'])) {
|
||
unset($mention['list_slug']);
|
||
}
|
||
$usernamesOnly[] = $mention;
|
||
}
|
||
|
||
return $usernamesOnly;
|
||
}
|
||
|
||
/**
|
||
* Extracts all the usernames and the indices they occur at from the tweet.
|
||
*
|
||
* @return array The username elements in the tweet.
|
||
*
|
||
* @deprecated since version 1.1.0
|
||
*/
|
||
public function extractMentionedUsernamesWithIndices()
|
||
{
|
||
return $this->extractMentionedScreennamesWithIndices();
|
||
}
|
||
|
||
/**
|
||
* Extracts all the usernames and the indices they occur at from the tweet.
|
||
*
|
||
* @param string $tweet The tweet to extract.
|
||
*
|
||
* @return array The username elements in the tweet.
|
||
*/
|
||
public function extractMentionsOrListsWithIndices($tweet = null)
|
||
{
|
||
if (is_null($tweet)) {
|
||
$tweet = $this->tweet;
|
||
}
|
||
|
||
if (!preg_match('/[@＠]/iu', $tweet)) {
|
||
return [];
|
||
}
|
||
|
||
preg_match_all(self::$patterns['valid_mentions_or_lists'], $tweet, $matches, PREG_SET_ORDER | PREG_OFFSET_CAPTURE);
|
||
$results = [];
|
||
|
||
foreach ($matches as $match) {
|
||
list($all, $before, $at, $username, $list_slug, $outer) = array_pad($match, 6, ['', 0]);
|
||
$start_position = $at[1] > 0 ? StringUtils::strlen(substr($tweet, 0, $at[1])) : $at[1];
|
||
$end_position = $start_position + StringUtils::strlen($at[0]) + StringUtils::strlen($username[0]);
|
||
$entity = [
|
||
'screen_name' => $username[0],
|
||
'list_slug' => $list_slug[0],
|
||
'indices' => [$start_position, $end_position],
|
||
];
|
||
|
||
if (preg_match(self::$patterns['end_mention_match'], $outer[0])) {
|
||
continue;
|
||
}
|
||
|
||
if (!empty($list_slug[0])) {
|
||
$entity['indices'][1] = $end_position + StringUtils::strlen($list_slug[0]);
|
||
}
|
||
|
||
$results[] = $entity;
|
||
}
|
||
|
||
return $results;
|
||
}
|
||
|
||
/**
|
||
* Extracts all the usernames and the indices they occur at from the tweet.
|
||
*
|
||
* @return array The username elements in the tweet.
|
||
*
|
||
* @deprecated since version 1.1.0
|
||
*/
|
||
public function extractMentionedUsernamesOrListsWithIndices()
|
||
{
|
||
return $this->extractMentionsOrListsWithIndices();
|
||
}
|
||
|
||
/**
|
||
* setter/getter for extractURLWithoutProtocol.
|
||
*
|
||
* @param bool $flag
|
||
*
|
||
* @return Extractor
|
||
*/
|
||
public function extractURLWithoutProtocol($flag = null)
|
||
{
|
||
if (is_null($flag)) {
|
||
return $this->extractURLWithoutProtocol;
|
||
}
|
||
$this->extractURLWithoutProtocol = (bool) $flag;
|
||
|
||
return $this;
|
||
}
|
||
|
||
/**
|
||
* Remove overlapping entities.
|
||
* This returns a new array with no overlapping entities.
|
||
*
|
||
* @param array $entities
|
||
*
|
||
* @return array
|
||
*/
|
||
public function removeOverlappingEntities($entities)
|
||
{
|
||
$result = [];
|
||
usort($entities, [$this, 'sortEntites']);
|
||
|
||
$prev = null;
|
||
foreach ($entities as $entity) {
|
||
if (isset($prev) && $entity['indices'][0] < $prev['indices'][1]) {
|
||
continue;
|
||
}
|
||
$prev = $entity;
|
||
$result[] = $entity;
|
||
}
|
||
|
||
return $result;
|
||
}
|
||
|
||
/**
|
||
* sort by entity start index.
|
||
*
|
||
* @param array $a
|
||
* @param array $b
|
||
*
|
||
* @return int
|
||
*/
|
||
protected function sortEntites($a, $b)
|
||
{
|
||
if ($a['indices'][0] == $b['indices'][0]) {
|
||
return 0;
|
||
}
|
||
|
||
return ($a['indices'][0] < $b['indices'][0]) ? -1 : 1;
|
||
}
|
||
}
|