001 /* 002 * Copyright (C) 2003-2009 eXo Platform SAS. 003 * 004 * This is free software; you can redistribute it and/or modify it 005 * under the terms of the GNU Lesser General Public License as 006 * published by the Free Software Foundation; either version 2.1 of 007 * the License, or (at your option) any later version. 008 * 009 * This software is distributed in the hope that it will be useful, 010 * but WITHOUT ANY WARRANTY; without even the implied warranty of 011 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 012 * Lesser General Public License for more details. 013 * 014 * You should have received a copy of the GNU Lesser General Public 015 * License along with this software; if not, write to the Free 016 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 017 * 02110-1301 USA, or see the FSF site: http://www.fsf.org. 018 */ 019 package org.crsh.jcr; 020 021 import org.crsh.util.Safe; 022 import org.slf4j.Logger; 023 import org.slf4j.LoggerFactory; 024 import org.xml.sax.Attributes; 025 import org.xml.sax.ContentHandler; 026 import org.xml.sax.SAXException; 027 import org.xml.sax.helpers.DefaultHandler; 028 029 import javax.xml.parsers.SAXParser; 030 import javax.xml.parsers.SAXParserFactory; 031 import java.io.IOException; 032 import java.io.InputStream; 033 import java.util.ArrayList; 034 import java.util.LinkedList; 035 import java.util.List; 036 037 /** 038 * @author <a href="mailto:julien.viet@exoplatform.com">Julien Viet</a> 039 * @version $Revision$ 040 */ 041 public class Importer implements FileSystem { 042 043 /** . */ 044 private static final Logger log = LoggerFactory.getLogger(Importer.class); 045 046 /** . */ 047 private final ContentHandler handler; 048 049 /** . */ 050 private final LinkedList<EndElement> stack; 051 052 /** . */ 053 private final List<String> prefixes; 054 055 /** . */ 056 private final DefaultHandler attributesHandler = new DefaultHandler() { 057 058 @Override 059 public void startPrefixMapping(String prefix, String uri) throws SAXException { 060 if (stack.isEmpty()) { 061 log.debug("Adding prefix mapping " + prefix + " for " + uri); 062 handler.startPrefixMapping(prefix, uri); 063 prefixes.add(prefix); 064 } 065 } 066 067 @Override 068 public void startElement(String uri, String localName, String qName, Attributes attributes) throws SAXException { 069 log.debug("Creating element " + qName); 070 handler.startElement(uri, localName, qName, attributes); 071 stack.addLast(new EndElement(uri, localName, qName)); 072 } 073 }; 074 075 public Importer(ContentHandler handler) { 076 this.handler = handler; 077 this.stack = new LinkedList<EndElement>(); 078 this.prefixes = new ArrayList<String>(); 079 } 080 081 public void startDirectory(String directoryName) throws IOException { 082 } 083 084 public void file(String fileName, int length, InputStream data) throws IOException { 085 try { 086 SAXParserFactory factory = SAXParserFactory.newInstance(); 087 factory.setNamespaceAware(true); 088 factory.setValidating(true); 089 SAXParser parser = factory.newSAXParser(); 090 parser.parse(data, attributesHandler); 091 } 092 catch (Exception e) { 093 Safe.rethrow(IOException.class, e); 094 } 095 } 096 097 public void endDirectory(String directoryName) throws IOException { 098 try { 099 EndElement end = stack.removeLast(); 100 handler.endElement(end.uri, end.localName, end.qName); 101 if (stack.isEmpty()) { 102 for (String prefix : prefixes) { 103 log.debug("Removing prefix mapping " + prefix); 104 handler.endPrefixMapping(prefix); 105 } 106 prefixes.clear(); 107 } 108 } 109 catch (Exception e) { 110 Safe.rethrow(IOException.class, e); 111 } 112 } 113 114 private static class EndElement { 115 116 /** . */ 117 private final String uri; 118 119 /** . */ 120 private final String localName; 121 122 /** . */ 123 private final String qName; 124 125 private EndElement(String uri, String localName, String qName) { 126 this.uri = uri; 127 this.localName = localName; 128 this.qName = qName; 129 } 130 } 131 } 132