package software.amazon.databrew.dataset;

import com.google.common.collect.Lists;
import software.amazon.awssdk.services.databrew.model.Dataset;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.Map;

import static org.assertj.core.api.Assertions.assertThat;

public class TestUtil {

    public static final String DATASET_NAME = "dataset-test";
    public static final String INVALID_DATASET_NAME = "invalid-dataset-test";
    public static final String S3_BUCKET = "s3://test-dataset-input-bucket";
    public static final String S3_KEY = "input.json";
    public static final String S3_FOLDER_KEY = "abc/";
    public static final String S3_REGEX_KEY = "abc/<\\w+>/def";
    public static final String S3_PARAM_KEY = "abc/{str1}/def";
    public static final String UPDATED_S3_KEY = "input.xlsx";
    public static final String CSV_S3_KEY = "input.csv";
    public static final String TSV_S3_KEY = "input.tsv";
    public static final String EXCEL_S3_KEY = "input.xlsx";
    public static final String ORC_S3_KEY = "input.orc";
    public static final String METADATA_SOURCE_ARN = "arn:aws:appflow:us-east-1:123456789012:flow/test-flow";
    public static final String JSON_S3_KEY_EXTENSIONLESS = "input";
    public static final String PIPE_CSV_DELIMITER = "|";
    public static final String TAB_CSV_DELIMITER = "\t";
    public static final String INVALID_CSV_DELIMITER = "*";
    public static final String CSV_FORMAT = "CSV";
    public static final String EXCEL_FORMAT = "EXCEL";
    public static final String JSON_FORMAT = "JSON";
    public static final String PARQUET_FORMAT = "PARQUET";
    public static final String ORC_FORMAT = "ORC";
    public static final String GLUE_CONNECTION_NAME = "test-connection-name";
    public static final String DATABASE_TABLE_NAME = "test-table-name";
    public static final String DATABASE_INPUT_SQL_STR = "SELECT * FROM SCHEMA.TABLE";
    public static final S3Location s3InputDefinition = S3Location.builder()
            .bucket(S3_BUCKET)
            .key(S3_KEY)
            .build();
    public static final DatabaseInputDefinition databaseSQLInputDefinition = DatabaseInputDefinition.builder()
            .glueConnectionName(GLUE_CONNECTION_NAME)
            .databaseTableName(DATABASE_TABLE_NAME)
            .queryString(DATABASE_INPUT_SQL_STR)
            .tempDirectory(s3InputDefinition)
            .build();
    public static final Metadata metadataInput = Metadata.builder()
            .sourceArn(METADATA_SOURCE_ARN)
            .build();
    public static final DatabaseInputDefinition databaseInputDefinition = DatabaseInputDefinition.builder()
            .glueConnectionName(GLUE_CONNECTION_NAME)
            .databaseTableName(DATABASE_TABLE_NAME)
            .tempDirectory(s3InputDefinition)
            .build();
    public static final Input DATABASE_SQL_INPUT = Input.builder()
            .databaseInputDefinition(databaseSQLInputDefinition)
            .build();
    public static final Input METADATA_INPUT_DATASET = Input.builder()
            .s3InputDefinition(s3InputDefinition)
            .metadata(metadataInput)
            .build();
    public static final Input DATABASE_INPUT = Input.builder()
            .databaseInputDefinition(databaseInputDefinition)
            .build();
    public static final DatabaseInputDefinition updatedDatabaseInputDefinition = DatabaseInputDefinition.builder()
            .glueConnectionName(GLUE_CONNECTION_NAME)
            .databaseTableName(DATABASE_TABLE_NAME)
            .build();
    public static final Input UPDATED_DATABASE_INPUT = Input.builder()
            .databaseInputDefinition(updatedDatabaseInputDefinition)
            .build();
    public static final Input S3_INPUT = Input.builder()
            .s3InputDefinition(s3InputDefinition)
            .build();
    public static final S3Location updatedS3InputDefinition = S3Location.builder()
            .bucket(S3_BUCKET)
            .key(UPDATED_S3_KEY)
            .build();
    public static final Input UPDATED_S3_INPUT = Input.builder()
            .s3InputDefinition(updatedS3InputDefinition)
            .build();

