// Copyright 2020 Lukas Haubaum
//
// Licensed under the GNU Affero General Public License, Version 3;
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at

//     https://www.gnu.org/licenses/agpl-3.0.html
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
#include <string.h>
#include <time.h>
#include <limits.h>

#include "esp_err.h"
#include "esp_log.h"

#include "ena-crypto.h"
#include "ena-storage.h"
#include "ena-beacons.h"

#include "pb_decode.h"
#include "TemporaryExposureKeyExport.pb.h"

#include "ena-exposure.h"

static ena_exposure_config_t DEFAULT_ENA_EXPOSURE_CONFIG = {
    // transmission_risk_values
    {
        MINIMAL,   // UNKNOWN
        LOW,       // CONFIRMED_TEST_LOW
        MEDIUM,    // CONFIRMED_TEST_STANDARD
        HIGH,      // CONFIRMED_TEST_HIGH
        VERY_HIGH, // CONFIRMED_CLINICAL_DIAGNOSIS
        VERY_LOW,  // SELF_REPORT
        ZERO,      // NEGATIVE
        MINIMAL    // RECURSIVE
    },
    // duration_risk_values
    {
        MINIMAL,   // D = 0 min
        MINIMAL,   // D <= 5 min
        MEDIUM,    // D <= 10 min
        VERY_HIGH, // D <= 15 min
        VERY_HIGH, // D <= 20 min
        MAXIMUM,   // D <= 25 min
        MAXIMUM,   // D <= 30 min
        MAXIMUM    //  D > 30 min
    },
    // days_risk_values
    {
        MINIMAL,  // >= 14 days
        VERY_LOW, // 12-13 days
        VERY_LOW, // 10-11 days
        MEDIUM,   // 8-9 days
        HIGH,     // 6-7 days
        MAXIMUM,  // 4-5 days
        MAXIMUM,  // 2-3 days
        MAXIMUM   // 0-1 days
    },
    // attenuation_risk_values
    {
        MINIMAL, // A > 73 dB
        MINIMAL, // 73 >= A > 63
        MINIMAL, // 63 >= A > 61
        MAXIMUM, // 51 >= A > 33
        MAXIMUM, // 33 >= A > 27
        MAXIMUM, // 27 >= A > 15
        MAXIMUM, // 15 >= A > 10
        MAXIMUM  // A <= 10
    },
};

static const char kFileHeader[] = "EK Export v1    ";
static size_t kFileHeaderSize = sizeof(kFileHeader) - 1;

extern uint8_t export_bin_start[] asm("_binary_export_bin_start"); // test data from Google or https://svc90.main.px.t-online.de/version/v1/diagnosis-keys/country/DE/date/2020-07-22
extern uint8_t export_bin_end[] asm("_binary_export_bin_end");

int ena_exposure_risk_score(ena_exposure_config_t *config, ena_exposure_parameter_t params)
{
    int score = 1;
    score *= config->transmission_risk_values[params.report_type];

    // calc duration level
    int duration_level = MINUTES_0;
    if (params.duration > 0)
    {
        if (params.duration <= 5)
        {
            duration_level = MINUTES_5;
        }
        else if (params.duration <= 10)
        {
            duration_level = MINUTES_10;
        }
        if (params.duration <= 15)
        {
            duration_level = MINUTES_15;
        }
        if (params.duration <= 20)
        {
            duration_level = MINUTES_20;
        }
        if (params.duration <= 25)
        {
            duration_level = MINUTES_25;
        }
        if (params.duration <= 30)
        {
            duration_level = MINUTES_30;
        }
        else
        {
            duration_level = MINUTES_LONGER;
        }
    }
    score *= config->duration_risk_values[duration_level];

    // calc days level
    int days_level = DAYS_14;

    if (params.days < 2)
    {
        days_level = DAYS_0;
    }
    else if (params.days < 4)
    {
        days_level = DAYS_3;
    }
    else if (params.days < 6)
    {
        days_level = DAYS_5;
    }
    else if (params.days < 8)
    {
        days_level = DAYS_7;
    }
    else if (params.days < 10)
    {
        days_level = DAYS_9;
    }
    else if (params.days < 12)
    {
        days_level = DAYS_11;
    }
    else if (params.days < 14)
    {
        days_level = DAYS_13;
    }

    score *= config->days_risk_values[days_level];

    // calc attenuation level
    int attenuation_level = ATTENUATION_73;

    if (params.attenuation <= 10)
    {
        attenuation_level = ATTENUATION_LOWER;
    }
    else if (params.attenuation <= 15)
    {
        attenuation_level = ATTENUATION_10;
    }
    else if (params.attenuation <= 27)
    {
        attenuation_level = ATTENUATION_15;
    }
    else if (params.attenuation <= 33)
    {
        attenuation_level = ATTENUATION_27;
    }
    else if (params.attenuation <= 51)
    {
        attenuation_level = ATTENUATION_33;
    }
    else if (params.attenuation <= 63)
    {
        attenuation_level = ATTENUATION_51;
    }
    else if (params.attenuation <= 73)
    {
        attenuation_level = ATTENUATION_63;
    }

    score *= config->attenuation_risk_values[attenuation_level];

    if (score > 255)
    {
        score = 255;
    }

    return score;
}

