#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import json

from generator import download, download_to_file, get_abspath_list_file, get_version


def process(file, dst):
    with open(file, 'r') as json_file:
        amazon_aws_ip_list = json.load(json_file)
    l = []

    for prefix in amazon_aws_ip_list['prefixes']:
        l.append(prefix['ip_prefix'])

    for prefix in amazon_aws_ip_list['ipv6_prefixes']:
        l.append(prefix['ipv6_prefix'])

    warninglist = {}
    warninglist['name'] = 'List of known Amazon AWS IP address ranges'
    warninglist['version'] = get_version()
    warninglist['description'] = 'Amazon AWS IP address ranges (https://ip-ranges.amazonaws.com/ip-ranges.json)'
    warninglist['type'] = 'cidr'
    warninglist['list'] = sorted(set(l))
    warninglist['matching_attributes'] = ["ip-src", "ip-dst", "domain|ip"]

    with open(get_abspath_list_file(dst), 'w') as data_file:
        json.dump(warninglist, data_file, indent=2, sort_keys=True)
        data_file.write('\n')


if __name__ == '__main__':
    amazon_url = "https://ip-ranges.amazonaws.com/ip-ranges.json"
    amazon_file = "amazon_ip-ranges.json"
    amazon_dst = "amazon-aws"

    download_to_file(amazon_url, amazon_file)
    process(amazon_file, amazon_dst)