package com.product.print.util;

import com.deepoove.poi.exception.RenderException;
import com.deepoove.poi.render.RenderContext;
import com.deepoove.poi.template.run.RunTemplate;
import com.deepoove.poi.util.TableTools;
import com.product.common.lang.StringUtils;
import com.product.core.exception.BaseException;
import com.product.print.config.CmnCode;
import org.apache.poi.xwpf.usermodel.*;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

/**
 * @Author cheng
 * @Date 2023/6/27 15:50
 * @Desc
 */
public class DynamicTableRenderPolicy extends com.deepoove.poi.policy.DynamicTableRenderPolicy {

	private String replaceKey;

	private final String indexKey = "~index~";

	public DynamicTableRenderPolicy(String replaceKey) {
		this.replaceKey = replaceKey;

	}

	private XWPFRun run;

	@Override
	public void doRender(RenderContext<Object> context) throws Exception {
		RunTemplate runTemplate = (RunTemplate) context.getEleTemplate();
		XWPFRun run = runTemplate.getRun();
		try {
			if (!TableTools.isInsideTable(run)) {
				throw new IllegalStateException(
						"The template tag " + runTemplate.getSource() + " must be inside a table");
			}
			XWPFTableCell cell = (XWPFTableCell) ((XWPFParagraph) run.getParent()).getBody();
			XWPFTable table = cell.getTableRow().getTable();
			render(table, context.getData());
		} catch (Exception e) {
			throw new RenderException("Dynamic render table error:" + e.getMessage(), e);
		}
	}

	@Override
	public void render(XWPFTable xwpfTable, Object data) throws Exception {
		if (xwpfTable == null) {
			return;
		}
		List<Map<String, Object>> subTableData = (List<Map<String, Object>>) data;
		List<XWPFTableRow> rows = xwpfTable.getRows();
		//读取rows中的内容
		//获取最后一行
		XWPFTableRow xwpfTableRow = rows.get(rows.size() - 1);
		String tableExpression = "{{" + this.replaceKey + "}}";
		//获取表格起始行和结束行
		int startRowIndex = -1;
		int endRowIndex = -1;
		rows:
		for (int i = 0; i < rows.size(); i++) {
			XWPFTableRow row = rows.get(i);
			for (int j = 0; j < row.getTableCells().size(); j++) {
				if (startRowIndex > -1 && endRowIndex > -1) {
					break rows;
				}
				XWPFTableCell cell = row.getTableCells().get(j);
				String text = cell.getText();
				if (tableExpression.equals(text)) {
					startRowIndex = i;
				}
				if ("{{$~end~}}".equals(text)) {
					endRowIndex = i;
				}

			}
		}
		//表格起始行不在第一行且没有找到结束标记
		if (startRowIndex > 0 && endRowIndex == -1) {
			throw new BaseException(CmnCode.NOT_FIND_END_FLAG);
		}

		if (startRowIndex == 0 && endRowIndex > 0) {
			//删除结束标记所在行
			xwpfTable.removeRow(endRowIndex);
		}

		//读取最后行每个单元格的值，调用getReplaceKey方法获取表达式中的值
		String[] fieldNames = new String[xwpfTableRow.getTableCells().size()];
//		for (int i = 0; i < xwpfTableRow.getTableCells().size(); i++) {
//			XWPFTableCell cell = xwpfTableRow.getTableCells().get(i);
//			String text = cell.getText();
//			String replaceKey = getReplaceKey(text);
//			//获取表达式中的值
//			if (StringUtils.isEmpty(replaceKey)) {
//				//设置单元格为空值
//				replaceKey = "";
//			}
//			fieldNames[i] = replaceKey;
//		}
//		//读取完毕后删除最后一行
//		xwpfTable.removeRow(rows.size() - 1);
		//获取所有的行判断单元格是否有值没有就删除该行
//		List<XWPFTableRow> tableRows = xwpfTable.getRows();
//		List<XWPFTableRow> deleteRows = new ArrayList<>();
//		for (int i = 0; i < tableRows.size(); i++) {
//			XWPFTableRow row = tableRows.get(i);
//			boolean flag = false;
//			for (int j = 0; j < row.getTableCells().size(); j++) {
//				XWPFTableCell cell = row.getTableCells().get(j);
//				if (StringUtils.isNotEmpty(cell.getText())) {
//					flag = true;
//					break;
//				}
//			}
//			if (!flag) {
//				deleteRows.add(row);
//			}
//		}
//		//遍历要删除的行获取所在的下标进行删除
//		for (int i = 0; i < deleteRows.size(); i++) {
//			XWPFTableRow row = deleteRows.get(i);
//			int index = xwpfTable.getRows().indexOf(row);
//			xwpfTable.removeRow(index);
//		}
//		//遍历数据集合，每个map对应一行数据
//		for (int i = 0; i < subTableData.size(); i++) {
//			Map<String, Object> map = subTableData.get(i);
//			//创建一行
//			XWPFTableRow row = xwpfTable.createRow();
//			//遍历字段每个字段创建一个单元格
//			for (int j = 0; j < fieldNames.length; j++) {
//				//当前单元格
//				XWPFTableCell cell;
//				//判断row中第j个单元格是否存在
//				if (row.getTableCells().size() > j) {
//					cell = row.getTableCells().get(j);
//				} else {
//					cell = row.createCell();
//				}
//				//设置单元格的值从map中取出
//				//判断是否是序号列
//				if (indexKey.equals(fieldNames[j])) {
//					cell.setText(String.valueOf(i + 1));
//					continue;
//				}
//				Object value = map.get(fieldNames[j]);
//				if (value == null) {
//					value = "";
//				}
//				//设置单元格的值
//				cell.setText(value.toString());
//			}
//		}

	}

	/**
	 * 解析表达式中的值获取 以{{$开头，以}}结尾的字符串取出$后面的值截止到}}
	 */
	private String getReplaceKey(String text) {
		//使用正则解析表达式中的值获取 以{{$开头，以}}结尾的字符串取出$后面的值截止到}}
		Pattern pattern = Pattern.compile("\\{\\{\\$(.*?)\\}\\}");
		Matcher matcher = pattern.matcher(text);
		if (matcher.find()) {
			return matcher.group(1);
		}
		return null;
	}
}