/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements. See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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 org.apache.hadoop.ozone.container.common.impl;

import static org.apache.hadoop.ozone.container.common.impl.ContainerLayoutVersion.FILE_PER_CHUNK;
import static org.assertj.core.api.Assertions.assertThat;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.junit.jupiter.api.Assertions.assertTrue;

import java.io.File;
import java.io.IOException;
import java.nio.charset.Charset;
import java.time.Instant;
import java.util.UUID;
import org.apache.commons.io.FileUtils;
import org.apache.hadoop.conf.StorageUnit;
import org.apache.hadoop.hdds.HddsConfigKeys;
import org.apache.hadoop.hdds.conf.ConfigurationSource;
import org.apache.hadoop.hdds.conf.OzoneConfiguration;
import org.apache.hadoop.hdds.protocol.datanode.proto.ContainerProtos;
import org.apache.hadoop.ozone.OzoneConsts;
import org.apache.hadoop.ozone.container.common.helpers.ContainerUtils;
import org.apache.hadoop.ozone.container.keyvalue.ContainerLayoutTestInfo;
import org.apache.hadoop.ozone.container.keyvalue.KeyValueContainerData;
import org.apache.hadoop.ozone.container.upgrade.VersionedDatanodeFeatures;
import org.junit.jupiter.api.io.TempDir;

/**
 * This class tests create/read .container files.
 */
public class TestContainerDataYaml {

  private long testContainerID = 1234;

  @TempDir
  private File testRoot;

  private static final long MAXSIZE = (long) StorageUnit.GB.toBytes(5);
  private static final Instant SCAN_TIME = Instant.now();

  private static final String VOLUME_OWNER = "hdfs";
  private static final String CONTAINER_DB_TYPE = "RocksDB";

  private ContainerLayoutVersion layoutVersion;
  private OzoneConfiguration conf = new OzoneConfiguration();

  private void setLayoutVersion(ContainerLayoutVersion layoutVersion) {
    this.layoutVersion = layoutVersion;
  }

  /**
   * Creates a .container file.
   */
  private File createContainerFile(long containerID, int replicaIndex)
      throws IOException {
    String containerPath = containerID + ".container";

    KeyValueContainerData keyValueContainerData = new KeyValueContainerData(
        containerID, layoutVersion, MAXSIZE,
        UUID.randomUUID().toString(),
        UUID.randomUUID().toString());
    keyValueContainerData.setContainerDBType(CONTAINER_DB_TYPE);
    keyValueContainerData.setMetadataPath(testRoot.getAbsolutePath());
    keyValueContainerData.setChunksPath(testRoot.getAbsolutePath());
    keyValueContainerData.updateDataScanTime(SCAN_TIME);
    keyValueContainerData.setSchemaVersion(
        VersionedDatanodeFeatures.SchemaV2.chooseSchemaVersion());
    keyValueContainerData.setReplicaIndex(replicaIndex);
    keyValueContainerData.setDataChecksum(12345);

    File containerFile = new File(testRoot, containerPath);

    // Create .container file with ContainerData
    ContainerDataYaml.createContainerFile(keyValueContainerData, containerFile);

    //Check .container file exists or not.
    assertTrue(containerFile.exists());

    return containerFile;
  }

