oreodave/alisp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

stream: a few further refactors due to new sv functions

Browse Source
This commit is contained in:
Aryadev Chavali
2026-02-09 07:48:58 +00:00
committed by oreodave
parent 8426f04734
commit 2855536bdd
1 changed files with 20 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
src/stream.c
View File

@@ -162,23 +162,21 @@ bool stream_eoc(stream_t *stream)
bool stream_chunk(stream_t *stream) bool stream_chunk(stream_t *stream)
{ {
assert(stream); assert(stream);
u64 to_read = STREAM_DEFAULT_CHUNK;
switch (stream->type) switch (stream->type)
{ {
case STREAM_TYPE_STRING: case STREAM_TYPE_STRING:
// vacuously true // nothing to chunk, hence false
return true; return false;
case STREAM_TYPE_PIPE: case STREAM_TYPE_PIPE:
to_read = 1;
// fallthrough // fallthrough
case STREAM_TYPE_FILE: case STREAM_TYPE_FILE:
{ {
if (feof(stream->pipe.file)) if (feof(stream->pipe.file))
// We can't read anymore. End of the line // We can't read anymore. End of the line
return false; return false;
vec_ensure_free(&stream->pipe.cache, to_read); vec_ensure_free(&stream->pipe.cache, STREAM_DEFAULT_CHUNK);
int read = fread(vec_data(&stream->pipe.cache) + stream->pipe.cache.size, 1, int read = fread(vec_data(&stream->pipe.cache) + stream->pipe.cache.size, 1,
to_read, stream->pipe.file); STREAM_DEFAULT_CHUNK, stream->pipe.file);
// If we read something it's a good thing // If we read something it's a good thing
if (read > 0) if (read > 0)
@@ -219,8 +217,7 @@ char stream_peek(stream_t *stream)
// Cached already? We are done. // Cached already? We are done.
if (stream->position < stream->pipe.cache.size) if (stream->position < stream->pipe.cache.size)
{ {
const char *const str = (char *)vec_data(&stream->pipe.cache); return stream_sv(stream).data[0];
return str[stream->position];
} }
// Try to read chunks in till we've reached it or we're at the end of the // Try to read chunks in till we've reached it or we're at the end of the
@@ -233,7 +230,7 @@ char stream_peek(stream_t *stream)
// Same principle as the stream_eos(stream) check. // Same principle as the stream_eos(stream) check.
if (stream->position >= stream->pipe.cache.size) if (stream->position >= stream->pipe.cache.size)
return '\0'; return '\0';
return ((char *)vec_data(&stream->pipe.cache))[stream->position]; return stream_sv(stream).data[0];
} }
default: default:
FAIL("Unreachable"); FAIL("Unreachable");
@@ -248,8 +245,9 @@ u64 stream_seek(stream_t *stream, i64 offset)
else if (offset > 0) else if (offset > 0)
return stream_seek_forward(stream, offset); return stream_seek_forward(stream, offset);
else else
// vacuously successful {
return true; return 0;
}
} }
u64 stream_seek_forward(stream_t *stream, u64 offset) u64 stream_seek_forward(stream_t *stream, u64 offset)
@@ -358,31 +356,21 @@ sv_t stream_substr(stream_t *stream, u64 size)
} }
sv_t stream_substr_abs(stream_t *stream, u64 index, u64 size) sv_t stream_substr_abs(stream_t *stream, u64 index, u64 size)
{
switch (stream->type)
{
case STREAM_TYPE_STRING:
return sv_truncate(sv_chop_left(stream_sv_abs(stream), index), size);
case STREAM_TYPE_PIPE:
case STREAM_TYPE_FILE:
{ {
if (index + size > stream_size(stream)) if (index + size > stream_size(stream))
{
// => try reading chunks till either we drop or we have enough space // => try reading chunks till either we drop or we have enough space
for (bool read_chunk = stream_chunk(stream); for (bool read_chunk = stream_chunk(stream);
read_chunk && index + size >= stream->pipe.cache.size; read_chunk && index + size >= stream->pipe.cache.size;
read_chunk = stream_chunk(stream)) read_chunk = stream_chunk(stream))
continue; continue;
}
sv_t sv = stream_sv_abs(stream); sv_t sv = stream_sv_abs(stream);
sv = sv_chop_left(sv, index); sv = sv_chop_left(sv, index);
sv = sv_truncate(sv, size); sv = sv_truncate(sv, size);
return sv; return sv;
} }
default:
assert("Unreachable");
return SV(NULL, 0);
}
}
sv_t stream_till(stream_t *stream, const char *str) sv_t stream_till(stream_t *stream, const char *str)
{ {