1 files changed, 319 insertions, 0 deletions
diff --git a/sca-java-2.x/tags/2.0.1-RC1/modules/contribution/src/main/java/org/apache/tuscany/sca/contribution/processor/DefaultURLArtifactProcessorExtensionPoint.java b/sca-java-2.x/tags/2.0.1-RC1/modules/contribution/src/main/java/org/apache/tuscany/sca/contribution/processor/DefaultURLArtifactProcessorExtensionPoint.java
new file mode 100644
index 0000000000..812695575d
--- /dev/null
+++ b/sca-java-2.x/tags/2.0.1-RC1/modules/contribution/src/main/java/org/apache/tuscany/sca/contribution/processor/DefaultURLArtifactProcessorExtensionPoint.java
@@ -0,0 +1,319 @@
+/*
+ * 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.sca.contribution.processor;
+
+import java.io.IOException;
+import java.lang.reflect.Constructor;
+import java.net.URI;
+import java.net.URL;
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.Collections;
+import java.util.List;
+import java.util.Map;
+import java.util.regex.Pattern;
+
+import javax.xml.stream.XMLInputFactory;
+import javax.xml.stream.XMLOutputFactory;
+
+import org.apache.tuscany.sca.contribution.resolver.ModelResolver;
+import org.apache.tuscany.sca.core.ExtensionPointRegistry;
+import org.apache.tuscany.sca.core.FactoryExtensionPoint;
+import org.apache.tuscany.sca.extensibility.ServiceDeclaration;
+
+/**
+ * The default implementation of a URL artifact processor extension point.
+ *
+ * @version $Rev$ $Date$
+ */
+public class DefaultURLArtifactProcessorExtensionPoint extends
+    DefaultArtifactProcessorExtensionPoint<URLArtifactProcessor<?>> implements URLArtifactProcessorExtensionPoint {
+
+    private StAXArtifactProcessor<?> staxProcessor;
+    private boolean loaded;
+
+    /**
+     * Constructs a new extension point.
+     */
+    public DefaultURLArtifactProcessorExtensionPoint(ExtensionPointRegistry extensionPoints) {
+        super(extensionPoints);
+        FactoryExtensionPoint modelFactories = this.registry.getExtensionPoint(FactoryExtensionPoint.class);
+        XMLInputFactory inputFactory = modelFactories.getFactory(XMLInputFactory.class);
+        XMLOutputFactory outputFactory = modelFactories.getFactory(XMLOutputFactory.class);
+        StAXArtifactProcessorExtensionPoint staxProcessors =
+            extensionPoints.getExtensionPoint(StAXArtifactProcessorExtensionPoint.class);
+        staxProcessor = new ExtensibleStAXArtifactProcessor(staxProcessors, inputFactory, outputFactory);
+    }
+
+    public void addArtifactProcessor(URLArtifactProcessor<?> artifactProcessor) {
+        if (artifactProcessor.getArtifactType() != null) {
+            Pattern pattern = Pattern.compile(wildcard2regex(artifactProcessor.getArtifactType()));
+            processorsByArtifactType.put(pattern, artifactProcessor);
+        }
+        if (artifactProcessor.getModelType() != null) {
+            processorsByModelType.put(artifactProcessor.getModelType(), artifactProcessor);
+        }
+    }
+
+    public void removeArtifactProcessor(URLArtifactProcessor<?> artifactProcessor) {
+        if (artifactProcessor.getArtifactType() != null) {
+            String regex = wildcard2regex(artifactProcessor.getArtifactType());
+            for (Object key : processorsByArtifactType.keySet()) {
+                if ((key instanceof Pattern) && ((Pattern)key).pattern().equals(regex)) {
+                    processorsByArtifactType.remove(key);
+                }
+            }
+            processorsByArtifactType.remove((Object)artifactProcessor.getArtifactType());
+        }
+        if (artifactProcessor.getModelType() != null) {
+            processorsByModelType.remove(artifactProcessor.getModelType());
+        }
+    }
+
+    @SuppressWarnings("unchecked")
+	@Override
+    public <T> URLArtifactProcessor<T> getProcessor(Class<T> modelType) {
+        loadProcessors();
+        return (URLArtifactProcessor<T>)super.getProcessor(modelType);
+    }
+
+    public Collection<URLArtifactProcessor<?>> getProcessors(Object artifactType) {
+        loadProcessors();
+        String uri = (String)artifactType;
+        if (uri.endsWith("/")) {
+            // Ignore directories
+            return Collections.emptyList();
+        }
+        if (!uri.startsWith("/")) {
+            uri = "/" + uri;
+        }
+        List<URLArtifactProcessor<?>> processors = new ArrayList<URLArtifactProcessor<?>>();
+        for (Map.Entry<Object, URLArtifactProcessor<?>> e : processorsByArtifactType.entrySet()) {
+            Pattern pattern = (Pattern)e.getKey();
+            if (pattern.matcher(uri).matches()) {
+                processors.add(e.getValue());
+            }
+        }
+        return processors;
+    }
+
+    @SuppressWarnings("unchecked")
+	public URLArtifactProcessor<?> getProcessor(Object artifactType) {
+        Collection<URLArtifactProcessor<?>> processors = getProcessors(artifactType);
+        return processors.isEmpty() ? null : processors.iterator().next();
+    }
+
+    private static String wildcard2regex(String pattern) {
+        String wildcard = pattern;
+        if (wildcard.endsWith("/")) {
+            // Directory: xyz/ --> xyz/**
+            wildcard = wildcard + "**";
+        }
+        if (wildcard.startsWith(".")) {
+            // File extension: .xyz --> **/*.xyz
+            wildcard = "**/*" + wildcard;
+        } else if (wildcard.indexOf('/') == -1) {
+            // File name: abc.txt --> **/abc.txt
+            wildcard = "**/" + wildcard;
+        } else if (!(wildcard.startsWith("/") || wildcard.startsWith("**"))) {
+            wildcard = '/' + wildcard;
+        }
+        StringBuffer regex = new StringBuffer();
+        char[] chars = wildcard.toCharArray();
+        for (int i = 0; i < chars.length; i++) {
+            switch (chars[i]) {
+                case '*':
+                    if (i < chars.length - 1 && chars[i + 1] == '*') {
+                        // Next char is '*'
+                        if (i < chars.length - 2) {
+                            if (chars[i + 2] == '/') {
+                                // The wildcard is **/, it matches zero or more directories
+                                regex.append("(.*/)*");
+                                i += 2; // Skip */
+                            } else {
+                                // ** can only be followed by /
+                                throw new IllegalArgumentException("** can only be used as the name for a directory");
+                            }
+                        } else {
+                            regex.append(".*");
+                            i++; // Skip next *
+                        }
+                    } else {
+                        // Non-directory
+                        regex.append("[^/]*");
+                    }
+                    break;
+                case '?':
+                    regex.append("[^/]");
+                    break;
+                case '\\':
+                case '|':
+                case '(':
+                case ')':
+                    // case '[':
+                    // case ']':
+                    // case '{':
+                    // case '}':
+                case '^':
+                case '$':
+                case '+':
+                case '.':
+                case '<':
+                case '>':
+                    regex.append("\\").append(chars[i]);
+                    break;
+                default:
+                    regex.append(chars[i]);
+                    break;
+            }
+        }
+        return regex.toString();
+    }
+
+    /**
+     * Lazily load artifact processors registered in the extension point.
+     */
+    private synchronized void loadProcessors() {
+        if (loaded)
+            return;
+
+        // Get the processor service declarations
+        Collection<ServiceDeclaration> processorDeclarations;
+        try {
+            processorDeclarations =
+                registry.getServiceDiscovery().getServiceDeclarations(URLArtifactProcessor.class.getName());
+        } catch (IOException e) {
+            IllegalStateException ie = new IllegalStateException(e);
+            throw ie;
+        }
+
+        for (ServiceDeclaration processorDeclaration : processorDeclarations) {
+            Map<String, String> attributes = processorDeclaration.getAttributes();
+            // Load a URL artifact processor
+            String artifactType = attributes.get("type");
+            String modelTypeName = attributes.get("model");
+
+            // Create a processor wrapper and register it
+            URLArtifactProcessor<?> processor =
+                new LazyURLArtifactProcessor(artifactType, modelTypeName, processorDeclaration, registry,
+                                             staxProcessor);
+            addArtifactProcessor(processor);
+        }
+
+        loaded = true;
+    }
+
+    /**
+     * A wrapper around an Artifact processor class allowing lazy loading and
+     * initialization of artifact processors.
+     */
+    private static class LazyURLArtifactProcessor implements ExtendedURLArtifactProcessor {
+
+        private ExtensionPointRegistry extensionPoints;
+        private String artifactType;
+        private String modelTypeName;
+        private ServiceDeclaration processorDeclaration;
+        private URLArtifactProcessor<?> processor;
+        private Class<?> modelType;
+        private StAXArtifactProcessor<?> staxProcessor;
+
+        LazyURLArtifactProcessor(String artifactType,
+                                 String modelTypeName,
+                                 ServiceDeclaration processorDeclaration,
+                                 ExtensionPointRegistry extensionPoints,
+                                 StAXArtifactProcessor<?> staxProcessor) {
+            this.artifactType = artifactType;
+            this.modelTypeName = modelTypeName;
+            this.processorDeclaration = processorDeclaration;
+            this.extensionPoints = extensionPoints;
+            this.staxProcessor = staxProcessor;
+        }
+
+        public String getArtifactType() {
+            return artifactType;
+        }
+
+        @SuppressWarnings("unchecked")
+        private URLArtifactProcessor getProcessor() {
+            if (processor == null) {
+                try {
+                    FactoryExtensionPoint modelFactories =
+                        extensionPoints.getExtensionPoint(FactoryExtensionPoint.class);
+                    Class<URLArtifactProcessor> processorClass =
+                        (Class<URLArtifactProcessor>)processorDeclaration.loadClass();
+                    try {
+                        Constructor<URLArtifactProcessor> constructor =
+                            processorClass.getConstructor(FactoryExtensionPoint.class);
+                        processor = constructor.newInstance(modelFactories);
+                    } catch (NoSuchMethodException e) {
+                        try {
+                            Constructor<URLArtifactProcessor> constructor =
+                                processorClass.getConstructor(FactoryExtensionPoint.class,
+                                                              StAXArtifactProcessor.class);
+                            processor = constructor.newInstance(modelFactories, staxProcessor);
+                        } catch (NoSuchMethodException e2) {
+                            Constructor<URLArtifactProcessor> constructor =
+                                processorClass.getConstructor(ExtensionPointRegistry.class,
+                                                              StAXArtifactProcessor.class);
+                            processor = constructor.newInstance(extensionPoints, staxProcessor);
+                        }
+                    }
+                } catch (Throwable e) {
+                    IllegalStateException ie = new IllegalStateException("Exception during getProcessor() for " + 
+                    		                                             processorDeclaration.getClassName(), e);
+                    throw ie;
+                }
+            }
+            return processor;
+        }
+
+        public Object read(URL contributionURL, URI artifactURI, URL artifactURL, ProcessorContext context) throws ContributionReadException {
+            return getProcessor().read(contributionURL, artifactURI, artifactURL, context);
+        }
+
+        public Class<?> getModelType() {
+            if (modelTypeName != null && modelType == null) {
+                try {
+                    modelType = processorDeclaration.loadClass(modelTypeName);
+                } catch (ClassNotFoundException e) {
+                    IllegalStateException ie = new IllegalStateException(e);
+                    throw ie;
+                }
+            }
+            return modelType;
+        }
+
+        @SuppressWarnings("unchecked")
+        public void resolve(Object model, ModelResolver resolver, ProcessorContext context) throws ContributionResolveException {
+            getProcessor().resolve(model, resolver, context);
+        } // end method resolve
+        
+        /**
+         * Preresolve phase, for ExtendedURLArtifactProcessors only
+         */
+        @SuppressWarnings("unchecked")
+        public void preResolve( Object model, ModelResolver resolver, ProcessorContext context) throws ContributionResolveException {
+        	URLArtifactProcessor<?> processor = getProcessor();
+        	if( processor instanceof ExtendedURLArtifactProcessor ) {
+        		((ExtendedURLArtifactProcessor)processor).preResolve(model, resolver, context);
+        	} // end if
+        } // end method resolve
+
+    } // end class LazyURLArtifactProcessor
+} // end class DefaultURLArtifactProcessorExtensionPoint