// Copyright (c) 2020, 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.ir.analysis.fieldvalueanalysis;

import com.android.tools.r8.graph.DexEncodedField;
import com.android.tools.r8.graph.DexField;
import com.android.tools.r8.graph.DexItemFactory;
import com.android.tools.r8.graph.DexProgramClass;
import com.android.tools.r8.ir.analysis.value.AbstractValue;
import com.android.tools.r8.ir.analysis.value.ObjectState;
import com.google.common.collect.ImmutableMap;

public abstract class StaticFieldValues {

  public boolean isEnumStaticFieldValues() {
    return false;
  }

  public EnumStaticFieldValues asEnumStaticFieldValues() {
    return null;
  }

  public static Builder builder(DexProgramClass clazz) {
    return clazz.isEnum() ? EnumStaticFieldValues.builder() : EmptyStaticValues.builder();
  }

  public abstract static class Builder {

    public abstract void recordStaticField(
        DexEncodedField staticField, AbstractValue value, DexItemFactory factory);

    public abstract StaticFieldValues build();
  }

  // All the abstract values stored here may match a pinned field, using them requires therefore
  // to check the field is not pinned or prove it is no longer pinned.
  public static class EnumStaticFieldValues extends StaticFieldValues {
    private final ImmutableMap<DexField, ObjectState> enumAbstractValues;

    public EnumStaticFieldValues(ImmutableMap<DexField, ObjectState> enumAbstractValues) {
      this.enumAbstractValues = enumAbstractValues;
    }

    static StaticFieldValues.Builder builder() {
      return new Builder();
    }

    public static class Builder extends StaticFieldValues.Builder {
      private final ImmutableMap.Builder<DexField, ObjectState> enumObjectStateBuilder =
          ImmutableMap.builder();
      private AbstractValue valuesCandidateAbstractValue;

      Builder() {}

      @Override
      public void recordStaticField(
          DexEncodedField staticField, AbstractValue value, DexItemFactory factory) {
        if (factory.enumMembers.isValuesFieldCandidate(staticField, staticField.getHolderType())) {
          if (value.isSingleFieldValue()
              && value.asSingleFieldValue().getState().isEnumValuesObjectState()) {
            assert valuesCandidateAbstractValue == null
                || valuesCandidateAbstractValue.equals(value);
            valuesCandidateAbstractValue = value;
            enumObjectStateBuilder.put(staticField.field, value.asSingleFieldValue().getState());
          }
        } else if (factory.enumMembers.isEnumField(staticField, staticField.getHolderType())) {
          if (value.isSingleFieldValue() && !value.asSingleFieldValue().getState().isEmpty()) {
            enumObjectStateBuilder.put(staticField.field, value.asSingleFieldValue().getState());
          }
        }
      }

      @Override
      public StaticFieldValues build() {
        ImmutableMap<DexField, ObjectState> enumAbstractValues = enumObjectStateBuilder.build();
        if (enumAbstractValues.isEmpty()) {
          return EmptyStaticValues.getInstance();
        }
        assert enumAbstractValues.values().stream().noneMatch(ObjectState::isEmpty);
        return new EnumStaticFieldValues(enumAbstractValues);
      }
    }

    @Override
    public boolean isEnumStaticFieldValues() {
      return true;
    }

    @Override
    public EnumStaticFieldValues asEnumStaticFieldValues() {
      return this;
    }

    public ObjectState getObjectStateForPossiblyPinnedField(DexField field) {
      return enumAbstractValues.get(field);
    }
  }

  public static class EmptyStaticValues extends StaticFieldValues {
    private static EmptyStaticValues INSTANCE = new EmptyStaticValues();

    private EmptyStaticValues() {}

    public static EmptyStaticValues getInstance() {
      return INSTANCE;
    }

    static StaticFieldValues.Builder builder() {
      return new Builder();
    }

    public static class Builder extends StaticFieldValues.Builder {

      @Override
      public void recordStaticField(
          DexEncodedField staticField, AbstractValue value, DexItemFactory factory) {
        // Do nothing.
      }

      @Override
      public StaticFieldValues build() {
        return EmptyStaticValues.getInstance();
      }
    }
  }
}