/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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.    
 */

package org.apache.tuscany.spi.implementation.impl;

import static javax.xml.stream.XMLStreamConstants.END_ELEMENT;

import javax.xml.namespace.QName;
import javax.xml.stream.XMLStreamException;
import javax.xml.stream.XMLStreamReader;
import javax.xml.stream.XMLStreamWriter;

import org.apache.tuscany.assembly.ComponentType;
import org.apache.tuscany.assembly.Implementation;
import org.apache.tuscany.assembly.Property;
import org.apache.tuscany.assembly.Reference;
import org.apache.tuscany.assembly.Service;
import org.apache.tuscany.assembly.impl.DefaultAssemblyFactory;
import org.apache.tuscany.contribution.processor.StAXArtifactProcessorExtension;
import org.apache.tuscany.contribution.resolver.ArtifactResolver;
import org.apache.tuscany.contribution.service.ContributionReadException;
import org.apache.tuscany.contribution.service.ContributionResolveException;
import org.apache.tuscany.contribution.service.ContributionWireException;
import org.apache.tuscany.contribution.service.ContributionWriteException;
import org.apache.tuscany.spi.implementation.ImplementationActivator;

public class ImplementationArtifactProcessor implements StAXArtifactProcessorExtension {

    private ImplementationActivator ia;

    public ImplementationArtifactProcessor(ImplementationActivator ia) {
        this.ia = ia;
    }

    public QName getArtifactType() {
        return ia.getModelQName();
    }

    public Object read(XMLStreamReader reader) throws ContributionReadException {
        // Read an <implementation.crud> element
        try {

            Implementation implementation = (Implementation)ia.getImplementationClass().newInstance();
            ia.getSCDLProcessor().read(implementation, reader);

            // Skip to end element
            while (reader.hasNext()) {
                if (reader.next() == END_ELEMENT && ia.getModelQName().equals(reader.getName())) {
                    break;
                }
            }
            
            return implementation;

        } catch (XMLStreamException e) {
            throw new ContributionReadException(e);
        } catch (InstantiationException e) {
            throw new ContributionReadException(e);
        } catch (IllegalAccessException e) {
            throw new ContributionReadException(e);
        }
    }

    public void write(Object arg0, XMLStreamWriter arg1) throws ContributionWriteException {
    }

    public Class getModelType() {
        return ia.getImplementationClass();
    }

    public void resolve(Object arg0, ArtifactResolver resolver) throws ContributionResolveException {
        ClassLoader cl = Thread.currentThread().getContextClassLoader();
        Implementation impl = (Implementation)arg0;
        String scriptURI = cl.getResource(((Implementation)arg0).getURI()).toString();
        int lastDot = scriptURI.lastIndexOf('.');
        String ctURI = scriptURI.substring(0, lastDot) + ".componentType";
        ComponentType componentType = new DefaultAssemblyFactory().createComponentType();
        componentType.setUnresolved(true);
        componentType.setURI(ctURI);
        componentType = resolver.resolve(ComponentType.class, componentType);
        if (componentType.isUnresolved()) {
            throw new ContributionResolveException("missing .componentType side file");
        }
        for (Reference reference : componentType.getReferences()) {
            impl.getReferences().add(reference);
        }
        for (Service service : componentType.getServices()) {
            impl.getServices().add(service);
        }
        for (Property property : componentType.getProperties()) {
            impl.getProperties().add(property);
        }
    }

    public void wire(Object arg0) throws ContributionWireException {
    }

}