import datetime

import pytest
from dateutil.tz import tzutc

import brightsky
from brightsky.export import DBExporter, SYNOPExporter


SOURCES = [
    # Ordered by their distance to (52, 7.6)
    {
        'observation_type': 'forecast',
        'lat': 51.97,
        'lon': 7.63,
        'height': 60,
        'station_name': '<NAME>',
        'dwd_station_id': None,
        'wmo_station_id': 'P0036',
    },
    {
        'observation_type': 'current',
        'lat': 52.13,
        'lon': 7.7,
        'height': 48,
        'station_name': 'MUENSTER/OSNABR.',
        'dwd_station_id': '01766',
        'wmo_station_id': '10315',
    },
    {
        'observation_type': 'recent',
        'lat': 52.1344,
        'lon': 7.6969,
        'height': 47.8,
        'station_name': 'Münster/Osnabrück',
        'dwd_station_id': '01766',
        'wmo_station_id': '10315',
    },
]
RECENT_RECORDS = [
    {
        'timestamp': (
            datetime.datetime(2020, 8, 19, 12, tzinfo=tzutc())
            + datetime.timedelta(hours=i)),
        **SOURCES[2],
        'temperature': 100 + i,
    }
    for i in range(40)
]
CURRENT_RECORDS = [
    {
        'timestamp': (
            datetime.datetime(2020, 8, 20, 12, tzinfo=tzutc())
            + datetime.timedelta(hours=i)),
        **SOURCES[1],
        'temperature': 200 + i,
    }
    for i in range(24)
]
FORECAST_RECORDS = [
    {
        'timestamp': (
            datetime.datetime(2020, 8, 21, 8, tzinfo=tzutc())
            + datetime.timedelta(hours=i)),
        **SOURCES[0],
        'temperature': 300 + i,
    }
    for i in range(40)
]

ALL_FIELDS_RECORD = {
    'temperature': 294.35,
    'precipitation': 1.4,
    'pressure_msl': 100620,
    'sunshine': 1740,
    'wind_direction': 130,
    'wind_speed': 2.2,
    'cloud_cover': 87,
    'dew_point': 287.65,
    'visibility': 40000,
    'wind_gust_direction': 130,
    'wind_gust_speed': 3.9,
    'condition': 'fog',
}
RECENT_RECORDS[12].update(ALL_FIELDS_RECORD)

CONDITION_FIELDS = [
    {'condition': 'fog', '_expected_icon': 'fog'},
    {'condition': 'dry', '_expected_icon': 'clear-night'},
    {'condition': 'dry', 'cloud_cover': 50.,
     '_expected_icon': 'partly-cloudy-night'},
    {'condition': 'sleet', '_expected_icon': 'sleet'},
    {'condition': 'snow', '_expected_icon': 'snow'},
    {'condition': 'hail', '_expected_icon': 'hail'},
    {'condition': 'thunderstorm', '_expected_icon': 'thunderstorm'},
    {'condition': 'dry', '_expected_icon': 'clear-day'},
    {'condition': 'rain', '_expected_icon': 'rain'},
    {'condition': 'dry', 'cloud_cover': 50.,
     '_expected_icon': 'partly-cloudy-day'},
    {'condition': 'dry', 'cloud_cover': 100., '_expected_icon': 'cloudy'},
    {'precipitation': 0.3, '_expected_icon': 'clear-day'},
    {'precipitation': 10., '_expected_icon': 'rain'},
]
for i, fields in enumerate(CONDITION_FIELDS):
    RECENT_RECORDS[12+i].update(fields)


