Add 'isPunctuatorToken' to internal API

Author: IvanGoncharov

src/language/__tests__/lexer-test.js

@@ -15,7 +15,7 @@ import dedent from '../../jsutils/dedent';
 import inspect from '../../jsutils/inspect';
 import { GraphQLError } from '../../error';
 import { Source } from '../source';
-import { createLexer, TokenKind } from '../lexer';
+import { createLexer, TokenKind, isPunctuatorToken } from '../lexer';
 
 function lexOne(str) {
   const lexer = createLexer(new Source(str));
@@ -766,3 +766,31 @@ describe('Lexer', () => {
     ]);
   });
 });
+
+describe('isPunctuatorToken', () => {
+  it('returns true for punctuator tokens', () => {
+    expect(isPunctuatorToken(lexOne('!'))).to.equal(true);
+    expect(isPunctuatorToken(lexOne('$'))).to.equal(true);
+    expect(isPunctuatorToken(lexOne('&'))).to.equal(true);
+    expect(isPunctuatorToken(lexOne('('))).to.equal(true);
+    expect(isPunctuatorToken(lexOne(')'))).to.equal(true);
+    expect(isPunctuatorToken(lexOne('...'))).to.equal(true);
+    expect(isPunctuatorToken(lexOne(':'))).to.equal(true);
+    expect(isPunctuatorToken(lexOne('='))).to.equal(true);
+    expect(isPunctuatorToken(lexOne('@'))).to.equal(true);
+    expect(isPunctuatorToken(lexOne('['))).to.equal(true);
+    expect(isPunctuatorToken(lexOne(']'))).to.equal(true);
+    expect(isPunctuatorToken(lexOne('{'))).to.equal(true);
+    expect(isPunctuatorToken(lexOne('|'))).to.equal(true);
+    expect(isPunctuatorToken(lexOne('}'))).to.equal(true);
+  });
+
+  it('returns false for non-punctuator tokens', () => {
+    expect(isPunctuatorToken(lexOne(''))).to.equal(false);
+    expect(isPunctuatorToken(lexOne('name'))).to.equal(false);
+    expect(isPunctuatorToken(lexOne('1'))).to.equal(false);
+    expect(isPunctuatorToken(lexOne('3.14'))).to.equal(false);
+    expect(isPunctuatorToken(lexOne('"str"'))).to.equal(false);
+    expect(isPunctuatorToken(lexOne('"""str"""'))).to.equal(false);
+  });
+});

src/language/lexer.js

@@ -129,6 +129,27 @@ export const TokenKind = Object.freeze({
  */
 export type TokenKindEnum = $Values<typeof TokenKind>;
 
+// @internal
+export function isPunctuatorToken(token: Token) {
+  const kind = token.kind;
+  return (
+    kind === TokenKind.BANG ||
+    kind === TokenKind.DOLLAR ||
+    kind === TokenKind.AMP ||
+    kind === TokenKind.PAREN_L ||
+    kind === TokenKind.PAREN_R ||
+    kind === TokenKind.SPREAD ||
+    kind === TokenKind.COLON ||
+    kind === TokenKind.EQUALS ||
+    kind === TokenKind.AT ||
+    kind === TokenKind.BRACKET_L ||
+    kind === TokenKind.BRACKET_R ||
+    kind === TokenKind.BRACE_L ||
+    kind === TokenKind.PIPE ||
+    kind === TokenKind.BRACE_R
+  );
+}
+
 /**
  * A helper function to describe a token as a string for debugging
  */