pixelfed/app/Jobs/StatusPipeline/StatusEntityLexer.php

<?php

namespace App\Jobs\StatusPipeline;

use App\Hashtag;
use App\Jobs\MentionPipeline\MentionPipeline;
use App\Mention;
use App\Profile;
use App\Status;
use App\StatusHashtag;
use App\Services\PublicTimelineService;
use App\Util\Lexer\Autolink;
use App\Util\Lexer\Extractor;
use App\Util\Sentiment\Bouncer;
use DB;
use Illuminate\Bus\Queueable;
use Illuminate\Contracts\Queue\ShouldQueue;
use Illuminate\Foundation\Bus\Dispatchable;
use Illuminate\Queue\InteractsWithQueue;
use Illuminate\Queue\SerializesModels;

class StatusEntityLexer implements ShouldQueue
{
	use Dispatchable, InteractsWithQueue, Queueable, SerializesModels;

	protected $status;
	protected $entities;
	protected $autolink;

	/**
	 * Delete the job if its models no longer exist.
	 *
	 * @var bool
	 */
	public $deleteWhenMissingModels = true;

	/**
	 * Create a new job instance.
	 *
	 * @return void
	 */
	public function __construct(Status $status)
	{
		$this->status = $status;
	}

	/**
	 * Execute the job.
	 *
	 * @return void
	 */
	public function handle()
	{
		$profile = $this->status->profile;
		$status = $this->status;

		if(in_array($status->type, ['photo', 'photo:album', 'video', 'video:album', 'photo:video:album'])) {
			$profile->status_count = $profile->status_count + 1;
			$profile->save();
		}

		if($profile->no_autolink == false) {
			$this->parseEntities();
		}
	}

	public function parseEntities()
	{
		$this->extractEntities();
	}

	public function extractEntities()
	{
		$this->entities = Extractor::create()->extract($this->status->caption);
		$this->autolinkStatus();
	}

	public function autolinkStatus()
	{
		$this->autolink = Autolink::create()->autolink($this->status->caption);
		$this->storeEntities();
	}

	public function storeEntities()
	{
		$this->storeHashtags();
		DB::transaction(function () {
			$status = $this->status;
			$status->rendered = nl2br($this->autolink);
			$status->entities = json_encode($this->entities);
			$status->save();
		});
	}

	public function storeHashtags()
	{
		$tags = array_unique($this->entities['hashtags']);
		$status = $this->status;

		foreach ($tags as $tag) {
			if(mb_strlen($tag) > 124) {
				continue;
			}
			DB::transaction(function () use ($status, $tag) {
				$slug = str_slug($tag, '-', false);
				$hashtag = Hashtag::where('slug', $slug)->first();
				if (!$hashtag) {
					$hashtag = Hashtag::create(
						['name' => $tag, 'slug' => $slug]
					);
				}

				StatusHashtag::firstOrCreate(
					[
						'status_id' => $status->id,
						'hashtag_id' => $hashtag->id,
						'profile_id' => $status->profile_id,
						'status_visibility' => $status->visibility,
					]
				);
			});
		}
		$this->storeMentions();
	}

	public function storeMentions()
	{
		$mentions = array_unique($this->entities['mentions']);
		$status = $this->status;

		foreach ($mentions as $mention) {
			$mentioned = Profile::whereUsername($mention)->first();

			if (empty($mentioned) || !isset($mentioned->id)) {
				continue;
			}

			DB::transaction(function () use ($status, $mentioned) {
				$m = new Mention();
				$m->status_id = $status->id;
				$m->profile_id = $mentioned->id;
				$m->save();

				MentionPipeline::dispatch($status, $m);
			});
		}
		$this->deliver();
	}

	public function deliver()
	{
		$status = $this->status;
		$types = [
			'photo',
			'photo:album',
			'video',
			'video:album',
			'photo:video:album'
		];

		if(config_cache('pixelfed.bouncer.enabled')) {
			Bouncer::get($status);
		}

		if( $status->uri == null &&
			$status->scope == 'public' &&
			in_array($status->type, $types) &&
			$status->in_reply_to_id === null &&
			$status->reblog_of_id === null
		) {
			PublicTimelineService::add($status->id);
		}

		if(config_cache('federation.activitypub.enabled') == true && config('app.env') == 'production') {
			StatusActivityPubDeliver::dispatch($status);
		}
	}
}