{% macro get_schemas_used(schema_name) %}
    {% set schemas = [
        schema_name,
        schema_name + '_re',
        schema_name + '_re_internal',
        schema_name + '_raw',
        schema_name + '_expected',
        schema_name + '_dbt_test__audit',
        schema_name + '_seeds'
    ] %}
    {{ return (schemas) }}
{% endmacro %}

{% macro drop_all_schemas(schema_name) %}
    {% set schemas_to_drop = get_schemas_used(schema_name) %}
    {{ adapter.dispatch('drop_all_schemas')(schemas_to_drop) }}
{% endmacro %}

{% macro default__drop_all_schemas(schemas_to_drop) %}
    {% for schema in schemas_to_drop %}
        {% set relation = api.Relation.create(database=target.database, schema=schema) %}
        {% do adapter.drop_schema(relation) %}
    {% endfor %}
{% endmacro %}

{% macro redshift__drop_all_schemas(schemas_to_drop) %}
    {# 
        dropping schemas with adapter.drop_schema doesn't seem to work with redshift
        so we default to issuing DDL commands to redshift
    #}
    {% set drop_query %}
        {% for schema in schemas_to_drop %}
            drop schema if exists {{schema}} cascade;
        {% endfor %}
    {% endset %}
    {% do run_query(drop_query) %}
{% endmacro %}

{% macro create_required_schemas(schema_name) %}
    {# required to manually create schemas used for redshift tests #}
    {% set schemas_to_drop = get_schemas_used(schema_name) %}
    {% set create_query %}
        {% for schema in schemas_to_drop %}
            create schema if not exists {{schema}};
        {% endfor %}
    {% endset %}
    {% do run_query(create_query) %}
{% endmacro %}