org.aya.resolve.context.ModuleSymbol Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of base Show documentation

The Aya proof assistant

There is a newer version: 0.36.0

// Copyright (c) 2020-2024 Tesla (Yinsen) Zhang.
// Use of this source code is governed by the MIT license that can be found in the LICENSE.md file.
package org.aya.resolve.context;

import kala.collection.MapView;
import kala.collection.SetView;
import kala.collection.mutable.MutableMap;
import org.aya.syntax.concrete.stmt.ModuleName;
import org.jetbrains.annotations.NotNull;

import java.util.function.BiConsumer;

/**
 * A container of symbols.
 */
public record ModuleSymbol(@NotNull MutableMap> table) {
  public ModuleSymbol() { this(MutableMap.create()); }

  public ModuleSymbol(@NotNull ModuleSymbol other) {
    this(MutableMap.from(other.table));
  }

  public @NotNull Candidate get(@NotNull String name) {
    return table.getOrPut(name, Candidate.Imported::empty);
  }

  public boolean contains(@NotNull String name) { return !get(name).isEmpty(); }

  /**
   * @param name   name for symbol
   * @param symbol the symbol
   */
  public void add(@NotNull String name, T symbol, ModuleName fromModule) {
    var candy = Candidate.of(fromModule, symbol);
    var old = get(name);
    table.put(name, old.merge(candy));
  }

  public @NotNull SetView keysView() { return table.keysView(); }
  public @NotNull MapView> view() { return table.view(); }

  public void forEach(@NotNull BiConsumer> action) {
    table.forEach(action);
  }
}