package org.apache.jackrabbit.oak.osgi;

import static com.google.common.base.Preconditions.checkNotNull;
import static com.google.common.base.Preconditions.checkState;
import static com.google.common.collect.Maps.newHashMap;

import java.io.Closeable;
import java.util.Map;

import javax.annotation.Nonnull;

import com.google.common.io.Closeables;
import org.apache.jackrabbit.oak.spi.commit.Observable;
import org.apache.jackrabbit.oak.spi.commit.Observer;
import org.osgi.framework.BundleContext;
import org.osgi.framework.ServiceReference;
import org.osgi.util.tracker.ServiceTracker;
import org.osgi.util.tracker.ServiceTrackerCustomizer;

public class ObserverTracker implements ServiceTrackerCustomizer {
    private final Observable observable;
    private final Map<ServiceReference, Closeable> subscriptions = newHashMap();

    private BundleContext bundleContext;
    private ServiceTracker tracker;

    public ObserverTracker(@Nonnull Observable observable) {
        this.observable = checkNotNull(observable);
    }

    public void start(@Nonnull BundleContext bundleContext) {
        checkState(this.bundleContext == null);
        this.bundleContext = checkNotNull(bundleContext);
        tracker = new ServiceTracker(bundleContext, Observer.class.getName(), this);
        tracker.open();
    }

    public void stop() {
        checkState(this.bundleContext != null);
        tracker.close();
    }

    //------------------------< ServiceTrackerCustomizer >----------------------

    @Override
    public Object addingService(ServiceReference reference) {
        Object service = bundleContext.getService(reference);

        if (service instanceof Observer) {
            Closeable subscription = observable.addObserver((Observer) service);
            subscriptions.put(reference, subscription);
            return service;
        } else {
            bundleContext.ungetService(reference);
            return null;
        }
    }

    @Override
    public void modifiedService(ServiceReference reference, Object service) {
        // nothing to do
    }

    @Override
    public void removedService(ServiceReference reference, Object service) {
        Closeables.closeQuietly(subscriptions.get(reference));
        bundleContext.ungetService(reference);
    }

}

@@ -1,8 +1,817 @@
+/*%0A * Licensed to the Apache Software Foundation (ASF) under one%0A * or more contributor license agreements.  See the NOTICE file%0A * distributed with this work for additional information%0A * regarding copyright ownership.  The ASF licenses this file%0A * to you under the Apache License, Version 2.0 (the%0A * %22License%22); you may not use this file except in compliance%0A * with the License.  You may obtain a copy of the License at%0A *%0A *   http://www.apache.org/licenses/LICENSE-2.0%0A *%0A * Unless required by applicable law or agreed to in writing,%0A * software distributed under the License is distributed on an%0A * %22AS IS%22 BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY%0A * KIND, either express or implied.  See the License for the%0A * specific language governing permissions and limitations%0A * under the License.%0A */%0A%0A
 package 

Enhance observation mechanism to support Apache Sling the dreaded license header...