main: command line arguments for filename, and allow stdin parsing
- Now take a single command line argument for the filename to read and compile. - If filename is "--", then read stdin until EOF using a different read handler (using ~vec_t~ along with buffered reading).
This commit is contained in:
65
src/main.c
65
src/main.c
@@ -35,12 +35,71 @@ int read_file(const char *filename, sv_t *ret)
|
|||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
int main(void)
|
int read_pipe(FILE *pipe, sv_t *ret)
|
||||||
|
{
|
||||||
|
// Keep reading in chunks, keeping it altogether in a vector
|
||||||
|
vec_t contents = {0};
|
||||||
|
char buffer[1024];
|
||||||
|
while (!feof(pipe))
|
||||||
|
{
|
||||||
|
size_t bytes_read = fread(buffer, 1, sizeof(buffer), pipe);
|
||||||
|
vec_append(&contents, buffer, bytes_read);
|
||||||
|
}
|
||||||
|
if (contents.size == 0)
|
||||||
|
return 1;
|
||||||
|
|
||||||
|
ret->size = contents.size;
|
||||||
|
vec_append_byte(&contents, '\0');
|
||||||
|
// NOTE: vec_data(&contents) may be stack allocated; let's create a copy if
|
||||||
|
// that's the case.
|
||||||
|
if (contents.not_inlined)
|
||||||
|
{
|
||||||
|
ret->data = vec_data(&contents);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
ret->data = calloc(1, contents.size);
|
||||||
|
memcpy(ret->data, vec_data(&contents), contents.size);
|
||||||
|
}
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
void usage(FILE *fp)
|
||||||
|
{
|
||||||
|
fprintf(fp, "Usage: arl [FILE]\n"
|
||||||
|
"Compiles [FILE] as ARL source code.\n"
|
||||||
|
" [FILE]: File to compile.\n"
|
||||||
|
"If FILE is \"--\", then read from stdin.\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
int main(int argc, char *argv[])
|
||||||
{
|
{
|
||||||
int ret = 0;
|
int ret = 0;
|
||||||
const char *filename = "./examples/hello-world.arl";
|
char *filename = "";
|
||||||
|
if (argc == 1)
|
||||||
|
{
|
||||||
|
usage(stderr);
|
||||||
|
ret = 1;
|
||||||
|
goto end;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
filename = argv[1];
|
||||||
|
}
|
||||||
|
|
||||||
|
int read_err = 0;
|
||||||
sv_t contents = {0};
|
sv_t contents = {0};
|
||||||
if (read_file(filename, &contents))
|
if (strcmp(filename, "--") == 0)
|
||||||
|
{
|
||||||
|
filename = "stdin";
|
||||||
|
read_err = read_pipe(stdin, &contents);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
read_err = read_file(filename, &contents);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (read_err)
|
||||||
{
|
{
|
||||||
LOG_ERR("ERROR: Reading `%s`: ", filename);
|
LOG_ERR("ERROR: Reading `%s`: ", filename);
|
||||||
perror("");
|
perror("");
|
||||||
|
|||||||
Reference in New Issue
Block a user