Add filter to userAgent service

Author: pazakharov

Src/Console/SeleniumToolsController.php

@@ -20,12 +20,14 @@ public function actionIndex()
     }
     /**
      * Fill the base with data for the StepBrowser profiles e.g: UserAgents
-     *
+     * You ca use params in your request for filter user agents:
+     * `php yii seleniumTools/fill-base chrome win`
+     * `php yii seleniumTools/fill-base chrome 102.0`
      * @return void
      */
-    public function actionFillBase()
+    public function actionFillBase(...$filterKeys)
     {
         $userAgentService = Yii::$container->get(UserAgentService::class);
-        $userAgentService->fillBaseWithUserAgents();
+        $userAgentService->fillBaseWithUserAgents($filterKeys);
     }
 }

Src/Utils/UserAgent/UserAgentService.php

@@ -2,8 +2,8 @@
 
 namespace Zakharov\Yii2SeleniumTools\Utils\UserAgent;
 
+use PDO;
 use Yii;
-use yii\helpers\ArrayHelper;
 use Zakharov\Yii2SeleniumTools\models\UserAgent;
 use Zakharov\Yii2SeleniumTools\Utils\UserAgent\UserAgentProvider;
 
@@ -24,19 +24,39 @@ public function __construct(UserAgentProvider $userAgentProvider)
     /**
      * fill base with user agents by specified in container userAgentProvider
      *
+     * @param string[] $filterKey
      * @return int number of inserts
      */
-    public function fillBaseWithUserAgents()
+    public function fillBaseWithUserAgents($filterKeys = [])
     {
         $arrayOfUserAgents = array_values($this->userAgentProvider->getArrayOfUserAgents());
         if (empty($arrayOfUserAgents)) {
             return 0;
         }
+        $arrayOfUserAgents = $this->filterArrayOfUserAgentsWithFilter($arrayOfUserAgents, $filterKeys);
         $createdAt = (new \DateTime())->setTimezone(new \DateTimeZone('UTC'))->format('Y-m-d H:i:s');
         $updateArray = array_map(fn ($item) => [$item, $createdAt], $arrayOfUserAgents);
+
         return Yii::$app->db
             ->createCommand()
             ->batchInsert(UserAgent::tableName(), ['ua', 'created_at'], $updateArray)
             ->execute();
     }
+
+    /**
+     * filterArrayOfUserAgentsWithFilter
+     *
+     * @param  array $arrayOfUserAgents
+     * @param  array $filterKeys
+     * @return void
+     */
+    protected function filterArrayOfUserAgentsWithFilter(array $arrayOfUserAgents, array $filterKeys = [])
+    {
+        if (!empty($filterKeys)) {
+            foreach ($filterKeys as $key) {
+                $arrayOfUserAgents = array_filter($arrayOfUserAgents, fn ($item) => stripos($item, $key) !== false);
+            }
+        }
+        return $arrayOfUserAgents;
+    }
 }