corrosion-admin-panel/backend-nest/src/services/nats-bridge.service.ts

import { Injectable, OnModuleInit, Logger } from '@nestjs/common';
import { NatsService } from './nats.service';

@Injectable()
export class NatsBridgeService implements OnModuleInit {
  private readonly logger = new Logger(NatsBridgeService.name);
  private listeners: Map<string, Set<(event: string, data: unknown) => void>> = new Map();

  constructor(private nats: NatsService) {}

  onModuleInit() {
    this.nats.subscribe('corrosion.*.companion.heartbeat', (data, subject) => {
      const licenseId = subject.split('.')[1];
      this.emit(licenseId, 'heartbeat', data);
    });

    this.nats.subscribe('corrosion.*.console.output', (data, subject) => {
      const licenseId = subject.split('.')[1];
      this.emit(licenseId, 'console_output', data);
    });

    this.nats.subscribe('corrosion.*.files.response', (data, subject) => {
      const licenseId = subject.split('.')[1];
      this.emit(licenseId, 'files_response', data);
    });

    this.nats.subscribe('corrosion.*.wipe.status', (data, subject) => {
      const licenseId = subject.split('.')[1];
      this.emit(licenseId, 'wipe_status', data);
    });

    this.nats.subscribe('corrosion.*.server.status', (data, subject) => {
      const licenseId = subject.split('.')[1];
      this.emit(licenseId, 'server_status', data);
    });

    this.nats.subscribe('corrosion.*.deploy.status', (data, subject) => {
      const licenseId = subject.split('.')[1];
      this.emit(licenseId, 'deploy_status', data);
    });

    this.nats.subscribe('corrosion.*.oxide.status', (data, subject) => {
      const licenseId = subject.split('.')[1];
      this.emit(licenseId, 'oxide_status', data);
    });

    this.logger.log('NATS bridge subscriptions initialized');
  }

  addListener(licenseId: string, callback: (event: string, data: unknown) => void): void {
    if (!this.listeners.has(licenseId)) {
      this.listeners.set(licenseId, new Set());
    }
    this.listeners.get(licenseId)!.add(callback);
  }

  removeListener(licenseId: string, callback: (event: string, data: unknown) => void): void {
    this.listeners.get(licenseId)?.delete(callback);
  }

  private emit(licenseId: string, event: string, data: unknown): void {
    const callbacks = this.listeners.get(licenseId);
    if (callbacks) {
      for (const cb of callbacks) {
        cb(event, data);
      }
    }
  }
}