/src_configDesign/org/hxzon/configdesigner/core/CfgParser.java

package org.hxzon.configdesigner.core;



import java.util.ArrayList;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import java.util.concurrent.atomic.AtomicInteger;



import org.dom4j.Element;

import org.hxzon.util.Dom4jUtil;

import org.hxzon.util.Dt;



import com.alibaba.fastjson.JSONArray;

import com.alibaba.fastjson.JSONObject;



public class CfgParser {

    public static final Object Null = null;//JSONObject.NULL;//有定义，值为null

    public static final Object Undefined = null;//未定义



    private static Map<String, CfgInfo> typeInfos = new HashMap<String, CfgInfo>();

    private static Map<String, CfgInfo> entityTypeInfos = new HashMap<String, CfgInfo>();

    private static Map<String, AtomicInteger> entityIdSeqs = new HashMap<String, AtomicInteger>();



    public static CfgInfo parseSchema(String xmlStr) {

        Element root = Dom4jUtil.getRoot(xmlStr);

        return toCfgInfo_root(root);

    }



    private static CfgInfo toCfgInfo_root(Element e) {

        CfgInfo cfgInfo = new CfgInfo();

        String id = Dom4jUtil.getText(e, "@id");

        if (id == null) {

            id = e.getName();

        }

        cfgInfo.setId(id);

        cfgInfo.setLabel(Dom4jUtil.getText(e, "@label"));

        //

        cfgInfo.setType(CfgType.Struct);

        fillPartsInfo(cfgInfo, e);

        return cfgInfo;

    }



    private static CfgInfo toCfgInfo(Element e) {

        CfgInfo cfgInfo = new CfgInfo();

        String typeDef = Dom4jUtil.getText(e, "@typeDef");

        if (typeDef != null) {//类型定义

            if (!typeDef.isEmpty()) {

                cfgInfo.setId(typeDef);

            } else {

                cfgInfo.setId(e.getName());

            }

            typeInfos.put(cfgInfo.getId(), cfgInfo);

        } else {//非类型定义

            String id = Dom4jUtil.getText(e, "@id");

            if (id == null || id.isEmpty()) {

                id = e.getName();

            }

            cfgInfo.setId(id);

        }

        cfgInfo.setLabel(Dom4jUtil.getText(e, "@label"));



        String typeStr = Dom4jUtil.getText(e, "@type");

        CfgInfo typeRef = typeInfos.get(typeStr);

        if (typeRef != null) {

            cfgInfo.setTypeRef(typeRef);

            cfgInfo.setType(typeRef.getType());

        } else {

            CfgType type = CfgType.parseCfgType(typeStr);

            if (type == null) {

                type = CfgType.Struct;

            }

            cfgInfo.setType(type);

            if (type.isStruct()) {

                fillPartsInfo(cfgInfo, e);

            }

            if (type.isElementContainer()) {

                cfgInfo.setElementInfo(toCfgInfo(Dom4jUtil.getElement(e)));

            }

        }

        CfgType type = cfgInfo.getType();

        if (type.isSimple()) {

            cfgInfo.setDefaultValue(converValue(type, Dom4jUtil.getText(e, "@value")));

            String limit = Dom4jUtil.getText(e, "@limit");

            if (limit != null) {

                cfgInfo.setValidator(new CfgValueValidator(cfgInfo.getType(), limit));

            }

        }

        if (type == CfgType.Struct) {

            //vst不能直接作为元素的类型，所以不需要 labelKey

            cfgInfo.setLabelKey(Dom4jUtil.getText(e, "@labelKey"));

            cfgInfo.setKeyKey(Dom4jUtil.getText(e, "@keyKey"));

        }

        if (type == CfgType.String) {

            cfgInfo.setTextArea(tristateValue(e, "@textarea"));

        }

        cfgInfo.setEmbed(tristateValue(e, "@embed"));

        cfgInfo.setOptional(tristateValue(e, "@optional"));

        //

        if (typeDef != null) {

            return null;//not part

        }

        //实体定义

        String idPrefix = Dom4jUtil.getText(e, "@idPrefix");

        if (idPrefix != null) {

            cfgInfo.setIdPrefix(idPrefix);

            entityTypeInfos.put(cfgInfo.getId(), cfgInfo);

            entityIdSeqs.put(idPrefix, new AtomicInteger(0));

            //实体不能作为类型，否则导致实体在别处定义？

            CfgInfo listMapInfo = new CfgInfo();

            listMapInfo.setType(CfgType.ListMap);

            listMapInfo.setElementInfo(cfgInfo);

            listMapInfo.setId(cfgInfo.getId());

            listMapInfo.setLabel(cfgInfo.getLabel());

            return listMapInfo;

        }

        return cfgInfo;

    }