  @ContainerLayoutTestInfo.ContainerTest
  public void testCreateContainerFile(ContainerLayoutVersion layout)
      throws IOException {
    setLayoutVersion(layout);
    long containerID = testContainerID++;

    File containerFile = createContainerFile(containerID, 7);

    // Read from .container file, and verify data.
    KeyValueContainerData kvData = (KeyValueContainerData) ContainerDataYaml
        .readContainerFile(containerFile);
    assertEquals(containerID, kvData.getContainerID());
    assertEquals(ContainerProtos.ContainerType.KeyValueContainer, kvData
        .getContainerType());
    assertEquals(CONTAINER_DB_TYPE, kvData.getContainerDBType());
    assertEquals(containerFile.getParent(), kvData.getMetadataPath());
    assertEquals(containerFile.getParent(), kvData.getChunksPath());
    assertEquals(ContainerProtos.ContainerDataProto.State.OPEN, kvData
        .getState());
    assertEquals(layout, kvData.getLayoutVersion());
    assertEquals(0, kvData.getMetadata().size());
    assertEquals(MAXSIZE, kvData.getMaxSize());
    assertEquals(MAXSIZE, kvData.getMaxSize());
    assertTrue(kvData.lastDataScanTime().isPresent());
    assertEquals(SCAN_TIME.toEpochMilli(),
        kvData.lastDataScanTime().get().toEpochMilli());
    assertEquals(SCAN_TIME.toEpochMilli(),
        kvData.getDataScanTimestamp().longValue());
    assertEquals(VersionedDatanodeFeatures.SchemaV2.chooseSchemaVersion(),
        kvData.getSchemaVersion());
    assertEquals(7, kvData.getReplicaIndex());

    // Update ContainerData.
    kvData.addMetadata(OzoneConsts.VOLUME, VOLUME_OWNER);
    kvData.addMetadata(OzoneConsts.OWNER, OzoneConsts.OZONE);
    kvData.setState(ContainerProtos.ContainerDataProto.State.CLOSED);


    ContainerDataYaml.createContainerFile(kvData, containerFile);

    // Reading newly updated data from .container file
    kvData = (KeyValueContainerData) ContainerDataYaml.readContainerFile(
        containerFile);

    // verify data.
    assertEquals(containerID, kvData.getContainerID());
    assertEquals(ContainerProtos.ContainerType.KeyValueContainer, kvData
        .getContainerType());
    assertEquals(CONTAINER_DB_TYPE, kvData.getContainerDBType());
    assertEquals(containerFile.getParent(), kvData.getMetadataPath());
    assertEquals(containerFile.getParent(), kvData.getChunksPath());
    assertEquals(ContainerProtos.ContainerDataProto.State.CLOSED, kvData
        .getState());
    assertEquals(layout, kvData.getLayoutVersion());
    assertEquals(2, kvData.getMetadata().size());
    assertEquals(VOLUME_OWNER, kvData.getMetadata().get(OzoneConsts.VOLUME));
    assertEquals(OzoneConsts.OZONE,
        kvData.getMetadata().get(OzoneConsts.OWNER));
    assertEquals(MAXSIZE, kvData.getMaxSize());
    assertTrue(kvData.lastDataScanTime().isPresent());
    assertEquals(SCAN_TIME.toEpochMilli(),
        kvData.lastDataScanTime().get().toEpochMilli());
    assertEquals(SCAN_TIME.toEpochMilli(),
        kvData.getDataScanTimestamp().longValue());
  }

  @ContainerLayoutTestInfo.ContainerTest
  public void testCreateContainerFileWithoutReplicaIndex(
      ContainerLayoutVersion layout) throws IOException {
    setLayoutVersion(layout);
    long containerID = testContainerID++;

    File containerFile = createContainerFile(containerID, 0);

    final String content =
        FileUtils.readFileToString(containerFile, Charset.defaultCharset());

    assertThat(content)
        .withFailMessage("ReplicaIndex shouldn't be persisted if zero")
        .doesNotContain("replicaIndex");
  }

  @ContainerLayoutTestInfo.ContainerTest
  public void testIncorrectContainerFile(ContainerLayoutVersion layout) {
    setLayoutVersion(layout);
    String containerFile = "incorrect.container";

    // Get file from resource folder
    ClassLoader classLoader = getClass().getClassLoader();
    File file = new File(classLoader.getResource(containerFile).getFile());
    IllegalArgumentException exception = assertThrows(IllegalArgumentException.class,
        () -> ContainerDataYaml.readContainerFile(file));

    assertThat(exception).hasMessageContaining("No enum constant");
  }

