c3c/lib/std/io/stream/limitreader.c3

module std::io;

struct LimitReader (InStream)
{
	InStream wrapped_stream;
	usz limit;
}

<*
 @param [&inout] wrapped_stream : "The stream to read from"
 @param limit : "The max limit to read"
*>
fn LimitReader* LimitReader.init(&self, InStream wrapped_stream, usz limit)
{
	*self = { .wrapped_stream = wrapped_stream, .limit = limit };
	return self;
}

fn void? LimitReader.close(&self) @dynamic
{
	if (&self.wrapped_stream.close) return self.wrapped_stream.close();
}


fn usz? LimitReader.read(&self, char[] bytes) @dynamic
{
	if (self.limit == 0) return io::EOF~;
	usz m = min(bytes.len, self.limit);
	usz n = self.wrapped_stream.read(bytes[:m])!;
	self.limit -= n;
	return n;
}

fn char? LimitReader.read_byte(&self) @dynamic
{
	if (self.limit == 0) return io::EOF~;
	defer try self.limit--;
	return self.wrapped_stream.read_byte();
}

fn ulong? LimitReader.available(&self) @inline @dynamic
{
	return self.limit;
}