    private static String tristateValue(Element e, String path) {

        String value = Dom4jUtil.getText(e, path);

        if (value == null) {

            return null;

        }

        return CfgInfo.True.equals(value) ? CfgInfo.True : CfgInfo.False;

    }



    private static void fillPartsInfo(CfgInfo cfgInfo, Element e) {

        List<CfgInfo> partsInfo = new ArrayList<CfgInfo>();

        List<Element> childrenEle = Dom4jUtil.getElements(e);

        for (Element childEle : childrenEle) {

            CfgInfo partInfo = toCfgInfo(childEle);

            if (partInfo != null) {

                partsInfo.add(partInfo);

            }

        }

        cfgInfo.setPartsInfo(partsInfo);

    }



    public static Object converValue(CfgType type, String strValue) {

        if (type == CfgType.Boolean) {

            return "true".equals(strValue);

        }

        if (type == CfgType.Integer) {

            return strValue == null ? 0 : Integer.valueOf(strValue);

        }

        if (type == CfgType.Real) {

            return strValue == null ? 0.0 : Double.valueOf(strValue);

        }

        if (type == CfgType.String) {

            return strValue;

        }

        return Undefined;

    }



    //==============

    public static CfgValue buildCfgValue(CfgInfo cfgInfo, Object json, //

            int notNullValueDeep, int nullValueDeep) {

        if (cfgInfo.isOptional() && notNullValueDeep <= 0) {

            return null;

        }

        if (cfgInfo.isOptional() && json == null && nullValueDeep <= 0) {

            return null;

        }

        CfgType type = cfgInfo.getType();

        if (type.isSimple()) {

            return buildSimpleCfgValue(cfgInfo, json, notNullValueDeep, nullValueDeep);

        }

        if (type == CfgType.List) {

            return buildListCfgValue(cfgInfo, json, notNullValueDeep, nullValueDeep);

        }

        if (type == CfgType.ListMap) {

            return buildListMapCfgValue(cfgInfo, json, notNullValueDeep, nullValueDeep);

        }

        if (type == CfgType.Map) {

            return buildMapCfgValue(cfgInfo, json, notNullValueDeep, nullValueDeep);

        }

        if (type.isStruct()) {

            return buildStructCfgValue(cfgInfo, json, notNullValueDeep, nullValueDeep);

        }

        throw new RuntimeException("type error");

    }



    private static CfgValue buildSimpleCfgValue(CfgInfo cfgInfo, Object json, //

            int notNullValueDeep, int nullValueDeep) {

        CfgValue cfgValue = new CfgValue(cfgInfo);

        cfgValue.setValue(json != null ? json : cfgInfo.getDefaultValue());

        return cfgValue;

    }



    private static CfgValue buildStructCfgValue(CfgInfo mapCfgInfo, Object mapJson, //

            int notNullValueDeep, int nullValueDeep) {

        CfgValue cfgValue = new CfgValue(mapCfgInfo);

        for (CfgInfo partInfo : mapCfgInfo.getPartsInfo()) {

            Object partJson = null;

            if (partInfo.getType() == CfgType.ViewStruct) {

                partJson = mapJson;

            } else if (mapJson != null && mapJson instanceof JSONObject) {

                JSONObject jsonObj = (JSONObject) mapJson;

                partJson = jsonObj.get(partInfo.getId());

            }

            CfgValue part = buildCfgValue(partInfo, partJson, notNullValueDeep - 1, nullValueDeep - 1);

            if (part != null) {

                cfgValue.addValue(part);//add part

            }

        }

        return cfgValue;

    }



    private static CfgValue buildListCfgValue(CfgInfo listCfgInfo, Object listJson, //

            int notNullValueDeep, int nullValueDeep) {

        CfgValue listCfgValue = new CfgValue(listCfgInfo);

        CfgInfo eCfgInfo = listCfgInfo.getElementInfo();

        if (listJson != null && listJson instanceof JSONArray) {

            JSONArray jsonArray = (JSONArray) listJson;

            for (Object eleJson : jsonArray) {

                CfgValue e = buildCfgValue(eCfgInfo, eleJson, notNullValueDeep - 1, nullValueDeep - 1);

                listCfgValue.addValue(e);//add element

            }

        }

        return listCfgValue;

    }



    private static CfgValue buildMapCfgValue(CfgInfo mapsCfgInfo, Object mapsJson, //

            int notNullValueDeep, int nullValueDeep) {

        CfgValue mapsCfgValue = new CfgValue(mapsCfgInfo);

        CfgInfo eCfgInfo = mapsCfgInfo.getElementInfo();

        if (mapsJson != null && mapsJson instanceof JSONObject) {

            JSONObject jsonObj = (JSONObject) mapsJson;

            for (String key : jsonObj.keySet()) {

                CfgValue e = buildCfgValue(eCfgInfo, jsonObj.get(key), notNullValueDeep - 1, nullValueDeep - 1);

                e.setKey(key);

                mapsCfgValue.addValue(e);//add element

            }

        }

        return mapsCfgValue;

    }



