// Copyright (c) 2022, 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.retrace.internal;

import com.android.tools.r8.DiagnosticsHandler;
import com.android.tools.r8.naming.ClassNameMapper;
import com.android.tools.r8.references.ClassReference;
import com.android.tools.r8.retrace.InvalidMappingFileException;
import com.android.tools.r8.retrace.ProguardMapProducer;
import com.android.tools.r8.retrace.ProguardMappingProvider;
import java.io.BufferedReader;
import java.util.HashSet;
import java.util.Set;

public class ProguardMappingProviderBuilderImpl extends ProguardMappingProvider.Builder {

  private ProguardMapProducer proguardMapProducer;
  private boolean allowExperimental = false;
  private Set<String> allowedLookup = new HashSet<>();
  private boolean allowLookupAllClasses = false;
  private DiagnosticsHandler diagnosticsHandler = new DiagnosticsHandler() {};

  @Override
  public ProguardMappingProvider.Builder self() {
    return this;
  }

  @Override
  public ProguardMappingProvider.Builder setAllowExperimental(boolean allowExperimental) {
    this.allowExperimental = allowExperimental;
    return self();
  }

  @Override
  public ProguardMappingProvider.Builder setDiagnosticsHandler(
      DiagnosticsHandler diagnosticsHandler) {
    this.diagnosticsHandler = diagnosticsHandler;
    return self();
  }

  @Override
  public ProguardMappingProvider.Builder registerUse(ClassReference classReference) {
    allowedLookup.add(classReference.getTypeName());
    return self();
  }

  @Override
  public ProguardMappingProvider.Builder allowLookupAllClasses() {
    allowLookupAllClasses = true;
    return self();
  }

  @Override
  public ProguardMappingProvider.Builder setProguardMapProducer(
      ProguardMapProducer proguardMapProducer) {
    this.proguardMapProducer = proguardMapProducer;
    return self();
  }

  @Override
  public ProguardMappingProvider build() {
    try {
      if (allowLookupAllClasses) {
        return new ProguardMappingProviderImpl(
            ClassNameMapper.mapperFromBufferedReader(
                new BufferedReader(proguardMapProducer.get()),
                diagnosticsHandler,
                true,
                allowExperimental));
      } else {
        return new ProguardMappingProviderImpl(
            ClassNameMapper.mapperFromBufferedReaderWithFiltering(
                new BufferedReader(proguardMapProducer.get()),
                diagnosticsHandler,
                true,
                allowExperimental,
                allowedLookup),
            allowedLookup);
      }
    } catch (Exception e) {
      throw new InvalidMappingFileException(e);
    }
  }
}