From da1f89075252e2601bb82e195f696a58b17a9f91 Mon Sep 17 00:00:00 2001
From: Amaan Qureshi <amaanq12@gmail.com>
Date: Sat, 17 Feb 2024 00:53:59 -0500
Subject: [PATCH] chore: error out when multiple arguments are passed to
 `token`/`token.immediate`

---
 cli/src/generate/dsl.js | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/cli/src/generate/dsl.js b/cli/src/generate/dsl.js
index 32b961f7..3c26de0c 100644
--- a/cli/src/generate/dsl.js
+++ b/cli/src/generate/dsl.js
@@ -157,6 +157,7 @@ function sym(name) {
 }
 
 function token(value) {
+  checkArguments(arguments.length, token, 'token', '', 'literal');
   return {
     type: "TOKEN",
     content: normalize(value)
@@ -164,6 +165,7 @@ function token(value) {
 }
 
 token.immediate = function(value) {
+  checkArguments(arguments.length, token.immediate, 'token.immediate', '', 'literal');
   return {
     type: "IMMEDIATE_TOKEN",
     content: normalize(value)
@@ -404,11 +406,11 @@ function grammar(baseGrammar, options) {
   return { grammar: { name, word, rules, extras, conflicts, precedences, externals, inline, supertypes } };
 }
 
-function checkArguments(ruleCount, caller, callerName, suffix = '') {
+function checkArguments(ruleCount, caller, callerName, suffix = '', argType = 'rule') {
   if (ruleCount > 1) {
     const error = new Error([
-      `The \`${callerName}\` function only takes one rule argument${suffix}.`,
-      'You passed multiple rules. Did you mean to call `seq`?\n'
+      `The \`${callerName}\` function only takes one ${argType} argument${suffix}.`,
+      `You passed in multiple ${argType}s. Did you mean to call \`seq\`?\n`
     ].join('\n'));
     Error.captureStackTrace(error, caller);
     throw error