/*
 * 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.myfaces.extensions.validator.core.property;

import org.apache.myfaces.extensions.validator.internal.UsageCategory;
import org.apache.myfaces.extensions.validator.internal.UsageInformation;
import org.apache.myfaces.extensions.validator.core.metadata.MetaDataEntry;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import java.util.HashMap;
import java.util.List;
import java.util.ArrayList;
import java.util.Map;

/**
 * @author Gerhard Petracek
 * @since 1.x.1
 */
@UsageInformation(UsageCategory.INTERNAL)
public class DefaultPropertyInformation implements PropertyInformation
{
    protected final Log logger = LogFactory.getLog(getClass());

    private Map<String, Object> informationMap = new HashMap<String, Object>();
    private List<MetaDataEntry> metaDataList = new ArrayList<MetaDataEntry>();

    public boolean containsInformation(String key)
    {
        return informationMap.containsKey(key);
    }

    public Object getInformation(String key)
    {
        return informationMap.get(key);
    }

    public <T> T getInformation(String key, Class<T> targetClass)
    {
        return (T) getInformation(key);
    }

    public void setInformation(String key, Object value)
    {
        if(this.logger.isTraceEnabled())
        {
            this.logger.trace("new information added key: " + key + " value: " + value);
        }

        informationMap.put(key, value);
    }

    public MetaDataEntry[] getMetaDataEntries()
    {
        return metaDataList.toArray(new MetaDataEntry[metaDataList.size()]);
    }

    public void addMetaDataEntry(MetaDataEntry metaDataEntry)
    {
        metaDataEntry.setProperties(this.informationMap);
        this.metaDataList.add(metaDataEntry);
    }

    public void resetMetaDataEntries()
    {
        if(this.logger.isTraceEnabled())
        {
            this.logger.trace("resetting meta-data entries");
        }

        this.metaDataList.clear();
    }
}
