smoldot-nextjs.ts

import { startFromWorker } from "polkadot-api/smoldot/from-node-worker";
import { Worker } from "worker_threads";
import { fileURLToPath } from "url";
import { dirname, join } from "path";
import type { Client } from "polkadot-api/smoldot";

// Singleton worker instance and nonce cache
let worker: Worker | null = null;
let smoldot: Client | null = null;
const nonceCache: Map<string, number> = new Map();

// Initialize worker once
async function initializeWorker() {
  if (worker && smoldot) return { worker, smoldot };

  const currentFilePath = fileURLToPath(import.meta.url);
  const currentDir = dirname(currentFilePath);
  const workerPath = join(
    currentDir,
    "..",
    "node_modules",
    "polkadot-api",
    "dist",
    "reexports",
    "smoldot_node-worker.js"
  );

  worker = new Worker(workerPath);
  smoldot = startFromWorker(worker);

  return { worker, smoldot };
}

// Cleanup function to be called on server shutdown
export async function cleanupWorker() {
  if (smoldot) {
    await smoldot.terminate();
    smoldot = null;
  }
  if (worker) {
    worker.terminate();
    worker = null;
  }
}