// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.

package com.microsoft.typespec.http.client.generator.core.template;

import com.azure.core.util.CoreUtils;
import com.microsoft.typespec.http.client.generator.core.extension.plugin.JavaSettings;
import com.microsoft.typespec.http.client.generator.core.model.clientmodel.Annotation;
import com.microsoft.typespec.http.client.generator.core.model.clientmodel.ClassType;
import com.microsoft.typespec.http.client.generator.core.model.clientmodel.ClientMethod;
import com.microsoft.typespec.http.client.generator.core.model.clientmodel.ClientMethodParameter;
import com.microsoft.typespec.http.client.generator.core.model.clientmodel.ClientMethodType;
import com.microsoft.typespec.http.client.generator.core.model.clientmodel.PrimitiveType;
import com.microsoft.typespec.http.client.generator.core.model.clientmodel.ProxyMethod;
import com.microsoft.typespec.http.client.generator.core.model.javamodel.JavaBlock;
import com.microsoft.typespec.http.client.generator.core.model.javamodel.JavaClass;
import com.microsoft.typespec.http.client.generator.core.model.javamodel.JavaJavadocComment;
import com.microsoft.typespec.http.client.generator.core.model.javamodel.JavaType;
import com.microsoft.typespec.http.client.generator.core.model.javamodel.JavaVisibility;
import com.microsoft.typespec.http.client.generator.core.util.TemplateUtil;
import java.util.List;
import java.util.Map;
import java.util.function.Consumer;
import java.util.stream.Collectors;

/**
 * Template to generate client methods that are wrappers around the client methods generated by
 * {@link ClientMethodTemplate}.
 *
 */
public class WrapperClientMethodTemplate extends ClientMethodTemplateBase {

    private static final WrapperClientMethodTemplate INSTANCE = new WrapperClientMethodTemplate();

    protected WrapperClientMethodTemplate() {
    }

    public static WrapperClientMethodTemplate getInstance() {
        return INSTANCE;
    }

    @Override
    public void write(ClientMethod clientMethod, JavaType typeBlock) {
        JavaSettings settings = JavaSettings.getInstance();

        if (clientMethod.getType() == ClientMethodType.PagingAsyncSinglePage
            || clientMethod.getType() == ClientMethodType.PagingSyncSinglePage
            || clientMethod.getMethodVisibilityInWrapperClient() == null
            || clientMethod.getType() == ClientMethodType.LongRunningSync) {
            return;
        }

        ProxyMethod restAPIMethod = clientMethod.getProxyMethod();
        if (settings.isDataPlaneClient()) {
            typeBlock.javadocComment(comment -> generateProtocolMethodJavadoc(clientMethod, comment));
        } else {
            generateJavadoc(clientMethod, typeBlock, restAPIMethod);
        }

        addGeneratedAnnotation(typeBlock);
        TemplateUtil.writeClientMethodServiceMethodAnnotation(clientMethod, typeBlock);

        String methodName = clientMethod.getName();
        if (clientMethod.getType().name().contains("Async") && methodName.endsWith("Async")) {
            methodName = methodName.substring(0, methodName.length() - "Async".length());
        }

        String declaration = String.format("%1$s %2$s(%3$s)", clientMethod.getReturnValue().getType(), methodName,
            clientMethod.getParametersDeclaration());
        Consumer<JavaBlock> method = function -> {

            // API comment
            if (clientMethod.getImplementationDetails() != null
                && !CoreUtils.isNullOrEmpty(clientMethod.getImplementationDetails().getComment())) {
                function.line("// " + clientMethod.getImplementationDetails().getComment());
            }

            boolean shouldReturn = true;
            if (clientMethod.getReturnValue() != null
                && clientMethod.getReturnValue().getType() instanceof PrimitiveType) {
                PrimitiveType type = (PrimitiveType) clientMethod.getReturnValue().getType();
                if (type.asNullable() == ClassType.VOID) {
                    shouldReturn = false;
                }
            }

            writeMethodInvocation(clientMethod, function, shouldReturn);
        };
        if (clientMethod.getMethodVisibilityInWrapperClient() == JavaVisibility.Public) {
            typeBlock.publicMethod(declaration, method);
        } else if (typeBlock instanceof JavaClass) {
            JavaClass classBlock = (JavaClass) typeBlock;
            classBlock.method(clientMethod.getMethodVisibilityInWrapperClient(), null, declaration, method);
        }

    }

    /**
     * Extension to write the client method invocation.
     *
     * @param clientMethod the client method
     * @param function the method block to write the method invocation
     * @param shouldReturn whether method need return value
     */
    protected void writeMethodInvocation(ClientMethod clientMethod, JavaBlock function, boolean shouldReturn) {
        List<ClientMethodParameter> parameters = clientMethod.getMethodInputParameters();
        function.line((shouldReturn ? "return " : "") + "this.serviceClient.%1$s(%2$s);", clientMethod.getName(),
            parameters.stream().map(ClientMethodParameter::getName).collect(Collectors.joining(", ")));
    }

    protected void generateJavadoc(ClientMethod clientMethod, JavaType typeBlock, ProxyMethod restAPIMethod) {
        typeBlock.javadocComment(comment -> {
            comment.description(clientMethod.getDescription());
            List<ClientMethodParameter> methodParameters = clientMethod.getMethodInputParameters();
            for (ClientMethodParameter parameter : methodParameters) {
                comment.param(parameter.getName(), parameter.getDescription());
            }
            if (clientMethod.hasParameterDeclaration()) {
                comment.methodThrows("IllegalArgumentException", "thrown if parameters fail the validation");
            }
            if (restAPIMethod != null) {
                generateJavadocExceptions(clientMethod, comment, false);
                comment.methodThrows("RuntimeException",
                    "all other wrapped checked exceptions if the request fails to be sent");
            }
            comment.methodReturns(clientMethod.getReturnValue().getDescription());
        });
    }

    protected static void generateJavadocExceptions(ClientMethod clientMethod, JavaJavadocComment commentBlock,
        boolean useFullClassName) {
        ProxyMethod restAPIMethod = clientMethod.getProxyMethod();
        if (JavaSettings.getInstance().isAzureV1()) {
            if (restAPIMethod != null && restAPIMethod.getUnexpectedResponseExceptionType() != null) {
                commentBlock.methodThrows(
                    useFullClassName
                        ? restAPIMethod.getUnexpectedResponseExceptionType().getFullName()
                        : restAPIMethod.getUnexpectedResponseExceptionType().getName(),
                    "thrown if the request is rejected by server");
            }
            if (restAPIMethod != null && restAPIMethod.getUnexpectedResponseExceptionTypes() != null) {
                for (Map.Entry<ClassType, List<Integer>> exception : restAPIMethod.getUnexpectedResponseExceptionTypes()
                    .entrySet()) {
                    commentBlock.methodThrows(
                        useFullClassName ? exception.getKey().getFullName() : exception.getKey().getName(),
                        String.format("thrown if the request is rejected by server on status code %s",
                            exception.getValue().stream().map(String::valueOf).collect(Collectors.joining(", "))));
                }
            }
        } else {
            if (restAPIMethod != null
                && (restAPIMethod.getUnexpectedResponseExceptionType() != null
                    || restAPIMethod.getUnexpectedResponseExceptionTypes() != null)) {
                commentBlock.methodThrows("HttpResponseException", "thrown if the service returns an error");
            }
        }
    }

    protected void addGeneratedAnnotation(JavaType typeBlock) {
        typeBlock.annotation(Annotation.GENERATED.getName());
    }
}
