*/

package org.gradle.api.internal.rules;

import org.gradle.api.NamedDomainObjectFactory;
import org.gradle.model.internal.core.rule.describe.ModelRuleDescriptor;

public interface RuleAwareNamedDomainObjectFactoryRegistry<T> extends NamedDomainObjectFactoryRegistry<T> {

    <U extends T> void registerFactory(Class<U> type, NamedDomainObjectFactory<? extends U> factory, ModelRuleDescriptor descriptor);

public class DefaultRuleAwarePolymorphicNamedEntityInstantiator<T> implements RuleAwarePolymorphicNamedEntityInstantiator<T> {

    private final PolymorphicNamedEntityInstantiator<T> instantiator;
    private final RuleAwareNamedDomainObjectFactoryRegistry<T> registry;

    public DefaultRuleAwarePolymorphicNamedEntityInstantiator(PolymorphicNamedEntityInstantiator<T> instantiator) {

import org.gradle.api.Named
import org.gradle.api.NamedDomainObjectFactory
import org.gradle.api.Namer
import org.gradle.api.internal.DefaultPolymorphicDomainObjectContainer
import org.gradle.api.internal.rules.RuleAwareNamedDomainObjectFactoryRegistry
import org.gradle.internal.reflect.DirectInstantiator
import org.gradle.internal.reflect.Instantiator
import org.gradle.api.reflect.ObjectInstantiationException
import org.gradle.model.internal.core.*

import org.gradle.model.internal.core.NamedEntityInstantiator;

public interface RuleAwarePolymorphicNamedEntityInstantiator<T> extends NamedEntityInstantiator<T>, PolymorphicNamedEntityInstantiator<T>, RuleAwareNamedDomainObjectFactoryRegistry<T> {

import org.gradle.model.internal.core.rule.describe.ModelRuleDescriptor;

import java.util.HashMap;
import java.util.Map;

public class DefaultRuleAwareNamedDomainObjectFactoryRegistry<T> implements RuleAwareNamedDomainObjectFactoryRegistry<T> {

    private final Map<Class<? extends T>, Optional<ModelRuleDescriptor>> creators = new HashMap<>();
    private final NamedDomainObjectFactoryRegistry<T> delegate;