net.sourceforge.jweb.mybatis.generator.plugins.PluginUtil Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jweb-maven-plugin Show documentation
maven plugins and mybatis generator plugins
There is a newer version: 1.0.13
package net.sourceforge.jweb.mybatis.generator.plugins;
/*
 * Copyright 2002-2016 the original author or authors.
 *
 * 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.
 */
import java.util.ArrayList;
import java.util.Date;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;

import org.mybatis.generator.api.dom.xml.Attribute;
import org.mybatis.generator.api.dom.xml.TextElement;
import org.mybatis.generator.api.dom.xml.XmlElement;
import org.w3c.dom.Comment;
import org.w3c.dom.Element;
import org.w3c.dom.NamedNodeMap;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;
import org.w3c.dom.Text;

/**
 * plugin util for transform w3c.DOM and mybatis generator DOM
 * 
 * events
 * 
 * @author Alex
 *
 */
public class PluginUtil {
	private PluginUtil(){
		
	}
    public static Attribute cloneAttribute(Node node){
    	return new Attribute(node.getNodeName(), node.getNodeValue());
    }
    
    public static org.mybatis.generator.api.dom.xml.Element cloneElement(Node w3cEl){
    	return cloneElement(w3cEl, null);
    }
    
    public static org.mybatis.generator.api.dom.xml.Element cloneElement(Node w3cEl, String idAttrValue){
    	if(!(w3cEl instanceof Element)){
    		return new TextElement(w3cEl.getNodeValue());
    	}
    	XmlElement root=new XmlElement(w3cEl.getNodeName());
    	Queue queue=new LinkedList();
    	queue.offer(new ElementPeer(root, w3cEl));
    	while(!queue.isEmpty()){
    		ElementPeer cur=queue.poll();
    		NamedNodeMap attrs = cur.w3cElement.getAttributes();
    		if(attrs!=null){
    			for(int i=0;i0)cur.mybatisElement.addElement(new TextElement(text.trim()));
    				}
    				else if(child instanceof Comment){
    					String text = child.getNodeValue().trim();
    					if(text.length()>0) cur.mybatisElement.addElement(new TextElement(""));
    				}
    				else if(child instanceof Element){
    					XmlElement mybatisChild=new XmlElement(child.getNodeName());
    					cur.mybatisElement.addElement(mybatisChild);
    					queue.offer(new ElementPeer(mybatisChild, (Element)child));
    				}
    			}
    		}
    	}
    	return root;
    }
    /**
	 * find first child element of given name
	 * @param element
	 * @param name
	 * @return
	 */
    public static XmlElement findFirst(XmlElement element, String name){
		Queue queue=new LinkedList();
		queue.offer(element);
		
		while(!queue.isEmpty()){
			XmlElement top=queue.poll();
			if(name.equals(top.getName())){
				return top;
			}
			
			for(org.mybatis.generator.api.dom.xml.Element ch:top.getElements()){
				if(ch instanceof XmlElement){
					queue.offer((XmlElement)ch);
				}
			}
		}
		
		return null;
	}
	
	public static List findAll(XmlElement element, String name){
		List ret=new ArrayList();
		
		Queue queue=new LinkedList();
		queue.offer(element);
		
		while(!queue.isEmpty()){
			XmlElement top=queue.poll();
			if(name.equals(top.getName())){
				ret.add(top);
			}
			
			for(org.mybatis.generator.api.dom.xml.Element ch:top.getElements()){
				if(ch instanceof XmlElement){
					queue.offer((XmlElement)ch);
				}
			}
		}
		
		return ret;
	}
   
    private final static class ElementPeer {
    	XmlElement mybatisElement;
    	Node w3cElement;
		public ElementPeer(XmlElement mybatisElement, Node w3cElement) {
			super();
			this.mybatisElement = mybatisElement;
			this.w3cElement = w3cElement;
		}
    }
}