import type { ConnectionController } from '../ConnectionController';
import type { IncomingMessage } from '../../messages/IncomingMessage';
import type { OutgoingMessage } from '../../messages/OutgoingMessage';
import type { BinaryMessage } from '../../messages/BinaryMessage';

export abstract class Connection {

  public headers!: Record<string, string | string[] | undefined>;

  private connectionController!: ConnectionController;

  public setConnectionController(cc: ConnectionController) {
    this.connectionController = cc;
  }

  public abstract reply(message: OutgoingMessage): Promise<void>;

  public abstract send(message: OutgoingMessage): Promise<void>;

  public abstract getSocket(): any;

  public abstract close(data?: string): Promise<void>;

  public getStore(): Record<string, any> {
    const socket = this.getSocket() as any;

    if (!socket.boogerSessionUserStore) {
      socket.boogerSessionUserStore = {};
    }

    return socket.boogerSessionUserStore;
  }

  protected async onMessage(message: IncomingMessage): Promise<void> {
    await this.connectionController.onMessage(message);
  }

  protected async onBinaryMessage(message: BinaryMessage): Promise<void> {
    await this.connectionController.onBinaryMessage(message);
  }

  protected async onClose() {
    await this.connectionController.onClose();
  }

}