src/main/java/com/android/tools/r8/diagnostic/internal/DefinitionClassContextImpl.java

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

import com.android.tools.r8.diagnostic.DefinitionClassContext;
import com.android.tools.r8.origin.Origin;
import com.android.tools.r8.references.ClassReference;

public class DefinitionClassContextImpl extends DefinitionContextBase
    implements DefinitionClassContext {

  private final ClassReference classReference;

  private DefinitionClassContextImpl(ClassReference classReference, Origin origin) {
    super(origin);
    this.classReference = classReference;
  }

  public static Builder builder() {
    return new Builder();
  }

  @Override
  public ClassReference getClassReference() {
    return classReference;
  }

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

    private ClassReference classReference;

    private Builder() {}

    public Builder setClassContext(ClassReference classReference) {
      this.classReference = classReference;
      return this;
    }

    @Override
    Builder self() {
      return this;
    }

    @Override
    public DefinitionClassContextImpl build() {
      assert validate();
      return new DefinitionClassContextImpl(classReference, origin);
    }

    @Override
    public boolean validate() {
      super.validate();
      assert classReference != null;
      return true;
    }
  }
}