#!/usr/bin/env python

import sys
import os

dir = sys.argv[1]
teamno = int(sys.argv[2])

def start_flag():
    os.system('cd flag_server && sh docker.sh')    
    print '[*] start docker flag_server' 

def gen_host_lists():
    res = ''
    for i in range(teamno):
	res += "team%d:172.17.0.%d\n" % (i+1 , i+2)
    open('check_server/host.lists','w').write(res)
	

def start_check():
    gen_host_lists()
    os.system('cd check_server && sh docker.sh')
    print '[*] start docker check_server'

def start_team(teamno):
    team_dir = 'team' + str(teamno)
    os.system('cd %s/%s/ && sh docker.sh'%(dir,team_dir))
    print '[*] start docker %s' % team_dir 

if __name__ == '__main__':
    for i in range(teamno):
	start_team(i+1)
    
    start_check()
    start_flag()

for i in range(teamno):
        os.system('touch ./flag_server/submitted/team%s.txt' % (str(i+1)))
os.system("chmod 777 ./flag_server/submitted/*")