platform/script-debugger/protocol/protocol-model-generator/src/org/jetbrains/protocolReader/TypeMap.java

package org.jetbrains.protocolReader;

import com.intellij.openapi.util.Pair;
import gnu.trove.THashMap;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

/**
 * Keeps track of all referenced types.
 * A type may be used and resolved (generated or hard-coded).
 */
class TypeMap {
  private final Map<Pair<String, String>, TypeData> map = new THashMap<>();
  private Map<String, DomainGenerator> domainGeneratorMap;
  private final List<StandaloneTypeBinding> typesToGenerate = new ArrayList<>();

  void setDomainGeneratorMap(Map<String, DomainGenerator> domainGeneratorMap) {
    this.domainGeneratorMap = domainGeneratorMap;
  }

  @Nullable
  BoxableType resolve(String domainName, String typeName, TypeData.Direction direction) {
    DomainGenerator domainGenerator = domainGeneratorMap.get(domainName);
    if (domainGenerator == null) {
      throw new RuntimeException("Failed to find domain generator: " + domainName);
    }
    return getTypeData(domainName, typeName).get(direction).resolve(this, domainGenerator);
  }

  void addTypeToGenerate(@NotNull StandaloneTypeBinding binding) {
    typesToGenerate.add(binding);
  }

  public void generateRequestedTypes() throws IOException {
    // Size may grow during iteration.
    //noinspection ForLoopReplaceableByForEach
    for (int i = 0; i < typesToGenerate.size(); i++) {
      typesToGenerate.get(i).generate();
    }

    for (TypeData typeData : map.values()) {
      typeData.checkComplete();
    }
  }

  @NotNull
  TypeData getTypeData(@NotNull String domainName, @NotNull String typeName) {
    Pair<String, String> key = Pair.create(domainName, typeName);
    TypeData result = map.get(key);
    if (result == null) {
      result = new TypeData(typeName);
      map.put(key, result);
    }
    return result;
  }
}