#!/usr/bin/env python3

import database
import argparse
import sys
import ipaddress


if __name__ == '__main__':

    # Parsing arguments
    parser = argparse.ArgumentParser(
        description="TODO")
    parser.add_argument(
        'type',
        choices={'subdomains', 'ip4network'},
        help="Type of rule inputed")
    parser.add_argument(
        '-i', '--input', type=argparse.FileType('r'), default=sys.stdin,
        help="List of domains domains to block (with their subdomains)")
    parser.add_argument(
        '-f', '--first-party', action='store_true',
        help="The input only comes from verified first-party sources")
    args = parser.parse_args()

    database.open_db()

    if args.type == 'subdomains':
        for rule in args.input:
            database.feed_rule_subdomains(
                rule.strip(), first_party=args.first_party)
    elif args.type == 'ip4network':
        for rule in args.input:
            network = ipaddress.ip_network(rule.strip())
            database.feed_rule_ip4network(
                network, first_party=args.first_party)
    else:
        assert False

    database.close_db()