import csv, sys, json, glob, re
from os import path
from bs4 import BeautifulSoup as bs4

directions = ['W', 'N', 'E', 'S']

def csv_to_list(file_path):
    file_data = []
    with file(file_path) as csv_file:
        csv_data = csv.reader(csv_file)
        for line in csv_data:
            file_data.append(line)
    return file_data

def format_bidding(bidding):
    html_output = '<table>'
    html_output = html_output + '<tr>'
    for dir in directions:
        html_output = html_output + '<th>' + dir + '</th>'
    html_output = html_output + '</tr>'
    for bid_round in bidding:
        html_output = html_output + '<tr>'
        for bid in bid_round:
            bid_match = re.match(r'(\d)([SHDCN])', bid)
            if bid_match:
		        bid = bid_match.group(1) + '<img src="images/' + bid_match.group(2) + '.gif" />'
            html_output = html_output + '<td>' + bid + '</td>'
        html_output = html_output + '</tr>'
    html_output = html_output + '</table>'
    return html_output

def parse_lineup_data(sitting_data):
    round_lineups = {}
    for sitting in sitting_data[1:]:
        round_no = int(sitting[2])
        table_no = sitting[0] + '_' + sitting[1]
        if not round_lineups.has_key(round_no):
            round_lineups[round_no] = {}
        round_lineups[round_no][table_no] = sorted([int(sitting[3]), int(sitting[4])])
    return round_lineups

def parse_bidding_data(bidding_data):
    bids = {}
    for bid in bidding_data[1:]:
        board_no = int(bid[4])
        round_no = int(bid[3])
        table_no = bid[1] + '_' + bid[2]
        bid_counter = int(bid[5])
        bid_erased = int(bid[10])
        if not bids.has_key(board_no):
            bids[board_no] = {}
        if not bids[board_no].has_key(table_no):
            bids[board_no][table_no] = {}
        if not bids[board_no][table_no].has_key(round_no):
            bids[board_no][table_no][round_no] = {}
        if bid_erased == 1:
            if bids[board_no][table_no][round_no].has_key(bid_counter):
                if bids[board_no][table_no][round_no][bid_counter]['direction'] == bid[6]:
                    bids[board_no][table_no][round_no].pop(bid_counter, None)
        else:
            bids[board_no][table_no][round_no][bid_counter] = {'direction': bid[6], 'bid': bid[7] }
    return bids

bidding_data = csv_to_list(sys.argv[1])
sitting_data = csv_to_list(sys.argv[2])
round_lineups = parse_lineup_data(sitting_data)
bids = parse_bidding_data(bidding_data)

tournament_path_prefix = sys.argv[3] + '.html'
output_path = path.dirname(tournament_path_prefix)
tournament_prefix = path.splitext(path.realpath(tournament_path_prefix))[0]

for board_no, board_data in bids.items():
    for table_no, table_data in board_data.items():
        for round_no, round_data in table_data.items():
            bidding = sorted(round_data)
            dealer = round_data[bidding[0]]['direction']
            bidding_table = [[], [], [], []]
            for bid_index in bidding:
                bid = round_data[bid_index]
                bidding_table[directions.index(bid['direction'])].append(bid['bid'])
                last_bidder = bid['direction']
            for pos in range(0, directions.index(dealer)):
                bidding_table[pos].insert(0, '')
            for pos in range(directions.index(last_bidder), len(directions)):
                bidding_table[pos].append('')
            bidding_table = map(list, zip(*bidding_table))
            # TODO: file_number, a nie deal_number
            bidding_file_path = tournament_prefix + '_bidding_' + str(board_no) + '_' + '_'.join(map(str, round_lineups[round_no][table_no])) + '.txt'
            with file(bidding_file_path, 'w') as bidding_file:
                bidding_file.write(format_bidding(bidding_table))

tournament_files_match = re.compile(re.escape(tournament_prefix) + '([0-9]{3})\.html')
tournament_files = [f for f in glob.glob(tournament_prefix + '*.html') if re.search(tournament_files_match, f)]

deal_numbers = {}

for tournament_file in tournament_files:
    file_number = re.match(tournament_files_match, tournament_file).group(1)
    with file(tournament_file, 'r+') as board_html:
        board_content = bs4(board_html, from_encoding='utf-8')
        header_scripts = board_content.select('head script')
        jquery_scripts = [script for script in header_scripts if script['src'] == 'javas/jquery.js']
        if not len(jquery_scripts):
            board_content.head.append(bs4('<script src="javas/jquery.js" type="text/javascript"></script>').script)
        bidding_scripts = [script for script in header_scripts if script['src'] == 'javas/bidding.js']
        if not len(bidding_scripts):
            board_content.head.append(bs4('<script src="javas/bidding.js" type="text/javascript"></script>').script)
        board_number = board_content.select('h4')[0].contents[0].strip().replace('ROZDANIE ', '')
        deal_numbers[file_number] = board_number
        board_html.seek(0)
        board_html.write(board_content.prettify('utf-8', formatter='html'))
    board_text_path = path.splitext(tournament_file)[0] + '.txt'
    with file(board_text_path, 'r+') as board_text:
        board_text_content = bs4(board_text, from_encoding='iso-8859-2')
        for row in board_text_content.select('tr'):
            cells = row.select('td')
            if len(cells) == 11:
                pair_numbers = sorted([int(cells[1].contents[0]), int(cells[2].contents[0])])
                bidding_link = bs4('<a href="#" class="biddingLink">[lic]</a>')
                # TODO: file_number, a nie deal_number
                bidding_link.a['data-bidding-link'] = path.basename(tournament_prefix) + '_bidding_' + str(deal_numbers[file_number]) + '_' + '_'.join(map(str, pair_numbers)) + '.txt'
                for link in cells[3].select('a.biddingLink'):
                    link.extract()
                cells[3].append(bidding_link)
        board_text.seek(0)
        board_text.write(board_text_content.table.prettify('iso-8859-2', formatter='html'))