<?php
declare(strict_types=1);
/*
 * sysPass
 *
 * @author nuxsmin
 * @link https://syspass.org
 * @copyright 2012-2024, Rubén Domínguez nuxsmin@$syspass.org
 *
 * This file is part of sysPass.
 *
 * sysPass 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 3 of the License, or
 * (at your option) any later version.
 *
 * sysPass 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 sysPass.  If not, see <http://www.gnu.org/licenses/>.
 */

namespace SP\Infrastructure\Account\Repositories;

use Aura\SqlQuery\Common\SelectInterface;
use Aura\SqlQuery\QueryFactory;
use SP\Domain\Account\Dtos\AccountSearchFilterDto;
use SP\Domain\Account\Models\AccountSearchView as AccountSearchViewModel;
use SP\Domain\Account\Ports\AccountFilterBuilder;
use SP\Domain\Account\Ports\AccountSearchConstants;
use SP\Domain\Account\Ports\AccountSearchRepository;
use SP\Domain\Common\Providers\Filter;
use SP\Domain\Core\Context\Context;
use SP\Domain\Core\Events\EventDispatcherInterface;
use SP\Domain\Core\Exceptions\ConstraintException;
use SP\Domain\Core\Exceptions\QueryException;
use SP\Domain\Database\Ports\DatabaseInterface;
use SP\Infrastructure\Common\Repositories\BaseRepository;
use SP\Infrastructure\Database\QueryData;
use SP\Infrastructure\Database\QueryResult;

/**
 * Class AccountSearch
 */
final class AccountSearch extends BaseRepository implements AccountSearchRepository
{
    private readonly SelectInterface $query;

    public function __construct(
        DatabaseInterface        $database,
        Context                  $session,
        EventDispatcherInterface $eventDispatcher,
        QueryFactory             $queryFactory,
        private readonly AccountFilterBuilder $accountFilterUser
    ) {
        parent::__construct($database, $session, $eventDispatcher, $queryFactory);

        $this->initQuery();
    }

    /**
     * @return void
     */
    private function initQuery(): void
    {
        $cols = [
            'id',
            'clientId',
            'categoryId',
            'name',
            'login',
            'url',
            'notes',
            'userId',
            'userGroupId',
            'otherUserEdit',
            'otherUserGroupEdit',
            'isPrivate',
            'isPrivateGroup',
            'passDate',
            'passDateChange',
            'parentId',
            'countView',
            'dateEdit',
            'userName',
            'userLogin',
            'userGroupName',
            'categoryName',
            'clientName',
            'num_files',
            'publicLinkHash',
            'publicLinkDateExpire',
            'publicLinkTotalCountViews',
        ];
        $this->query = $this->queryFactory
            ->newSelect()
            ->cols($cols)
            ->from(sprintf('%s AS Account', AccountSearchViewModel::TABLE))
            ->distinct();
    }

    /**
     * Obtener las cuentas de una búsqueda.
     *
     * @param AccountSearchFilterDto $accountSearchFilter
     *
     * @return QueryResult
     * @throws ConstraintException
     * @throws QueryException
     */
    public function getByFilter(AccountSearchFilterDto $accountSearchFilter): QueryResult
    {
        $this->accountFilterUser->buildFilter($accountSearchFilter->getGlobalSearch(), $this->query);
        $this->filterByText($accountSearchFilter);
        $this->filterByCategory($accountSearchFilter);
        $this->filterByClient($accountSearchFilter);
        $this->filterByFavorite($accountSearchFilter);
        $this->filterByTags($accountSearchFilter);
        $this->setOrder($accountSearchFilter);

        if ($accountSearchFilter->getLimitCount() > 0) {
            $this->query->limit($accountSearchFilter->getLimitCount());
            $this->query->offset($accountSearchFilter->getLimitStart());
        }

        return $this->db->runQuery(
            QueryData::build($this->query)->setMapClassName(AccountSearchViewModel::class),
            true
        );
    }

    /**
     * @param AccountSearchFilterDto $accountSearchFilter
     * @return void
     */
    private function filterByText(AccountSearchFilterDto $accountSearchFilter): void
    {
        // Sets the search text depending on whether special search filters are being used
        $searchText = $accountSearchFilter->getCleanTxtSearch();

        if (!empty($searchText)) {
            $searchTextLike = '%' . $searchText . '%';

            $this->query
                ->where(
                    '(Account.name LIKE :name OR Account.login LIKE :login OR Account.url LIKE :url OR Account.notes LIKE :notes)',
                    [
                        'name' => $searchTextLike,
                        'login' => $searchTextLike,
                        'url' => $searchTextLike,
                        'notes' => $searchTextLike,
                    ]
                );
        }
    }

