pkg/wasm_builder/lib/src/builder/tables.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 '../ir/ir.dart' as ir;
import 'builder.dart';
import 'util.dart';

/// The interface for the tables in a module.
class TablesBuilder with Builder<ir.Tables> {
  final ModuleBuilder _moduleBuilder;
  final _tableBuilders = <TableBuilder>[];
  final _importedTables = <ir.ImportedTable>[];

  TablesBuilder(this._moduleBuilder);

  /// Defines a new table in this module.
  TableBuilder define(ir.RefType type, int minSize, [int? maxSize]) {
    final table = TableBuilder(
      _moduleBuilder,
      ir.FinalizableIndex(),
      type,
      minSize,
      maxSize,
    );
    _tableBuilders.add(table);
    return table;
  }

  /// Imports a table into this module.
  ir.ImportedTable import(
    String module,
    String name,
    ir.RefType type,
    int minSize, [
    int? maxSize,
  ]) {
    final table = ir.ImportedTable(
      _moduleBuilder.module,
      module,
      name,
      ir.FinalizableIndex(),
      type,
      minSize,
      maxSize,
    );
    _importedTables.add(table);
    return table;
  }

  @override
  ir.Tables forceBuild() {
    final built = finalizeImportsAndBuilders<ir.DefinedTable>(
      _importedTables,
      _tableBuilders,
    );
    return ir.Tables(_importedTables, built);
  }

  void collectUsedTypes(Set<ir.DefType> types) {
    for (final table in _tableBuilders) {
      final defType = table.type.containedDefType;
      if (defType != null) types.add(defType);
    }
    for (final table in _importedTables) {
      final defType = table.type.containedDefType;
      if (defType != null) types.add(defType);
    }
  }
}