// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.model.application.validation;

import com.yahoo.collections.Pair;
import com.yahoo.config.application.api.ApplicationPackage;
import com.yahoo.config.model.api.ApplicationClusterEndpoint;
import com.yahoo.config.model.api.ConfigChangeAction;
import com.yahoo.config.model.api.ContainerEndpoint;
import com.yahoo.config.model.api.Provisioned;
import com.yahoo.config.model.deploy.DeployState;
import com.yahoo.config.model.deploy.TestProperties;
import com.yahoo.config.model.provision.InMemoryProvisioner;
import com.yahoo.config.model.test.MockApplicationPackage;
import com.yahoo.config.provision.Environment;
import com.yahoo.config.provision.RegionName;
import com.yahoo.config.provision.SystemName;
import com.yahoo.config.provision.Zone;
import com.yahoo.vespa.model.VespaModel;
import com.yahoo.vespa.model.test.utils.VespaModelCreatorWithMockPkg;

import java.time.Instant;
import java.time.LocalDate;
import java.time.ZoneOffset;
import java.time.format.DateTimeFormatter;
import java.util.Arrays;
import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;
import java.util.stream.Stream;

import static com.yahoo.config.model.test.MockApplicationPackage.BOOK_SCHEMA;
import static com.yahoo.config.model.test.MockApplicationPackage.MUSIC_SCHEMA;

/**
 * @author bratseth
 */
public class ValidationTester {

    private final TestProperties properties;
    private final InMemoryProvisioner hostProvisioner;

    /** Creates a validation tester with 1 node available (in addition to cluster controllers) */
    public ValidationTester() {
        this(5);
    }

    /** Creates a validation tester with number of nodes available and the given test properties */
    public ValidationTester(int nodeCount, boolean sharedHosts, TestProperties properties) {
        this(new InMemoryProvisioner(nodeCount, sharedHosts), properties);
    }

    /** Creates a validation tester with a given host provisioner */
    public ValidationTester(InMemoryProvisioner hostProvisioner) {
        this(hostProvisioner, new TestProperties().setHostedVespa(true));
    }

    /** Creates a validation tester with a number of nodes available */
    public ValidationTester(int nodeCount) {
        this(new InMemoryProvisioner(nodeCount, false), new TestProperties().setHostedVespa(true));
    }

    /** Creates a validation tester with a given host provisioner */
    public ValidationTester(InMemoryProvisioner hostProvisioner, TestProperties testProperties) {
        this.hostProvisioner = hostProvisioner;
        this.properties = testProperties;
        hostProvisioner.setEnvironment(testProperties.zone().environment());
    }

    /**
     * Deploys an application
     *
     * @param previousModel the previous model, or null if no previous
     * @param services the services file content
     * @param environment the environment this deploys to
     * @param validationOverrides the validation overrides file content, or null if none
     * @param containerCluster container cluster(s) which are declared in services
     * @return the new model and any change actions
     */
    public Pair<VespaModel, List<ConfigChangeAction>> deploy(VespaModel previousModel,
                                                             String services,
                                                             Environment environment,
                                                             String validationOverrides,
                                                             String... containerCluster) {
        Instant now = LocalDate.parse("2000-01-01", DateTimeFormatter.ISO_DATE).atStartOfDay().atZone(ZoneOffset.UTC).toInstant();
        Provisioned provisioned = hostProvisioner.startProvisionedRecording();
        ApplicationPackage newApp = new MockApplicationPackage.Builder()
                .withServices(services)
                .withSchemas(List.of(MUSIC_SCHEMA, BOOK_SCHEMA))
                .withValidationOverrides(validationOverrides)
                .build();
        VespaModelCreatorWithMockPkg newModelCreator = new VespaModelCreatorWithMockPkg(newApp);
        Stream<String> clusters = containerCluster.length == 0 ? Stream.of("default") : Arrays.stream(containerCluster);
        Set<ContainerEndpoint> containerEndpoints = clusters.map(name -> new ContainerEndpoint(name,
                                                                                               ApplicationClusterEndpoint.Scope.zone,
                                                                                               List.of(name + ".example.com")))
                                                            .collect(Collectors.toSet());
        DeployState.Builder deployStateBuilder = new DeployState.Builder()
                                                             .zone(new Zone(SystemName.defaultSystem(),
                                                                            environment,
                                                                            RegionName.defaultName()))
                                                             .endpoints(containerEndpoints)
                                                             .applicationPackage(newApp)
                                                             .properties(properties)
                                                             .modelHostProvisioner(hostProvisioner)
                                                             .provisioned(provisioned)
                                                             .now(now);
        if (previousModel != null)
            deployStateBuilder.previousModel(previousModel);
        VespaModel newModel = newModelCreator.create(deployStateBuilder);
        return new Pair<>(newModel, newModelCreator.configChangeActions);
    }

    public static String censorNumbers(String s) {
        return s.replaceAll("\\d", "-");
    }

}