/*
* Geotoolkit - An Open Source Java GIS Toolkit
* http://www.geotoolkit.org
*
* (C) 2009-2010, Geomatys
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation;
* version 2.1 of the License.
*
* This library 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
* Lesser General Public License for more details.
*/
package org.geotoolkit.data.wfs;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.StringWriter;
import java.net.MalformedURLException;
import java.net.URL;
import java.net.URLConnection;
import java.util.ArrayList;
import java.util.List;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.xml.bind.JAXBException;
import javax.xml.bind.Marshaller;
import javax.xml.namespace.QName;
import org.geotoolkit.client.AbstractRequest;
import org.geotoolkit.util.NamesExt;
import org.geotoolkit.filter.visitor.SimplifyingFilterVisitor;
import org.geotoolkit.ogc.xml.v110.FilterType;
import org.geotoolkit.security.ClientSecurity;
import org.geotoolkit.sld.xml.StyleXmlIO;
import org.apache.sis.util.logging.Logging;
import org.geotoolkit.wfs.xml.GetFeature;
import org.geotoolkit.wfs.xml.Query;
import org.geotoolkit.wfs.xml.WFSMarshallerPool;
import org.geotoolkit.wfs.xml.ResultTypeType;
import org.geotoolkit.wfs.xml.WFSXmlFactory;
import org.opengis.util.GenericName;
import org.opengis.filter.Filter;
/**
* Abstract Get feature request.
*
* @author Johann Sorel (Geomatys)
* @author Cédric Briançon (Geomatys)
* @module
*/
public abstract class AbstractGetFeature extends AbstractRequest implements GetFeatureRequest{
protected static final Logger LOGGER = Logging.getLogger("org.geotoolkit.data.wfs");
protected final String version;
private QName typeName = null;
private Filter filter = null;
private Integer maxFeatures = null;
private GenericName[] propertyNames = null;
private String outputFormat = null;
protected AbstractGetFeature(final String serverURL, final String version, final ClientSecurity security){
super(serverURL,security,null);
this.version = version;
}
/**
* {@inheritDoc }
*/
@Override
public QName getTypeName() {
return typeName;
}
/**
* {@inheritDoc }
*/
@Override
public void setTypeName(final QName type) {
this.typeName = type;
}
/**
* {@inheritDoc }
*/
@Override
public Filter getFilter(){
return filter;
}
/**
* {@inheritDoc }
*/
@Override
public void setFilter(final Filter filter){
this.filter = filter;
}
/**
* {@inheritDoc }
*/
@Override
public Integer getMaxFeatures(){
return maxFeatures;
}
/**
* {@inheritDoc }
*/
@Override
public void setMaxFeatures(final Integer max){
maxFeatures = max;
}
/**
* {@inheritDoc }
*/
@Override
public GenericName[] getPropertyNames() {
return propertyNames;
}
/**
* {@inheritDoc }
*/
@Override
public void setPropertyNames(final GenericName[] properties) {
this.propertyNames = properties;
}
/**
* {@inheritDoc }
*/
@Override
public String getOutputFormat() {
return outputFormat;
}
/**
* {@inheritDoc }
*/
@Override
public void setOutputFormat(final String outputFormat) {
this.outputFormat = outputFormat;
}
/**
* {@inheritDoc }
*/
@Override
public URL getURL() throws MalformedURLException {
requestParameters.put("SERVICE", "WFS");
requestParameters.put("REQUEST", "GETFEATURE");
requestParameters.put("VERSION", version);
if(maxFeatures != null){
requestParameters.put("MAXFEATURES", maxFeatures.toString());
}
if(typeName != null){
final StringBuilder sbN = new StringBuilder();
final StringBuilder sbNS = new StringBuilder();
String prefix = typeName.getPrefix();
if (prefix == null || prefix.isEmpty()) {
prefix = "ut";
}
sbN.append(prefix).append(':').append(typeName.getLocalPart()).append(',');
sbNS.append("xmlns(").append(prefix).append('=').append(typeName.getNamespaceURI()).append(')').append(',');
if(sbN.length() > 0 && sbN.charAt(sbN.length()-1) == ','){
sbN.deleteCharAt(sbN.length()-1);
}
if(sbNS.length() > 0 && sbNS.charAt(sbNS.length()-1) == ','){
sbNS.deleteCharAt(sbNS.length()-1);
}
requestParameters.put("TYPENAME",sbN.toString());
requestParameters.put("NAMESPACE",sbNS.toString());
}
Filter filter = Filter.INCLUDE;
if(this.filter != null && this.filter != Filter.INCLUDE){
final SimplifyingFilterVisitor visitor = new SimplifyingFilterVisitor();
filter = (Filter) this.filter.accept(visitor, null);
}
if(filter != null && filter != Filter.INCLUDE){
final StyleXmlIO util = new StyleXmlIO();
final StringWriter writer = new StringWriter();
try {
util.writeFilter(writer, filter, org.geotoolkit.sld.xml.Specification.Filter.V_1_1_0);
} catch (JAXBException ex) {
LOGGER.log(Level.WARNING, ex.getLocalizedMessage(), ex);
}
final String strFilter = writer.toString();
try {
writer.close();
} catch (IOException ex) {
LOGGER.log(Level.FINER, ex.getLocalizedMessage(), ex);
}
requestParameters.put("FILTER", strFilter);
}
if(propertyNames != null){
final StringBuilder sb = new StringBuilder();
for(final GenericName prop : propertyNames){
final String propNs = NamesExt.getNamespace(prop);
if(typeName != null && propNs != null
&& propNs.equals(typeName.getNamespaceURI())){
sb.append(typeName.getPrefix()).append(':').append(prop.tip().toString()).append(',');
}else{
sb.append(NamesExt.toExtendedForm(prop)).append(',');
}
}
if(sb.length() > 0 && sb.charAt(sb.length()-1) == ','){
sb.deleteCharAt(sb.length()-1);
}
requestParameters.put("PROPERTYNAME", sb.toString());
}
if (outputFormat != null) {
requestParameters.put("OUTPUTFORMAT",outputFormat);
}
return super.getURL();
}
/**
* {@inheritDoc }
*/
@Override
public InputStream getResponseStream() throws IOException {
final List<QName> typeNames = new ArrayList<QName>();
if(typeName != null){
typeNames.add(typeName);
}
Filter filter = Filter.INCLUDE;
if(this.filter != null && this.filter != Filter.INCLUDE){
final SimplifyingFilterVisitor visitor = new SimplifyingFilterVisitor();
filter = (Filter) this.filter.accept(visitor, null);
}
FilterType xmlFilter;
if(filter != null && filter != Filter.INCLUDE){
final StyleXmlIO util = new StyleXmlIO();
xmlFilter = util.getTransformerXMLv110().visit(filter);
} else {
xmlFilter = null;
}
final List<String> propName = new ArrayList<String>();
if(propertyNames != null){
// TODO handle prefix/namespace
for(final GenericName prop : propertyNames){
propName.add(prop.tip().toString());
}
}
final Query query = WFSXmlFactory.buildQuery(version, xmlFilter, typeNames, "1.1.0", null, null, propName);
final GetFeature request = WFSXmlFactory.buildGetFeature(version, "WFS", null, null, maxFeatures, query, ResultTypeType.RESULTS, outputFormat);
final URL url = new URL(serverURL);
URLConnection conec = url.openConnection();
conec = security.secure(conec);
conec.setDoOutput(true);
conec.setRequestProperty("Content-Type", "text/xml");
OutputStream stream = conec.getOutputStream();
stream = security.encrypt(stream);
try {
Marshaller marshaller = WFSMarshallerPool.getInstance().acquireMarshaller();
marshaller.marshal(request, stream);
WFSMarshallerPool.getInstance().recycle(marshaller);
} catch (JAXBException ex) {
throw new IOException(ex);
}
stream.close();
return security.decrypt(conec.getInputStream());
}
}