content: /*******************************************************************************
   * Copyright 2017 Bstek
   *
   * Licensed under the Apache License, Version 2.0 (the "License"); you may not
   * use this file except in compliance with the License.  You may obtain a copy
   * of the License at
   *
   *   http://www.apache.org/licenses/LICENSE-2.0
   *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
   * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  See the
   * License for the specific language governing permissions and limitations under
   * the License.
   ******************************************************************************/
  package com.bstek.ureport.parser.impl;
  
  import com.bstek.ureport.definition.dataset.*;
  import com.bstek.ureport.definition.datasource.*;
  import com.bstek.ureport.expression.ExpressionUtils;
  import com.bstek.ureport.expression.model.Expression;
  import com.bstek.ureport.parser.Parser;
  import org.dom4j.Element;
  
  import java.util.ArrayList;
  import java.util.List;
  
  /**
   * @author
   * @since 2016年12月30日
   */
  public class DatasourceParser implements Parser<DatasourceDefinition> {
  	@Override
  	public DatasourceDefinition parse(Element element) {
  		String type=element.attributeValue("type");
  		if(type.equals("jdbc")){
  			JdbcDatasourceDefinition ds=new JdbcDatasourceDefinition();
  			ds.setName(element.attributeValue("name"));
  			ds.setDriver(element.attributeValue("driver"));
  			ds.setUrl(element.attributeValue("url"));
  			ds.setUsername(element.attributeValue("username"));
  			ds.setPassword(element.attributeValue("password"));
  			ds.setDatasets(parseDatasets(element));
  			return ds;
  		}else if(type.equals("spring")){
  			SpringBeanDatasourceDefinition ds=new SpringBeanDatasourceDefinition();
  			ds.setName(element.attributeValue("name"));
  			ds.setBeanId(element.attributeValue("bean"));
  			ds.setDatasets(parseDatasets(element));
  			return ds;
  		}else if(type.equals("buildin")){
  			BuildinDatasourceDefinition ds=new BuildinDatasourceDefinition();
  			ds.setName(element.attributeValue("name"));
  			ds.setDatasets(parseDatasets(element));
  			return ds;
  		}
  		return null;
  	}
  
  	private List<DatasetDefinition> parseDatasets(Element element){
  		List<DatasetDefinition> list=new ArrayList<DatasetDefinition>();
  		for(Object obj:element.elements()){
  			if(obj==null || !(obj instanceof Element)){
  				continue;
  			}
  			Element ele=(Element)obj;
  			String type=ele.attributeValue("type");
  			if(type.equals("sql")){
  				SqlDatasetDefinition dataset=new SqlDatasetDefinition();
  				dataset.setName(ele.attributeValue("name"));
  				dataset.setSql(parseSql(ele,dataset));
  				dataset.setFields(parseFields(ele));
  				dataset.setParameters(parseParameters(ele));
  				list.add(dataset);
  			}else if(type.equals("bean")){
  				BeanDatasetDefinition dataset=new BeanDatasetDefinition();
  				dataset.setName(ele.attributeValue("name"));
  				dataset.setMethod(ele.attributeValue("method"));
  				dataset.setFields(parseFields(ele));
  				dataset.setClazz(ele.attributeValue("clazz"));
  				list.add(dataset);
  			}
  		}
  		return list;
  	}
  
  	private List<Parameter> parseParameters(Element element){
  		List<Parameter> parameters=new ArrayList<Parameter>();
  		for(Object obj:element.elements()){
  			if(obj==null || !(obj instanceof Element)){
  				continue;
  			}
  			Element ele=(Element)obj;
  			if(!ele.getName().equals("parameter")){
  				continue;
  			}
  			Parameter param=new Parameter();
  			param.setName(ele.attributeValue("name"));
  			param.setDefaultValue(ele.attributeValue("default-value"));
  			param.setType(DataType.valueOf(ele.attributeValue("type")));
  			parameters.add(param);
  		}
  		return parameters;
  	}
  
  	private List<Field> parseFields(Element element){
  		List<Field> fields=new ArrayList<Field>();
  		for(Object obj:element.elements()){
  			if(obj==null || !(obj instanceof Element)){
  				continue;
  			}
  			Element ele=(Element)obj;
  			if(!ele.getName().equals("field")){
  				continue;
  			}
  			Field field=new Field(ele.attributeValue("name"));
  			fields.add(field);
  		}
  		return fields;
  	}
  
  	private String parseSql(Element element,SqlDatasetDefinition dataset){
  		for(Object obj:element.elements()){
  			if(obj==null || !(obj instanceof Element)){
  				continue;
  			}
  			Element ele=(Element)obj;
  			if(ele.getName().equals("sql")){
  				String sql=ele.getText().trim();
  				if(sql.startsWith(ExpressionUtils.EXPR_PREFIX) && sql.endsWith(ExpressionUtils.EXPR_SUFFIX)){
  					String s=sql.substring(2,sql.length()-1);
  					Expression expr=ExpressionUtils.parseExpression(s);
  					dataset.setSqlExpression(expr);
  				}
  				return ele.getText();
  			}
  		}
  		return null;
  	}
  }
  