pkg/scrape/example/generic_classes.dart

// Copyright (c) 2020, 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/ast.dart';
import 'package:scrape/scrape.dart';

void main(List<String> arguments) {
  Scrape()
    ..addHistogram('Classes', order: SortOrder.numeric)
    ..addVisitor(GenericClassVisitor.new)
    ..runCommandLine(arguments);
}

class GenericClassVisitor extends ScrapeVisitor {
  @override
  void visitClassDeclaration(ClassDeclaration node) {
    if (node.namePart.typeParameters == null) {
      record('Classes', 0);
    } else {
      record('Classes', node.namePart.typeParameters!.typeParameters.length);
    }
    super.visitClassDeclaration(node);
  }
}