SYNOP_SOURCE = {
    'observation_type': 'synop',
    'lat': 52.1344,
    'lon': 7.69685,
    'height': 47.8,
    'station_name': 'Muenster/Osnabrueck',
    'dwd_station_id': '01766',
    'wmo_station_id': '10315',
}
SYNOP_NOW = datetime.datetime.utcnow().replace(
    minute=0, second=0, microsecond=0, tzinfo=tzutc()
) + datetime.timedelta(hours=1)
SYNOP_RECORDS = [
    {
        'timestamp': SYNOP_NOW - datetime.timedelta(minutes=90),
        **SYNOP_SOURCE,
        'temperature': 300,
    },
    {
        'timestamp': SYNOP_NOW - datetime.timedelta(minutes=50),
        **SYNOP_SOURCE,
        'precipitation_10': 0.2,
        'wind_speed_10': 2.0,
        'wind_direction_10': 10,
        'wind_gust_speed_10': 2.3,
        'wind_gust_direction_10': 130,
    },
    {
        'timestamp': SYNOP_NOW - datetime.timedelta(minutes=40),
        **SYNOP_SOURCE,
        'precipitation_10': 0.3,
        'wind_speed_10': 1.8,
        'wind_direction_10': 350,
        'wind_gust_speed_10': 5.7,
        'wind_gust_direction_10': 60,
    },
    {
        'timestamp': SYNOP_NOW - datetime.timedelta(minutes=30),
        **SYNOP_SOURCE,
        'cloud_cover': None,
        'dew_point': 292.17,
        'precipitation_10': 0.,
        'pressure_msl': 100880,
        'relative_humidity': 80,
        'sunshine_30': 1440,
        'temperature': 296.15,
        'visibility': 35000,
        'wind_direction_10': 340,
        'wind_speed_10': 1.2,
        'wind_gust_direction_10': 20,
        'wind_gust_speed_10': 2.1,
    },
    {
        'timestamp': SYNOP_NOW - datetime.timedelta(minutes=20),
        **SYNOP_SOURCE,
        'precipitation_10': 0.,
        'wind_speed_10': 0.8,
        'wind_direction_10': 30,
        'wind_gust_speed_10': 3.3,
        'wind_gust_direction_10': 30,
    },
    {
        'timestamp': SYNOP_NOW - datetime.timedelta(minutes=10),
        **SYNOP_SOURCE,
        'precipitation_10': 0.1,
        'wind_speed_10': 2.2,
        'wind_direction_10': 20,
        'wind_gust_speed_10': 5.1,
        'wind_gust_direction_10': 40,
    },
    {
        'timestamp': SYNOP_NOW,
        **SYNOP_SOURCE,
        'cloud_cover': 88,
        'dew_point': 292.47,
        'precipitation_10': 0,
        'pressure_msl': 100920,
        'relative_humidity': 80,
        'sunshine_60': 3000,
        'temperature': 296.05,
        'wind_direction_10': 350,
        'wind_speed_10': 2.1,
        'wind_gust_direction_10': 30,
        'wind_gust_speed_10': 3.1,
        'condition': 'rain',
    },
]


@pytest.fixture
def data(db):
    records = RECENT_RECORDS + CURRENT_RECORDS + FORECAST_RECORDS
    DBExporter().export(records)


@pytest.fixture
def synop_data(db):
    SYNOPExporter().export(SYNOP_RECORDS)


def test_sources_required_parameters(data, api):
    assert api.simulate_get('/sources').status_code == 400
    assert api.simulate_get('/sources?lat=52').status_code == 400
    assert api.simulate_get('/sources?lon=7.6').status_code == 400
    assert api.simulate_get('/sources?lat=52&lon=7.6').status_code == 200
    assert api.simulate_get('/sources?wmo_station_id=10315').status_code == 200
    assert api.simulate_get('/sources?dwd_station_id=01766').status_code == 200


def test_sources_response(data, api):
    resp = api.simulate_get('/sources?lat=52&lon=7.6')
    assert resp.status_code == 200
    resp_sources = resp.json['sources']
    assert len(resp_sources) == 3
    for resp_source, source in zip(resp_sources, SOURCES):
        for k, v in source.items():
            assert resp_source[k] == v
    assert [s['distance'] for s in resp.json['sources']] == [
        3922, 16008, 16365]
    assert resp_sources[0]['first_record'] == (
        FORECAST_RECORDS[0]['timestamp'].isoformat())
    assert resp_sources[0]['last_record'] == (
        FORECAST_RECORDS[-1]['timestamp'].isoformat())


def test_sources_max_dist(data, api):
    resp = api.simulate_get('/sources?lat=52&lon=7.6')
    assert len(resp.json['sources']) == 3
    resp = api.simulate_get('/sources?lat=52&lon=7.6&max_dist=5000')
    assert len(resp.json['sources']) == 1


def test_sources_by_station_id(data, api):
    paths = [
        '/sources?wmo_station_id=10315',
        '/sources?dwd_station_id=01766',
    ]
    for path in paths:
        resp = api.simulate_get(path)
        assert resp.status_code == 200
        assert len(resp.json['sources']) == 2


