eveseat/web

<?php

/*
 * This file is part of SeAT
 *
 * Copyright (C) 2015 to 2022 Leon Jacobs
 *
 * 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 2 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, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */

namespace Seat\Web\Http\DataTables\Scopes;

use Illuminate\Support\Facades\Gate;
use Seat\Eveapi\Models\Character\CharacterInfo;
use Yajra\DataTables\Contracts\DataTableScope;

/**
 * Class CharacterMailScope.
 *
 * This is a specific scope for character mails table.
 * It will restrict returned data based on both requested character ID and granted access.
 *
 * @package Seat\Web\Http\DataTables\Scopes
 */
class CharacterMailScope implements DataTableScope
{
    /**
     * @var int[]
     */
    private $requested_characters;

    /**
     * CharacterMailScope constructor.
     *
     * @param  int[]|null  $character_ids
     */
    public function __construct(?array $character_ids = null)
    {
        $this->requested_characters = $character_ids;
    }

    /**
     * Apply a query scope.
     *
     * @param  \Illuminate\Database\Query\Builder|\Illuminate\Database\Eloquent\Builder  $query
     * @return mixed
     */
    public function apply($query)
    {
        return $query->whereHas('recipients', function ($sub_query) {

            if ($this->requested_characters != null) {
                $character_ids = collect($this->requested_characters)->filter(function ($item) {
                    return Gate::allows('character.mail', [$item]);
                });

                return $character_ids->count() == count($this->requested_characters) ?
                    $sub_query->whereIn('recipient_id', $this->requested_characters) :
                    $sub_query->whereIn('recipient_id', []);
            }

            if (auth()->user()->isAdmin())
                return $sub_query;

            // collect metadata related to required permission
            $permissions = auth()->user()->roles()->with('permissions')->get()
                ->pluck('permissions')
                ->flatten()
                ->filter(function ($permission) {
                    return $permission->title == 'character.mail';
                });

            // in case at least one permission is granted without restrictions, return all
            if ($permissions->filter(function ($permission) { return ! $permission->hasFilters(); })->isNotEmpty())
                return $sub_query;

            // extract entity ids and group by entity type
            $map = $permissions->map(function ($permission) {
                $filters = json_decode($permission->pivot->filters);

                return [
                    'characters'   => collect($filters->character ?? [])->pluck('id')->toArray(),
                    'corporations' => collect($filters->corporation ?? [])->pluck('id')->toArray(),
                    'alliances'    => collect($filters->alliance ?? [])->pluck('id')->toArray(),
                ];
            });

            // collect at least user owned characters
            $owned_range = auth()->user()->associatedCharacterIds();

            $characters_range = $map->pluck('characters')->flatten()->toArray();

            $corporations_range = CharacterInfo::whereHas('affiliation', function ($affiliation) use ($map) {
                $affiliation->whereIn('corporation_id', $map->pluck('corporations')->flatten()->toArray());
            })->select('character_id')->get()->pluck('character_id')->toArray();

            $alliances_range = CharacterInfo::whereHas('affiliation', function ($affiliation) use ($map) {
                $affiliation->whereIn('alliance_id', $map->pluck('alliances')->flatten()->toArray());
            })->select('character_id')->get()->pluck('character_id')->toArray();

            // sharelink
            $sharelink = session()->get('user_sharing') ?: [];

            // merge all collected characters IDs in a single array and apply filter
            $character_ids = array_merge(
                $characters_range, $corporations_range, $alliances_range, $owned_range, $sharelink,
                $map->pluck('corporations')->flatten()->toArray(), $map->pluck('alliances')->flatten()->toArray());

            return $sub_query->whereIntegerInRaw('recipient_id', $character_ids);
        });
    }
}