    /**
     * @param AccountSearchFilterDto $accountSearchFilter
     * @return void
     */
    private function filterByCategory(AccountSearchFilterDto $accountSearchFilter): void
    {
        if ($accountSearchFilter->getCategoryId() !== null) {
            $this->query
                ->where(
                    'Account.categoryId = :categoryId',
                    [
                        'categoryId' => $accountSearchFilter->getCategoryId(),
                    ]
                );
        }
    }

    /**
     * @param AccountSearchFilterDto $accountSearchFilter
     * @return void
     */
    private function filterByClient(AccountSearchFilterDto $accountSearchFilter): void
    {
        if ($accountSearchFilter->getClientId() !== null) {
            $this->query
                ->where(
                    'Account.categoryId = :clientId',
                    [
                        'categoryId' => $accountSearchFilter->getClientId(),
                    ]
                );
        }
    }

    /**
     * @param AccountSearchFilterDto $accountSearchFilter
     * @return void
     */
    private function filterByFavorite(AccountSearchFilterDto $accountSearchFilter): void
    {
        if ($accountSearchFilter->isSearchFavorites() === true) {
            $this->query
                ->join(
                    'INNER',
                    'AccountToFavorite',
                    'AccountToFavorite.accountId = Account.id AND AccountToFavorite.userId = :userId',
                    [
                        'userId' => $this->context->getUserData()->getId(),
                    ]
                );
        }
    }

    /**
     * @param AccountSearchFilterDto $accountSearchFilter
     * @return void
     */
    private function filterByTags(AccountSearchFilterDto $accountSearchFilter): void
    {
        if ($accountSearchFilter->hasTags()) {
            $this->query->join(
                'INNER',
                'AccountToTag',
                'AccountToTag.accountId = Account.id'
            );

            $this->query
                ->where(
                    'AccountToTag.tagId IN (:tagId)',
                    [
                        'tagId' => $accountSearchFilter->getTagsId(),
                    ]
                );

            if (AccountSearchConstants::FILTER_CHAIN_AND === $accountSearchFilter->getFilterOperator()) {
                $this->query
                    ->groupBy(['Account.id'])
                    ->having(
                        'COUNT(DISTINCT AccountToTag.tagId) = :tagsCount',
                        [
                            'tagsCount' => count($accountSearchFilter->getTagsId()),
                        ]
                    );
            }
        }
    }

    /**
     * Devuelve la cadena de ordenación de la consulta
     */
    private function setOrder(AccountSearchFilterDto $filter): void
    {
        $orderKey = match ($filter->getSortKey()) {
            AccountSearchConstants::SORT_NAME => 'Account.name',
            AccountSearchConstants::SORT_CATEGORY => 'Account.categoryName',
            AccountSearchConstants::SORT_LOGIN => 'Account.login',
            AccountSearchConstants::SORT_URL => 'Account.url',
            AccountSearchConstants::SORT_CLIENT => 'Account.clientName',
            default => 'Account.clientName, Account.name',
        };

        if ($filter->isSortViews() && !$filter->getSortKey()) {
            $this->query->orderBy(['Account.countView DESC']);
        } else {
            $sortOrder = match ($filter->getSortOrder()) {
                AccountSearchConstants::SORT_DIR_ASC => 'ASC',
                AccountSearchConstants::SORT_DIR_DESC => 'DESC',
            };

            $this->query->orderBy([
                                      sprintf('%s %s', $orderKey, $sortOrder),
                                  ]);
        }
    }

    /**
     * @param int $userId
     * @param int $userGroupId
     *
     * @return SelectInterface
     */
    public function withFilterForUser(int $userId, int $userGroupId): SelectInterface
    {
        $where = [
            'Account.userId = :userId',
            'Account.userGroupId = :userGroupId',
            'Account.id IN (SELECT AccountToUser.accountId FROM AccountToUser WHERE AccountToUser.accountId = Account.id AND AccountToUser.userId = :userId
                                    UNION
                                    SELECT AccountToUserGroup.accountId FROM AccountToUserGroup WHERE AccountToUserGroup.accountId = Account.id AND AccountToUserGroup.userGroupId = :userGroupId)',
        ];