def test_source_by_source_id(db, data, api):
    source_id = db.fetch(
        "SELECT id FROM sources WHERE observation_type = 'forecast'")[0]['id']
    resp = api.simulate_get(f'/sources?source_id={source_id}')
    assert len(resp.json['sources']) == 1
    assert resp.json['sources'][0]['observation_type'] == 'forecast'


def test_no_sources_available(data, api):
    assert api.simulate_get('/sources?lat=0&lon=0').status_code == 404
    assert api.simulate_get('/sources?wmo_station_id=12345').status_code == 404
    assert api.simulate_get('/sources?dwd_station_id=12345').status_code == 404


def test_weather_required_parameters(data, api):
    assert api.simulate_get('/weather').status_code == 400
    assert api.simulate_get('/weather?lat=52&lon=7.6').status_code == 400
    assert api.simulate_get(
        '/weather?lat=52&lon=7.6&date=2020-08-20').status_code == 200
    assert api.simulate_get(
        '/weather?lat=52&lon=7.6&last_date=2020-08-20').status_code == 400


def test_weather_response(data, api):
    resp = api.simulate_get('/weather?lat=52&lon=7.6&date=2020-08-20')
    assert len(resp.json['weather']) == 25
    assert len(resp.json['sources']) == 1
    assert all(
        w['source_id'] == resp.json['sources'][0]['id']
        for w in resp.json['weather'])
    assert resp.json['weather'][0]['timestamp'] == '2020-08-20T00:00:00+00:00'
    assert resp.json['weather'][-1]['timestamp'] == '2020-08-21T00:00:00+00:00'
    for w, record in zip(resp.json['weather'], RECENT_RECORDS[12:]):
        assert w['temperature'] == round(record['temperature'] - 273.15, 2)


def test_weather_date_range(data, api):
    resp = api.simulate_get(
        '/weather?lat=52&lon=7.6&date=2020-08-20&last_date=2020-08-22')
    assert len(resp.json['weather']) == 49
    assert resp.json['weather'][0]['timestamp'] == '2020-08-20T00:00:00+00:00'
    assert resp.json['weather'][-1]['timestamp'] == '2020-08-22T00:00:00+00:00'
    resp = api.simulate_get('/weather?lat=52&lon=7.6&date=2020-08-20T12:00')
    assert len(resp.json['weather']) == 25
    assert resp.json['weather'][0]['timestamp'] == '2020-08-20T12:00:00+00:00'
    assert resp.json['weather'][-1]['timestamp'] == '2020-08-21T12:00:00+00:00'


def test_weather_source_selection(data, api):
    resp = api.simulate_get(
        '/weather?lat=52&lon=7.6&date=2020-08-20&last_date=2020-08-22')
    assert len(resp.json['sources']) == 3
    observation_types = {
        s['id']: s['observation_type'] for s in resp.json['sources']}
    for w in resp.json['weather'][:28]:
        assert observation_types[w['source_id']] == 'recent'
    for w in resp.json['weather'][28:36]:
        assert observation_types[w['source_id']] == 'current'
    for w in resp.json['weather'][36:]:
        assert observation_types[w['source_id']] == 'forecast'


def test_weather_units(data, api):
    resp_dwd = api.simulate_get('/weather?lat=52&lon=7.6&date=2020-08-20')
    resp_si = api.simulate_get(
        '/weather?lat=52&lon=7.6&date=2020-08-20&units=si')
    expected_conversions = {
        'temperature': 21.2,
        'pressure_msl': 1006.2,
        'sunshine': 29,
        'wind_speed': 7.9,
        'dew_point': 14.5,
        'wind_gust_speed': 14.,
    }
    for k, v in ALL_FIELDS_RECORD.items():
        assert resp_dwd.json['weather'][0][k] == expected_conversions.get(k, v)
        assert resp_si.json['weather'][0][k] == v


