more refactoring and error checking

This commit is contained in:
2026-07-21 20:45:06 +02:00
parent 09b552ed8c
commit 3f04adcd6b
3 changed files with 75 additions and 57 deletions
+42 -30
View File
@@ -1,21 +1,21 @@
#include "lisp.h"
#define func_arithmetic_ex(name, operator, operator_name) \
int name(Nodes *args, Scopes *scopes, Node *result) { \
int name(Nodes args, Scopes *scopes, Node *result) { \
UNUSED(scopes); \
if (args->count != 2) { \
if (args.count != 2) { \
puts("'" operator_name "' expects two arguments!\n"); \
return 1; \
} \
if (args->items[0].kind != NODE_KIND_NUMBER || \
args->items[1].kind != NODE_KIND_NUMBER) { \
if (args.items[0].kind != NODE_KIND_NUMBER || \
args.items[1].kind != NODE_KIND_NUMBER) { \
puts("'" operator_name \
"' only works on numbers, you specified something " \
"else!\n"); \
return 1; \
} \
long long a = args->items[0].as_number; \
long long b = args->items[1].as_number; \
long long a = args.items[0].as_number; \
long long b = args.items[1].as_number; \
result->as_number = a operator b; \
result->kind = NODE_KIND_NUMBER; \
return 0; \
@@ -35,48 +35,60 @@ func_arithmetic(lisp_function_gt, >);
func_arithmetic(lisp_function_lt_eq, <=);
func_arithmetic(lisp_function_gt_eq, >=);
int lisp_function_println(Nodes *args, Scopes *scopes, Node *result) {
if (args->count != 1) {
int lisp_function_println(Nodes args, Scopes *scopes, Node *result) {
if (args.count != 1) {
printf("'+' expects one argument!\n");
return 1;
}
eval_one(args->items[0], scopes, result);
eval_one(args.items[0], scopes, result);
node_print(*result);
printf("\n");
return 0;
}
int lisp_function_list(Nodes *args, Scopes *scopes, Node *result) {
int lisp_function_list(Nodes args, Scopes *scopes, Node *result) {
UNUSED(scopes);
Nodes *nodes = malloc(sizeof(Nodes));
if (nodes == NULL) {
printf("Failed to allocate memory, something is very wrong!\n");
return 1;
}
memcpy(nodes, &args, sizeof(Nodes));
result->kind = NODE_KIND_LIST;
result->as_list = args;
result->as_list = nodes;
return 0;
}
int lisp_function_append(Nodes *args, Scopes *scopes, Node *result) {
int lisp_function_append(Nodes args, Scopes *scopes, Node *result) {
UNUSED(scopes);
if (args->count != 2) {
if (args.count != 2) {
printf("'append' expects two arguments!\n");
return 1;
}
if (args->items[0].kind != NODE_KIND_LIST ||
args->items[1].kind != NODE_KIND_LIST) {
if (args.items[0].kind != NODE_KIND_LIST ||
args.items[1].kind != NODE_KIND_LIST) {
printf("'append' can only append two lists!\n");
return 1;
}
Nodes *nodes = malloc(sizeof(Nodes));
if (nodes == NULL) {
printf("Failed to allocate memory, something is very wrong!\n");
return 1;
}
memset(nodes, 0, sizeof(Nodes));
Nodes *a = args->items[0].as_list;
Nodes *a = args.items[0].as_list;
nob_da_append_many(nodes, a->items, a->count);
Nodes *b = args->items[1].as_list;
Nodes *b = args.items[1].as_list;
nob_da_append_many(nodes, b->items, b->count);
result->as_list = nodes;
@@ -84,55 +96,55 @@ int lisp_function_append(Nodes *args, Scopes *scopes, Node *result) {
return 0;
}
int lisp_function_length(Nodes *args, Scopes *scopes, Node *result) {
int lisp_function_length(Nodes args, Scopes *scopes, Node *result) {
UNUSED(scopes);
if (args->count != 1) {
if (args.count != 1) {
printf("'length' expects one argument!\n");
return 1;
}
if (args->items[0].kind != NODE_KIND_LIST) {
if (args.items[0].kind != NODE_KIND_LIST) {
printf("'length' can only be used on lists!\n");
return 1;
}
result->kind = NODE_KIND_NUMBER;
result->as_number = args->items[0].as_list->count;
result->as_number = args.items[0].as_list->count;
return 0;
}
int lisp_function_isnull(Nodes *args, Scopes *scopes, Node *result) {
int lisp_function_isnull(Nodes args, Scopes *scopes, Node *result) {
UNUSED(scopes);
if (args->count != 1) {
if (args.count != 1) {
printf("'null?' expects one argument!\n");
return 1;
}
if (args->items[0].kind != NODE_KIND_LIST) {
if (args.items[0].kind != NODE_KIND_LIST) {
printf("'null?' can only be used on lists!\n");
return 1;
}
result->kind = NODE_KIND_NUMBER;
result->as_number = args->items[0].as_list->count == 0 ? 1 : 0;
result->as_number = args.items[0].as_list->count == 0 ? 1 : 0;
return 0;
}
int lisp_function_builtins(Nodes *args, Scopes *scopes, Node *result) {
if (args->count != 0) {
int lisp_function_builtins(Nodes args, Scopes *scopes, Node *result) {
if (args.count != 0) {
printf("'builtins' expects no arguments!\n");
return 1;
}
Scope *global_scope = nob_da_first(scopes);
Nodes *nodes = malloc(sizeof(Node));
Nodes nodes = {0};
nob_da_foreach(ScopeObject, scope_object, global_scope) {
nob_da_append(nodes, scope_object->node);
nob_da_append(&nodes, scope_object->node);
}
result->as_list = nodes;
*result->as_list = nodes;
result->kind = NODE_KIND_LIST;
return 0;
}
+14 -10
View File
@@ -164,6 +164,8 @@ int parse_one(Tokens *tokens, Node *node) {
return 1;
}
// equivalent to queue.poll
// TODO: implement a 'da_poll' macro
Token tok = tokens->items[0];
tokens->items++;
tokens->count--;
@@ -171,6 +173,10 @@ int parse_one(Tokens *tokens, Node *node) {
switch (tok.kind) {
case TOKEN_KIND_PAR_LEFT: {
Nodes *nodes = malloc(sizeof(Nodes));
if (nodes == NULL) {
printf("Failed to allocate memory, something is very wrong!\n");
return 1;
}
memset(nodes, 0, sizeof(Nodes));
while (true) {
@@ -182,13 +188,13 @@ int parse_one(Tokens *tokens, Node *node) {
if (tokens->items[0].kind == TOKEN_KIND_PAR_RIGHT) {
tokens->items++;
tokens->count--;
if (tokens->count > 0) {
}
break;
}
parse_one(tokens, node);
int ret = parse_one(tokens, node);
if (ret != 0)
return ret;
nob_da_append(nodes, *node);
}
@@ -213,6 +219,8 @@ int parse_one(Tokens *tokens, Node *node) {
node->as_symbol = tok.data;
} break;
default: {
// unknown token kind, therefore an error
// should I use UNREACHABLE instead?
return 1;
}
}
@@ -242,10 +250,6 @@ int parse(Tokens *tokens, Nodes *nodes) {
Tokens toks = {.count = sub_size,
.capacity = sub_size,
.items = tokens->items + last_idx};
printf("Parsing:\n");
nob_da_foreach(Token, token, &toks) {
printf(" %s\n", token_kind_name(token->kind));
}
Node result = {0};
int ret = parse_one(&toks, &result);
@@ -336,14 +340,14 @@ int eval_one(Node input, Scopes *scopes, Node *result) {
return 0;
}
Nodes *args = malloc(sizeof(Nodes));
Nodes args = {0};
for (size_t i = 1; i < children->count; i++) {
Node n = {0};
int ret = eval_one(children->items[i], scopes, &n);
if (ret != 0) {
return ret;
}
nob_da_append(args, n);
nob_da_append(&args, n);
}
Node func_node = {0};
+19 -17
View File
@@ -32,7 +32,9 @@ typedef struct s_Node Node;
typedef struct s_Nodes Nodes;
typedef struct s_Scopes Scopes;
typedef int (*LispFunction)(Nodes *args, Scopes *scopes, Node *result);
// args is specifically not a pointer,
// functions shouldn't be able to mess with their arguments
typedef int (*LispFunction)(Nodes args, Scopes *scopes, Node *result);
struct s_Node {
NodeKind kind;
@@ -84,23 +86,23 @@ int eval_one(Node input, Scopes *scopes, Node *result);
int eval(Nodes inputs, Scopes *scopes, Nodes *results);
// builtins
int lisp_function_add(Nodes *args, Scopes *scopes, Node *result);
int lisp_function_sub(Nodes *args, Scopes *scope, Node *result);
int lisp_function_div(Nodes *args, Scopes *scope, Node *result);
int lisp_function_mul(Nodes *args, Scopes *scope, Node *result);
int lisp_function_mod(Nodes *args, Scopes *scope, Node *result);
int lisp_function_add(Nodes args, Scopes *scopes, Node *result);
int lisp_function_sub(Nodes args, Scopes *scope, Node *result);
int lisp_function_div(Nodes args, Scopes *scope, Node *result);
int lisp_function_mul(Nodes args, Scopes *scope, Node *result);
int lisp_function_mod(Nodes args, Scopes *scope, Node *result);
int lisp_function_eq(Nodes *args, Scopes *scope, Node *result);
int lisp_function_lt(Nodes *args, Scopes *scope, Node *result);
int lisp_function_gt(Nodes *args, Scopes *scope, Node *result);
int lisp_function_lt_eq(Nodes *args, Scopes *scope, Node *result);
int lisp_function_gt_eq(Nodes *args, Scopes *scope, Node *result);
int lisp_function_eq(Nodes args, Scopes *scope, Node *result);
int lisp_function_lt(Nodes args, Scopes *scope, Node *result);
int lisp_function_gt(Nodes args, Scopes *scope, Node *result);
int lisp_function_lt_eq(Nodes args, Scopes *scope, Node *result);
int lisp_function_gt_eq(Nodes args, Scopes *scope, Node *result);
int lisp_function_println(Nodes *args, Scopes *scopes, Node *result);
int lisp_function_list(Nodes *args, Scopes *scopes, Node *result);
int lisp_function_append(Nodes *args, Scopes *scope, Node *result);
int lisp_function_length(Nodes *args, Scopes *scope, Node *result);
int lisp_function_isnull(Nodes *args, Scopes *scope, Node *result);
int lisp_function_builtins(Nodes *args, Scopes *scopes, Node *result);
int lisp_function_println(Nodes args, Scopes *scopes, Node *result);
int lisp_function_list(Nodes args, Scopes *scopes, Node *result);
int lisp_function_append(Nodes args, Scopes *scope, Node *result);
int lisp_function_length(Nodes args, Scopes *scope, Node *result);
int lisp_function_isnull(Nodes args, Scopes *scope, Node *result);
int lisp_function_builtins(Nodes args, Scopes *scopes, Node *result);
#endif // LISP_H