org.gradle.platform.base.internal.registry.ModelMapBasedRule.java Source code

Introduction

Here is the source code for org.gradle.platform.base.internal.registry.ModelMapBasedRule.java

Source

/*
 * Copyright 2014 the original author or authors.
 *
 * Licensed 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.gradle.platform.base.internal.registry;

import com.google.common.base.Predicate;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.Iterables;
import org.gradle.model.ModelMap;
import org.gradle.model.internal.core.AbstractModelActionWithView;
import org.gradle.model.internal.core.ModelReference;
import org.gradle.model.internal.core.ModelView;
import org.gradle.model.internal.inspect.MethodRuleDefinition;
import org.gradle.model.internal.type.ModelType;

import java.util.Arrays;
import java.util.List;

public abstract class ModelMapBasedRule<R, S, T, C> extends AbstractModelActionWithView<C> {
    private final MethodRuleDefinition<R, ?> ruleDefinition;
    protected int baseTypeParameterIndex;

    public ModelMapBasedRule(ModelReference<C> subject, final Class<? extends T> baseType,
            MethodRuleDefinition<R, ?> ruleDefinition, ModelReference<?>... additionalInputs) {
        super(subject, ruleDefinition.getDescriptor(),
                calculateInputs(baseType,
                        ruleDefinition.getReferences().subList(1, ruleDefinition.getReferences().size()),
                        Arrays.asList(additionalInputs)));
        this.ruleDefinition = ruleDefinition;
        this.baseTypeParameterIndex = 1 + Iterables.indexOf(
                ruleDefinition.getReferences().subList(1, ruleDefinition.getReferences().size()),
                new Predicate<ModelReference<?>>() {
                    @Override
                    public boolean apply(ModelReference<?> element) {
                        return element.getType().equals(ModelType.of(baseType));
                    }
                });
    }

    private static ImmutableList<ModelReference<?>> calculateInputs(final Class<?> baseType,
            final List<ModelReference<?>> references, List<ModelReference<?>> modelReferences) {
        Iterable<ModelReference<?>> filteredReferences = Iterables.filter(references,
                new Predicate<ModelReference<?>>() {
                    @Override
                    public boolean apply(ModelReference<?> element) {
                        return !element.getType().equals(ModelType.of(baseType));
                    }
                });

        ImmutableList.Builder<ModelReference<?>> allInputs = ImmutableList.builder();
        allInputs.addAll(modelReferences);
        allInputs.addAll(filteredReferences);
        return allInputs.build();
    }

    protected void invoke(List<ModelView<?>> inputs, ModelMap<S> modelMap, T baseTypeParameter,
            Object... ignoredInputs) {
        List<Object> ignoredInputsList = Arrays.asList(ignoredInputs);
        Object[] args = new Object[inputs.size() + 2 - ignoredInputs.length];
        args[0] = modelMap;
        args[baseTypeParameterIndex] = baseTypeParameter;

        for (ModelView<?> view : inputs) {
            Object instance = view.getInstance();
            if (ignoredInputsList.contains(instance)) {
                continue;
            }
            for (int i = 0; i < args.length; i++) {
                if (args[i] == null) {
                    args[i] = instance;
                    break;
                }
            }
        }
        ruleDefinition.getRuleInvoker().invoke(args);
    }
}