/**
 * 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.cxf.transport.http.asyncclient;

import java.util.Dictionary;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.Hashtable;
import java.util.Map;

import org.apache.cxf.Bus;
import org.osgi.framework.BundleActivator;
import org.osgi.framework.BundleContext;
import org.osgi.framework.Constants;
import org.osgi.framework.ServiceReference;
import org.osgi.service.cm.ManagedService;
import org.osgi.util.tracker.ServiceTracker;

public class Activator implements BundleActivator {
    private ConduitConfigurer conduitConfigurer;

    @Override
    public void start(BundleContext context) throws Exception {
        conduitConfigurer = new ConduitConfigurer(context);
        conduitConfigurer.open();
        Dictionary<String, Object> properties = new Hashtable<>();
        properties.put(Constants.SERVICE_PID, "org.apache.cxf.transport.http.async");
        context.registerService(ManagedService.class.getName(), conduitConfigurer, properties);
    }

    @Override
    public void stop(BundleContext context) throws Exception {
        conduitConfigurer.close();
    }

    public static class ConduitConfigurer extends ServiceTracker<Bus, Bus> implements ManagedService {

        private Map<String, Object> currentConfig;

        public ConduitConfigurer(BundleContext context) {
            super(context, Bus.class, null);
        }

        @Override
        public void updated(Dictionary<String, ?> properties) {
            this.currentConfig = toMap(properties);
            Object[] buses = getServices();
            if (buses == null) {
                return;
            }
            for (Object bus : buses) {
                configureConduitFactory((Bus) bus);
            }
        }

        @Override
        public Bus addingService(ServiceReference<Bus> reference) {
            Bus bus = super.addingService(reference);
            configureConduitFactory(bus);
            return bus;
        }

        private Map<String, Object> toMap(Dictionary<String, ?> properties) {
            Map<String, Object> props = new HashMap<>();
            if (properties == null) {
                return props;
            }
            Enumeration<String> keys = properties.keys();
            while (keys.hasMoreElements()) {
                String key = keys.nextElement();
                props.put(key, properties.get(key));
            }
            return props;
        }

        private void configureConduitFactory(Bus bus) {
            AsyncHTTPConduitFactory conduitFactory = bus.getExtension(AsyncHTTPConduitFactory.class);
            conduitFactory.update(this.currentConfig);
        }

    }
}
