import glob
import getopt
import os
import sys

print("Anti-CF hosts file generator v1.0")
print("Licensed under CC0 1.0")
print("")

#functions
def progress(value, filename):
    sys.stdout.write('\033[2K\033[1G') #wipe the line before printing
    sys.stdout.write("Compiling rules [%s] %s" % ("{:,}".format(value), filename))
    sys.stdout.flush()

opts, argv = getopt.getopt(sys.argv[1:], "o:d:")

def process(outputf, inputd):
    if not os.path.isdir(inputd):
        print("Input not directory");
        sys.exit(1)

    with open(outputf, 'a') as outfile:
        rulescom = 0 #counter for counting how much rules compiled

        files = glob.glob(os.path.join(inputd, "*.txt")) #filter out anything but txt files

        for f in files:
            with open(f) as ruleso:
                for line in ruleso:
                    outfile.write("0.0.0.0  {}\n".format(line.rstrip())) #this does the job
                    rulescom = rulescom + 1
                    progress(rulescom, f)

if len(sys.argv) < 5: #check if theres argv
    print("Usage: generate.py -o <output file> -d <input directory>")
    print("Example: generate.py -o filter.txt -d ../../cloudflare_users/domains/")
    sys.exit(1)

#argv parsing
for k, v in opts:
    if k == '-o':
        outputfile = v
    if k == '-d':
        inputdirectory = v

process(outputfile, inputdirectory)