77 lines
1.7 KiB
TypeScript
77 lines
1.7 KiB
TypeScript
import { LavaSDK } from "@lavanet/lava-sdk";
|
|
import type { ActiveQuery } from "@lumeweb/libkernel/module";
|
|
import {
|
|
addHandler,
|
|
defer,
|
|
getKey,
|
|
handlePresentKey as handlePresentKeyModule,
|
|
} from "@lumeweb/libkernel/module";
|
|
import {
|
|
createClient as createRpcClient,
|
|
RpcNetwork,
|
|
} from "@lumeweb/kernel-rpc-client";
|
|
import { bytesToHex, deriveChildKey } from "@lumeweb/libweb";
|
|
import { hyperTransport } from "./transport.js";
|
|
|
|
addHandler("presentKey", handlePresentKey);
|
|
addHandler("query", handleQuery);
|
|
|
|
const chainInstances = new Map<string, LavaSDK>();
|
|
const moduleReadyDefer = defer();
|
|
let rpc: RpcNetwork;
|
|
|
|
async function handlePresentKey(aq: ActiveQuery) {
|
|
handlePresentKeyModule({
|
|
callerInput: {
|
|
key: aq.callerInput.rootKey,
|
|
},
|
|
} as ActiveQuery);
|
|
rpc = createRpcClient();
|
|
moduleReadyDefer.resolve();
|
|
}
|
|
|
|
async function handleQuery(aq: ActiveQuery) {
|
|
if (!("chain" in aq.callerInput)) {
|
|
aq.reject("chain missing");
|
|
return;
|
|
}
|
|
if (!("query" in aq.callerInput)) {
|
|
aq.reject("query missing");
|
|
return;
|
|
}
|
|
|
|
const { chain, query, rpcInterface = undefined } = aq.callerInput;
|
|
|
|
let lava: LavaSDK;
|
|
|
|
if (!chainInstances.has(chain)) {
|
|
lava = await setupRelayChain(chain, rpcInterface);
|
|
} else {
|
|
lava = chainInstances.get(chain) as LavaSDK;
|
|
}
|
|
|
|
await rpc.ready;
|
|
|
|
try {
|
|
aq.respond(lava.sendRelay(query));
|
|
} catch (e) {
|
|
aq.reject(e);
|
|
}
|
|
}
|
|
|
|
async function setupRelayChain(chain: string, rpcInterface?: string) {
|
|
const instance = await LavaSDK.create({
|
|
chainID: chain,
|
|
privateKey: bytesToHex(deriveChildKey(await getKey(), "lavanet")),
|
|
badge: {
|
|
// @ts-ignore
|
|
transport: hyperTransport(rpc),
|
|
},
|
|
rpcInterface,
|
|
});
|
|
|
|
chainInstances.set(chain, instance);
|
|
|
|
return instance;
|
|
}
|