// Copyright (c) 2017, the R8 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.
// Copyright (c) 2016, the R8 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.
package com.android.tools.r8.graph;

import com.google.common.collect.ImmutableMap;
import com.google.common.collect.ImmutableSet;
import com.google.common.collect.Iterables;

import com.android.tools.r8.naming.ClassNameMapper;
import com.android.tools.r8.utils.ProgramClassCollection;
import com.android.tools.r8.utils.Timing;

import java.util.Collection;
import java.util.IdentityHashMap;
import java.util.Map;

public class DirectMappedDexApplication extends DexApplication {

  private final ImmutableMap<DexType, DexLibraryClass> libraryClasses;

  private DirectMappedDexApplication(ClassNameMapper proguardMap,
      ProgramClassCollection programClasses,
      ImmutableMap<DexType, DexLibraryClass> libraryClasses,
      ImmutableSet<DexType> mainDexList, byte[] deadCode,
      DexItemFactory dexItemFactory, DexString highestSortingString,
      Timing timing) {
    super(proguardMap, programClasses, mainDexList, deadCode,
        dexItemFactory, highestSortingString, timing);
    this.libraryClasses = libraryClasses;
  }

  public Collection<DexLibraryClass> libraryClasses() {
    return libraryClasses.values();
  }

  @Override
  public DexClass definitionFor(DexType type) {
    DexClass result = programClasses.get(type);
    if (result == null) {
      result = libraryClasses.get(type);
    }
    return result;
  }

  @Override
  public Builder builder() {
    return new Builder(this);
  }

  @Override
  public DirectMappedDexApplication toDirect() {
    return this;
  }

  @Override
  public DirectMappedDexApplication asDirect() {
    return this;
  }

  public String toString() {
    return "DexApplication (direct)";
  }

  public static class Builder extends DexApplication.Builder<Builder> {

    private Map<DexType, DexLibraryClass> libraryClasses = new IdentityHashMap<>();

    Builder(LazyLoadedDexApplication application) {
      super(application);
      // As a side-effect, this will force-load all classes.
      Map<DexType, DexClass> allClasses = application.getFullClassMap();
      Iterables.filter(allClasses.values(), DexLibraryClass.class)
          .forEach(k -> libraryClasses.put(k.type, k));

    }

    private Builder(DirectMappedDexApplication application) {
      super(application);
      this.libraryClasses.putAll(application.libraryClasses);
    }

    @Override
    Builder self() {
      return this;
    }

    @Override
    public DexApplication build() {
      return new DirectMappedDexApplication(proguardMap,
          ProgramClassCollection.create(programClasses),
          ImmutableMap.copyOf(libraryClasses), ImmutableSet.copyOf(mainDexList), deadCode,
          dexItemFactory, highestSortingString, timing);
    }
  }
}