pkg/dart2wasm/lib/js/method_collector.dart

// Copyright (c) 2023, 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:dart2wasm/js/util.dart';
import 'package:kernel/ast.dart';

typedef JSMethods = Map<Procedure, String>;

/// Collects JS methods and adds Dart stubs to libraries.
class MethodCollector {
  final CoreTypesUtil _util;
  final JSMethods jsMethods = {};
  late Library _library;
  int _methodN = 1;

  MethodCollector(this._util);

  void enterLibrary(Library library) => _library = library;

  // We could generate something more human readable, but for now we just
  // generate something short and unique.
  String generateMethodName() => '_${_methodN++}';

  Procedure addInteropProcedure(String name, String pragmaOptionString,
      FunctionNode function, Uri fileUri, AnnotationType type,
      {required bool isExternal}) {
    final procedure = Procedure(
        Name(name, _library), ProcedureKind.Method, function,
        isStatic: true, isExternal: isExternal, fileUri: fileUri)
      ..isNonNullableByDefault = true;
    _util.annotateProcedure(procedure, pragmaOptionString, type);
    _library.addProcedure(procedure);
    return procedure;
  }

  void addMethod(Procedure dartProcedure, String methodName, String code) {
    jsMethods[dartProcedure] = "$methodName: $code";
  }
}