kz.greetgo.security.crypto.MongoContentAccess Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of greetgo.security Show documentation

Security infrastructure used in greetgo!

There is a newer version: 0.0.12

package kz.greetgo.security.crypto;

import com.mongodb.client.MongoCollection;
import com.mongodb.client.model.UpdateOptions;
import kz.greetgo.security.util.Base64Util;
import org.bson.Document;
import org.bson.conversions.Bson;

import static com.mongodb.client.model.Filters.eq;
import static kz.greetgo.security.util.MongoUtil.toStr;

public class MongoContentAccess implements ContentAccess {
  private final MongoCollection collection;
  private final CryptoBuilderKeysInMongo.Names names;

  public MongoContentAccess(MongoCollection collection, CryptoBuilderKeysInMongo.Names names) {
    this.collection = collection;
    this.names = names;
  }

  @Override
  public byte[] downloadBytes() {

    Document found = collection.find(eq(names.idFieldName, names.idValue)).limit(1).first();

    if (found == null) {
      return null;
    }

    return Base64Util.base64ToBytes(toStr(found.get(names.keyFieldName)));
  }

  @Override
  public void uploadBytes(byte[] bytes) {
    if (bytes == null) {
      removeBytes();
      return;
    }

    String base64 = Base64Util.bytesToBase64(bytes);

    Bson filter = eq(names.idFieldName, names.idValue);

    Document set = new Document();
    set.append(names.keyFieldName, base64);

    Document update = new Document();
    update.append("$set", set);

    UpdateOptions options = new UpdateOptions();
    options.upsert(true);

    collection.updateOne(filter, update, options);
  }

  private void removeBytes() {
    collection.deleteOne(eq(names.idFieldName, names.idValue));
  }

  @Override
  public boolean exists() {
    return null != collection.find(eq(names.idFieldName, names.idValue)).limit(1).first();
  }
}