  @ContainerLayoutTestInfo.ContainerTest
  void testCheckBackWardCompatibilityOfContainerFile(
      ContainerLayoutVersion layout) throws Exception {
    setLayoutVersion(layout);
    // This test is for if we upgrade, and then .container files added by new
    // server will have new fields added to .container file, after a while we
    // decided to rollback. Then older ozone can read .container files
    // created or not.

    String containerFile = "additionalfields.container";
    //Get file from resources folder
    ClassLoader classLoader = getClass().getClassLoader();
    File file = new File(classLoader.getResource(containerFile).getFile());
    KeyValueContainerData kvData = (KeyValueContainerData) ContainerDataYaml
        .readContainerFile(file);
    ContainerUtils.verifyContainerFileChecksum(kvData, conf);

    //Checking the Container file data is consistent or not
    assertEquals(ContainerProtos.ContainerDataProto.State.CLOSED, kvData
        .getState());
    assertEquals(CONTAINER_DB_TYPE, kvData.getContainerDBType());
    assertEquals(ContainerProtos.ContainerType.KeyValueContainer, kvData
        .getContainerType());
    assertEquals(9223372036854775807L, kvData.getContainerID());
    assertEquals("/hdds/current/aed-fg4-hji-jkl/containerDir0/1", kvData
        .getChunksPath());
    assertEquals("/hdds/current/aed-fg4-hji-jkl/containerDir0/1", kvData
        .getMetadataPath());
    assertEquals(FILE_PER_CHUNK, kvData.getLayoutVersion());
    assertEquals(2, kvData.getMetadata().size());
  }

  /**
   * Test to verify {@link ContainerUtils#verifyContainerFileChecksum(ContainerData, ConfigurationSource)}.
   */
  @ContainerLayoutTestInfo.ContainerTest
  public void testChecksumInContainerFile(ContainerLayoutVersion layout) throws IOException {
    setLayoutVersion(layout);
    long containerID = testContainerID++;

    File containerFile = createContainerFile(containerID, 0);

    // Read from .container file, and verify data.
    KeyValueContainerData kvData = (KeyValueContainerData) ContainerDataYaml.readContainerFile(containerFile);
    ContainerUtils.verifyContainerFileChecksum(kvData, conf);
  }

  /**
   * The container's data checksum is stored in a separate file with its Merkle hash tree. It should not be persisted
   * to the .container file.
   */
  @ContainerLayoutTestInfo.ContainerTest
  public void testDataChecksumNotInContainerFile(ContainerLayoutVersion layout) throws IOException {
    setLayoutVersion(layout);
    long containerID = testContainerID++;

    File containerFile = createContainerFile(containerID, 0);

    // Read from .container file. The kvData object should not have a data hash because it was not persisted in this
    // file.
    KeyValueContainerData kvData = (KeyValueContainerData) ContainerDataYaml.readContainerFile(containerFile);
    assertEquals(0, kvData.getDataChecksum());
  }

  /**
   * Test to verify {@link ContainerUtils#verifyContainerFileChecksum(ContainerData, ConfigurationSource)}.
   */
  @ContainerLayoutTestInfo.ContainerTest
  public void testChecksumInContainerFileWithReplicaIndex(
      ContainerLayoutVersion layout) throws IOException {
    setLayoutVersion(layout);
    long containerID = testContainerID++;

    File containerFile = createContainerFile(containerID, 10);

    // Read from .container file, and verify data.
    KeyValueContainerData kvData = (KeyValueContainerData) ContainerDataYaml
        .readContainerFile(containerFile);
    ContainerUtils.verifyContainerFileChecksum(kvData, conf);
  }

  private KeyValueContainerData getKeyValueContainerData() throws IOException {
    String containerFile = "incorrect.checksum.container";
    //Get file from resources folder
    ClassLoader classLoader = getClass().getClassLoader();
    File file = new File(classLoader.getResource(containerFile).getFile());
    return (KeyValueContainerData) ContainerDataYaml.readContainerFile(file);
  }

  /**
   * Test to verify incorrect checksum is detected.
   */
  @ContainerLayoutTestInfo.ContainerTest
  public void testIncorrectChecksum(ContainerLayoutVersion layout) {
    setLayoutVersion(layout);
    Exception ex = assertThrows(Exception.class, () -> {
      KeyValueContainerData kvData = getKeyValueContainerData();
      ContainerUtils.verifyContainerFileChecksum(kvData, conf);
    });

    assertThat(ex).hasMessageStartingWith("Container checksum error for ContainerID:");
  }

  /**
   * Test to verify disabled checksum with incorrect checksum.
   */
  @ContainerLayoutTestInfo.ContainerTest
  public void testDisabledChecksum(ContainerLayoutVersion layout)
      throws IOException {
    setLayoutVersion(layout);
    KeyValueContainerData kvData = getKeyValueContainerData();
    conf.setBoolean(HddsConfigKeys.
        HDDS_CONTAINER_CHECKSUM_VERIFICATION_ENABLED, false);
    ContainerUtils.verifyContainerFileChecksum(kvData, conf);
  }
}