    public static final Input S3_FOLDER_INPUT = Input.builder().s3InputDefinition(S3Location.builder()
            .bucket(S3_BUCKET)
            .key(S3_FOLDER_KEY)
            .build()).build();

    public static final Input S3_REGEX_INPUT = Input.builder().s3InputDefinition(S3Location.builder()
            .bucket(S3_BUCKET)
            .key(S3_REGEX_KEY)
            .build()).build();

    public static final Input S3_PARAM_INPUT = Input.builder().s3InputDefinition(S3Location.builder()
            .bucket(S3_BUCKET)
            .key(S3_PARAM_KEY)
            .build()).build();

    public static final S3Location csvS3InputDefinition = S3Location.builder()
            .bucket(S3_BUCKET)
            .key(CSV_S3_KEY)
            .build();
    public static final Input CSV_S3_INPUT = Input.builder()
            .s3InputDefinition(csvS3InputDefinition)
            .build();
    public static final S3Location tsvS3InputDefinition = S3Location.builder()
            .bucket(S3_BUCKET)
            .key(TSV_S3_KEY)
            .build();
    public static final Input TSV_S3_INPUT = Input.builder()
            .s3InputDefinition(tsvS3InputDefinition)
            .build();
    public static final S3Location excelS3InputDefinition = S3Location.builder()
            .bucket(S3_BUCKET)
            .key(EXCEL_S3_KEY)
            .build();
    public static final S3Location orcS3InputDefinition = S3Location.builder()
            .bucket(S3_BUCKET)
            .key(ORC_S3_KEY)
            .build();
    public static final Input EXCEL_S3_INPUT = Input.builder()
            .s3InputDefinition(excelS3InputDefinition)
            .build();
    public static final Input ORC_S3_INPUT = Input.builder()
            .s3InputDefinition(orcS3InputDefinition)
            .build();
    public static final S3Location jsonS3InputDefinition = S3Location.builder()
            .bucket(S3_BUCKET)
            .key(JSON_S3_KEY_EXTENSIONLESS)
            .build();
    public static final Input JSON_S3_INPUT_EXTENSIONLESS = Input.builder()
            .s3InputDefinition(jsonS3InputDefinition)
            .build();
    public static final JsonOptions JSON_OPTIONS = JsonOptions.builder()
            .multiLine(true)
            .build();
    public static final FormatOptions JSON_FORMAT_OPTIONS = FormatOptions.builder()
            .json(JSON_OPTIONS)
            .build();
    public static final ExcelOptions EXCEL_OPTIONS_INDEXES = ExcelOptions.builder()
            .sheetIndexes(new ArrayList<Integer>() {{
                add(1);
            }})
            .build();
    public static final FormatOptions EXCEL_FORMAT_OPTIONS_INDEXES = FormatOptions.builder()
            .excel(EXCEL_OPTIONS_INDEXES)
            .build();
    public static final ExcelOptions EXCEL_OPTIONS_NAMES = ExcelOptions.builder()
            .sheetNames(new ArrayList<String>() {{
                add("test");
            }})
            .build();
    public static final FormatOptions EXCEL_FORMAT_OPTIONS_NAMES = FormatOptions.builder()
            .excel(EXCEL_OPTIONS_NAMES)
            .build();
    public static final CsvOptions CSV_OPTIONS = CsvOptions.builder()
            .delimiter(PIPE_CSV_DELIMITER)
            .build();
    public static final FormatOptions CSV_FORMAT_OPTIONS = FormatOptions.builder()
            .csv(CSV_OPTIONS)
            .build();
    public static final CsvOptions INVALID_CSV_OPTIONS = CsvOptions.builder()
            .delimiter(INVALID_CSV_DELIMITER)
            .build();
    public static final FormatOptions INVALID_CSV_FORMAT_OPTIONS = FormatOptions.builder()
            .csv(INVALID_CSV_OPTIONS)
            .build();
    public static final CsvOptions TSV_OPTIONS = CsvOptions.builder()
            .delimiter(TAB_CSV_DELIMITER)
            .build();
    public static final FormatOptions TSV_FORMAT_OPTIONS = FormatOptions.builder()
            .csv(TSV_OPTIONS)
            .build();
    public static final CsvOptions CSV_OPTIONS_HEADERLESS = CsvOptions.builder()
            .delimiter(PIPE_CSV_DELIMITER)
            .headerRow(false)
            .build();
    public static final FormatOptions CSV_FORMAT_OPTIONS_HEADERLESS = FormatOptions.builder()
            .csv(CSV_OPTIONS_HEADERLESS)
            .build();
    public static final ExcelOptions EXCEL_OPTIONS_HEADERLESS = ExcelOptions.builder()
            .sheetNames(new ArrayList<String>() {{
                add("test");
            }})
            .headerRow(false)
            .build();
    public static final FormatOptions EXCEL_FORMAT_OPTIONS_HEADERLESS = FormatOptions.builder()
            .excel(EXCEL_OPTIONS_HEADERLESS)
            .build();