void ena_exposure_summary(ena_exposure_config_t *config, ena_exposure_summary_t *summary)
{
    // XXX TEST key export (should be called on other location though)
    ESP_ERROR_CHECK_WITHOUT_ABORT(ena_exposure_check_export(export_bin_start, (export_bin_end - export_bin_start)));

    uint32_t count = ena_storage_exposure_information_count();
    uint32_t current_time = (uint32_t)time(NULL);

    summary->days_since_last_exposure = INT_MAX;
    summary->max_risk_score = 0;
    summary->risk_score_sum = 0;
    summary->num_exposures = count;

    if (count == 0)
    {
        summary->days_since_last_exposure = -1;
    }

    ena_exposure_information_t exposure_info;
    ena_exposure_parameter_t params;
    for (int i = 0; i < count; i++)
    {
        ena_storage_get_exposure_information(i, &exposure_info);
        params.days = (current_time - exposure_info.day) / (60 * 60 * 24); // difference in days
        if (params.days < summary->days_since_last_exposure)
        {
            summary->days_since_last_exposure = params.days;
        }
        params.duration = exposure_info.duration_minutes;
        params.attenuation = exposure_info.typical_attenuation;
        int score = ena_exposure_risk_score(config, params);
        if (score > summary->max_risk_score)
        {
            summary->max_risk_score = score;
        }
        summary->risk_score_sum += score;
    }
}

ena_exposure_config_t *ena_exposure_default_config(void)
{
    return &DEFAULT_ENA_EXPOSURE_CONFIG;
}

bool ena_exposure_decode_key_data(pb_istream_t *stream, const pb_field_t *field, void **arg)
{
    uint8_t *key_data = (uint8_t *)*arg;

    if (!pb_read(stream, key_data, stream->bytes_left))
    {
        ESP_LOGW(ENA_EXPOSURE_LOG, "Decoding failed: %s\n", PB_GET_ERROR(stream));
        return false;
    }

    return true;
}

bool ena_exposure_decode_key(pb_istream_t *stream, const pb_field_t *field, void **arg)
{
    uint8_t key_data[ENA_KEY_LENGTH] = {0};
    TemporaryExposureKey tek = TemporaryExposureKey_init_zero;

    tek.key_data = (pb_callback_t){
        .funcs.decode = ena_exposure_decode_key_data,
        .arg = &key_data,
    };

    if (!pb_decode(stream, TemporaryExposureKey_fields, &tek))
    {
        ESP_LOGW(ENA_EXPOSURE_LOG, "Decoding failed: %s\n", PB_GET_ERROR(stream));
        return false;
    }

    ESP_LOGD(ENA_EXPOSURE_LOG,
             "check reported tek: rolling_start_interval_number %d, rolling_period %d, days_since_last_exposure %d, report_type %d, transmission_risk_values %d",
             tek.rolling_start_interval_number, tek.rolling_period, tek.days_since_onset_of_symptoms, tek.report_type, tek.transmission_risk_level);

    ESP_LOG_BUFFER_HEXDUMP(ENA_EXPOSURE_LOG, &key_data, ENA_KEY_LENGTH, ESP_LOG_DEBUG);

    bool match = false;
    ena_beacon_t beacon;
    ena_exposure_information_t exposure_info;
    exposure_info.duration_minutes = 0;
    exposure_info.min_attenuation = INT_MAX;
    exposure_info.typical_attenuation = 0;
    exposure_info.report_type = tek.report_type;
    uint8_t rpi[ENA_KEY_LENGTH];
    uint8_t rpik[ENA_KEY_LENGTH];
    ena_crypto_rpik(rpik, key_data);
    uint32_t beacons_count = ena_storage_beacons_count();
    for (int i = 0; i < tek.rolling_period; i++)
    {
        ena_crypto_rpi(rpi, rpik, tek.rolling_start_interval_number + i);
        for (int y = 0; y < beacons_count; y++)
        {
            ena_storage_get_beacon(y, &beacon);
            if (memcmp(beacon.rpi, rpi, sizeof(ENA_KEY_LENGTH)) == 0)
            {
                match = true;
                exposure_info.day = tek.rolling_start_interval_number * ENA_TIME_WINDOW;
                exposure_info.duration_minutes += (ENA_BEACON_TRESHOLD / 60);
                exposure_info.typical_attenuation = (exposure_info.typical_attenuation + beacon.rssi) / 2;
                if (beacon.rssi < exposure_info.min_attenuation)
                {
                    exposure_info.min_attenuation = beacon.rssi;
                }
            }
        }
    }

    if (match)
    {
        ena_storage_add_exposure_information(&exposure_info);
    }

    return true;
}

esp_err_t ena_exposure_check_export(uint8_t *buf, size_t size)
{
    // validate header
    if (memcmp(kFileHeader, buf, kFileHeaderSize) != 0)
    {
        ESP_LOGW(ENA_EXPOSURE_LOG, "Wrong or missing header!");
        return ESP_FAIL;
    }

    TemporaryExposureKeyExport tek_export = TemporaryExposureKeyExport_init_zero;

    tek_export.keys = (pb_callback_t){
        .funcs.decode = ena_exposure_decode_key,
    };

    pb_istream_t stream = pb_istream_from_buffer(&buf[kFileHeaderSize], (size - kFileHeaderSize));

    if (!pb_decode(&stream, TemporaryExposureKeyExport_fields, &tek_export))
    {
        ESP_LOGW(ENA_EXPOSURE_LOG, "Decoding failed: %s\n", PB_GET_ERROR(&stream));
        return ESP_FAIL;
    }

    return ESP_OK;
}