    private static CfgValue buildListMapCfgValue(CfgInfo listMapsCfgInfo, Object mapsJson, //

            int notNullValueDeep, int nullValueDeep) {

        CfgValue listMapsCfgValue = new CfgValue(listMapsCfgInfo);

        CfgInfo eCfgInfo = listMapsCfgInfo.getElementInfo();

        if (mapsJson != null && mapsJson instanceof JSONObject) {

            JSONObject jsonObj = (JSONObject) mapsJson;

            for (String key : jsonObj.keySet()) {

                CfgValue e = buildCfgValue(eCfgInfo, jsonObj.get(key), notNullValueDeep - 1, nullValueDeep - 1);

                //e.setKey(key);//not use

                listMapsCfgValue.addValue(e);//add element

            }

        }

        return listMapsCfgValue;

    }



    public static CfgValue buildListElementCfgValue(CfgValue parent, int deep) {

        if (deep <= 0) {

            return null;

        }

        CfgInfo listCfgInfo = parent.getCfgInfo();

        if (listCfgInfo.getType().isElementContainer()) {

            CfgInfo eCfgInfo = listCfgInfo.getElementInfo();

            CfgValue r = buildCfgValue(eCfgInfo, null, 1, deep);

            r.setParent(parent);

            return r;

        }

        throw new RuntimeException("type error");

    }



    public static Object toJson(CfgValue root, boolean trim) {

        CfgInfo cfgInfo = root.getCfgInfo();

        CfgType type = cfgInfo.getType();

        if (type == CfgType.Boolean) {

            boolean rb = Dt.toBoolean(root.getValue(), false);

            return (trim && !rb) ? Undefined : rb;

        }

        if (type == CfgType.Integer) {

            long rl = Dt.toLong(root.getValue(), 0);

            return (trim && rl == 0) ? Undefined : rl;

        }

        if (type == CfgType.Real) {

            double rr = Dt.toDouble(root.getValue(), 0);

            return (trim && rr == 0) ? Undefined : rr;

        }

        if (type == CfgType.String) {

            String rs = Dt.toString(root.getValue(), "");

            return (trim && rs.isEmpty()) ? Undefined : rs;//new MyString(rs);

        }

        if (type.isStruct()) {

            JSONObject json = new JSONObject();

            for (CfgValue e : root.getValues()) {

                Object re = toJson(e, trim);

                if (!trim || re != Undefined) {

                    if (e.getCfgInfo().getType() == CfgType.ViewStruct) {

                        JSONObject jsonRe = (JSONObject) re;

                        for (String key : jsonRe.keySet()) {

                            json.put(key, jsonRe.get(key));

                        }

                    } else {

                        json.put(e.getCfgInfo().getId(), re);

                    }

                }

            }

            return (trim && json.isEmpty()) ? Undefined : json;

        }

        if (type == CfgType.List) {

            JSONArray jsonA = new JSONArray();

            for (CfgValue e : root.getValues()) {

                jsonA.add(toJson(e, trim));

            }

            return (trim && jsonA.isEmpty()) ? Undefined : jsonA;

        }

        if (type == CfgType.Map) {

            JSONObject json = new JSONObject();

            for (CfgValue e : root.getValues()) {

                json.put(e.getKey(), toJson(e, trim));

            }

            return (trim && json.isEmpty()) ? Undefined : json;

        }

        if (type == CfgType.ListMap) {

            JSONObject json = new JSONObject();

            for (CfgValue e : root.getValues()) {

                CfgValue keyValue = e.findCfgValue(cfgInfo.getElementInfo().getKeyKey());

                Object key = (keyValue == null) ? null : keyValue.getValue();

                json.put(Dt.toString(key, "error"), toJson(e, trim));

            }

            return (trim && json.isEmpty()) ? Undefined : json;

        }

        return Undefined;

    }



    public static CfgValue copy(CfgValue origCfgValue) {

        CfgInfo cfgInfo = origCfgValue.getCfgInfo();

        CfgValue r = new CfgValue(cfgInfo);

        if (cfgInfo.getType().isCombo()) {

            for (CfgValue childCfgValue : origCfgValue.getValues()) {

                r.addValue(copy(childCfgValue));

            }

        } else {

            r.setValue(origCfgValue.getValue());

        }

        if (origCfgValue.isMapElement()) {

            r.setKey(origCfgValue.getKey());

        }

        r.setParent(origCfgValue.getParent());

        return r;

    }



    //===========

    public static CfgInfo getEntityCfgInfo(String id) {

        return entityTypeInfos.get(id);

    }



    public static int nextEntityId(String idPrefix) {

        return entityIdSeqs.get(idPrefix).incrementAndGet();

    }

}