        return $this->query
            ->where(sprintf('(%s)', join(sprintf(' %s ', AccountSearchConstants::FILTER_CHAIN_OR), $where)))
            ->bindValues([
                             'userId' => $userId,
                             'userGroupId' => $userGroupId,
                         ]);
    }

    /**
     * @param int $userGroupId
     *
     * @return SelectInterface
     */
    public function withFilterForGroup(int $userGroupId): SelectInterface
    {
        return $this->query
            ->where('Account.userGroupId = :userGroupId')
            ->orWhere(
                '(Account.id IN (SELECT AccountToUserGroup.accountId FROM AccountToUserGroup WHERE AccountToUserGroup.accountId = id AND AccountToUserGroup.userGroupId = :userGroupId))'
            )
            ->bindValues([
                             'userGroupId' => $userGroupId,
                         ]);
    }

    /**
     * @param string $userGroupName
     *
     * @return SelectInterface
     */
    public function withFilterForMainGroup(string $userGroupName): SelectInterface
    {
        $userGroupNameLike = '%' . Filter::safeSearchString($userGroupName) . '%';

        return $this->query
            ->where('Account.userGroupName LIKE :userGroupName')
            ->bindValues([
                             'userGroupName' => $userGroupNameLike,
                         ]);
    }

    /**
     * @param string $owner
     *
     * @return SelectInterface
     */
    public function withFilterForOwner(string $owner): SelectInterface
    {
        $ownerLike = '%' . Filter::safeSearchString($owner) . '%';

        return $this->query
            ->where('(Account.userLogin LIKE :userLogin OR Account.userName LIKE :userName)')
            ->bindValues([
                             'userLogin' => $ownerLike,
                             'userName' => $ownerLike,
                         ]);
    }

    /**
     * @param string $fileName
     *
     * @return SelectInterface
     */
    public function withFilterForFile(string $fileName): SelectInterface
    {
        $fileNameLike = '%' . Filter::safeSearchString($fileName) . '%';

        return $this->query
            ->where(
                '(Account.id IN (SELECT AccountFile.accountId FROM AccountFile WHERE AccountFile.name LIKE :fileName))'
            )
            ->bindValues([
                             'fileName' => $fileNameLike,
                         ]);
    }

    /**
     * @param int $accountId
     *
     * @return SelectInterface
     */
    public function withFilterForAccountId(int $accountId): SelectInterface
    {
        return $this->query
            ->where('Account.id = :accountId')
            ->bindValues([
                             'accountId' => $accountId,
                         ]);
    }

    /**
     * @param string $clientName
     *
     * @return SelectInterface
     */
    public function withFilterForClient(string $clientName): SelectInterface
    {
        $clientNameLike = '%' . Filter::safeSearchString($clientName) . '%';

        return $this->query
            ->where('Account.clientName LIKE :clientName')
            ->bindValues([
                             'clientName' => $clientNameLike,
                         ]);
    }

    /**
     * @param string $categoryName
     *
     * @return SelectInterface
     */
    public function withFilterForCategory(string $categoryName): SelectInterface
    {
        $categoryNameLike = '%' . Filter::safeSearchString($categoryName) . '%';

        return $this->query
            ->where('Account.categoryName LIKE :categoryName')
            ->bindValues([
                             'categoryName' => $categoryNameLike,
                         ]);
    }

    /**
     * @param string $accountName
     *
     * @return SelectInterface
     */
    public function withFilterForAccountNameRegex(string $accountName): SelectInterface
    {
        return $this->query
            ->where('Account.name REGEXP :name')
            ->bindValues([
                             'name' => $accountName,
                         ]);
    }

    public function withFilterForIsExpired(): SelectInterface
    {
        return $this->query
            ->where('(Account.passDateChange > 0 AND UNIX_TIMESTAMP() > Account.passDateChange)');
    }

    public function withFilterForIsNotExpired(): SelectInterface
    {
        return $this->query
            ->where(
                '(Account.passDateChange = 0 OR Account.passDateChange IS NULL OR UNIX_TIMESTAMP() < Account.passDateChange)'
            );
    }

    /**
     * @param int $userId
     * @param int $userGroupId
     *
     * @return SelectInterface
     */
    public function withFilterForIsPrivate(int $userId, int $userGroupId): SelectInterface
    {
        return $this->query
            ->where(
                '(Account.isPrivate = 1 AND Account.userId = :userId) OR (Account.isPrivateGroup = 1 AND Account.userGroupId = :userGroupId)'
            )
            ->bindValues([
                             'userId' => $userId,
                             'userGroupId' => $userGroupId,
                         ]);
    }

    public function withFilterForIsNotPrivate(): SelectInterface
    {
        return $this->query
            ->where(
                '(Account.isPrivate = 0 OR Account.isPrivate IS NULL) AND (Account.isPrivateGroup = 0 OR Account.isPrivateGroup IS NULL)'
            );
    }
}