js-waku/packages/run/web/index.ts

/* eslint-disable no-console */
import { createEncoder } from "@waku/core";
import type { LightNode } from "@waku/interfaces";
import { createLightNode, Protocols } from "@waku/sdk";
import { createRoutingInfo } from "@waku/utils";

declare global {
  interface Window {
    wakuBrowser: WakuBrowser;
  }
}

class WakuBrowser {
  private waku: LightNode | null = null;

  public async createAndStartNode(config: {
    bootstrapPeers: string[];
    networkConfig: { clusterId: number; numShardsInCluster: number };
  }): Promise<{ success: boolean }> {
    console.log("Creating light node...");
    this.waku = await createLightNode({
      defaultBootstrap: false,
      bootstrapPeers: config.bootstrapPeers,
      networkConfig: config.networkConfig
    });

    console.log("Starting node...");
    await this.waku.start();

    // Explicitly dial each bootstrap peer
    console.log("Dialing bootstrap peers...");
    for (const peer of config.bootstrapPeers) {
      try {
        console.log(`Dialing ${peer}...`);
        await this.waku.dial(peer);
        console.log(`Successfully dialed ${peer}`);
      } catch (error) {
        console.warn(`Failed to dial ${peer}:`, error);
      }
    }

    console.log("Waiting for peers...");
    try {
      await this.waku.waitForPeers([Protocols.LightPush], 30000); // 30 second timeout
      console.log("Peers found!");
    } catch (error) {
      console.warn("Timeout waiting for peers, continuing anyway:", error);
      // Continue anyway - we can still try to send messages
    }

    // Check connected peers
    const peers = this.waku.libp2p.getPeers();
    console.log(
      `Connected to ${peers.length} peers:`,
      peers.map((p) => p.toString())
    );

    return { success: true };
  }

  public async sendLightPushMessage(
    contentTopic: string,
    message: string
  ): Promise<{
    success: boolean;
    successCount: number;
    failureCount: number;
  }> {
    if (!this.waku) {
      throw new Error("Waku node not started");
    }

    const networkConfig = {
      clusterId: 0,
      numShardsInCluster: 8
    };

    const routingInfo = createRoutingInfo(networkConfig, { contentTopic });
    const encoder = createEncoder({ contentTopic, routingInfo });

    const result = await this.waku.lightPush.send(encoder, {
      payload: new TextEncoder().encode(message)
    });

    return {
      success: result.successes.length > 0,
      successCount: result.successes.length,
      failureCount: result.failures?.length || 0
    };
  }

  public async stop(): Promise<{ success: boolean }> {
    if (this.waku) {
      await this.waku.stop();
      this.waku = null;
    }
    return { success: true };
  }
}

// Expose to window
window.wakuBrowser = new WakuBrowser();
console.log("WakuBrowser initialized");