def test_weather_timezone(data, api):
    resp = api.simulate_get(
        '/weather?lat=52&lon=7.6&date=2020-08-20&tz=Europe/Berlin')
    assert resp.json['weather'][0]['timestamp'] == '2020-08-20T00:00:00+02:00'
    assert resp.json['weather'][-1]['timestamp'] == '2020-08-21T00:00:00+02:00'
    # date offset should be used if tz not supplied
    resp = api.simulate_get(
        '/weather?lat=52&lon=7.6&date=2020-08-20T00:00:00%2b02:00')
    assert resp.json['weather'][0]['timestamp'] == '2020-08-20T00:00:00+02:00'
    assert resp.json['weather'][-1]['timestamp'] == '2020-08-21T00:00:00+02:00'
    # but tz should always be preferred
    resp = api.simulate_get(
        '/weather?lat=52&lon=7.6&date=2020-08-20T00:00:00-04:00'
        '&tz=Europe/Berlin')
    assert resp.json['weather'][0]['timestamp'] == '2020-08-20T06:00:00+02:00'
    assert resp.json['weather'][-1]['timestamp'] == '2020-08-21T06:00:00+02:00'
    # last_date's offset should be used for range selection but not as response
    # timezone
    resp = api.simulate_get(
        '/weather?lat=52&lon=7.6&date=2020-08-20'
        '&last_date=2020-08-20T12:00-04:00')
    assert resp.json['weather'][0]['timestamp'] == '2020-08-20T00:00:00+00:00'
    assert resp.json['weather'][-1]['timestamp'] == '2020-08-20T16:00:00+00:00'
    # date's offset should NOT be used as default last_date offset
    resp = api.simulate_get(
        '/weather?lat=52&lon=7.6&date=2020-08-20T00:00:00%2b02:00'
        '&last_date=2020-08-22')
    assert resp.json['weather'][0]['timestamp'] == '2020-08-20T00:00:00+02:00'
    assert resp.json['weather'][-1]['timestamp'] == '2020-08-22T02:00:00+02:00'


def test_weather_icon(data, api):
    resp = api.simulate_get('/weather?lat=52&lon=7.6&date=2020-08-20')
    for condition, record in zip(CONDITION_FIELDS, resp.json['weather']):
        assert record['icon'] == condition['_expected_icon']


def test_synop_disallows_lat_lon(data, api):
    resp = api.simulate_get(
        '/synop',
        params={
            'lat': 52,
            'lon': 7.6,
            'date': SYNOP_RECORDS[0]['timestamp'].isoformat(),
        })
    assert resp.status_code == 400


def test_synop_response(synop_data, api):
    resp = api.simulate_get(
        '/synop',
        params={
            'wmo_station_id': SYNOP_SOURCE['wmo_station_id'],
            'date': SYNOP_RECORDS[0]['timestamp'].isoformat(),
        })
    assert resp.status_code == 200
    assert len(resp.json['sources']) == 1
    assert len(resp.json['weather']) == len(SYNOP_RECORDS)
    for w, record in zip(resp.json['weather'], SYNOP_RECORDS):
        assert w['timestamp'] == record['timestamp'].isoformat()


def test_current_weather_response(synop_data, api, db):
    # XXX: This test may be flaky as the concurrent refresh may not have
    #      finished yet. Can we somehow wait until the lock is released?
    resp = api.simulate_get('/current_weather?lat=52&lon=7.6')
    assert resp.status_code == 200
    assert len(resp.json['sources']) == 1
    expected_weather = {
        # From latest record
        'timestamp': SYNOP_NOW.isoformat(),
        'cloud_cover': 88,
        'condition': 'rain',
        'dew_point': 19.32,
        'precipitation_10': 0.,
        'pressure_msl': 1009.2,
        'relative_humidity': 80,
        'temperature': 22.9,
        'wind_direction_10': 350,
        'wind_speed_10': 7.6,
        'wind_gust_direction_10': 30,
        'wind_gust_speed_10': 11.2,
        # Summed from latest three/six records
        'precipitation_30': 0.1,
        'precipitation_60': 0.6,
        # Averaged from latest three/six records
        'wind_speed_30': 6.1,
        'wind_speed_60': 6.1,
        # Averaged (for circular quantities) from latest three/six records
        'wind_direction_30': 13,
        'wind_direction_60': 3,
        # Filled up from previous record
        'visibility': 35000,
        # Maximum from previous three/six records
        'wind_gust_direction_30': 40,
        'wind_gust_direction_60': 60,
        'wind_gust_speed_30': 18.4,
        'wind_gust_speed_60': 20.5,
        # Calculated from latest available _30 and _60 values
        'sunshine_30': 26,
        'sunshine_60': 50,
    }
    for k, v in expected_weather.items():
        assert resp.json['weather'][k] == v, k


def test_status_response(api):
    resp = api.simulate_get('/')
    assert resp.status_code == 200
    assert resp.json['name'] == 'brightsky'
    assert resp.json['version'] == brightsky.__version__
    assert resp.json['status'] == 'ok'
