Add token kind matching lexer function

This function reads the next token, compares its kind to a parameter and
returns a boolean indicating whether it matched. If it did not match,
the token is pushed back into the lexer for later reuse.

1 files changed, 11 insertions, 0 deletions

diff --git a/src/lex.c b/src/lex.c
index 80c1c5b..89d1fc6 100644
--- a/src/lex.c
+++ b/src/lex.c
@@ -392,3 +392,14 @@ unlex(struct lexer *lexer, const struct token *in)
 	assert(lexer->un.token == T_NONE);
 	lexer->un = *in;
 }
+
+bool
+match(struct lexer *lexer, enum lexical_token token)
+{
+	struct token tmp;
+	enum lexical_token kind = lex(lexer, &tmp);
+	if (kind != token) {
+		unlex(lexer, &tmp);
+	}
+	return kind == token;
+}