/*
 * 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.geronimo.components.jaspi.model;

import java.lang.reflect.Array;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.security.auth.callback.CallbackHandler;
import javax.xml.bind.annotation.adapters.XmlAdapter;

/**
 * @version $Rev: 939768 $ $Date: 2010-04-30 11:26:46 -0700 (Fri, 30 Apr 2010) $
 */
public class KeyedObjectMapAdapter<T extends KeyedObject> extends XmlAdapter<List<T>, Map<String, T>> {
    public static CallbackHandler staticCallbackHandler;
    private final CallbackHandler callbackHandler;
//    private final Class<T> type;

    public KeyedObjectMapAdapter(CallbackHandler callbackHandler) {
        this.callbackHandler = callbackHandler;
//        this.type = type;
    }

    public KeyedObjectMapAdapter() {
        this(staticCallbackHandler);
    }

    public Map<String, T> unmarshal(List<T> configProviderTypes) throws Exception {
        Map<String, T> map = new HashMap<String, T>();
        if (configProviderTypes != null) {
            for (T configProviderType : configProviderTypes) {
                if (configProviderType != null) {
                    String key = configProviderType.getKey();
                    map.put(key, configProviderType);
                }
            }
        }
        return map;
    }

    public ArrayList<T> marshal(Map<String, T> stringConfigProviderTypeMap) throws Exception {
        if (stringConfigProviderTypeMap == null) {
            return null;
        }
        ArrayList<T> list = new ArrayList<T>();
        for (T configProviderType : stringConfigProviderTypeMap.values()) {
            list.add(configProviderType);
        }
        return list;
    }
}