    public static final PathOptions PATH_OPTIONS_WITH_VALID_FILES_LIMIT = PathOptions.builder()
            .filesLimit(FilesLimit.builder().maxFiles(3).build()).build();

    public static final PathOptions PATH_OPTIONS_WITH_INVALID_FILES_LIMIT = PathOptions.builder()
            .filesLimit(FilesLimit.builder().maxFiles(-3).build()).build();

    public static final PathOptions PATH_OPTIONS_WITH_VALID_LAST_MODIFIED = PathOptions.builder()
            .lastModifiedDateCondition(getFilterExpression("relative_after", ":date1", "-2w")
            ).build();

    public static final PathOptions PATH_OPTIONS_WITH_INVALID_LAST_MODIFIED = PathOptions.builder()
            .lastModifiedDateCondition(getFilterExpression("relative_after", ":date1", "two weeks")
            ).build();

    private static FilterExpression getFilterExpression(String condtion, String valueRef, String value) {
        return FilterExpression.builder()
                .expression(String.format("%s %s", condtion, valueRef))
                .valuesMap(Lists.newArrayList(FilterValue.builder()
                        .valueReference(valueRef)
                        .value(value)
                        .build())
                ).build();
    }

    public static final PathOptions PATH_OPTIONS_WITH_VALID_PARAM = PathOptions.builder()
            .parameters(Lists.newArrayList(PathParameter.builder()
                    .pathParameterName("str1")
                    .datasetParameter(DatasetParameter.builder()
                            .name("str1")
                            .type("String")
                            .createColumn(true)
                            .filter(getFilterExpression("contains", ":val1", "abc"))
                            .build())
                    .build())
            ).build();

    public static final PathOptions PATH_OPTIONS_WITH_INVALID_PARAM = PathOptions.builder()
            .parameters(Lists.newArrayList(PathParameter.builder()
                    .pathParameterName("str1")
                    .datasetParameter(DatasetParameter.builder()
                            .name("str1")
                            .type("String")
                            .createColumn(true)
                            .filter(getFilterExpression("after", ":val1", "abc"))
                            .build())
                    .build())
            ).build();

    public static final Map<String, String> sampleTags() {
        Map<String, String> tagMap = new HashMap<>();
        tagMap.put("test1Key", "test1Value");
        tagMap.put("test2Key", "test12Value");
        return tagMap;
    }

    public static void assertThatDatasetModelsAreEqual(final Object rawModel,
                                                       final Dataset sdkModel) {
        assertThat(rawModel).isInstanceOf(ResourceModel.class);
        ResourceModel model = (ResourceModel) rawModel;
        assertThat(model.getName()).isEqualTo(sdkModel.name());
    }
}