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

package org.gradle.internal.instrumentation.model;

public interface CallInterceptionRequest {
    CallableInfo getInterceptedCallable();
    ImplementationInfo getImplementationInfo();
    RequestExtrasContainer getRequestExtras();

    protected abstract String classNameForRequest(CallInterceptionRequest request);

    protected abstract Consumer<TypeSpec.Builder> classContentForClass(
        String className,
        List<CallInterceptionRequest> requestsClassGroup,
        Consumer<? super CallInterceptionRequest> onProcessedRequest,
        Consumer<? super HasFailures.FailureInfo> onFailure
    );

    @Override

        this.produceFlagForGeneratedClassName = produceFlagForGeneratedClassName;
    }


    @Override
    public Collection<CallInterceptionRequest> postProcessRequest(CallInterceptionRequest originalRequest) {
        Optional<ExecutableElement> maybeOriginatingElement = originalRequest.getRequestExtras().getByType(OriginatingElement.class)
            .map(OriginatingElement::getElement);

 */
public class InstrumentedTypesResourceGenerator implements InstrumentationResourceGenerator {
    @Override
    public Collection<CallInterceptionRequest> filterRequestsForResource(Collection<CallInterceptionRequest> interceptionRequests) {
        return interceptionRequests.stream()
            .filter(request -> request.getInterceptedCallable().getOwner().isInterceptSubtypes())

import java.util.Collection;
import java.util.Collections;

public class WithExtensionReferencesReader implements RequestPostProcessorExtension {
    @Override
    public Collection<CallInterceptionRequest> postProcessRequest(CallInterceptionRequest originalRequest) {
        if (shouldPostProcess(originalRequest)) {
            originalRequest.getRequestExtras().getByType(RequestExtra.OriginatingElement.class).ifPresent(originatingElement -> {

        BytecodeInterceptorType getInterceptorType();

        List<CallInterceptionRequest> getRequests();

        class NamedCallableInterceptorSpec implements CallInterceptorSpec {

            private final String name;
            private final String className;
            private final String fullClassName;
            private final List<CallInterceptionRequest> requests;
            private final BytecodeInterceptorType interceptorType;

    protected Consumer<TypeSpec.Builder> classContentForClass(
        String className,
        List<CallInterceptionRequest> requestsClassGroup,
        Consumer<? super CallInterceptionRequest> onProcessedRequest,
        Consumer<? super FailureInfo> onFailure
    ) {
        Map<Type, FieldSpec> typeFieldByOwner = generateFieldsForImplementationOwners(requestsClassGroup);

import org.gradle.internal.instrumentation.model.CallInterceptionRequest;

import java.io.OutputStream;
import java.util.Collection;
import java.util.List;

public interface InstrumentationResourceGenerator {

    /**
     * Filter the requests to only those that are relevant to the resource being generated.
     */
    Collection<CallInterceptionRequest> filterRequestsForResource(

import java.util.stream.Stream;

public class WithExtensionReferencesPostProcessor implements RequestPostProcessorExtension {

    @Override
    public Collection<CallInterceptionRequest> postProcessRequest(CallInterceptionRequest originalRequest) {
        Optional<WithExtensionReferencesExtra> extra = originalRequest.getRequestExtras().getByType(WithExtensionReferencesExtra.class);
        return extra

package org.gradle.internal.instrumentation.processor.extensibility;

import org.gradle.internal.instrumentation.model.CallInterceptionRequest;

import java.util.Collection;

public interface RequestPostProcessorExtension extends InstrumentationProcessorExtension {
    Collection<CallInterceptionRequest> postProcessRequest(CallInterceptionRequest originalRequest);