DecryptionMaterials.java

package com.amazonaws.encryptionsdk.model;

import com.amazonaws.encryptionsdk.DataKey;
import java.security.PublicKey;
import java.util.Collections;
import java.util.Map;

public final class DecryptionMaterials {
  private final DataKey<?> dataKey;
  private final PublicKey trailingSignatureKey;
  private final Map<String, String> encryptionContext;

  private DecryptionMaterials(Builder b) {
    dataKey = b.getDataKey();
    trailingSignatureKey = b.getTrailingSignatureKey();
    if (b.getEncryptionContext() != null) {
      encryptionContext = b.getEncryptionContext();
    } else {
      encryptionContext = Collections.emptyMap();
    }
  }

  public DataKey<?> getDataKey() {
    return dataKey;
  }

  public PublicKey getTrailingSignatureKey() {
    return trailingSignatureKey;
  }

  public Map<String, String> getEncryptionContext() {
    return encryptionContext;
  }

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

  public Builder toBuilder() {
    return new Builder(this);
  }

  public static final class Builder {
    private DataKey<?> dataKey;
    private PublicKey trailingSignatureKey;
    private Map<String, String> encryptionContext;

    private Builder(DecryptionMaterials result) {
      this.dataKey = result.getDataKey();
      this.trailingSignatureKey = result.getTrailingSignatureKey();
      this.encryptionContext = result.getEncryptionContext();
    }

    private Builder() {}

    public DataKey<?> getDataKey() {
      return dataKey;
    }

    public Builder setDataKey(DataKey<?> dataKey) {
      this.dataKey = dataKey;
      return this;
    }

    public PublicKey getTrailingSignatureKey() {
      return trailingSignatureKey;
    }

    public Builder setTrailingSignatureKey(PublicKey trailingSignatureKey) {
      this.trailingSignatureKey = trailingSignatureKey;
      return this;
    }

    public Map<String, String> getEncryptionContext() {
      return encryptionContext;
    }

    public Builder setEncryptionContext(Map<String, String> encryptionContext) {
      this.encryptionContext = encryptionContext;
      return this;
    }

    public DecryptionMaterials build() {
      return new DecryptionMaterials(this);
    }
  }
}