TTK-26902 update AccountConnectionCommand

teltek · Mar 7, 2024 · cbace1c · cbace1c
1 parent d3e745c
commit cbace1c
Showing 1 changed file with 16 additions and 17 deletions.
diff --git a/Command/AccountConnectionCommand.php b/Command/AccountConnectionCommand.php
@@ -30,11 +30,12 @@ protected function configure()
             ->setName('pumukit:youtube:account:test')
             ->setDescription('Test connection accounts')
             ->addOption('account', null, InputOption::VALUE_REQUIRED, 'Name of account')
+            ->addOption('channel', null, InputOption::VALUE_REQUIRED, 'UID of channel')
             ->setHelp(
                 <<<'EOT'
 Check:
 
-    php bin/console pumukit:youtube:account:test --account={login}
+    php bin/console pumukit:youtube:account:test --account={login} --channel={channelId}
 
 EOT
             )
@@ -45,33 +46,31 @@ protected function execute(InputInterface $input, OutputInterface $output): int
     {
         $output->writeln('<info>----- Testing connection with YouTube -----</info>');
 
-        $account = $this->accountExists($input->getOption('account'));
-        if (!$account) {
-            $output->writeln('<notice>Account not in DB</notice>');
+        $youtubeAccount = $this->ensureYouTubeAccountExists($input);
 
-            return 0;
-        }
-
-        $service = $this->googleAccountService->googleServiceFromAccount($account);
+        $service = $this->googleAccountService->googleServiceFromAccount($youtubeAccount);
 
-        $response = $service->playlists->listPlaylists('snippet,contentDetails', [
+        $queryParams = [
+            'channelId' => $input->getOption('channel'),
             'maxResults' => 5,
-            'mine' => true,
-        ]);
+        ];
 
-        print_r($response);
+        $response = $service->playlists->listPlaylists('snippet', $queryParams);
+        $output->writeln('Number of playlist of account: '.$response->pageInfo->getTotalResults());
 
         return 0;
     }
 
-    private function accountExists(string $login): ?Tag
+    private function ensureYouTubeAccountExists(InputInterface $input): Tag
     {
-        $account = $this->documentManager->getRepository(Tag::class)->findOneBy(['properties.login' => $login]);
+        $youtubeAccount = $this->documentManager->getRepository(Tag::class)->findOneBy([
+            'properties.login' => $input->getOption('account'),
+        ]);
 
-        if ($account->getProperty('access_token') || $account->getProperty('refresh_token')) {
-            return $account;
+        if (!$youtubeAccount) {
+            throw new \Exception('Account not found');
         }
 
-        return null;
+        return $youtubeAccount;
     }
 }