package shkd.repc.repmd.template;
import kd.bos.context.RequestContext;
import kd.bos.dataentity.entity.DynamicObject;
import kd.bos.dataentity.entity.DynamicObjectCollection;
import kd.bos.dataentity.utils.StringUtils;
import kd.bos.entity.ExtendedDataEntity;
import kd.bos.entity.plugin.AbstractOperationServicePlugIn;
import kd.bos.entity.plugin.AddValidatorsEventArgs;
import kd.bos.entity.plugin.args.AfterOperationArgs;
import kd.bos.entity.validate.AbstractValidator;
import kd.bos.form.*;
import kd.bos.form.control.Control;
import kd.bos.form.control.EntryGrid;
import kd.bos.form.control.events.ItemClickEvent;
import kd.bos.form.events.MessageBoxClosedEvent;
import kd.bos.form.plugin.AbstractFormPlugin;
import kd.bos.orm.query.QCP;
import kd.bos.orm.query.QFilter;
import kd.bos.servicehelper.BusinessDataServiceHelper;
import kd.sdk.plugin.Plugin;
import org.apache.poi.xssf.usermodel.XSSFCell;
import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import shkd.repc.repmd.template.util.ExportTemplateUtil;
import shkd.repc.repmd.template.util.Template;
import java.math.BigDecimal;
import java.util.*;
/**
* 面积数据导出-二开插件
*/
public class AreaDataExportPlugin extends AbstractFormPlugin implements Plugin {
@Override
public void registerListener(EventObject e) {
super.registerListener(e);
this.addItemClickListeners("qeug_advcontoolbarap");
}
@Override
public void itemClick(ItemClickEvent evt) {
super.itemClick(evt);
if (StringUtils.equals("qeug_areadataexport", evt.getItemKey())) {
DynamicObjectCollection productEntry = this.getModel().getEntryEntity("productentry");
if (null==productEntry||productEntry.size()==0){
this.getView().showMessage("暂无可导出数据！");
return;
}
EntryGrid entryGrid = this.getControl("productentry");
int[] selectRows = entryGrid.getSelectRows();
String top="导出确认提示";
String body;
//默认导出
if (selectRows.length==0){
body = "默认导出所有【产品构成】数据？";
this.showExportMessage(top,body,"default");
}else {
body = "导出 " + selectRows.length + " 行【产品构成】数据？";
this.showExportMessage(top,body,"select");
}
}
}
@Override
public void confirmCallBack(MessageBoxClosedEvent evt) {
super.confirmCallBack(evt);
String callBackId = evt.getCallBackId();
StringBuilder tipsBuffer = new StringBuilder();
if (StringUtils.equalsIgnoreCase("export_area", callBackId) && evt.getResult().getValue() == 6) {
String customValue = evt.getCustomVaule();
boolean isDefaultExport = "default".equals(customValue);
// 获取模板
DynamicObject templateObj = getTemplate();
if (templateObj == null) {
return;
}
// 读取系统模板
Control productView = this.getView().getControl("productentry");
Control areaView = this.getView().getControl("qeug_subentryentity");
List<Template> templateList = ExportTemplateUtil.readSysTemplate(templateObj, productView, areaView, tipsBuffer);
if (templateList.isEmpty()) {
return;
}
// 创建 Excel
XSSFWorkbook workbook = new XSSFWorkbook();
XSSFSheet sheet = createExcelSheet(workbook, templateList, areaView);
// 获取表头信息（第三行）
List<String> columnHeaders = getColumnHeaders(sheet);
// 获取数据
DynamicObjectCollection productEntry = this.getModel().getEntryEntity("productentry");
List<DynamicObject> exportData = isDefaultExport ? new ArrayList<>(productEntry) : getSelectedData(productEntry);
// 填充数据
fillExcelData(sheet, exportData, columnHeaders);
// 上传文件并提供下载链接
exportAndDownload(workbook, customValue);
tipsBuffer.append("导出成功");
this.getView().showSuccessNotification(tipsBuffer.toString());
}
}
/**
* 获取模板对象
*/
private DynamicObject getTemplate() {
QFilter numberFilter = new QFilter("number", QCP.equals, "repmd_projectbill_IMPT_ENTRY");
return BusinessDataServiceHelper.loadSingle("bos_importentry_template", numberFilter.toArray());
}
/**
* 创建 Excel Sheet 并填充标题、描述、表头等
*/
private XSSFSheet createExcelSheet(XSSFWorkbook workbook, List<Template> templateList, Control areaView) {
XSSFSheet sheet = workbook.createSheet("产品构成 # productentry");
ExportTemplateUtil.createTitleRow(sheet, workbook);
ExportTemplateUtil.createDescriptionRow(sheet, workbook);
ExportTemplateUtil.createTemplateHeaders(sheet, workbook, templateList);
ExportTemplateUtil.otherStyleSet(sheet, areaView);
return sheet;
}
/**
* 获取第三行的表头信息
*/
private List<String> getColumnHeaders(XSSFSheet sheet) {
List<String> columnHeaders = new ArrayList<>();
XSSFRow thirdRow = sheet.getRow(2);
if (thirdRow != null) {
for (int cellIndex = 0; cellIndex < thirdRow.getLastCellNum(); cellIndex++) {
XSSFCell cell = thirdRow.getCell(cellIndex);
columnHeaders.add(cell != null ? cell.getStringCellValue().trim() : ""); // 避免 null
}
}
return columnHeaders;
}
/**
* 获取选中的数据
*/
private List<DynamicObject> getSelectedData(DynamicObjectCollection productEntry) {
List<DynamicObject> selectedData = new ArrayList<>();
EntryGrid entryGrid = this.getControl("productentry");
int[] selectedRows = entryGrid.getSelectRows();
for (int row : selectedRows) {
selectedData.add(productEntry.get(row));
}
return selectedData;
}
/**
* 填充 Excel 数据
*/
private void fillExcelData(XSSFSheet sheet, List<DynamicObject> productEntries, List<String> columnHeaders) {
int rowIndex = 4; // 从 Excel 第 5 行开始填充
for (DynamicObject productObj : productEntries) {
if (productObj == null) continue;
String productTypeNumber = productObj.getString("productentry_producttype.number");
String productTypeName = productObj.getString("productentry_producttype.name");
DynamicObjectCollection areaEntries = productObj.getDynamicObjectCollection("qeug_subentryentity");
if (areaEntries != null && !areaEntries.isEmpty()) {
for (DynamicObject areaObj : areaEntries) {
XSSFRow dataRow = sheet.createRow(rowIndex);
int cellIndex = 0;
for (String header : columnHeaders) {
XSSFCell cell = dataRow.createCell(cellIndex);
if ("productentry_producttype.number".equals(header)) {
cell.setCellValue(productTypeNumber);
} else if ("productentry_producttype.name".equals(header)) {
cell.setCellValue(productTypeName);
} else {
Object value = areaObj.get(header);
setCellValue(cell, value);
}
cellIndex++;
}
rowIndex++;
}
}
}
}
/**
* 设置 Excel 单元格值
*/
private void setCellValue(XSSFCell cell, Object value) {
if (value instanceof String) {
cell.setCellValue((String) value);
} else if (value instanceof BigDecimal) {
cell.setCellValue(((BigDecimal) value).doubleValue());
} else if (value instanceof Integer) {
cell.setCellValue((Integer) value);
} else if (value instanceof Double) {
cell.setCellValue((Double) value);
} else if (value != null) {
cell.setCellValue(value.toString());
}
}
/**
* 导出 Excel 并提供下载链接
*/
private void exportAndDownload(XSSFWorkbook workbook, String fileName) {
String uploadedFilePath = ExportTemplateUtil.uploadExcel(workbook, fileName);
String url = RequestContext.get().getClientFullContextPath() + "/attachment/download.do?path=" + uploadedFilePath;
this.getView().openUrl(url);
}
private void showExportMessage(String top,String body,String customValue){
// 基本信息
// 备注: \r\n 可换行
//String msg = "消息头\r\n message header...";
// 详细消息
//String detail = "消息体\r\nmessage body...";
// 消息框按钮类型
// None:不启用 Toast:简短提示,几秒后消失
MessageBoxOptions options = MessageBoxOptions.OKCancel;
// 确认提示类型
// Default:默认提示 Save:保存提交类提示 Delete:删除类提示 Wait:等待类提示 Fail:失败类提示
ConfirmTypes confirmTypes = ConfirmTypes.Default;
// 确认框回调
ConfirmCallBackListener callBack = new ConfirmCallBackListener("export_area", this);
// 按钮名称
// 注意: 按钮名称参数Map的key值必须和 options 参数相对应, 否则按钮名称修改不生效!
// 例: options 使用 MessageBoxOptions.OKCancel, 则 按钮名称参数Map的key必须为2 or 6
// options 使用 MessageBoxOptions.AbortRetryIgnore, 则 按钮名称参数Map的key必须为3 or 4 or 5
// none-0; ok-1; cancel-2; abort-3; retry-4; ignore-5; yes-6; no-7; custom-8
// -----------------------------------------------------------------
// options | btnNameMaps的key的取值
// MessageBoxOptions.None | 2
// MessageBoxOptions.OK | 2
// MessageBoxOptions.OKCancel | 2 & 6
// MessageBoxOptions.AbortRetryIgnore | 3 & 4 & 5
// MessageBoxOptions.YesNoCancel | 6 & 7 & 2
// MessageBoxOptions.YesNo | 6 & 7
// MessageBoxOptions.RetryCancel | 4 & 2
// MessageBoxOptions.Toast | 2
// -----------------------------------------------------------------
Map<Integer, String> btnNameMaps = new HashMap<>();
btnNameMaps.put(2, "取消");
btnNameMaps.put(6, "确认");
// 用户自定义参数,前端会在afterConfirm中返回
//String customValue = "allowExport";
this.getView().showConfirm(top, body, options, confirmTypes, callBack, btnNameMaps,customValue);
}
}