/*
 *    GeoTools - The Open Source Java GIS Toolkit
 *    http://geotools.org
 *
 *    (C) 2006-2008, Open Source Geospatial Foundation (OSGeo)
 *
 *    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.geotools.xml.gml;

import org.geotools.feature.AttributeTypeBuilder;
import org.geotools.feature.SchemaException;
import org.geotools.feature.simple.SimpleFeatureTypeBuilder;
import org.opengis.feature.simple.SimpleFeatureType;
import org.opengis.feature.type.AttributeDescriptor;
import org.opengis.feature.type.GeometryDescriptor;
import org.opengis.referencing.crs.CoordinateReferenceSystem;

/**
 * A sad hack class until the new Feature Model comes around. 
 * 
 * @see ChoiceAttributeType
 * @author Jesse
 * @since 1.1.0
 *
 * @source $URL$
 */
public class WFSFeatureTypeTransformer {

    public static SimpleFeatureType transform( SimpleFeatureType schema, CoordinateReferenceSystem crs ) throws SchemaException {
        SimpleFeatureTypeBuilder build = new SimpleFeatureTypeBuilder();
        build.setName( schema.getName() );
        //build.setDefaultGeometry( schema.getDefaultGeometry().getLocalName() );
        
        GeometryDescriptor defaultGeometryType = null;
        for( int i = 0; i < schema.getAttributeCount(); i++ ) {
            AttributeDescriptor attributeType = schema.getDescriptor(i);
            if( attributeType instanceof ChoiceGeometryType ){
                defaultGeometryType = handleChoiceGeometryAttribute(schema, crs, build, defaultGeometryType, attributeType);
            }
            else if (attributeType instanceof GeometryDescriptor) {
                defaultGeometryType = handleGeometryAttribute(schema, crs, build, defaultGeometryType, attributeType);
            }
            else {
                build.add(attributeType);
            }
        }
        //Only try to set default geometry when there actually is a geometry type
        if(defaultGeometryType!=null){
           build.setDefaultGeometry(defaultGeometryType.getLocalName());  
        }
        return build.buildFeatureType();
    }

    private static GeometryDescriptor handleGeometryAttribute( SimpleFeatureType schema, CoordinateReferenceSystem crs, SimpleFeatureTypeBuilder factory, GeometryDescriptor defaultGeometryType, AttributeDescriptor attributeType ) {
        GeometryDescriptor geometryType = (GeometryDescriptor) attributeType;
        GeometryDescriptor geometry;

        AttributeTypeBuilder builder = new AttributeTypeBuilder();
        builder.setName( geometryType.getLocalName()  );
        builder.setBinding( geometryType.getType().getBinding() );
        builder.setNillable( geometryType.isNillable() );
        
        //builder.setDefaultValue(defaultValue);
        builder.setCRS( crs );
        
        geometry = builder.buildDescriptor( geometryType.getLocalName(), builder.buildGeometryType() );
        
//        geometry = (GeometryDescriptor) AttributeTypeFactory.newAttributeType(
//                geometryType.getLocalName(), geometryType.getBinding(), geometryType.isNillable(),
//                0, geometryType.createDefaultValue(), crs);

        if (defaultGeometryType == null || geometryType == schema.getGeometryDescriptor()) {
            defaultGeometryType = geometry;
        }
        factory.add(geometry);
        return defaultGeometryType;
    }

    private static GeometryDescriptor handleChoiceGeometryAttribute( SimpleFeatureType schema, CoordinateReferenceSystem crs, SimpleFeatureTypeBuilder factory, GeometryDescriptor defaultGeometryType, AttributeDescriptor attributeType ) {
        ChoiceGeometryTypeImpl geometryType = (ChoiceGeometryTypeImpl) attributeType;
        ChoiceGeometryTypeImpl geometry;

        geometry = new ChoiceGeometryTypeImpl(
            geometryType.getName(), geometryType.getChoices(), geometryType.getBinding(), geometryType.isNillable(),
            geometryType.getMinOccurs(), geometryType.getMaxOccurs(), geometryType.createDefaultValue(), crs, geometryType.getRestrictions());

        if (defaultGeometryType == null || geometryType == schema.getGeometryDescriptor()) {
            defaultGeometryType = geometry;
        }
        factory.add(geometry);
        return defaultGeometryType;
    }

}