/*
 * Copyright 2015 Amazon.com, Inc. or its affiliates. All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except
 * in compliance with the License. A copy of the License is located at
 *
 * http://aws.amazon.com/apache2.0
 *
 * or in the "license" file accompanying this file. This file 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 com.amazonaws.services.dynamodbv2.mapper.integration;

import com.amazonaws.services.dynamodbv2.AmazonDynamoDB;
import com.amazonaws.services.dynamodbv2.model.AmazonDynamoDBException;
import com.amazonaws.services.dynamodbv2.model.AttributeDefinition;
import com.amazonaws.services.dynamodbv2.model.CreateTableRequest;
import com.amazonaws.services.dynamodbv2.model.DeleteTableRequest;
import com.amazonaws.services.dynamodbv2.model.DescribeTableRequest;
import com.amazonaws.services.dynamodbv2.model.KeySchemaElement;
import com.amazonaws.services.dynamodbv2.model.KeyType;
import com.amazonaws.services.dynamodbv2.model.LocalSecondaryIndex;
import com.amazonaws.services.dynamodbv2.model.Projection;
import com.amazonaws.services.dynamodbv2.model.ProjectionType;
import com.amazonaws.services.dynamodbv2.model.ProvisionedThroughput;
import com.amazonaws.services.dynamodbv2.model.ScalarAttributeType;
import com.amazonaws.services.dynamodbv2.model.TableDescription;
import com.amazonaws.services.dynamodbv2.util.TableUtils;
import java.util.ArrayList;
import java.util.List;
import java.util.logging.Logger;
import org.testng.annotations.BeforeClass;

public class DynamoDBCryptoIntegrationTestBase extends DynamoDBTestBase {
  protected static final boolean DEBUG = false;
  protected static final String KEY_NAME = "key";
  protected static final String TABLE_NAME = "aws-java-sdk-util-crypto";

  protected static long startKey = System.currentTimeMillis();

  protected static final String TABLE_WITH_RANGE_ATTRIBUTE = "aws-java-sdk-range-test-crypto";
  protected static final String TABLE_WITH_INDEX_RANGE_ATTRIBUTE =
      "aws-java-sdk-index-range-test-crypto";

  protected static Logger log = Logger.getLogger("DynamoDBCryptoITCaseBase");

  @BeforeClass
  public static void setUp() throws Exception {
    // Create a table
    DynamoDBTestBase.setUpTestBase();
    String keyName = KEY_NAME;
    CreateTableRequest createTableRequest =
        new CreateTableRequest()
            .withTableName(TABLE_NAME)
            .withKeySchema(
                new KeySchemaElement().withAttributeName(keyName).withKeyType(KeyType.HASH))
            .withAttributeDefinitions(
                new AttributeDefinition()
                    .withAttributeName(keyName)
                    .withAttributeType(ScalarAttributeType.S));
    createTableRequest.setProvisionedThroughput(
        new ProvisionedThroughput().withReadCapacityUnits(10L).withWriteCapacityUnits(5L));

    if (TableUtils.createTableIfNotExists(dynamo, createTableRequest)) {
      TableUtils.waitUntilActive(dynamo, TABLE_NAME);
    }
  }

  /** Utility method to delete tables used in the integration test */
  public static void deleteCryptoIntegrationTestTables() {
    List<String> integrationTestTables = new ArrayList<>();
    integrationTestTables.add(TABLE_NAME);
    integrationTestTables.add(TABLE_WITH_INDEX_RANGE_ATTRIBUTE);
    integrationTestTables.add(TABLE_WITH_RANGE_ATTRIBUTE);
    for (String name : integrationTestTables) {
      dynamo.deleteTable(new DeleteTableRequest().withTableName(name));
    }
  }

  protected static void setUpTableWithRangeAttribute() throws Exception {
    setUp();

    String keyName = DynamoDBCryptoIntegrationTestBase.KEY_NAME;
    String rangeKeyAttributeName = "rangeKey";

    CreateTableRequest createTableRequest =
        new CreateTableRequest()
            .withTableName(TABLE_WITH_RANGE_ATTRIBUTE)
            .withKeySchema(
                new KeySchemaElement().withAttributeName(keyName).withKeyType(KeyType.HASH),
                new KeySchemaElement()
                    .withAttributeName(rangeKeyAttributeName)
                    .withKeyType(KeyType.RANGE))
            .withAttributeDefinitions(
                new AttributeDefinition()
                    .withAttributeName(keyName)
                    .withAttributeType(ScalarAttributeType.N),
                new AttributeDefinition()
                    .withAttributeName(rangeKeyAttributeName)
                    .withAttributeType(ScalarAttributeType.N));
    createTableRequest.setProvisionedThroughput(
        new ProvisionedThroughput().withReadCapacityUnits(10L).withWriteCapacityUnits(5L));

    if (TableUtils.createTableIfNotExists(dynamo, createTableRequest)) {
      TableUtils.waitUntilActive(dynamo, TABLE_WITH_RANGE_ATTRIBUTE);
    }
  }

  protected static void setUpTableWithIndexRangeAttribute(boolean recreateTable) throws Exception {
    setUp();
    if (recreateTable) {
      dynamo.deleteTable(new DeleteTableRequest().withTableName(TABLE_WITH_INDEX_RANGE_ATTRIBUTE));
      waitForTableToBecomeDeleted(TABLE_WITH_INDEX_RANGE_ATTRIBUTE);
    }

    String keyName = DynamoDBCryptoIntegrationTestBase.KEY_NAME;
    String rangeKeyAttributeName = "rangeKey";
    String indexFooRangeKeyAttributeName = "indexFooRangeKey";
    String indexBarRangeKeyAttributeName = "indexBarRangeKey";
    String multipleIndexRangeKeyAttributeName = "multipleIndexRangeKey";
    String indexFooName = "index_foo";
    String indexBarName = "index_bar";
    String indexFooCopyName = "index_foo_copy";
    String indexBarCopyName = "index_bar_copy";

    CreateTableRequest createTableRequest =
        new CreateTableRequest()
            .withTableName(TABLE_WITH_INDEX_RANGE_ATTRIBUTE)
            .withKeySchema(
                new KeySchemaElement().withAttributeName(keyName).withKeyType(KeyType.HASH),
                new KeySchemaElement()
                    .withAttributeName(rangeKeyAttributeName)
                    .withKeyType(KeyType.RANGE))
            .withLocalSecondaryIndexes(
                new LocalSecondaryIndex()
                    .withIndexName(indexFooName)
                    .withKeySchema(
                        new KeySchemaElement().withAttributeName(keyName).withKeyType(KeyType.HASH),
                        new KeySchemaElement()
                            .withAttributeName(indexFooRangeKeyAttributeName)
                            .withKeyType(KeyType.RANGE))
                    .withProjection(new Projection().withProjectionType(ProjectionType.ALL)),
                new LocalSecondaryIndex()
                    .withIndexName(indexBarName)
                    .withKeySchema(
                        new KeySchemaElement().withAttributeName(keyName).withKeyType(KeyType.HASH),
                        new KeySchemaElement()
                            .withAttributeName(indexBarRangeKeyAttributeName)
                            .withKeyType(KeyType.RANGE))
                    .withProjection(new Projection().withProjectionType(ProjectionType.ALL)),
                new LocalSecondaryIndex()
                    .withIndexName(indexFooCopyName)
                    .withKeySchema(
                        new KeySchemaElement().withAttributeName(keyName).withKeyType(KeyType.HASH),
                        new KeySchemaElement()
                            .withAttributeName(multipleIndexRangeKeyAttributeName)
                            .withKeyType(KeyType.RANGE))
                    .withProjection(new Projection().withProjectionType(ProjectionType.ALL)),
                new LocalSecondaryIndex()
                    .withIndexName(indexBarCopyName)
                    .withKeySchema(
                        new KeySchemaElement().withAttributeName(keyName).withKeyType(KeyType.HASH),
                        new KeySchemaElement()
                            .withAttributeName(multipleIndexRangeKeyAttributeName)
                            .withKeyType(KeyType.RANGE))
                    .withProjection(new Projection().withProjectionType(ProjectionType.ALL)))
            .withAttributeDefinitions(
                new AttributeDefinition()
                    .withAttributeName(keyName)
                    .withAttributeType(ScalarAttributeType.N),
                new AttributeDefinition()
                    .withAttributeName(rangeKeyAttributeName)
                    .withAttributeType(ScalarAttributeType.N),
                new AttributeDefinition()
                    .withAttributeName(indexFooRangeKeyAttributeName)
                    .withAttributeType(ScalarAttributeType.N),
                new AttributeDefinition()
                    .withAttributeName(indexBarRangeKeyAttributeName)
                    .withAttributeType(ScalarAttributeType.N),
                new AttributeDefinition()
                    .withAttributeName(multipleIndexRangeKeyAttributeName)
                    .withAttributeType(ScalarAttributeType.N));
    createTableRequest.setProvisionedThroughput(
        new ProvisionedThroughput().withReadCapacityUnits(10L).withWriteCapacityUnits(5L));

    if (TableUtils.createTableIfNotExists(dynamo, createTableRequest)) {
      TableUtils.waitUntilActive(dynamo, TABLE_WITH_INDEX_RANGE_ATTRIBUTE);
    }
  }

  protected static void waitForTableToBecomeDeleted(String tableName) {
    waitForTableToBecomeDeleted(dynamo, tableName);
  }

  public static void waitForTableToBecomeDeleted(AmazonDynamoDB dynamo, String tableName) {
    log.info(() -> "Waiting for " + tableName + " to become Deleted...");
    long startTime = System.currentTimeMillis();
    long endTime = startTime + (60_000);
    while (System.currentTimeMillis() < endTime) {
      try {
        Thread.sleep(5_000);
      } catch (Exception e) {
        // Ignored or expected.
      }
      try {
        DescribeTableRequest request = new DescribeTableRequest(tableName);
        TableDescription table = dynamo.describeTable(request).getTable();

        log.info(() -> "  - current state: " + table.getTableStatus());
        if (table.getTableStatus() == "DELETING") {
          continue;
        }
      } catch (AmazonDynamoDBException exception) {
        if (exception.getErrorCode().equalsIgnoreCase("ResourceNotFoundException")) {
          log.info(() -> "successfully deleted");
          return;
        }
      }
    }

    throw new RuntimeException("Table " + tableName + " never went deleted");
  }

  public static void main(String[] args) throws Exception {
    setUp();
    deleteCryptoIntegrationTestTables();
  }
}