package cloud.localstack.awssdkv2;
import org.junit.Assert;

import org.junit.Test;
import software.amazon.awssdk.auth.credentials.AwsBasicCredentials;
import software.amazon.awssdk.auth.credentials.StaticCredentialsProvider;
import software.amazon.awssdk.regions.Region;
import software.amazon.awssdk.services.cloudwatch.model.*;
import software.amazon.awssdk.services.cloudwatch.CloudWatchClient;

import java.time.Instant;
import java.time.ZoneOffset;
import java.time.ZonedDateTime;
import java.time.format.DateTimeFormatter;
import java.time.temporal.ChronoUnit;
import java.util.ArrayList;
import java.net.URI;
import java.util.List;

public class BasicFeaturesSDKV2Test {
    @Test
    public void testCWPutMetrics() throws Exception {
        final CloudWatchClient cwClient = CloudWatchClient.builder()
                .credentialsProvider(StaticCredentialsProvider.create(
                        AwsBasicCredentials.create("test", "test")))
                .region(Region.of("us-east-1"))
                .endpointOverride(URI.create("http://localhost.localstack.cloud:4566"))
                .build();

        Dimension dimension = Dimension.builder()
                .name("UNIQUE_PAGES")
                .value("URLS")
                .build();

        // Set an Instant object
        String time = ZonedDateTime.now( ZoneOffset.UTC ).format( DateTimeFormatter.ISO_INSTANT );
        Instant instant = Instant.parse(time);

        double dataPoint = 1.23423;

        MetricDatum datum = MetricDatum.builder()
                .metricName("PAGES_VISITED")
                .unit(StandardUnit.NONE)
                .value(dataPoint)
                .timestamp(instant)
                // .dimensions(dimension)
                .build();

        PutMetricDataRequest request = PutMetricDataRequest.builder()
                .namespace("SITE/TRAFFIC")
                .metricData(datum).build();

        PutMetricDataResponse response =  cwClient.putMetricData(request);
        Assert.assertNotNull(response);

        // GetMetricDataRequest request2 = GetMetricDataRequest.builder()
        // Set the current time and date range for metric query.
        Instant nowDate = Instant.now();
        long hours = 1;
        long minutes = 30;
        Instant startTime = nowDate.minus(5, ChronoUnit.MINUTES);
        Instant endTime = nowDate.plus(hours, ChronoUnit.HOURS).plus(minutes, ChronoUnit.MINUTES);

        Metric met = Metric.builder()
                .metricName("PAGES_VISITED")
                .namespace("SITE/TRAFFIC")
                .build();

        MetricStat metStat = MetricStat.builder()
                .stat("Sum")
                .period(60)  // Assuming period in seconds
                .metric(met)
                .build();

        MetricDataQuery dataQuery = MetricDataQuery.builder()
                .metricStat(metStat)
                .id("foo2")
                .returnData(true)
                .build();

        List<MetricDataQuery> dq = new ArrayList<>();
        dq.add(dataQuery);

        GetMetricDataRequest getMetricDataRequest = GetMetricDataRequest.builder()
                // .maxDatapoints(10)
                // .scanBy(ScanBy.TIMESTAMP_DESCENDING)
                .startTime(startTime)
                .endTime(endTime)
                .metricDataQueries(dq)
                .build();

        // Call the async method for CloudWatch data retrieval.
        GetMetricDataResponse responseGetData = cwClient.getMetricData(getMetricDataRequest);
        Assert.assertNotNull(responseGetData);
        System.out.println(responseGetData.metricDataResults());
    }
}