import { WebSocketError } from "./error";
import type { WebSocketMessage } from "./message";
import type { WebSocketPeer } from "./peer";

type WSHook<ArgsT extends Array<any> = []> = (
  peer: WebSocketPeer,
  ...args: ArgsT
) => void | Promise<void>;

// type WSGlobalHook<ArgsT extends Array<any> = []> = (
//   ...args: ArgsT
// ) => void | Promise<void>;

export type UserHooks = Partial<WebSocketHooks & AdapterHooks>;

export function defineWebSocketHooks<T extends UserHooks = UserHooks>(
  hooks: T,
): Partial<WebSocketHooks> {
  return hooks;
}

export interface WebSocketHooks {
  /** Catch-all handler */
  $: (name: keyof UserHooks, peer: WebSocketPeer, ...args: any[]) => void;

  /** A message is received */
  message: WSHook<[WebSocketMessage]>;

  /** A socket is opened */
  open: WSHook<[]>;

  /** A socket is closed */
  close: WSHook<[{ code?: number; reason?: string }]>;

  /** An error occurs */
  error: WSHook<[WebSocketError]>;
}

export interface AdapterHooks {
  // Bun
  "bun:message": WSHook<[ws: any, message: any]>;
  "bun:open": WSHook<[ws: any]>;
  "bun:close": WSHook<[ws: any]>;
  "bun:drain": WSHook<[]>;
  "bun:error": WSHook<[ws: any, error: any]>;
  "bun:ping": WSHook<[ws: any, data: any]>;
  "bun:pong": WSHook<[ws: any, data: any]>;

  // Cloudflare
  "cloudflare:accept": WSHook<[]>;
  "cloudflare:message": WSHook<[event: any]>;
  "cloudflare:error": WSHook<[event: any]>;
  "cloudflare:close": WSHook<[event: any]>;

  // Deno
  "deno:open": WSHook<[]>;
  "deno:message": WSHook<[event: any]>;
  "deno:close": WSHook<[]>;
  "deno:error": WSHook<[error: any]>;

  // ws (Node)
  "node:open": WSHook<[]>;
  "node:message": WSHook<[data: any, isBinary: boolean]>;
  "node:close": WSHook<[code: number, reason: Buffer]>;
  "node:error": WSHook<[error: any]>;
  "node:ping": WSHook<[data: Buffer]>;
  "node:pong": WSHook<[data: Buffer]>;
  "node:unexpected-response": WSHook<[req: any, res: any]>;
  "node:upgrade": WSHook<[req: any]>;
  // "node:server-error": WSGlobalHook<[error: any]>;
  // "node:server-listening": WSGlobalHook<[]>;
  // "node:server-close": WSGlobalHook<[]>;
  // "node:server-headers": WSGlobalHook<[headers: any, request: any]>;

  // uws (Node)
  "uws:open": WSHook<[ws: any]>;
  "uws:message": WSHook<[ws: any, message: any, isBinary: boolean]>;
  "uws:close": WSHook<[ws: any, code: number, message: any]>;
  "uws:ping": WSHook<[ws: any, message: any]>;
  "uws:pong": WSHook<[ws: any, message: any]>;
  "uws:drain": WSHook<[ws: any]>;
  "uws:upgrade": WSHook<[res: any, req: any, context: any]>;
  "uws:subscription": WSHook<
    [ws: any, topic: any, newCount: number, oldCount: number]
  >;
}