service/src/main/java/io/mifos/identity/internal/repository/Tenants.java

/*
 * Copyright 2017 The Mifos Initiative.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package io.mifos.identity.internal.repository;

import com.datastax.driver.core.BoundStatement;
import com.datastax.driver.core.DataType;
import com.datastax.driver.core.KeyspaceMetadata;
import com.datastax.driver.core.schemabuilder.Create;
import com.datastax.driver.core.schemabuilder.SchemaBuilder;
import io.mifos.core.cassandra.core.CassandraSessionProvider;
import io.mifos.core.cassandra.core.TenantAwareEntityTemplate;
import io.mifos.identity.internal.util.IdentityConstants;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.nio.ByteBuffer;
import java.util.Optional;

/**
 * @author Myrle Krantz
 */
@Component
public class Tenants {
  static final String TABLE_NAME = "isis_tenant";
  static final String VERSION_COLUMN = "version";
  static final String FIXED_SALT_COLUMN = "fixed_salt";
  static final String PASSWORD_EXPIRES_IN_DAYS_COLUMN = "password_expires_in_days";
  static final String TIME_TO_CHANGE_PASSWORD_AFTER_EXPIRATION_IN_DAYS = "time_to_change_password_after_expiration_in_days";

  private final CassandraSessionProvider cassandraSessionProvider;
  private final TenantAwareEntityTemplate tenantAwareEntityTemplate;

  @Autowired
  Tenants(final CassandraSessionProvider cassandraSessionProvider,
          final TenantAwareEntityTemplate tenantAwareEntityTemplate)
  {
    this.cassandraSessionProvider = cassandraSessionProvider;
    this.tenantAwareEntityTemplate = tenantAwareEntityTemplate;
  }

  public void buildTable() {
    final Create create = SchemaBuilder.createTable(TABLE_NAME)
        .ifNotExists()
        .addPartitionKey(VERSION_COLUMN, DataType.cint())
        .addColumn(FIXED_SALT_COLUMN, DataType.blob())
        .addColumn(PASSWORD_EXPIRES_IN_DAYS_COLUMN, DataType.cint())
        .addColumn(TIME_TO_CHANGE_PASSWORD_AFTER_EXPIRATION_IN_DAYS, DataType.cint());

    cassandraSessionProvider.getTenantSession().execute(create);
  }

  public void add(
          final byte[] fixedSalt,
          final int passwordExpiresInDays,
          final int timeToChangePasswordAfterExpirationInDays)
  {
    //There will only be one entry in this table.
    final BoundStatement tenantCreationStatement =
        cassandraSessionProvider.getTenantSession().prepare("INSERT INTO " + Tenants.TABLE_NAME + " ("
            + VERSION_COLUMN + ", "
            + FIXED_SALT_COLUMN + ", "
            + PASSWORD_EXPIRES_IN_DAYS_COLUMN + ", "
            + TIME_TO_CHANGE_PASSWORD_AFTER_EXPIRATION_IN_DAYS + ")"
            + "VALUES (?, ?, ?, ?)").bind();

    tenantCreationStatement.setInt(VERSION_COLUMN, IdentityConstants.CURRENT_VERSION);

    tenantCreationStatement.setBytes(FIXED_SALT_COLUMN, ByteBuffer.wrap(fixedSalt));
    tenantCreationStatement.setInt(PASSWORD_EXPIRES_IN_DAYS_COLUMN, passwordExpiresInDays);
    tenantCreationStatement.setInt(TIME_TO_CHANGE_PASSWORD_AFTER_EXPIRATION_IN_DAYS, timeToChangePasswordAfterExpirationInDays);


    cassandraSessionProvider.getTenantSession().execute(tenantCreationStatement);
  }

  public Optional<PrivateTenantInfoEntity> getPrivateTenantInfo()
  {
    return tenantAwareEntityTemplate
        .findById(PrivateTenantInfoEntity.class, IdentityConstants.CURRENT_VERSION);
  }

  public boolean currentTenantAlreadyProvisioned() {
    final String keyspace = cassandraSessionProvider.getTenantSession().getLoggedKeyspace();

    final KeyspaceMetadata keyspaceMetadata = cassandraSessionProvider.getTenantSession()
        .getCluster().getMetadata().getKeyspace(keyspace);

    return keyspaceMetadata != null && keyspaceMetadata.getTable(TABLE_NAME) != null;
  }
}