* See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.gradle.internal.instrumentation.api.types;

public enum BytecodeInterceptorType {

    /**
     * An interceptor that is applied to the bytecode always.
     */
    INSTRUMENTATION(
        FilterableBytecodeInterceptor.InstrumentationInterceptor.class,

    BytecodeInterceptorType getType();

    /**
     * A marker interface that indicates that a class is used for generating bytecode upgrades.
     */
    interface BytecodeUpgradeInterceptorFactory extends FilterableBytecodeInterceptorFactory {
        @Override
        default BytecodeInterceptorType getType() {
            return BytecodeInterceptorType.BYTECODE_UPGRADE;
        }
    }

public interface FilterableBytecodeInterceptor {

    BytecodeInterceptorType getType();

    /**
     * A marker interface that indicates that a class is used for bytecode upgrades.
     */
    interface BytecodeUpgradeInterceptor extends FilterableBytecodeInterceptor {
        @Override
        default BytecodeInterceptorType getType() {
            return BytecodeInterceptorType.BYTECODE_UPGRADE;
        }
    }

    /**

    INSTRUMENTATION_ONLY(EnumSet.of(BytecodeInterceptorType.INSTRUMENTATION)),
    ALL(EnumSet.of(BytecodeInterceptorType.INSTRUMENTATION, BytecodeInterceptorType.BYTECODE_UPGRADE));

    @SuppressWarnings("ImmutableEnumChecker")
    private final EnumSet<BytecodeInterceptorType> instrumentationTypes;

    BytecodeInterceptorFilter(EnumSet<BytecodeInterceptorType> instrumentationTypes) {

import org.gradle.internal.instrumentation.api.types.FilterableBytecodeInterceptor;
import org.gradle.internal.instrumentation.api.types.BytecodeInterceptorType;

/**
 * CallInterceptor that can be filtered by a {@link BytecodeInterceptorType} type.
 */
public interface FilterableCallInterceptor extends CallInterceptor, FilterableBytecodeInterceptor {

        assertThat(compilation)
            .generatedSourceFile(fqName(expectedJvmInterceptors))
            .containsElementsIn(expectedJvmInterceptors)

        where:
        type << [BytecodeInterceptorType.INSTRUMENTATION, BytecodeInterceptorType.BYTECODE_UPGRADE]
    }

    def "should group visitMethodInsn logic by call owner"() {
        given:
        def givenFirstSource = source """
            package org.gradle.test;

import org.gradle.internal.instrumentation.api.metadata.InstrumentationMetadata;
import org.gradle.internal.instrumentation.api.types.BytecodeInterceptorFilter;
import org.gradle.internal.instrumentation.api.types.BytecodeInterceptorType;
import org.gradle.internal.instrumentation.model.CallInterceptionRequest;
import org.gradle.internal.instrumentation.model.CallableInfo;
import org.gradle.internal.instrumentation.model.CallableKindInfo;

import com.squareup.javapoet.TypeName;
import com.squareup.javapoet.TypeSpec;
import com.squareup.javapoet.WildcardTypeName;
import org.gradle.internal.instrumentation.api.types.BytecodeInterceptorType;
import org.gradle.internal.instrumentation.model.CallInterceptionRequest;
import org.gradle.internal.instrumentation.model.CallableKindInfo;
import org.gradle.internal.instrumentation.model.ParameterInfo;

import static org.gradle.internal.instrumentation.api.declarations.InterceptorDeclaration.JVM_BYTECODE_GENERATED_CLASS_NAME_FOR_PROPERTY_UPGRADES;
import static org.gradle.internal.instrumentation.api.types.BytecodeInterceptorType.BYTECODE_UPGRADE;
import static org.gradle.internal.instrumentation.model.CallableKindInfo.GROOVY_PROPERTY_GETTER;
import static org.gradle.internal.instrumentation.model.CallableKindInfo.INSTANCE_METHOD;