//
// Copyright Amazon.com Inc. or its affiliates.
// All Rights Reserved.
//
// SPDX-License-Identifier: Apache-2.0
//

import Amplify
import Foundation

public extension AWSAPIPlugin {

    func get(request: RESTRequest, listener: RESTOperation.ResultListener?) -> RESTOperation {
        let operationRequest = RESTOperationRequest(request: request,
                                                    operationType: .get)

        let operation = AWSRESTOperation(request: operationRequest,
                                         session: session,
                                         mapper: mapper,
                                         pluginConfig: pluginConfig,
                                         resultListener: listener)

        queue.addOperation(operation)
        return operation
    }
    
    func get(request: RESTRequest) async throws -> RESTTask.Success {
        let operationRequest = RESTOperationRequest(request: request,
                                                    operationType: .get)

        let operation = AWSRESTOperation(request: operationRequest,
                                         session: session,
                                         mapper: mapper,
                                         pluginConfig: pluginConfig,
                                         resultListener: nil)

        let task = AmplifyOperationTaskAdapter(operation: operation)
        queue.addOperation(operation)
        return try await task.value
    }

    func put(request: RESTRequest, listener: RESTOperation.ResultListener?) -> RESTOperation {
        let operationRequest = RESTOperationRequest(request: request,
                                                    operationType: .put)

        let operation = AWSRESTOperation(request: operationRequest,
                                         session: session,
                                         mapper: mapper,
                                         pluginConfig: pluginConfig,
                                         resultListener: listener)

        queue.addOperation(operation)
        return operation
    }
    
    func put(request: RESTRequest) async throws -> RESTTask.Success {
        let operationRequest = RESTOperationRequest(request: request,
                                                    operationType: .put)

        let operation = AWSRESTOperation(request: operationRequest,
                                         session: session,
                                         mapper: mapper,
                                         pluginConfig: pluginConfig,
                                         resultListener: nil)

        let task = AmplifyOperationTaskAdapter(operation: operation)
        queue.addOperation(operation)
        return try await task.value
    }

    func post(request: RESTRequest, listener: RESTOperation.ResultListener?) -> RESTOperation {
        let operationRequest = RESTOperationRequest(request: request,
                                                    operationType: .post)

        let operation = AWSRESTOperation(request: operationRequest,
                                         session: session,
                                         mapper: mapper,
                                         pluginConfig: pluginConfig,
                                         resultListener: listener)

        queue.addOperation(operation)
        return operation
    }
    
    func post(request: RESTRequest) async throws -> RESTTask.Success {
        let operationRequest = RESTOperationRequest(request: request,
                                                    operationType: .post)

        let operation = AWSRESTOperation(request: operationRequest,
                                         session: session,
                                         mapper: mapper,
                                         pluginConfig: pluginConfig,
                                         resultListener: nil)

        let task = AmplifyOperationTaskAdapter(operation: operation)
        queue.addOperation(operation)
        return try await task.value
    }

    func patch(request: RESTRequest, listener: RESTOperation.ResultListener?) -> RESTOperation {
        let operationRequest = RESTOperationRequest(request: request, operationType: .patch)

        let operation = AWSRESTOperation(request: operationRequest,
                                         session: session,
                                         mapper: mapper,
                                         pluginConfig: pluginConfig,
                                         resultListener: listener)

        queue.addOperation(operation)
        return operation
    }
    
    func patch(request: RESTRequest) async throws -> RESTTask.Success {
        let operationRequest = RESTOperationRequest(request: request, operationType: .patch)

        let operation = AWSRESTOperation(request: operationRequest,
                                         session: session,
                                         mapper: mapper,
                                         pluginConfig: pluginConfig,
                                         resultListener: nil)

        let task = AmplifyOperationTaskAdapter(operation: operation)
        queue.addOperation(operation)
        return try await task.value
    }

    func delete(request: RESTRequest, listener: RESTOperation.ResultListener?) -> RESTOperation {
        let operationRequest = RESTOperationRequest(request: request,
                                                    operationType: .delete)

        let operation = AWSRESTOperation(request: operationRequest,
                                         session: session,
                                         mapper: mapper,
                                         pluginConfig: pluginConfig,
                                         resultListener: listener)

        queue.addOperation(operation)
        return operation
    }
    
    func delete(request: RESTRequest) async throws -> RESTTask.Success {
        let operationRequest = RESTOperationRequest(request: request,
                                                    operationType: .delete)

        let operation = AWSRESTOperation(request: operationRequest,
                                         session: session,
                                         mapper: mapper,
                                         pluginConfig: pluginConfig,
                                         resultListener: nil)

        let task = AmplifyOperationTaskAdapter(operation: operation)
        queue.addOperation(operation)
        return try await task.value
    }

    func head(request: RESTRequest, listener: RESTOperation.ResultListener?) -> RESTOperation {
        let operationRequest = RESTOperationRequest(request: request,
                                                    operationType: .head)

        let operation = AWSRESTOperation(request: operationRequest,
                                         session: session,
                                         mapper: mapper,
                                         pluginConfig: pluginConfig,
                                         resultListener: listener)

        queue.addOperation(operation)
        return operation
    }
    
    func head(request: RESTRequest) async throws -> RESTTask.Success {
        let operationRequest = RESTOperationRequest(request: request,
                                                    operationType: .head)

        let operation = AWSRESTOperation(request: operationRequest,
                                         session: session,
                                         mapper: mapper,
                                         pluginConfig: pluginConfig,
                                         resultListener: nil)

        let task = AmplifyOperationTaskAdapter(operation: operation)
        queue.addOperation(operation)
        return try await task.value
    }
}