gridpilot.gg/packages/social/infrastructure/inmemory/InMemorySocialAndFeed.ts

import type { Driver } from '@gridpilot/racing/domain/entities/Driver';
import type { FeedItem } from '@gridpilot/social/domain/types/FeedItem';
import type { IFeedRepository } from '@gridpilot/social/domain/repositories/IFeedRepository';
import type { ISocialGraphRepository } from '@gridpilot/social/domain/repositories/ISocialGraphRepository';
import type { ILogger } from '@gridpilot/shared/logging/ILogger';

export type Friendship = {
  driverId: string;
  friendId: string;
};

export type RacingSeedData = {
  drivers: Driver[];
  friendships: Friendship[];
  feedEvents: FeedItem[];
};

export class InMemoryFeedRepository implements IFeedRepository {
  private readonly feedEvents: FeedItem[];
  private readonly friendships: Friendship[];
  private readonly driversById: Map<string, Driver>;
  private readonly logger: ILogger;

  constructor(logger: ILogger, seed: RacingSeedData) {
    this.logger = logger;
    this.logger.info('InMemoryFeedRepository initialized.');
    this.feedEvents = seed.feedEvents;
    this.friendships = seed.friendships;
    this.driversById = new Map(seed.drivers.map((d) => [d.id, d]));
  }

  async getFeedForDriver(driverId: string, limit?: number): Promise<FeedItem[]> {
    this.logger.debug(`Getting feed for driver: ${driverId}, limit: ${limit}`);
    try {
      const friendIds = new Set(
        this.friendships
          .filter((f) => f.driverId === driverId)
          .map((f) => f.friendId),
      );

      const items = this.feedEvents.filter((item) => {
        if (item.actorDriverId && friendIds.has(item.actorDriverId)) {
          return true;
        }
        return false;
      });

      const sorted = items
        .slice()
        .sort((a, b) => b.timestamp.getTime() - a.timestamp.getTime());

      this.logger.info(`Found ${sorted.length} feed items for driver: ${driverId}.`);
      return typeof limit === 'number' ? sorted.slice(0, limit) : sorted;
    } catch (error) {
      this.logger.error(`Error getting feed for driver ${driverId}:`, error);
      throw error;
    }
  }

  async getGlobalFeed(limit?: number): Promise<FeedItem[]> {
    this.logger.debug(`Getting global feed, limit: ${limit}`);
    try {
      const sorted = this.feedEvents
        .slice()
        .sort((a, b) => b.timestamp.getTime() - a.timestamp.getTime());

      this.logger.info(`Found ${sorted.length} global feed items.`);
      return typeof limit === 'number' ? sorted.slice(0, limit) : sorted;
    } catch (error) {
      this.logger.error(`Error getting global feed:`, error);
      throw error;
    }
  }
}

export class InMemorySocialGraphRepository implements ISocialGraphRepository {
  private readonly friendships: Friendship[];
  private readonly driversById: Map<string, Driver>;
  private readonly logger: ILogger;

  constructor(logger: ILogger, seed: RacingSeedData) {
    this.logger = logger;
    this.logger.info('InMemorySocialGraphRepository initialized.');
    this.friendships = seed.friendships;
    this.driversById = new Map(seed.drivers.map((d) => [d.id, d]));
  }

  async getFriendIds(driverId: string): Promise<string[]> {
    this.logger.debug(`Getting friend IDs for driver: ${driverId}`);
    try {
      const friendIds = this.friendships
        .filter((f) => f.driverId === driverId)
        .map((f) => f.friendId);
      this.logger.info(`Found ${friendIds.length} friend IDs for driver: ${driverId}.`);
      return friendIds;
    } catch (error) {
      this.logger.error(`Error getting friend IDs for driver ${driverId}:`, error);
      throw error;
    }
  }

  async getFriends(driverId: string): Promise<Driver[]> {
    this.logger.debug(`Getting friends for driver: ${driverId}`);
    try {
      const ids = await this.getFriendIds(driverId);
      const friends = ids
        .map((id) => this.driversById.get(id))
        .filter((d): d is Driver => Boolean(d));
      this.logger.info(`Found ${friends.length} friends for driver: ${driverId}.`);
      return friends;
    } catch (error) {
      this.logger.error(`Error getting friends for driver ${driverId}:`, error);
      throw error;
    }
  }

  async getSuggestedFriends(driverId: string, limit?: number): Promise<Driver[]> {
    this.logger.debug(`Getting suggested friends for driver: ${driverId}, limit: ${limit}`);
    try {
      const directFriendIds = new Set(await this.getFriendIds(driverId));
      const suggestions = new Map<string, number>();

      for (const friendship of this.friendships) {
        if (!directFriendIds.has(friendship.driverId)) continue;
        const friendOfFriendId = friendship.friendId;
        if (friendOfFriendId === driverId) continue;
        if (directFriendIds.has(friendOfFriendId)) continue;

        suggestions.set(friendOfFriendId, (suggestions.get(friendOfFriendId) ?? 0) + 1);
      }

      const rankedIds = Array.from(suggestions.entries())
        .sort((a, b) => b[1] - a[1])
        .map(([id]) => id);

      const drivers = rankedIds
        .map((id) => this.driversById.get(id))
        .filter((d): d is Driver => Boolean(d));

      const result = typeof limit === 'number' ? drivers.slice(0, limit) : drivers;
      this.logger.info(`Found ${result.length} suggested friends for driver: ${driverId}.`);
      return result;
    } catch (error) {
      this.logger.error(`Error getting suggested friends for driver ${driverId}:`, error);
      throw error;
    }
  }
}