kz.greetgo.security.crypto.CryptoBuilderKeysInDb 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!
The newest version!
package kz.greetgo.security.crypto;

import kz.greetgo.db.DbType;
import kz.greetgo.db.Jdbc;
import kz.greetgo.security.crypto.errors.NotEqualsIdFieldLengths;
import kz.greetgo.security.crypto.errors.UnsupportedDb;
import kz.greetgo.security.crypto.jdbc.ContentNames;
import kz.greetgo.security.crypto.jdbc.DbDialect;
import kz.greetgo.security.crypto.jdbc.DbDialectOracle;
import kz.greetgo.security.crypto.jdbc.DbDialectPostgres;
import kz.greetgo.security.crypto.jdbc.JdbcContentAccess;
import kz.greetgo.security.crypto.jdbc.create_table.CreateTable;

import java.util.Objects;

public class CryptoBuilderKeysInDb {
  private final CryptoBuilder parent;
  private final DbType dbType;
  private final Jdbc jdbc;

  public CryptoBuilderKeysInDb(CryptoBuilder parent, DbType dbType, Jdbc jdbc) {
    this.parent = parent;
    this.dbType = dbType;
    this.jdbc = jdbc;
  }

  public CryptoBuilderKeysInDb setConfig(CryptoSourceConfig config) {
    parent.setConfig(config);
    return this;
  }

  private static class Names {
    public String tableName = "crypto_keys";
    public String idFieldName = "id";
    public String idValue;
    public String valueFieldName = "content";

    public Names(String idValue) {
      this.idValue = idValue;
    }

    ContentNames fix() {
      return new ContentNames(tableName, idFieldName, idValue, valueFieldName);
    }
  }

  private final Names namesForPrivateKey = new Names("private.key");
  private final Names namesForPublicKey = new Names("public.key");

  public CryptoBuilderKeysInDb setTableName(String tableName) {
    return setTableNameForPrivateKey(tableName).setTableNameForPublicKey(tableName);
  }

  public CryptoBuilderKeysInDb setTableNameForPrivateKey(String tableName) {
    Objects.requireNonNull(tableName);
    namesForPrivateKey.tableName = tableName;
    return this;
  }

  public CryptoBuilderKeysInDb setTableNameForPublicKey(String tableName) {
    Objects.requireNonNull(tableName);
    namesForPublicKey.tableName = tableName;
    return this;
  }

  public CryptoBuilderKeysInDb setIdFieldName(String idFieldName) {
    return setIdFieldNameForPrivateKey(idFieldName).setIdFieldNameForPublicKey(idFieldName);
  }

  public CryptoBuilderKeysInDb setIdFieldNameForPrivateKey(String idFieldName) {
    Objects.requireNonNull(idFieldName);
    namesForPrivateKey.idFieldName = idFieldName;
    return this;
  }

  public CryptoBuilderKeysInDb setIdFieldNameForPublicKey(String idFieldName) {
    Objects.requireNonNull(idFieldName);
    namesForPublicKey.idFieldName = idFieldName;
    return this;
  }

  public CryptoBuilderKeysInDb setPrivateKeyIdValue(String idValue) {
    Objects.requireNonNull(idValue);
    namesForPrivateKey.idValue = idValue;
    return this;
  }

  public CryptoBuilderKeysInDb setPublicKeyIdValue(String idValue) {
    Objects.requireNonNull(idValue);
    namesForPublicKey.idValue = idValue;
    return this;
  }

  public CryptoBuilderKeysInDb setValueFieldName(String valueFieldName) {
    return setValueFieldNameForPrivateKey(valueFieldName).setValueFieldNameForPublicKey(valueFieldName);
  }

  public CryptoBuilderKeysInDb setValueFieldNameForPrivateKey(String valueFieldName) {
    Objects.requireNonNull(valueFieldName);
    namesForPrivateKey.valueFieldName = valueFieldName;
    return this;
  }

  public CryptoBuilderKeysInDb setValueFieldNameForPublicKey(String valueFieldName) {
    Objects.requireNonNull(valueFieldName);
    namesForPublicKey.valueFieldName = valueFieldName;
    return this;
  }

  private int privateIdFieldLength = 50;
  private int publicIdFieldLength = 50;

  public CryptoBuilderKeysInDb setIdFieldLength(int idFieldLength) {
    return setPrivateIdFieldLength(idFieldLength).setPublicIdFieldLength(idFieldLength);
  }

  public CryptoBuilderKeysInDb setPrivateIdFieldLength(int privateIdFieldLength) {
    this.privateIdFieldLength = privateIdFieldLength;
    return this;
  }

  public CryptoBuilderKeysInDb setPublicIdFieldLength(int publicIdFieldLength) {
    this.publicIdFieldLength = publicIdFieldLength;
    return this;
  }

  String createTableDDL_privateKey;
  String createTableDDL_publicKey;

  DbDialect dialect;

  ContentNames privateKeyNames = namesForPrivateKey.fix();
  ContentNames publicKeyNames = namesForPublicKey.fix();

  private void prepareDDL() {
    if (Objects.equals(privateKeyNames.tableName, publicKeyNames.tableName)) {

      if (privateIdFieldLength != publicIdFieldLength) {
        throw new NotEqualsIdFieldLengths(privateIdFieldLength, publicIdFieldLength);
      }

      boolean sameValueFields = Objects.equals(privateKeyNames.valueFieldName, publicKeyNames.valueFieldName);

      CreateTable createTable = new CreateTable(privateKeyNames.tableName);
      createTable
        .newField(privateKeyNames.idFieldName)
        .typeString(privateIdFieldLength)
        .primaryKey();
      createTable
        .newField(privateKeyNames.valueFieldName)
        .setNotNull(sameValueFields)
        .typeBlob();

      if (!sameValueFields) {
        createTable
          .newField(publicKeyNames.valueFieldName)
          .typeBlob();
      }

      createTableDDL_privateKey = createTableDDL_publicKey = dialect.generateCreateTableDDL(createTable);
    } else {

      {
        CreateTable createTable = new CreateTable(privateKeyNames.tableName);
        createTable
          .newField(privateKeyNames.idFieldName)
          .primaryKey()
          .typeString(privateIdFieldLength);
        createTable
          .newField(privateKeyNames.valueFieldName)
          .typeBlob()
          .notNull();

        createTableDDL_privateKey = dialect.generateCreateTableDDL(createTable);
      }
      {
        CreateTable createTable = new CreateTable(publicKeyNames.tableName);
        createTable
          .newField(publicKeyNames.idFieldName)
          .primaryKey()
          .typeString(publicIdFieldLength);
        createTable
          .newField(publicKeyNames.valueFieldName)
          .typeBlob()
          .notNull();

        createTableDDL_publicKey = dialect.generateCreateTableDDL(createTable);
      }
    }
  }

  private DbDialect calcDialect() {
    switch (dbType) {
      case Postgres:
        return new DbDialectPostgres();
      case Oracle:
        return new DbDialectOracle();
      default:
        throw new UnsupportedDb(dbType);
    }
  }

  public Crypto build() {
    dialect = calcDialect();

    prepareDDL();

    JdbcContentAccess privateKeyAccess
      = new JdbcContentAccess(jdbc, privateKeyNames, createTableDDL_privateKey, dialect);

    JdbcContentAccess publicKeyAccess
      = new JdbcContentAccess(jdbc, publicKeyNames, createTableDDL_publicKey, dialect);

    return parent.build(privateKeyAccess, publicKeyAccess);
  }
}