package MY::A;

use strict;
use warnings;

use IO::Async::Loop;
use Future;
use Future::AsyncAwait;
use Net::Async::WebSocket::Client;

our $VERSION = '0.01';

sub new {
	my ($class, $params) = @_;
	$params //= {};
	my $self = {
		'max-hellos' => (exists($params->{'max-hellos'})&&defined($params->{'max-hellos'}))
			? $params->{'max-hellos'} : 4,
		'websocket-uri' => (exists($params->{'websocket-uri'})&&defined($params->{'websocket-uri'}))
			? $params->{'websocket-uri'} : 'wss://echo.websocket.org',
	};
	bless $self => $class;
	return $self;
}

async sub fetch {
	my ($self, $params) = @_;
	my $URI = $self->{'websocket-uri'};
	my $MAX_HELLOS = $self->{'max-hellos'};

	my $request = Protocol::WebSocket::Request->new(
		host => $URI,
		resource_name => __PACKAGE__
	);

	my $loop = IO::Async::Loop->new;
	my @data_received;
	my $num_items_received = 0;
	my $client = Net::Async::WebSocket::Client->new(
	  on_text_frame => sub {
		my ( $clientself, $frame ) = @_;
		return unless $frame =~ /hello/;
		$num_items_received++;
		push @data_received, $frame;
		print __PACKAGE__." : on_text_frame() : got # ${num_items_received}: ${frame}\n";
		# the 1st item is sent from the websocket like 'Request served by 4d896d95b55478'
		# so we will save this but don't count it in the below test
		if( $num_items_received >= $MAX_HELLOS ){
			print __PACKAGE__." : on_text_frame() : CLOSING client $clientself.\n";
			# we have received enough items,
			# stop the loop, is this the right way to do it?
			$loop->stop;
			$clientself->close_when_empty;
			# even if we stopped the loop,
			# data is kept added unless we return() here
			#return;
		}
	  },
	);
	print __PACKAGE__." : adding client $client to the loop ...\n";
	$loop->add( $client );

	await eval { $client->connect(
		url => $URI,
	) };
	if( $@ ){ print STDERR __PACKAGE__." : failed to connect to URI '${URI}'.\n"; return undef }
	# NOTE: $client will CLOSE after 4 items received
	# here we send 8 items, $client does not seem to be closed,
	# neither does $loop!
	for my $i (1..$MAX_HELLOS){
		my $ts = "hello from ".__PACKAGE__." #${i}";
		print __PACKAGE__." : sending data to socket : $ts\n";
		$client->send_text_frame($ts);
		sleep(1);
	}

	$loop->run;
	#$client->close_when_empty if $client;

	#my $retdata = Future->done(\@data_received);
	#return $retdata;
	return \@data_received;
}
1;