FIXME REORDER COMMIT -- Change main so it can parse the ast
FIXME THIS COMMIT NEEDS TO BE REORDERED FIXME THIS COMMIT NEEDS TO BE REORDERED FIXME THIS COMMIT NEEDS TO BE REORDERED FIXME THIS COMMIT NEEDS TO BE REORDERED
This commit is contained in:
83
src/main.c
83
src/main.c
@ -1,5 +1,6 @@
|
||||
#include "error.h"
|
||||
#include "lexer.h"
|
||||
#include "parser.h"
|
||||
#include "tokenlist.h"
|
||||
|
||||
#include <limits.h>
|
||||
@ -7,38 +8,57 @@
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
|
||||
bool print_token(lexer_token_t *token) {
|
||||
lexer_token_print(token);
|
||||
return true;
|
||||
typedef enum mode { MODE_AST, MODE_TEXT, MODE_TOKENS } mode_t;
|
||||
|
||||
void print_tokens(tokenlist_t *list) {
|
||||
for (auto entry = list->head; entry; entry = entry->next) {
|
||||
auto token = &entry->token;
|
||||
lexer_token_print(token);
|
||||
}
|
||||
}
|
||||
|
||||
bool print_value(lexer_token_t *token) {
|
||||
if (token->id == TOKEN_ERROR) {
|
||||
printf("%s\n", token->value);
|
||||
for (size_t i = 0; i < token->character_number; ++i)
|
||||
printf(" ");
|
||||
printf("^-- %s\n", token->explanation);
|
||||
} else {
|
||||
printf("%s", token->value);
|
||||
void print_text(tokenlist_t *list) {
|
||||
for (auto entry = list->head; entry; entry = entry->next) {
|
||||
auto token = &entry->token;
|
||||
if (token->id == TOKEN_ERROR) {
|
||||
printf("%s\n", token->value);
|
||||
for (size_t i = 0; i < token->character_number; ++i)
|
||||
printf(" ");
|
||||
printf("^-- %s\n", token->explanation);
|
||||
return;
|
||||
} else {
|
||||
printf("%s", token->value);
|
||||
}
|
||||
}
|
||||
return token->id != TOKEN_ERROR;
|
||||
}
|
||||
|
||||
void parse_ast(tokenlist_t *list) {
|
||||
parse_result_t result = parse(list->head);
|
||||
if (result.err) {
|
||||
puts(result.err->message);
|
||||
error_free(result.err);
|
||||
return;
|
||||
}
|
||||
ast_node_free(result.node);
|
||||
}
|
||||
|
||||
int get_execution_mode(int argc, char *argv[]) {
|
||||
if (argc != 3 || (strcmp(argv[1], "tokens") != 0 &&
|
||||
strcmp(argv[1], "text") != 0 && strcmp(argv[1], "ast"))) {
|
||||
puts("Usage: oas [tokens|text|ast] <filename>");
|
||||
exit(1);
|
||||
}
|
||||
|
||||
if (strcmp(argv[1], "tokens") == 0)
|
||||
return MODE_TOKENS;
|
||||
if (strcmp(argv[1], "text") == 0)
|
||||
return MODE_TEXT;
|
||||
return MODE_AST;
|
||||
}
|
||||
|
||||
int main(int argc, char *argv[]) {
|
||||
if (argc != 3 ||
|
||||
(strcmp(argv[1], "-tokens") != 0 && strcmp(argv[1], "-text") != 0)) {
|
||||
puts("Usage: oas -tokens <filename>");
|
||||
puts("Usage: oas -text <filename>");
|
||||
return 1;
|
||||
}
|
||||
|
||||
bool (*print_fn)(lexer_token_t *);
|
||||
mode_t mode = get_execution_mode(argc, argv);
|
||||
char *filename = argv[2];
|
||||
if (strcmp(argv[1], "-tokens") == 0) {
|
||||
print_fn = print_token;
|
||||
} else {
|
||||
print_fn = print_value;
|
||||
}
|
||||
|
||||
lexer_t *lex = &(lexer_t){};
|
||||
error_t *err = lexer_open(lex, filename);
|
||||
@ -54,9 +74,18 @@ int main(int argc, char *argv[]) {
|
||||
if (err)
|
||||
goto cleanup_tokens;
|
||||
|
||||
for (auto entry = list->head; entry; entry = entry->next) {
|
||||
print_fn(&entry->token);
|
||||
switch (mode) {
|
||||
case MODE_TOKENS:
|
||||
print_tokens(list);
|
||||
break;
|
||||
case MODE_TEXT:
|
||||
print_text(list);
|
||||
break;
|
||||
case MODE_AST:
|
||||
parse_ast(list);
|
||||
break;
|
||||
}
|
||||
|
||||
tokenlist_free(list);
|
||||
error_free(err);
|
||||
return 0;
|
||||
|
Reference in New Issue
Block a user