/*
 * Copyright Java Operator SDK Authors
 *
 * 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.javaoperatorsdk.operator.sample;

import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.util.ArrayList;
import java.util.List;

import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.RegisterExtension;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import io.fabric8.kubernetes.api.model.HasMetadata;
import io.fabric8.kubernetes.api.model.NamespaceBuilder;
import io.fabric8.kubernetes.api.model.ObjectMetaBuilder;
import io.fabric8.kubernetes.client.KubernetesClient;
import io.fabric8.kubernetes.client.KubernetesClientBuilder;
import io.javaoperatorsdk.operator.junit.AbstractOperatorExtension;
import io.javaoperatorsdk.operator.junit.ClusterDeployedOperatorExtension;
import io.javaoperatorsdk.operator.junit.LocallyRunOperatorExtension;
import io.javaoperatorsdk.operator.sample.dependent.SchemaDependentResource;

import static java.util.concurrent.TimeUnit.MINUTES;
import static org.assertj.core.api.Assertions.assertThat;
import static org.awaitility.Awaitility.await;

class MySQLSchemaOperatorE2E {

  static final Logger log = LoggerFactory.getLogger(MySQLSchemaOperatorE2E.class);

  static final KubernetesClient client = new KubernetesClientBuilder().build();

  static final String MY_SQL_NS = "mysql";

  private static final List<HasMetadata> infrastructure = new ArrayList<>();
  public static final String TEST_RESOURCE_NAME = "mydb1";

  static {
    infrastructure.add(
        new NamespaceBuilder().withNewMetadata().withName(MY_SQL_NS).endMetadata().build());
    try {
      infrastructure.addAll(client.load(new FileInputStream("k8s/mysql-db.yaml")).items());
    } catch (FileNotFoundException e) {
      e.printStackTrace();
    }
  }

  boolean isLocal() {
    String deployment = System.getProperty("test.deployment");
    boolean remote = (deployment != null && deployment.equals("remote"));
    log.info("Running the operator " + (remote ? "remotely" : "locally"));
    return !remote;
  }

  @RegisterExtension
  AbstractOperatorExtension operator =
      isLocal()
          ? LocallyRunOperatorExtension.builder()
              .withReconciler(new MySQLSchemaReconciler()) // configuration for schema comes from
              // SchemaDependentResource annotation
              .withInfrastructure(infrastructure)
              .withPortForward(MY_SQL_NS, "app", "mysql", 3306, SchemaDependentResource.LOCAL_PORT)
              .build()
          : ClusterDeployedOperatorExtension.builder()
              .withOperatorDeployment(client.load(new FileInputStream("k8s/operator.yaml")).items())
              .withInfrastructure(infrastructure)
              .build();

  public MySQLSchemaOperatorE2E() throws FileNotFoundException {}

  @Test
  void test() {

    MySQLSchema testSchema = new MySQLSchema();
    testSchema.setMetadata(
        new ObjectMetaBuilder()
            .withName(TEST_RESOURCE_NAME)
            .withNamespace(operator.getNamespace())
            .build());
    testSchema.setSpec(new SchemaSpec());
    testSchema.getSpec().setEncoding("utf8");

    log.info("Creating test MySQLSchema object: {}", testSchema);
    client.resource(testSchema).createOrReplace();

    log.info("Waiting 2 minutes for expected resources to be created and updated");
    await()
        .atMost(2, MINUTES)
        .ignoreExceptions()
        .untilAsserted(
            () -> {
              MySQLSchema updatedSchema =
                  client
                      .resources(MySQLSchema.class)
                      .inNamespace(operator.getNamespace())
                      .withName(testSchema.getMetadata().getName())
                      .get();
              assertThat(updatedSchema.getStatus()).isNotNull();
              assertThat(updatedSchema.getStatus().getStatus()).isEqualTo("CREATED");
              assertThat(updatedSchema.getStatus().getSecretName()).isNotNull();
              assertThat(updatedSchema.getStatus().getUserName()).isNotNull();
            });

    client
        .resources(MySQLSchema.class)
        .inNamespace(operator.getNamespace())
        .withName(testSchema.getMetadata().getName())
        .delete();

    await()
        .atMost(4, MINUTES)
        .ignoreExceptions()
        .untilAsserted(
            () -> {
              MySQLSchema updatedSchema =
                  client
                      .resources(MySQLSchema.class)
                      .inNamespace(operator.getNamespace())
                      .withName(testSchema.getMetadata().getName())
                      .get();
              assertThat(updatedSchema).isNull();
            });
  }
}
