/*
 * Copyright (c) 2006-2011 Nuxeo SA (http://nuxeo.com/) and others.
 *
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     bstefanescu
 */
package org.eclipse.ecr.web.jaxrs.scan;

import java.io.InputStream;
import java.net.URL;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

import org.objectweb.asm.ClassReader;
import org.osgi.framework.Bundle;

/**
 * @author <a href="mailto:bs@nuxeo.com">Bogdan Stefanescu</a>
 *
 */
public class Scanner {

    public final static String PATH_ANNO = "Ljavax/ws/rs/Path;";
    public final static String PROVIDER_ANNO = "Ljavax/ws/rs/ext/Provider;";

    protected Bundle bundle;

    protected String packageBase;

    protected Map<String, Collection<Class<?>>> collectors;

    public Scanner(Bundle bundle, String packageBase) {
        this (bundle, packageBase, PATH_ANNO, PROVIDER_ANNO);
    }

    public Scanner(Bundle bundle, String packageBase, String ... annotations) {
        this.bundle = bundle;
        this.packageBase = packageBase;
        this.collectors = new HashMap<String, Collection<Class<?>>>();
        for (String annotation  : annotations) {
            addCollector(annotation);
        }
    }

    public void addCollector(String annotation) {
        collectors.put(annotation, new ArrayList<Class<?>>());
    }

    public void addCollector(String annotation, Collection<Class<?>> collector) {
        collectors.put(annotation, collector);
    }

    public Collection<Class<?>> getCollector(String annotation) {
        return collectors.get(annotation);
    }

    public Map<String, Collection<Class<?>>> getCollectors() {
        return collectors;
    }

    @SuppressWarnings("unchecked")
    public void scan() throws Exception {
        Enumeration<URL> urls = bundle.findEntries(packageBase, "*.class", true);
        if (urls == null) {
            return;
        }
        Set<String> annotations = collectors.keySet();
        while (urls.hasMoreElements()) {
            URL url = urls.nextElement();
            InputStream in = url.openStream();
            try {
                ClassReader cr = new ClassReader(in);
                AnnotationReader reader = new AnnotationReader(annotations);
                cr.accept(reader, null,
                        ClassReader.SKIP_DEBUG | ClassReader.SKIP_CODE | ClassReader.SKIP_FRAMES);
                if (reader.hasResults()) {
                    String cname = reader.getClassName();
                    for (String anno : reader.getResults()) {
                        collectors.get(anno).add(bundle.loadClass(cname));
                    }
                }
            } finally {
                in.close();
            }
        }
    }

    public Set<Class<?>> getClasses() {
        HashSet<Class<?>> result = new HashSet<Class<?>>();
        for (Collection<Class<?>> c : collectors.values()) {
            result.addAll(c);
        }
        return result;
    }

    public Set<Class<?>> getClasses(String anno) {
        return new HashSet<Class<?>>(collectors.get(anno));
    }
}