#include <furi_hal_region.h>
#include <furi_hal_version.h>
#include <furi.h>

const FuriHalRegion furi_hal_region_zero = {
    .country_code = "00",
    .bands_count = 1,
    .bands = {
        {
            .start = 0,
            .end = 1000000000,
            .power_limit = 12,
            .duty_cycle = 50,
        },
    }};

const FuriHalRegion furi_hal_region_eu_ru = {
    .country_code = "EU",
    .bands_count = 2,
    .bands = {
        {
            .start = 433050000,
            .end = 434790000,
            .power_limit = 12,
            .duty_cycle = 50,
        },
        {
            .start = 868150000,
            .end = 868550000,
            .power_limit = 12,
            .duty_cycle = 50,
        }}};

const FuriHalRegion furi_hal_region_us_ca_au = {
    .country_code = "US",
    .bands_count = 3,
    .bands = {
        {
            .start = 304100000,
            .end = 321950000,
            .power_limit = 12,
            .duty_cycle = 50,
        },
        {
            .start = 433050000,
            .end = 434790000,
            .power_limit = 12,
            .duty_cycle = 50,
        },
        {
            .start = 915000000,
            .end = 928000000,
            .power_limit = 12,
            .duty_cycle = 50,
        }}};

const FuriHalRegion furi_hal_region_jp = {
    .country_code = "JP",
    .bands_count = 2,
    .bands = {
        {
            .start = 312000000,
            .end = 315250000,
            .power_limit = 12,
            .duty_cycle = 50,
        },
        {
            .start = 920500000,
            .end = 923500000,
            .power_limit = 12,
            .duty_cycle = 50,
        }}};

static const FuriHalRegion* furi_hal_region = NULL;

void furi_hal_region_init(void) {
    FuriHalVersionRegion region = furi_hal_version_get_hw_region();

    if(region == FuriHalVersionRegionUnknown) {
        furi_hal_region = &furi_hal_region_zero;
    } else if(region == FuriHalVersionRegionEuRu) {
        furi_hal_region = &furi_hal_region_eu_ru;
    } else if(region == FuriHalVersionRegionUsCaAu) {
        furi_hal_region = &furi_hal_region_us_ca_au;
    } else if(region == FuriHalVersionRegionJp) {
        furi_hal_region = &furi_hal_region_jp;
    }
}

const FuriHalRegion* furi_hal_region_get(void) {
    return furi_hal_region;
}

void furi_hal_region_set(FuriHalRegion* region) {
    furi_check(region);

    furi_hal_region = region;
}

bool furi_hal_region_is_provisioned(void) {
    return furi_hal_region != NULL;
}

const char* furi_hal_region_get_name(void) {
    if(furi_hal_region) {
        return furi_hal_region->country_code;
    } else {
        return "--";
    }
}

bool furi_hal_region_is_frequency_allowed(uint32_t frequency) {
    if(!furi_hal_region) {
        return false;
    }

    const FuriHalRegionBand* band = furi_hal_region_get_band(frequency);
    if(!band) {
        return false;
    }

    return true;
}

const FuriHalRegionBand* furi_hal_region_get_band(uint32_t frequency) {
    if(!furi_hal_region) {
        return NULL;
    }

    for(size_t i = 0; i < furi_hal_region->bands_count; i++) {
        if(furi_hal_region->bands[i].start <= frequency &&
           furi_hal_region->bands[i].end >= frequency) {
            return &furi_hal_region->bands[i];
        }
    }

    return NULL;
}