migrations/Version20241114200328.php line 1

Open in your IDE?
  1. <?php
  3. declare(strict_types=1);
  5. namespace DoctrineMigrations;
  7. use Doctrine\DBAL\Schema\Schema;
  8. use Doctrine\Migrations\AbstractMigration;
  10. /**
  11.  * Auto-generated Migration: Please modify to your needs!
  12.  */
  13. final class Version20241114200328 extends AbstractMigration
  14. {
  15.     public function getDescription(): string
  16.     {
  17.         return '';
  18.     }
  20.     public function up(Schema $schema): void
  21.     {
  22.         // this up() migration is auto-generated, please modify it to your needs
  24.         // Fetch all users with non-empty params
  25.         $users = $this->connection->fetchAllAssociative('SELECT id, params FROM user WHERE params IS NOT NULL');
  27.         foreach ($users as $user) {
  28.             $userId = $user['id'];
  29.             $params = unserialize($user['params']);
  31.             // Check if params is a valid array
  32.             if (!\is_array($params)) {
  33.                 continue;
  34.             }
  36.             // Extract all stream IDs referenced in this user's params
  37.             $streamIds = $this->getStreamIdsFromParams($params);
  39.             // Fetch only the direct message streams relevant to this user
  40.             $directMessageStreamIds = $this->getDirectMessageStreamIds($streamIds);
  42.             // Clean up the params for the current user
  43.             $updatedParams = $this->removeDirectMessageStreams($params, $directMessageStreamIds);
  45.             // Update the user params in the database if changes were made
  46.             if ($updatedParams !== $params) {
  47.                 $this->addSql('UPDATE user SET params = ? WHERE id = ?', [serialize($updatedParams), $userId]);
  48.             }
  49.         }
  50.     }
  52.     public function down(Schema $schema): void
  53.     {
  54.         // this down() migration is auto-generated, please modify it to your needs
  56.         // No rollback required
  57.     }
  59.     /**
  60.      * Extract stream IDs from user's params.
  61.      */
  62.     private function getStreamIdsFromParams(array $params): array
  63.     {
  64.         $streamIds = [];
  66.         if (isset($params['workspaces']) && \is_array($params['workspaces'])) {
  67.             foreach ($params['workspaces'] as $workspace) {
  68.                 // Collect stream IDs from favorites
  69.                 if (isset($workspace['favorites'])) {
  70.                     $streamIds = array_merge($streamIds, $workspace['favorites']);
  71.                 }
  73.                 // Collect stream IDs from folders
  74.                 if (isset($workspace['folders']) && \is_array($workspace['folders'])) {
  75.                     foreach ($workspace['folders'] as $folder) {
  76.                         $streamIds = array_merge($streamIds, $folder);
  77.                     }
  78.                 }
  80.                 // Collect stream IDs from unassigned
  81.                 if (isset($workspace['unassigned'])) {
  82.                     $streamIds = array_merge($streamIds, $workspace['unassigned']);
  83.                 }
  84.             }
  85.         }
  87.         return array_unique($streamIds);
  88.     }
  90.     /**
  91.      * Fetch direct message streams for a given set of stream IDs.
  92.      */
  93.     private function getDirectMessageStreamIds(array $streamIds): array
  94.     {
  95.         if (empty($streamIds)) {
  96.             return [];
  97.         }
  99.         // Fetch only the streams that are marked as direct messages
  100.         $placeholders = implode(',', array_fill(0, \count($streamIds), '?'));
  101.         $query = "SELECT id FROM w_stream WHERE direct_message = 1 AND id IN ($placeholders)";
  102.         $streams = $this->connection->fetchAllAssociative($query, $streamIds);
  104.         return array_column($streams, 'id');
  105.     }
  107.     /**
  108.      * Remove direct message streams from user params.
  109.      */
  110.     private function removeDirectMessageStreams(array $params, array $directMessageStreamIds): array
  111.     {
  112.         if (isset($params['workspaces']) && \is_array($params['workspaces'])) {
  113.             foreach ($params['workspaces'] as &$workspace) {
  114.                 // Remove from favorites
  115.                 if (isset($workspace['favorites'])) {
  116.                     $workspace['favorites'] = array_values(
  117.                         array_diff($workspace['favorites'], $directMessageStreamIds)
  118.                     );
  119.                 }
  121.                 // Remove from folders
  122.                 if (isset($workspace['folders']) && \is_array($workspace['folders'])) {
  123.                     foreach ($workspace['folders'] as &$folder) {
  124.                         $folder = array_values(
  125.                             array_diff($folder, $directMessageStreamIds)
  126.                         );
  127.                     }
  128.                 }
  130.                 // Remove from unassigned streams
  131.                 if (isset($workspace['unassigned'])) {
  132.                     $workspace['unassigned'] = array_values(
  133.                         array_diff($workspace['unassigned'], $directMessageStreamIds)
  134.                     );
  135.                 }
  136.             }
  137.         }
  139.         return $params;
  140.     }
  141. }