package com.vci.starter.word.bo;

import com.aspose.words.IMailMergeDataSource;
import com.aspose.words.ref.Ref;
import com.vci.starter.web.util.VciBaseUtil;
import org.apache.commons.lang3.StringUtils;
import org.springframework.util.CollectionUtils;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * 用于mergeField的表格形式的数据写入对象，因为aspose.word不支持HashMap
 * @author weidy,dangsn
 */
public class WordMergeListDataSource implements IMailMergeDataSource {
    /**
     * 数据对象
     */
	private List<Map<String, Object>> dataList;

    /**
     * 索引
     */
    private int index;

    /**
     * 表格名称
     */
    private String tableName;

    /**
     * 子表格的名称
     */
    private String childTableName;

    /**
     * @param dataList 数据集 ，数据集中Map的key，需要跟TableStart,TableEnd包含的名称对应
     * @param tableName 与模板中的TableStart,TableEnd对应
     */
    public WordMergeListDataSource(List<Map<String, Object>> dataList, String tableName) {
        this.dataList = dataList;
        this.tableName = tableName;
        index = -1;
        this.setChildTableName(listChildTableNameFromData(this.dataList));
    }

    /**
     * @param data 单个数据集
     * @param tableName 与模板中的Name对应
     */
    public WordMergeListDataSource(Map<String, Object> data, String tableName) {
        if(this.dataList == null) {  
            this.dataList = new ArrayList<Map<String,Object>>();
            this.dataList.add(data);  
        }  
        this.tableName = tableName;  
        index = -1;
        this.setChildTableName(listChildTableNameFromData(this.dataList));
    }

    /**
     * 获取子表名称
     * @return 子表名称，多个会以逗号分隔
     */
    public String getChildTableName() {
        return childTableName;
    }

    /**
     * 设置子表名称
     * @param childTableName 子表名称，多个以逗号分隔
     */
    public void setChildTableName(String childTableName) {
        this.childTableName = childTableName;
    }

    /**
     * 获取结果集总数 
     * @return 个数
     */  
    public int getCount() {
        return this.dataList.size();  
    }

    /**
     * 获取数据列表
     * @return 数据列表
     */
    public List<Map<String, Object>> getDataList(){
        return this.dataList;
    }

    /**
     * 获取表格名称
     * @return 表格的名称
     */
    public String getTableName() {
        return this.tableName;  
    }

    /**
     * 获取具体的值，接口所需要的
     * @param key 字段名称
     * @param ref 值
     * @return 是否有值
     * @throws Exception 获取的时候能出错
     */
    @Override
    public boolean getValue(String key, Ref<Object> ref) throws Exception {
        if(index < 0 || index >= this.getCount()){
            return false;
        }
        if(ref != null){
            ref.set(this.dataList.get(index).get(key));
            return true;
        }else{
            return false;
        }
    }

    /**
     * 从数据中找到list的作为子表的对象
     * @param dataList 数据列表
     * @return 包含子表的名称的连接字符串
     */
    private String listChildTableNameFromData(List<Map<String,Object>> dataList ){
        List<String> nextChildNameList = new ArrayList<>();
        if(!CollectionUtils.isEmpty(dataList)){
            dataList.get(0).forEach((k,v) ->{
                if(v != null && v instanceof List){
                    nextChildNameList.add(k);
                }
            });
        }
        return nextChildNameList.stream().collect(Collectors.joining(","));
    }

    /**
     * 实现的接口
     * @param tableName 子表的名称
     * @return 子表对象，不符合的会返回null
     * @throws Exception 获取出错的时候会抛出异常
     */
    @Override
    public IMailMergeDataSource getChildDataSource(String tableName) throws Exception {
        if(StringUtils.isNotBlank(getChildTableName())){
            List<String> childrenTableName = VciBaseUtil.str2List(getChildTableName());
            if(childrenTableName.contains(tableName)){
                Object childData = this.dataList.get(index).get(tableName);
                if(childData != null && childData instanceof List) {
                    List<Map<String,Object>> childDataList = (List<Map<String,Object>>)childData;
                    if(!CollectionUtils.isEmpty(childDataList)) {
                        WordMergeListDataSource childDataSource = new WordMergeListDataSource(childDataList, tableName);
                        childDataSource.setChildTableName(listChildTableNameFromData(childDataList));
                        return childDataSource;
                    }
                }
            }
        }
        return null;
    }

    /** 
     * 实现接口 
     * 判断是否还有下一条记录
     * @return true移动到下一条
     */  
    @Override
    public boolean moveNext() {
        index += 1;  
        if(index >= this.getCount()) {
            return false;  
        }  
        return true;  
    }

	
}