xmlmailmergedatasource.java

SWT开发的Java调用COM的工具集

/*
 * Copyright (c) 2007 Software Wizards Pty Ltd, Victoria, Australia.
 * mailto:[e-mail]. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted. See the GNU General Public License
 * for more details.
 *
 * Redistribution of the SWTtoCOM software is not permitted as part of any
 * commercial product. Licensing terms for such distribution may be
 * obtained from the copyright holder.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */
package au.com.swz.swttocom.example.msword;

import java.io.FileInputStream;
import java.util.List;

import org.dom4j.Document;
import org.dom4j.Node;
import org.dom4j.io.SAXReader;

/**
 * A very simple MailMergeDataSource that uses DOM4J xpath to
 * source the field values from an XML document.
 * 
 * @author Mark Richter
 */
public class XMLMailMergeDataSource implements IMailMergeDataSource {
	private Document xmlDoc;
	
	public XMLMailMergeDataSource(String xmlFilePath) throws Exception {
		SAXReader saxReader = new SAXReader();
		xmlDoc = saxReader.read(new FileInputStream(xmlFilePath));
	}
	
	public String getFieldText(String fieldCode) {
		Node node = xmlDoc.selectSingleNode(parseFieldCode(fieldCode));
		if (node == null) {
			return null;
		}
		return node.getText();
	}
	
	public String getTableFieldText(int row, String fieldCode) {
		List nodeList = xmlDoc.selectNodes(parseFieldCode(fieldCode));
		if (row >= nodeList.size()) {
			return null;
		}
		return ((Node)nodeList.get(row)).getText();
	}
	
	private String parseFieldCode(String fieldCode) {
		int endIdx = fieldCode.indexOf(' ', 13);
		return fieldCode.substring(13, endIdx);
	}
}