From 5742e9af6b74e70ba2e3ec6ec5c9dc02862050b1 Mon Sep 17 00:00:00 2001
From: tovjemam <gitea@deadfish.cz>
Date: Mon, 23 Dec 2024 14:52:28 +0100
Subject: [PATCH] expected tokens vararg

---
 parser.c | 34 +++++++++++++++-------------------
 1 file changed, 15 insertions(+), 19 deletions(-)

diff --git a/parser.c b/parser.c
index d6f9d26..f7a9404 100644
--- a/parser.c
+++ b/parser.c
@@ -1,5 +1,6 @@
 #include <stdlib.h>
 #include <stdio.h>
+#include <stdarg.h>
 #include "parser.h"
 
 /* Vrátí ukazatel na aktuální token */
@@ -42,8 +43,9 @@ static void error_bad_alloc(struct parser *parser) {
     error_printf(&parser->eb, "Out of memory\n");
 }
 
-static void error_expected_one_of(struct parser *parser, const int expected[]) {
+static void error_expected_tokens(struct parser *parser, size_t num_tokens, ...) {
     size_t i;
+    va_list ap;
 
     enum token_type got = get_token(parser)->type;
 
@@ -56,24 +58,22 @@ static void error_expected_one_of(struct parser *parser, const int expected[]) {
     error_set(&parser->eb, ERR_INVALID_FUNCTION);
     error_printf(&parser->eb, "Syntax error - expected ");
 
-    for (i = 0; expected[i] >= 0; ++i) {
+    va_start(ap, num_tokens);
+    for (i = 0; i < num_tokens; ++i) {
+        enum token_type tok = va_arg(ap, enum token_type);
+
         if (i > 0)
             error_printf(&parser->eb, ", ");
 
-        error_printf(&parser->eb, "%s", lex_token_str(expected[i]));
+        error_printf(&parser->eb, "%s", lex_token_str(tok));
     }
+    va_end(ap);
 
     error_printf(&parser->eb, " - but got %s\n", lex_token_str(got));
 
     lex_print_position(&parser->lexer, &parser->eb);
 }
 
-static void error_expected_single(struct parser *parser, enum token_type expected) {
-    static int expected_arr[] = { -1, -1 };
-    expected_arr[0] = expected;
-    error_expected_one_of(parser, expected_arr);
-}
-
 static struct expr_node* parse_subexpression(struct parser *parser);
 int parse_n_expressions(struct parser *parser, struct expr_node **out_nodes, size_t n);
 
@@ -81,7 +81,7 @@ static struct expr_node* parse_bracketed(struct parser *parser) {
     struct expr_node* node;
 
     if (!accept_token(parser, TOK_LEFT_PAREN)) {
-        error_expected_single(parser, TOK_LEFT_PAREN);
+        error_expected_tokens(parser, 1, TOK_LEFT_PAREN);
         return NULL;
     }
 
@@ -89,7 +89,7 @@ static struct expr_node* parse_bracketed(struct parser *parser) {
         return NULL;
 
     if (!accept_token(parser, TOK_RIGHT_PAREN)) {
-        error_expected_single(parser, TOK_RIGHT_PAREN);
+        error_expected_tokens(parser, 1, TOK_RIGHT_PAREN);
         node_free(node);
         return NULL;
     }
@@ -128,7 +128,7 @@ static struct expr_node *parse_base(struct parser *parser) {
         next_token(parser);
 
         if (!accept_token(parser, TOK_LEFT_PAREN)) {
-            error_expected_single(parser, TOK_LEFT_PAREN);
+            error_expected_tokens(parser, 1, TOK_LEFT_PAREN);
             return NULL;
         }
 
@@ -145,7 +145,7 @@ static struct expr_node *parse_base(struct parser *parser) {
         }
 
         if (!accept_token(parser, TOK_RIGHT_PAREN)) {
-            error_expected_single(parser, TOK_RIGHT_PAREN);
+            error_expected_tokens(parser, 1, TOK_RIGHT_PAREN);
             node_free(node);
             return NULL;
         }
@@ -157,11 +157,7 @@ static struct expr_node *parse_base(struct parser *parser) {
         return parse_bracketed(parser);
     }
 
-    {
-        static const int expected[] = { TOK_NUMBER, TOK_VARIABLE, TOK_FUNCTION, TOK_LEFT_PAREN, -1 };
-        error_expected_one_of(parser, expected);
-    }
-
+    error_expected_tokens(parser, 4, TOK_NUMBER, TOK_VARIABLE, TOK_FUNCTION, TOK_LEFT_PAREN);
     return NULL;
 }
 
@@ -317,7 +313,7 @@ int parser_parse_n(struct parser *parser, const char *str, const char *variable_
 
     if (!token_is(parser, TOK_EOF)) {
         size_t i;
-        error_expected_single(parser, TOK_EOF);
+        error_expected_tokens(parser, 1, TOK_EOF);
         
         for (i = 0; i < n; ++i) {
             node_free(out_nodes[i]);