<?php
namespace FireflyIII\Helper\Related;

use FireflyIII\Database\SwitchUser;
use Illuminate\Support\Collection;

/**
 * Class Related
 *
 * @package FireflyIII\Helper\Related
 */
class Related implements RelatedInterface
{
    use SwitchUser;

    /**
     *
     */
    public function __construct()
    {
        $this->setUser(\Auth::user());
    }

    /**
     * @param array $objectIds
     *
     * @return Collection
     */
    public function getJournalsByIds(array $objectIds)
    {
        /** @var \FireflyIII\Database\TransactionJournal\TransactionJournal $repository */
        $repository = \App::make('FireflyIII\Database\TransactionJournal\TransactionJournal');

        return $repository->getByIds($objectIds);
    }

    /**
     * @param string              $query
     * @param \TransactionJournal $journal
     *
     * @return Collection
     */
    public function search($query, \TransactionJournal $journal)
    {
        $start = clone $journal->date;
        $end   = clone $journal->date;
        $start->startOfMonth();
        $end->endOfMonth();

        // get already related transactions:
        $exclude = [$journal->id];
        foreach ($journal->transactiongroups()->get() as $group) {
            foreach ($group->transactionjournals() as $current) {
                $exclude[] = $current->id;
            }
        }
        $exclude = array_unique($exclude);

        /** @var Collection $collection */
        $collection = $this->getUser()->transactionjournals()
                           ->withRelevantData()
                           ->before($end)
                           ->where('encrypted', 0)
                           ->after($start)
                           ->whereNotIn('id', $exclude)
                           ->where('description', 'LIKE', '%' . $query . '%')
                           ->get();

        // manually search encrypted entries:
        /** @var Collection $encryptedCollection */
        $encryptedCollection = $this->getUser()->transactionjournals()
                                    ->withRelevantData()
                                    ->before($end)
                                    ->where('encrypted', 1)
                                    ->after($start)
                                    ->whereNotIn('id', $exclude)
                                    ->get();
        $encrypted           = $encryptedCollection->filter(
            function (\TransactionJournal $journal) use ($query) {
                $strPos = strpos($journal->description, $query);
                if ($strPos !== false) {
                    return $journal;
                }
            }
        );
        $collected           = $collection->merge($encrypted);


        return $collected;
    }
}