osu-web / app / Models / Traits / Es / ForumPostSearch.php

<?php

// Copyright (c) ppy Pty Ltd <[e-mail]>. Licensed under the GNU Affero General Public License v3.0.
// See the LICENCE file in the repository root for full licence text.

namespace App\Models\Traits\Es;

use App\Models\Forum\Forum;
use Carbon\Carbon;

trait ForumPostSearch
{
    use BaseDbIndexable;

    public static function esIndexName()
    {
        return $GLOBALS['cfg']['osu']['elasticsearch']['prefix'].'posts';
    }

    public static function esIndexingQuery()
    {
        $forumIds = Forum::where('enable_indexing', 1)->pluck('forum_id');

        return static::withoutGlobalScopes()->whereIn('forum_id', $forumIds)->with('forum')->with('topic');
    }

    public static function esSchemaFile()
    {
        return config_path('schemas/posts.json');
    }

    public function esRouting()
    {
        // Post and Topic should have the same routing for relationships to work.
        return $this->topic_id;
    }

    public function esShouldIndex()
    {
        return $this->forum->enable_indexing && $this->topic !== null;
    }

    public function getEsId()
    {
        return "post-{$this->post_id}";
    }

    public function toEsJson()
    {
        $mappings = static::esMappings();

        $document = [];
        foreach ($mappings as $field => $mapping) {
            switch ($field) {
                case 'is_deleted':
                    $value = $this->trashed() || $this->topic->trashed();
                    break;
                case 'topic_title':
                    if ($this->topic !== null && $this->topic->topic_first_post_id === $this->getKey()) {
                        $value = $this->topic->topic_title;
                    } else {
                        $value = null;
                    }
                    break;
                default:
                    $value = $this[$field];
            }

            if ($value instanceof Carbon) {
                $value = $value->toIso8601String();
            }

            $document[$field] = $value;
        }

        return $document;
    }
}