001/**
002 * Copyright (c) 2025-2026, Michael Yang 杨福海 (fuhai999@gmail.com).
003 * <p>
004 * Licensed under the GNU Lesser General Public License (LGPL) ,Version 3.0 (the "License");
005 * you may not use this file except in compliance with the License.
006 * You may obtain a copy of the License at
007 * <p>
008 * http://www.gnu.org/licenses/lgpl-3.0.txt
009 * <p>
010 * Unless required by applicable law or agreed to in writing, software
011 * distributed under the License is distributed on an "AS IS" BASIS,
012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 * See the License for the specific language governing permissions and
014 * limitations under the License.
015 */
016package dev.tinyflow.core.parser;
017
018
019import com.agentsflex.core.chain.*;
020import com.agentsflex.core.chain.node.BaseNode;
021import com.agentsflex.core.util.StringUtil;
022import com.alibaba.fastjson.JSONArray;
023import com.alibaba.fastjson.JSONObject;
024import dev.tinyflow.core.Tinyflow;
025
026import java.util.ArrayList;
027import java.util.Collections;
028import java.util.List;
029
030
031public abstract class BaseNodeParser implements NodeParser {
032
033    private static final JSONObject EMPTY_JSON_OBJECT = new JSONObject(Collections.emptyMap());
034
035    public JSONObject getData(JSONObject nodeObject) {
036        JSONObject jsonObject = nodeObject.getJSONObject("data");
037        return jsonObject != null ? jsonObject : EMPTY_JSON_OBJECT;
038    }
039
040    public void addParameters(BaseNode node, JSONObject data) {
041        List<Parameter> inputParameters = getParameters(data, "parameters");
042        node.setParameters(inputParameters);
043    }
044
045    public List<Parameter> getParameters(JSONObject data, String key) {
046        return getParameters(data.getJSONArray(key));
047    }
048
049    public List<Parameter> getParameters(JSONArray parametersJsonArray) {
050        if (parametersJsonArray == null || parametersJsonArray.isEmpty()) {
051            return Collections.emptyList();
052        }
053        List<Parameter> parameters = new ArrayList<>(parametersJsonArray.size());
054        for (int i = 0; i < parametersJsonArray.size(); i++) {
055            JSONObject parameterJsonObject = parametersJsonArray.getJSONObject(i);
056            Parameter parameter = new Parameter();
057            parameter.setId(parameterJsonObject.getString("id"));
058            parameter.setName(parameterJsonObject.getString("name"));
059            parameter.setDescription(parameterJsonObject.getString("description"));
060            parameter.setDataType(DataType.ofValue(parameterJsonObject.getString("dataType")));
061            parameter.setRef(parameterJsonObject.getString("ref"));
062            parameter.setRefType(RefType.ofValue(parameterJsonObject.getString("refType")));
063            parameter.setRequired(parameterJsonObject.getBooleanValue("required"));
064            parameter.setDefaultValue(parameterJsonObject.getString("defaultValue"));
065            parameter.setValue(parameterJsonObject.getString("value"));
066
067            JSONArray children = parameterJsonObject.getJSONArray("children");
068            if (children != null && !children.isEmpty()) {
069                parameter.addChildren(getParameters(children));
070            }
071
072            parameters.add(parameter);
073        }
074
075        return parameters;
076    }
077
078
079    public void addOutputDefs(BaseNode node, JSONObject data) {
080        JSONArray outputDefs = data.getJSONArray("outputDefs");
081        if (outputDefs == null || outputDefs.isEmpty()) {
082            return;
083        }
084
085        for (int i = 0; i < outputDefs.size(); i++) {
086            JSONObject outputDefJsonObject = outputDefs.getJSONObject(i);
087            Parameter parameter = new Parameter();
088            parameter.setId(outputDefJsonObject.getString("id"));
089            parameter.setName(outputDefJsonObject.getString("name"));
090            parameter.setDescription(outputDefJsonObject.getString("description"));
091            parameter.setRef(outputDefJsonObject.getString("ref"));
092            parameter.setValue(outputDefJsonObject.getString("value"));
093            parameter.setDefaultValue(outputDefJsonObject.getString("defaultValue"));
094            parameter.setRefType(RefType.ofValue(outputDefJsonObject.getString("refType")));
095            parameter.setDataType(DataType.ofValue(outputDefJsonObject.getString("dataType")));
096
097            JSONArray children = outputDefJsonObject.getJSONArray("children");
098            if (children != null && !children.isEmpty()) {
099                parameter.addChildren(getParameters(children));
100            }
101
102            node.addOutputDef(parameter);
103        }
104    }
105
106    @Override
107    public ChainNode parse(JSONObject nodeJSONObject, Tinyflow tinyflow) {
108        JSONObject data = getData(nodeJSONObject);
109        BaseNode node = doParse(nodeJSONObject, data, tinyflow);
110        if (node != null) {
111
112            node.setId(nodeJSONObject.getString("id"));
113            node.setName(nodeJSONObject.getString("label"));
114            node.setDescription(nodeJSONObject.getString("description"));
115
116            if (!data.isEmpty()) {
117
118                addParameters(node, data);
119                addOutputDefs(node, data);
120
121                String conditionString = data.getString("condition");
122
123                if (StringUtil.hasText(conditionString)) {
124                    node.setCondition(new JsCodeCondition(conditionString.trim()));
125                }
126
127                Boolean async = data.getBoolean("async");
128                if (async != null) {
129                    node.setAsync(async);
130                }
131
132                String name = data.getString("title");
133                if (StringUtil.hasText(name)) {
134                    node.setName(name);
135                }
136
137                String description = data.getString("description");
138                if (StringUtil.hasText(description)) {
139                    node.setDescription(description);
140                }
141
142                Boolean loopEnable = data.getBoolean("loopEnable");
143                if (loopEnable != null) {
144                    node.setLoopEnable(loopEnable);
145                }
146
147                Long loopIntervalMs = data.getLong("loopIntervalMs");
148                if (loopIntervalMs != null) {
149                    node.setLoopIntervalMs(loopIntervalMs);
150                }
151
152                Integer maxLoopCount = data.getInteger("maxLoopCount");
153                if (maxLoopCount != null) {
154                    node.setMaxLoopCount(maxLoopCount);
155                }
156
157                String loopBreakCondition = data.getString("loopBreakCondition");
158                if (StringUtil.hasText(loopBreakCondition)) {
159                    node.setLoopBreakCondition(new JsCodeCondition(loopBreakCondition.trim()));
160                }
161            }
162        }
163
164        return node;
165    }
166
167    protected abstract BaseNode doParse(JSONObject root, JSONObject data, Tinyflow tinyflow);
168}