jakim / ig-monitoring

components/AccountManager.php

<?php
/**
 * Created for IG Monitoring.
 * User: jakim <pawel@jakimowski.info>
 * Date: 28.04.2018
 */

namespace app\components;


use app\components\traits\BatchInsertCommandTrait;
use app\components\traits\FindOrCreateTrait;
use app\components\builders\AccountBuilder;
use app\models\Account;
use app\models\AccountCategory;
use app\models\Media;
use app\models\MediaAccount;
use DateTime;
use Yii;
use yii\base\Component;
use yii\helpers\StringHelper;
use function is_string;

class AccountManager extends Component
{
    use FindOrCreateTrait, BatchInsertCommandTrait;

The trait app\components\traits\BatchInsertCommandTrait requires some properties which are not provided by app\components\AccountManager: $db, $queryBuilder

    public function startMonitoring($account): Account
    {
        if (is_string($account)) {
            /** @var Account $account */
            $account = $this->findOrCreate(['username' => $account], Account::class);
        }

        $accountUpdater = Yii::createObject([
            'class' => AccountBuilder::class,
            'account' => $account,
        ]);
        $accountUpdater
            ->setMonitoring()
            ->setIsValid()
            ->save();

        return $account;
    }

    public function addToMedia(Media $media, array $usernames)
    {
        $this->saveUsernames($usernames);

        $accounts = Account::find()
            ->andWhere(['username' => $usernames])
            ->column();

        $rows = array_map(function ($id) use ($media) {
            return [
                $media->id,
                $id,
                $media->taken_at,
            ];
        }, $accounts);

        $this->batchInsertIgnoreCommand(MediaAccount::tableName(), ['media_id', 'account_id', 'created_at'], $rows)
            ->execute();
    }

    /**
     * @param array|string[] $usernames
     * @throws \yii\db\Exception
     */
    public function saveUsernames(array $usernames)
    {
        $createdAt = (new DateTime())->format('Y-m-d H:i:s');
        $rows = array_map(function ($username) use ($createdAt) {
            return [
                $username,
                $createdAt,
                $createdAt,
            ];
        }, $usernames);

        $this->batchInsertIgnoreCommand(Account::tableName(), ['username', 'updated_at', 'created_at'], $rows)
            ->execute();
    }

    /**
     * @param string|string[] $categories
     * @param null|int $userId
     * @return array|int[]
     */
    public function findByCategories($categories, $userId = null): array
    {
        if (is_string($categories)) {
            $categories = StringHelper::explode($categories, ',', true, true);
            $categories = array_unique($categories);
        }

        $ids = [];
        foreach ($categories as $category) {
            $ids[] = AccountCategory::find()
                ->distinct()
                ->select('account_id')
                ->innerJoinWith('category')
                ->andFilterWhere(['user_id' => $userId])
                ->andFilterWhere(['like', 'category.name', $category])
                ->column();
        }

        return array_intersect(...$ids, ...$ids);
    }
}