/*
* JBoss, Home of Professional Open Source.
* Copyright 2011, Red Hat, Inc., and individual contributors
* as indicated by the @author tags. See the copyright.txt file in the
* distribution for a full listing of individual contributors.
*
* This is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as
* published by the Free Software Foundation; either version 2.1 of
* the License, or (at your option) any later version.
*
* This software is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this software; if not, write to the Free
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
*/

package org.jboss.as.controller.operations.validation;


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

import org.jboss.as.controller.AttributeDefinition;
import org.jboss.as.controller.OperationFailedException;
import org.jboss.dmr.ModelNode;
import org.jboss.dmr.ModelType;

/**
* Date: 16.11.2011
*
* @author <a href="mailto:jperkins@redhat.com">James R. Perkins</a>
*/
public class ObjectTypeValidator extends ModelTypeValidator implements AllowedValuesValidator {
    private final Map<String, AttributeDefinition> allowedValues;
    private final List<ModelNode> nodeValues;

    public ObjectTypeValidator(final boolean nullable, final AttributeDefinition... attributes) {
        super(nullable, true, false, ModelType.OBJECT, findModelTypes(attributes));
        allowedValues = new HashMap<String, AttributeDefinition>(attributes.length);
        nodeValues = new ArrayList<ModelNode>(attributes.length);
        for (AttributeDefinition attribute : attributes) {
            allowedValues.put(attribute.getName(), attribute);
            nodeValues.add(new ModelNode(attribute.getName()));
        }
    }

    @Override
    public void validateParameter(final String parameterName, final ModelNode value) throws OperationFailedException {
        super.validateParameter(parameterName, value);
        if (value.isDefined()) {
            for (String key : value.keys()) {
                if (allowedValues.containsKey(key)) {
                    allowedValues.get(key).getValidator().validateParameter(key, value.get(key));
                } else {
                    throw new OperationFailedException(new ModelNode().set("invalid type key"));
                }
            }
        }
    }

    @Override
    public List<ModelNode> getAllowedValues() {
        return nodeValues;
    }

    private static ModelType[] findModelTypes(final AttributeDefinition... attributes) {
        final Set<ModelType> result = new HashSet<ModelType>();
        for (AttributeDefinition attr : attributes) {
            if (attr.getType() != ModelType.OBJECT)
                result.add(attr.getType());
        }
        return result.toArray(new ModelType[result.size()]);
    }
}
@@ -1112,34 +1112,8 @@
 ap;%0A
-import java.util.HashSet;%0A
 impo
@@ -1156,30 +1156,8 @@
 Map;
-%0Aimport java.util.Set;
 %0A%0Aim
@@ -1788,36 +1788,8 @@
 JECT
-, findModelTypes(attributes)
 );%0A 
@@ -2829,387 +2829,6 @@
    %7D
-%0A%0A    private static ModelType%5B%5D findModelTypes(final AttributeDefinition... attributes) %7B%0A        final Set%3CModelType%3E result = new HashSet%3CModelType%3E();%0A        for (AttributeDefinition attr : attributes) %7B%0A            if (attr.getType() != ModelType.OBJECT)%0A                result.add(attr.getType());%0A        %7D%0A        return result.toArray(new ModelType%5Bresult.size()%5D);%0A    %7D
 %0A%7D

ObjectTypeValidator should only allow for ModelType.OBJECT for it's value.