import type { Stream } from "@libp2p/interface-connection";
import type { PeerId } from "@libp2p/interface-peer-id";
import type { Multiaddr } from "@multiformats/multiaddr";
import type { Libp2p } from "libp2p";

import type { Protocols } from "./waku";
import type { WakuFilter } from "./waku_filter";
import type { WakuLightPush } from "./waku_light_push";
import type { DecryptionMethod } from "./waku_message";
import type { WakuRelay } from "./waku_relay";
import type { WakuStore } from "./waku_store";

export interface Waku {
  libp2p: Libp2p;
  relay?: WakuRelay;
  store?: WakuStore;
  filter?: WakuFilter;
  lightPush?: WakuLightPush;

  dial(peer: PeerId | Multiaddr, protocols?: Protocols[]): Promise<Stream>;

  addPeerToAddressBook(
    peerId: PeerId | string,
    multiaddrs: Multiaddr[] | string[]
  ): void;

  start(): Promise<void>;

  stop(): Promise<void>;

  isStarted(): boolean;

  addDecryptionKey(
    key: Uint8Array | string,
    options?: { method?: DecryptionMethod; contentTopics?: string[] }
  ): void;

  deleteDecryptionKey(key: Uint8Array | string): void;
}

export interface WakuLight extends Waku {
  relay: undefined;
  store: WakuStore;
  filter: WakuFilter;
  lightPush: WakuLightPush;
}

export interface WakuPrivacy extends Waku {
  relay: WakuRelay;
  store: undefined;
  filter: undefined;
  lightPush: undefined;
}

export interface WakuFull extends Waku {
  relay: WakuRelay;
  store: WakuStore;
  filter: WakuFilter;
  lightPush: WakuLightPush;
}