import type { Ws } from "../../packages";
import type {
  NContextService,
  NDiscoveryService,
  NSchemeService,
} from "../services";

export interface IAbstractWsAdapter {
  start(): Promise<void>;
  stop(): Promise<void>;

  send(connectionId: string, type: any, payload: any): void;
  broadcast(connectionId: string[], type: any, payload: any): void;
}

export namespace NAbstractWsAdapter {
  export type WsKind = "ws";

  export type AllEventType =
    | "handshake"
    | "handshake.error"
    | "validation.error.unknown_event"
    | "validation.error.unknown_event_kind"
    | "validation.error.invalid_data_structure"
    | "validation.error.service_not_found"
    | "validation.error.domain_not_found"
    | "validation.error.event_not_found"
    | "session:to:session"
    | "session:to:room"
    | "session:to:service";

  export type EventKind = "handshake" | "validation" | "communication";

  export type HandshakePayload = {
    code: string;
    message: string;
  };

  export type BaseCommunicationPayload = {
    service: string;
    domain: string;
    version: string;
    event: string;
    language?: string;
    data: any;
  };

  export type EventKindStructure<K extends EventKind> = K extends "handshake"
    ? HandshakePayload
    : K extends "validation"
    ? HandshakePayload
    : K extends "communication"
    ? any
    : never;

  export type ClientEventStructure<E extends EventKind> = {
    event: AllEventType;
    kind: E;
    payload: EventKindStructure<E>;
  };

  export type Instance<K extends WsKind> = K extends "ws"
    ? Ws.WebSocketServer
    : never;

  export type Config = Pick<
    NDiscoveryService.CoreConfig["adapters"]["ws"],
    "enable" | "protocol" | "host" | "port" | "kind"
  > &
    Pick<NDiscoveryService.CoreConfig["adapters"], "serverTag">;

  export type Context<
    USER_INFO = any,
    SYSTEM_INFO = any,
    AUTH_SCOPE extends NSchemeService.AuthScope = NSchemeService.AuthScope
  > = {
    store: NContextService.EventStore;
  } & (AUTH_SCOPE extends "public:route"
    ? {}
    : AUTH_SCOPE extends "private:user"
    ? {
        user: USER_INFO;
      }
    : AUTH_SCOPE extends "private:system"
    ? {
        user: USER_INFO;
        system: SYSTEM_INFO;
      }
    : never);

  export type Handler = (
    payload: any,
    agents: NSchemeService.Agents,
    context: NAbstractWsAdapter.Context
  ) => Promise<void>;
}