import { createClient } from "@connectrpc/connect";
import { createGrpcTransport } from "@connectrpc/connect-node";
import { SymbioticAPIService } from "@symbioticfi/relay-client-ts";
import {
  GetLastAllCommittedRequestSchema,
  SignMessageRequestSchema,
  ListenProofsRequestSchema,
} from "@symbioticfi/relay-client-ts";
import { create } from "@bufbuild/protobuf";

async function main() {
    const transport = createGrpcTransport({
        baseUrl: "http://localhost:8080",
    });
    const relayClient = createClient(SymbioticAPIService, transport);
    
    const getLastAllCommittedRequest = create(GetLastAllCommittedRequestSchema);
    const getLastAllCommittedResponse = await relayClient.getLastAllCommitted(getLastAllCommittedRequest);
    const epochInfos = getLastAllCommittedResponse.epochInfos;
    let suggestedEpoch = 0;
    for (const epochInfo of epochInfos) {
        if (suggestedEpoch === 0 || epochInfo.lastCommittedEpoch < suggestedEpoch) {
            suggestedEpoch = epochInfo.lastCommittedEpoch;
        }
    }

    const signMessageRequest = create(SignMessageRequestSchema, {
        keyTag: 15, // default key tag - BN254
        message: message.into(),
        requiredEpoch: suggestedEpoch,
    });
    const signMessageResponse = await relayClient.signMessage(signMessageRequest);

    const listenProofsRequest = create(ListenProofsRequestSchema, { startEpoch: suggestedEpoch });
    const proofsStream = await relayClient.listenProofs(listenProofsRequest);
    let aggregationProof;
    for await (const proofResponse of proofsStream) {
        if (proofResponse.requestId === signMessageResponse.requestHash) {
            aggregationProof = proofResponse.aggregationProof?.proof;
            break;
        }
    }

    await appContract.completeTask(taskId, signMessageResponse.epoch, aggregationProof);
}