pkg/analyzer/test/util/token_check.dart

// Copyright (c) 2021, the Dart project authors. Please see the AUTHORS file
// for details. All rights reserved. Use of this source code is governed by a
// BSD-style license that can be found in the LICENSE file.

import 'package:analyzer/dart/ast/token.dart';
import 'package:analyzer/src/dart/ast/token.dart';
import 'package:analyzer_utilities/check/check.dart';

extension KeywordTokenExtension on CheckTarget<KeywordToken> {
  CheckTarget<Keyword> get keyword {
    return nest(
      value.keyword,
      (selected) => 'has keyword ${valueStr(selected)}',
    );
  }
}

extension TokenExtension on CheckTarget<Token?> {
  CheckTarget<KeywordToken> get isKeyword {
    return isA<KeywordToken>();
  }

  void get isKeywordConst {
    isKeyword.keyword.isEqualTo(Keyword.CONST);
  }

  void get isKeywordSuper {
    isKeyword.keyword.isEqualTo(Keyword.SUPER);
  }

  void get isKeywordVar {
    isKeyword.keyword.isEqualTo(Keyword.VAR);
  }
}