import socket
import config_pb2


geoiplist = config_pb2.GeoIPList()
geoip = geoiplist.entry.add()
geoip.country_code = 'CN'
for l in open('china_ip_list.txt', 'r').readlines():
    l = l.strip()
    network, prefix = l.split('/')
    cidr = geoip.cidr.add()
    cidr.prefix = int(prefix)
    cidr.ip = socket.inet_aton(network)
open('china-ip-list.dat', 'wb').write(geoiplist.SerializeToString())

geositelist = config_pb2.GeoSiteList()
geosite = geositelist.entry.add()
geosite.country_code = 'CN'
for l in open('accelerated-domains.china.raw.txt', 'r').readlines():
    l = l.strip()
    domain = geosite.domain.add()
    domain.type = 2
    domain.value = l
open('dnsmasq-china-list.dat', 'wb').write(geositelist.SerializeToString())