com.clevercloud.biscuitpulsar.AuthenticationBiscuit Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of biscuit-pulsar Show documentation

There is a newer version: 3.7.1

package com.clevercloud.biscuitpulsar;

import com.google.common.base.Charsets;

import java.io.IOException;
import java.net.URI;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.Map;
import java.util.function.Supplier;

import org.apache.pulsar.client.api.Authentication;
import org.apache.pulsar.client.api.AuthenticationDataProvider;
import org.apache.pulsar.client.api.EncodedAuthenticationParameterSupport;
import org.apache.pulsar.client.api.PulsarClientException;

/**
 * Biscuit based authentication provider.
 */
public class AuthenticationBiscuit implements Authentication, EncodedAuthenticationParameterSupport {
  private Supplier biscuitSupplier;

  public AuthenticationBiscuit() {
  }

  public AuthenticationBiscuit(String biscuit) {
    this(() -> biscuit);
  }

  public AuthenticationBiscuit(Supplier biscuitSupplier) {
    this.biscuitSupplier = biscuitSupplier;
  }

  @Override
  public void close() throws IOException {
    // noop
  }

  @Override
  public String getAuthMethodName() {
    return "biscuit";
  }

  @Override
  public AuthenticationDataProvider getAuthData() throws PulsarClientException {
    return new AuthenticationDataBiscuit(biscuitSupplier);
  }

  @Override
  public void configure(String encodedAuthParamString) {
    // Interpret the whole param string as the biscuit. If the string contains the notation `biscuit:xxxxx` then strip
    // the prefix
    if (encodedAuthParamString.startsWith("biscuit:")) {
      this.biscuitSupplier = () -> encodedAuthParamString.substring("biscuit:".length());
    } else if (encodedAuthParamString.startsWith("file:")) {
      // Read biscuit from a file
      URI filePath = URI.create(encodedAuthParamString);
      this.biscuitSupplier = () -> {
        try {
          return new String(Files.readAllBytes(Paths.get(filePath)), Charsets.UTF_8).trim();
        } catch (IOException e) {
          throw new RuntimeException("Failed to read biscuit from file", e);
        }
      };
    } else {
      this.biscuitSupplier = () -> encodedAuthParamString;
    }
  }

  @Override
  public void configure(Map authParams) {
    // noop
  }

  @Override
  public void start() throws PulsarClientException {
    // noop
  }
}