import { createClient, ConnectError, Code } from '@connectrpc/connect';
import { createGrpcWebTransport } from '@connectrpc/connect-web';
import { ElizaService } from '@buf/connectrpc_eliza.connectrpc_es/eliza_connect.js';
import type { IntroduceRequest } from '@buf/connectrpc_eliza.connectrpc_es/eliza_pb.js';

// Client setup
const transport = createGrpcWebTransport({
  baseUrl: 'https://demo.connectrpc.com',
});

const client = createClient(ElizaService, transport);

// Function to call the service and cancel it
async function callAndCancel() {
  const controller = new AbortController();

  // Schedule cancellation
  setTimeout(() => {
    console.log('Aborting request...');
    controller.abort();
  }, 500);

  try {
    const request: IntroduceRequest = { name: 'User' };
    const stream = client.introduce(request, { signal: controller.signal });

    for await (const response of stream) {
      console.log('Received response:', response.sentence);
    }
    console.log('Stream finished without error.');
  } catch (e) {
    if (e instanceof ConnectError) {
      console.error('Caught a ConnectError:');
      console.error(' > Message:', e.message);
      console.error(' > Code:', e.code); // This is 20
      console.error(' > Is Code.Canceled?', e.code === Code.Canceled); // This is false
    } else {
      console.error('Caught an unexpected error:', e);
    }
  }
}

callAndCancel();