// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License. See License.txt in the project root for license information.
import { getHttpOperationParameter, shouldGenerateConvenient, shouldGenerateProtocol, } from "@azure-tools/typespec-client-generator-core";
import { getDeprecated, isErrorModel, NoTarget } from "@typespec/compiler";
import { getResourceOperation } from "@typespec/rest";
import { collectionFormatToDelimMap } from "../type/collection-format.js";
import { InputOperationParameterKind } from "../type/input-operation-parameter-kind.js";
import { convertLroFinalStateVia } from "../type/operation-final-state-via.js";
import { RequestLocation } from "../type/request-location.js";
import { parseHttpRequestMethod } from "../type/request-method.js";
import { ResponseLocation } from "../type/response-location.js";
import { getExternalDocs, getOperationId } from "./decorators.js";
import { fromSdkHttpExamples } from "./example-converter.js";
import { fromSdkModelType, fromSdkType } from "./type-converter.js";
import { getClientNamespaceString } from "./utils.js";
export function fromSdkServiceMethod(sdkContext, method, uri, rootApiVersions) {
    let generateConvenience = shouldGenerateConvenient(sdkContext, method.operation.__raw.operation);
    if (method.operation.verb === "patch" && generateConvenience) {
        sdkContext.logger.reportDiagnostic({
            code: "unsupported-patch-convenience-method",
            format: {
                methodCrossLanguageDefinitionId: method.crossLanguageDefinitionId,
            },
            target: method.__raw ?? NoTarget,
        });
        generateConvenience = false;
    }
    return {
        name: method.name,
        resourceName: getResourceOperation(sdkContext.program, method.operation.__raw.operation)?.resourceType
            .name ??
            getOperationGroupName(sdkContext, method.operation, getClientNamespaceString(sdkContext)),
        deprecated: getDeprecated(sdkContext.program, method.__raw),
        summary: method.summary,
        doc: method.doc,
        accessibility: method.access,
        parameters: fromSdkOperationParameters(sdkContext, method.operation, rootApiVersions),
        responses: fromSdkHttpOperationResponses(sdkContext, method.operation.responses),
        httpMethod: parseHttpRequestMethod(method.operation.verb),
        uri: uri,
        path: method.operation.path,
        externalDocsUrl: getExternalDocs(sdkContext, method.operation.__raw.operation)?.url,
        requestMediaTypes: getRequestMediaTypes(method.operation),
        bufferResponse: true,
        longRunning: loadLongRunningOperation(sdkContext, method),
        paging: loadOperationPaging(sdkContext, method, rootApiVersions, uri),
        generateProtocolMethod: shouldGenerateProtocol(sdkContext, method.operation.__raw.operation),
        generateConvenienceMethod: generateConvenience,
        crossLanguageDefinitionId: method.crossLanguageDefinitionId,
        decorators: method.decorators,
        examples: method.operation.examples
            ? fromSdkHttpExamples(sdkContext, method.operation.examples)
            : undefined,
    };
}
export function getParameterDefaultValue(sdkContext, clientDefaultValue, parameterType) {
    if (clientDefaultValue === undefined ||
        // a constant parameter should overwrite client default value
        parameterType.kind === "constant") {
        return undefined;
    }
    const kind = getValueType(sdkContext, clientDefaultValue);
    return {
        type: {
            kind: kind,
            name: kind,
            crossLanguageDefinitionId: `TypeSpec.${kind}`,
        },
        value: clientDefaultValue,
    };
}
function getValueType(sdkContext, value) {
    switch (typeof value) {
        case "string":
            return "string";
        case "number":
            return "int32";
        case "boolean":
            return "boolean";
        case "bigint":
            return "int64";
        default:
            sdkContext.logger.reportDiagnostic({
                code: "unsupported-default-value-type",
                format: { valueType: typeof value },
                target: NoTarget,
            });
            return "unknown";
    }
}
function fromSdkOperationParameters(sdkContext, operation, rootApiVersions) {
    const parameters = [];
    for (const p of operation.parameters) {
        if (p.kind === "cookie") {
            sdkContext.logger.reportDiagnostic({
                code: "unsupported-cookie-parameter",
                format: { parameterName: p.name, path: operation.path },
                target: NoTarget,
            });
            return parameters;
        }
        const param = fromSdkHttpOperationParameter(sdkContext, p, rootApiVersions);
        parameters.push(param);
    }
    if (operation.bodyParam) {
        const bodyParam = fromSdkHttpOperationParameter(sdkContext, operation.bodyParam, rootApiVersions);
        parameters.push(bodyParam);
    }
    return parameters;
}
export function fromSdkHttpOperationParameter(sdkContext, p, rootApiVersions) {
    let retVar = sdkContext.__typeCache.properties.get(p);
    if (retVar) {
        return retVar;
    }
    const isContentType = p.kind === "header" && p.serializedName.toLocaleLowerCase() === "content-type";
    const parameterType = fromSdkType(sdkContext, p.type);
    const format = p.kind === "header" || p.kind === "query" ? p.collectionFormat : undefined;
    // use serializedName if available, but fallback to name
    // special case for body as the name is incorrectly set to "body" https://github.com/Azure/typespec-azure/issues/2292
    const serializedName = "serializedName" in p && p.kind !== "body" ? (p.serializedName ?? p.name) : p.name;
    // TODO: In addition to checking if a path parameter is exploded, we should consider capturing the delimiter for
    // any path expansion to ensure the parameter values are delimited correctly during serialization.
    // https://github.com/microsoft/typespec/issues/5561
    const explode = isExplodedParameter(p);
    retVar = {
        name: p.name,
        nameInRequest: p.kind === "header" ? normalizeHeaderName(serializedName) : serializedName,
        summary: p.summary,
        doc: p.doc,
        type: parameterType,
        location: getParameterLocation(p),
        isApiVersion: p.name.toLocaleLowerCase() === "apiversion" || p.name.toLocaleLowerCase() === "api-version", // TODO -- we should use `isApiVersionParam` instead
        isContentType: isContentType,
        isEndpoint: false,
        explode: explode,
        arraySerializationDelimiter: format ? collectionFormatToDelimMap[format] : undefined,
        isRequired: !p.optional,
        kind: getParameterKind(p, parameterType, rootApiVersions.length > 0),
        defaultValue: getParameterDefaultValue(sdkContext, p.clientDefaultValue, parameterType),
        decorators: p.decorators,
        skipUrlEncoding: p.kind === "path" ? p.allowReserved : false,
    };
    sdkContext.__typeCache.updateSdkPropertyReferences(p, retVar);
    return retVar;
}
function loadLongRunningOperation(sdkContext, method) {
    if (method.kind !== "lro") {
        return undefined;
    }
    return {
        finalStateVia: convertLroFinalStateVia(method.lroMetadata.finalStateVia),
        finalResponse: {
            // in swagger, we allow delete to return some meaningful body content
            // for now, let assume we don't allow return type
            statusCodes: method.operation.verb === "delete" ? [204] : [200],
            bodyType: method.lroMetadata.finalResponse?.envelopeResult !== undefined
                ? fromSdkModelType(sdkContext, method.lroMetadata.finalResponse.envelopeResult)
                : undefined,
        },
        resultPath: method.lroMetadata.finalResponse?.resultPath,
    };
}
function fromSdkHttpOperationResponses(sdkContext, operationResponses) {
    const responses = [];
    for (const r of operationResponses) {
        responses.push(fromSdkHttpOperationResponse(sdkContext, r));
    }
    return responses;
}
export function fromSdkHttpOperationResponse(sdkContext, sdkResponse) {
    let retVar = sdkContext.__typeCache.responses.get(sdkResponse);
    if (retVar) {
        return retVar;
    }
    const range = sdkResponse.statusCodes;
    retVar = {
        statusCodes: toStatusCodesArray(range),
        bodyType: sdkResponse.type ? fromSdkType(sdkContext, sdkResponse.type) : undefined,
        headers: fromSdkServiceResponseHeaders(sdkContext, sdkResponse.headers),
        isErrorResponse: sdkResponse.type !== undefined && isErrorModel(sdkContext.program, sdkResponse.type.__raw),
        contentTypes: sdkResponse.contentTypes,
    };
    sdkContext.__typeCache.updateSdkResponseReferences(sdkResponse, retVar);
    return retVar;
}
function fromSdkServiceResponseHeaders(sdkContext, headers) {
    return headers.map((h) => ({
        name: h.__raw.name,
        nameInResponse: h.serializedName,
        summary: h.summary,
        doc: h.doc,
        type: fromSdkType(sdkContext, h.type),
    }));
}
function toStatusCodesArray(range) {
    if (typeof range === "number")
        return [range];
    const statusCodes = [range.end - range.start + 1];
    for (let i = range.start; i <= range.end; i++) {
        statusCodes.push(i);
    }
    return statusCodes;
}
function getRequestMediaTypes(op) {
    const contentTypes = op.parameters.filter((p) => p.kind === "header" && p.serializedName.toLocaleLowerCase() === "content-type");
    if (contentTypes.length === 0)
        return undefined;
    return contentTypes.map((p) => getMediaTypes(p.type)).flat();
}
function getMediaTypes(type) {
    if (type.kind === "constant") {
        if (type.valueType.kind !== "string") {
            throw `Media type in "content-type" should be string. But get ${type.valueType.kind}.`;
        }
        return [type.value];
    }
    else if (type.kind === "union") {
        const mediaTypes = [];
        for (const unionItem of type.variantTypes) {
            if (unionItem.kind === "constant" && unionItem.valueType.kind === "string") {
                mediaTypes.push(unionItem.value);
            }
            else {
                throw `Media type in "content-type" should be string. But get ${unionItem.kind}.`;
            }
        }
        return mediaTypes;
    }
    else if (type.kind === "enum") {
        if (type.valueType.kind !== "string") {
            throw `Media type in "content-type" should be string. But get ${type.valueType.kind}.`;
        }
        return type.values.map((v) => v.value);
    }
    return [];
}
function loadOperationPaging(context, method, rootApiVersions, uri) {
    if (method.kind !== "paging" || method.pagingMetadata === undefined) {
        return undefined;
    }
    let nextLink;
    if (method.pagingMetadata.nextLinkSegments) {
        nextLink = {
            responseSegments: method.pagingMetadata.nextLinkSegments.map((segment) => getResponseSegmentName(segment)),
            responseLocation: getResponseLocation(context, method, method.pagingMetadata.nextLinkSegments[0]),
        };
        if (method.pagingMetadata.nextLinkOperation) {
            nextLink.operation = fromSdkServiceMethod(context, method.pagingMetadata.nextLinkOperation, uri, rootApiVersions);
        }
    }
    let continuationToken;
    if (method.pagingMetadata.continuationTokenParameterSegments &&
        method.pagingMetadata.continuationTokenResponseSegments) {
        // The last segment of the service method parameter can be used to map back to the protocol parameter
        const lastParameterSegment = method.pagingMetadata.continuationTokenParameterSegments[method.pagingMetadata.continuationTokenParameterSegments.length - 1];
        continuationToken = {
            parameter: fromSdkHttpOperationParameter(context, getHttpOperationParameter(method, lastParameterSegment), rootApiVersions),
            responseSegments: method.pagingMetadata.continuationTokenResponseSegments.map((segment) => getResponseSegmentName(segment)),
            responseLocation: getResponseLocation(context, method, method.pagingMetadata.continuationTokenResponseSegments?.[0]),
        };
    }
    return {
        // TODO - this is hopefully temporary until TCGC provides the information directly on pagingMetadata https://github.com/Azure/typespec-azure/issues/2291
        itemPropertySegments: method.response.resultSegments.map((s) => s.name),
        nextLink: nextLink,
        continuationToken: continuationToken,
    };
}
function getResponseSegmentName(segment) {
    return segment.kind === "responseheader" || segment.kind === "body"
        ? segment.serializedName
        : segment.name;
}
function getResponseLocation(context, method, p) {
    switch (p?.kind) {
        case "responseheader":
            return ResponseLocation.Header;
        case "property":
            return ResponseLocation.Body;
        default:
            context.logger.reportDiagnostic({
                code: "unsupported-continuation-location",
                format: {
                    crossLanguageDefinitionId: method.crossLanguageDefinitionId,
                },
                target: NoTarget,
            });
            return ResponseLocation.None;
    }
}
// TODO: https://github.com/Azure/typespec-azure/issues/1441
function getParameterLocation(p) {
    switch (p?.kind) {
        case "path":
            return RequestLocation.Path;
        case "header":
            return RequestLocation.Header;
        case "query":
            return RequestLocation.Query;
        case "body":
            return RequestLocation.Body;
        default:
            return RequestLocation.None;
    }
}
function getParameterKind(p, type, hasGlobalApiVersion) {
    if (p.kind === "body") {
        /** TODO: remove this and use the spread metadata of parameter when https://github.com/Azure/typespec-azure/issues/1513 is resolved */
        if (type.kind === "model" && p.type !== p.correspondingMethodParams[0]?.type) {
            return InputOperationParameterKind.Spread;
        }
        return InputOperationParameterKind.Method;
    }
    return type.kind === "constant"
        ? InputOperationParameterKind.Constant
        : p.isApiVersionParam
            ? hasGlobalApiVersion
                ? InputOperationParameterKind.Client
                : InputOperationParameterKind.Method
            : p.onClient
                ? InputOperationParameterKind.Client
                : InputOperationParameterKind.Method;
}
function getOperationGroupName(context, operation, namespace) {
    const explicitOperationId = getOperationId(context, operation.__raw.operation);
    if (explicitOperationId) {
        const ids = explicitOperationId.split("_");
        if (ids.length > 1) {
            return ids.slice(0, -2).join("_");
        }
    }
    if (operation.__raw.operation.interface) {
        return operation.__raw.operation.interface.name;
    }
    if (operation.__raw.operation.namespace) {
        return operation.__raw.operation.namespace.name;
    }
    return namespace;
}
// TODO: remove after https://github.com/Azure/typespec-azure/issues/1227 is fixed
function normalizeHeaderName(name) {
    switch (name.toLocaleLowerCase()) {
        case "accept":
            return "Accept";
        case "content-type":
            return "Content-Type";
        default:
            return name;
    }
}
function isExplodedParameter(p) {
    return (p.kind === "path" || p.kind === "query") && p.explode === true;
}
//# sourceMappingURL=operation-converter.js.map