project-inter-server/src/repositories/shuttle/eta/BaseRedisETARepository.ts

import { IEta } from "../../../entities/ShuttleRepositoryEntities";
import { BaseRedisRepository } from "../../BaseRedisRepository";
import { ETAGetterRepository, ETARepositoryEvent, ETARepositoryEventListener, ETARepositoryEventName } from "./ETAGetterRepository";

export abstract class BaseRedisETARepository extends BaseRedisRepository implements ETAGetterRepository {
  private static readonly ETA_KEY_PREFIX = 'shuttle:eta:';

  // Helper methods
  protected createEtaKey = (shuttleId: string, stopId: string) =>
    `${BaseRedisETARepository.ETA_KEY_PREFIX}${shuttleId}:${stopId}`;

  createRedisHashFromEta = (eta: IEta): Record<string, string> => ({
    secondsRemaining: eta.secondsRemaining.toString(),
    shuttleId: eta.shuttleId,
    stopId: eta.stopId,
    systemId: eta.systemId,
    updatedTime: eta.updatedTime.toISOString(),
  });

  createEtaFromRedisData = (data: Record<string, string>): IEta => ({
    secondsRemaining: parseFloat(data.secondsRemaining),
    shuttleId: data.shuttleId,
    stopId: data.stopId,
    systemId: data.systemId,
    updatedTime: new Date(data.updatedTime),
  });

  // Getter implementations
  async getEtasForShuttleId(shuttleId: string): Promise<IEta[]> {
    const keys = await this.redisClient.keys(`${BaseRedisETARepository.ETA_KEY_PREFIX}${shuttleId}:*`);
    const etas: IEta[] = [];

    for (const key of keys) {
      const data = await this.redisClient.hGetAll(key);
      if (Object.keys(data).length > 0) {
        etas.push(this.createEtaFromRedisData(data));
      }
    }

    return etas;
  }

  async getEtasForStopId(stopId: string): Promise<IEta[]> {
    const keys = await this.redisClient.keys(`${BaseRedisETARepository.ETA_KEY_PREFIX}*`);
    const etas: IEta[] = [];

    for (const key of keys) {
      const data = await this.redisClient.hGetAll(key);
      if (Object.keys(data).length > 0 && data.stopId === stopId) {
        etas.push(this.createEtaFromRedisData(data));
      }
    }

    return etas;
  }

  async getEtaForShuttleAndStopId(shuttleId: string, stopId: string): Promise<IEta | null> {
    const key = this.createEtaKey(shuttleId, stopId);
    const data = await this.redisClient.hGetAll(key);

    if (Object.keys(data).length === 0) {
      return null;
    }

    return this.createEtaFromRedisData(data);
  }

  // Protected setter for internal use
  protected async addOrUpdateEta(eta: IEta): Promise<void> {
    const key = this.createEtaKey(eta.shuttleId, eta.stopId);
    const hash = this.createRedisHashFromEta(eta);
    await this.redisClient.hSet(key, hash);
    this.emit(ETARepositoryEvent.ETA_UPDATED, eta);
  }

  // EventEmitter override methods for type safety
  override on<T extends ETARepositoryEventName>(
    event: T,
    listener: ETARepositoryEventListener<T>,
  ): this;
  override on(event: string | symbol, listener: (...args: any[]) => void): this {
    return super.on(event, listener);
  }

  override once<T extends ETARepositoryEventName>(
    event: T,
    listener: ETARepositoryEventListener<T>,
  ): this;
  override once(event: string | symbol, listener: (...args: any[]) => void): this {
    return super.once(event, listener);
  }

  override off<T extends ETARepositoryEventName>(
    event: T,
    listener: ETARepositoryEventListener<T>,
  ): this;
  override off(event: string | symbol, listener: (...args: any[]) => void): this {
    return super.off(event, listener);
  }

  override addListener<T extends ETARepositoryEventName>(
    event: T,
    listener: ETARepositoryEventListener<T>,
  ): this;
  override addListener(event: string | symbol, listener: (...args: any[]) => void): this {
    return super.addListener(event, listener);
  }

  override removeListener<T extends ETARepositoryEventName>(
    event: T,
    listener: ETARepositoryEventListener<T>,
  ): this;
  override removeListener(event: string | symbol, listener: (...args: any[]) => void): this {
    return super.removeListener(event, listener);
  }

  override removeAllListeners(eventName?: string | symbol): this {
    return super.removeAllListeners(eventName);
  }
}