# -*- coding: utf-8 -*-
import io

path = r'C:\Users\jcxiong\Documents\Codex\MyProject\trafficAudit\traffic-audit-server\src\main\java\com\trafficaudit\reportexport\service\ReportExportService.java'
with io.open(path, encoding='utf-8') as f:
    content = f.read()
has_crlf = '\r\n' in content
lines = content.split('\n')

def func_span(name):
    start = next(i for i, l in enumerate(lines) if l.strip().startswith('public byte[] ' + name + '('))
    end = len(lines)
    for i in range(start+1, len(lines)):
        if lines[i].strip().startswith('public byte[] '):
            end = i
            break
    return start, end

def process_rank(func, metric, unit, title_suffix, city_writer, province_writer):
    fs, fe = func_span(func)
    i = next(i for i in range(fs, fe) if lines[i].strip() == 'int rowIdx = 3;')
    j = i
    while j > fs and 'Row head1 = sheet.createRow(' not in lines[j]:
        j -= 1
    assert 'Row head1' in lines[j], func + ' head1 not found'
    lines[j:i] = ['        rowIdx = writeRankHead(sheet, rowIdx, "%s", "%s");' % (metric, unit)]
    auto_i = next(m for m in range(j+1, fe) if lines[m].strip() == 'autoWidth(sheet, 18);')
    insert = [
        '',
        '        // 本月区块（当月标题 + 累计数据，与样例结构一致）',
        '        sheet.createRow(rowIdx++);',
        '        Row title2 = sheet.createRow(rowIdx++);',
        '        title2.createCell(0).setCellValue(year + "年" + month + "月全省分市州累计完成" + "%s");' % title_suffix,
        '        rowIdx = writeRankHead(sheet, rowIdx, "%s", "%s");' % (metric, unit),
        '        rowIdx = %s;' % province_writer,
        '        for (String city : RegionUtil.cityList()) {',
        '            rowIdx = %s;' % city_writer,
        '        }',
        '',
        '        // 末尾空行（与样例结构对齐）',
        '        for (int i = 0; i < 4; i++) sheet.createRow(rowIdx++);',
        '',
    ]
    lines[auto_i:auto_i] = insert
    print('processed', func)

process_rank('exportTurnoverRank', '周转量', '万吨公里', '公路货物运输周转量情况',
             'writeTurnoverRankRows(sheet, rowIdx, city, cumMap, false)',
             'writeTurnoverRankRows(sheet, rowIdx, null, cumMap, true)')
process_rank('exportFreightRank', '货运量', '万吨', '公路货运量情况',
             'writeRankRows(sheet, rowIdx, city, cumMap, h2032FreightCum, h2032FreightYoy, false)',
             'writeRankRows(sheet, rowIdx, null, cumMap, h2032FreightCum, h2032FreightYoy, true)')

# Change A: blank row before head2 in exportFreightTurnover
blank = [i for i, l in enumerate(lines) if l.strip() == 'sheet.createRow(rowIdx++);']
target = [i for i in blank if i+1 < len(lines) and 'Row head2 = sheet.createRow(rowIdx++);' in lines[i+1]]
assert len(target) == 1, target
del lines[target[0]]
print('removed blank row at', target[0]+1)

# helper before writeRankRows
anchor = next(i for i, l in enumerate(lines) if l.strip().startswith('private int writeRankRows('))
helper = [
    '    /** 排名表双行表头 */',
    '    private int writeRankHead(Sheet sheet, int rowIdx, String metric, String unit) {',
    '        Row h1 = sheet.createRow(rowIdx++);',
    '        h1.createCell(0).setCellValue("市州");',
    '        h1.createCell(1).setCellValue("规上" + metric);',
    '        h1.createCell(4).setCellValue("规上增速");',
    '        h1.createCell(6).setCellValue("规下" + metric);',
    '        h1.createCell(9).setCellValue("规下增速");',
    '        h1.createCell(11).setCellValue("合计" + metric);',
    '        h1.createCell(14).setCellValue("合计增速");',
    '        h1.createCell(16).setCellValue("分市州规上规下占比");',
    '',
    '        Row h2 = sheet.createRow(rowIdx++);',
    '        h2.createCell(1).setCellValue("累计完成    (万" + unit + "）");',
    '        h2.createCell(2).setCellValue("排名");',
    '        h2.createCell(3).setCellValue("占全省比重");',
    '        h2.createCell(4).setCellValue("同比");',
    '        h2.createCell(5).setCellValue("增速    排名");',
    '        h2.createCell(6).setCellValue("累计完成    (万" + unit + "）");',
    '        h2.createCell(7).setCellValue("排名");',
    '        h2.createCell(8).setCellValue("占全省比重");',
    '        h2.createCell(9).setCellValue("同比");',
    '        h2.createCell(10).setCellValue("增速排名");',
    '        h2.createCell(11).setCellValue("累计完成    (万" + unit + "）");',
    '        h2.createCell(12).setCellValue("排名");',
    '        h2.createCell(13).setCellValue("占全省比重");',
    '        h2.createCell(14).setCellValue("同比");',
    '        h2.createCell(15).setCellValue("增速排名");',
    '        return rowIdx;',
    '    }',
    '',
    '',
]
lines[anchor:anchor] = helper
print('helper inserted')

new_content = '\n'.join(lines)
if has_crlf:
    new_content = new_content.replace('\n', '\r\n')
with io.open(path, 'w', encoding='utf-8', newline='') as f:
    f.write(new_content)
print('saved')