// Copyright (C) 2019-2020 Zilliz. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance
// with the License. You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software distributed under the License
// is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
// or implied. See the License for the specific language governing permissions and limitations under the License

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

#pragma once

#include <aws/core/Core_EXPORTS.h>
#include <aws/core/auth/AWSCredentials.h>
#include <aws/core/client/ClientConfiguration.h>
#include <aws/core/utils/memory/stl/AWSString.h>

#include "aws/core/internal/AWSHttpResourceClient.h"
#include "storage/tencent/TencentCloudSTSClient.h"

namespace Aws {
namespace Http {
enum class HttpResponseCode;
}  // namespace Http

namespace Internal {
class TencentCloudSTSClientTestHelper;
/**
 * To support retrieving credentials from STS.
 * Note that STS accepts request with protocol of queryxml. Calling GetResource() will trigger
 * a query request using AWSHttpResourceClient under the hood.
 */
class AWS_CORE_API TencentCloudSTSCredentialsClient
    : public AWSHttpResourceClient {
    friend class ::Aws::Internal::TencentCloudSTSClientTestHelper;

 public:
    /**
     * Initializes the provider to retrieve credentials from STS when it expires.
     */
    explicit TencentCloudSTSCredentialsClient(
        const Client::ClientConfiguration& clientConfiguration);

    TencentCloudSTSCredentialsClient&
    operator=(TencentCloudSTSCredentialsClient& rhs) = delete;
    TencentCloudSTSCredentialsClient(
        const TencentCloudSTSCredentialsClient& rhs) = delete;
    TencentCloudSTSCredentialsClient&
    operator=(TencentCloudSTSCredentialsClient&& rhs) = delete;
    TencentCloudSTSCredentialsClient(
        const TencentCloudSTSCredentialsClient&& rhs) = delete;

    // If you want to make an AssumeRoleWithWebIdentity call to sts. use these classes to pass data to and get info from
    // TencentCloudSTSCredentialsClient client. If you want to make an AssumeRole call to sts, define the request/result
    // members class/struct like this.
    struct STSAssumeRoleWithWebIdentityRequest {
        Aws::String region;
        Aws::String providerId;
        Aws::String webIdentityToken;
        Aws::String roleArn;
        Aws::String roleSessionName;
    };

    struct STSAssumeRoleWithWebIdentityResult {
        Aws::Auth::AWSCredentials creds;
    };

    STSAssumeRoleWithWebIdentityResult
    GetAssumeRoleWithWebIdentityCredentials(
        const STSAssumeRoleWithWebIdentityRequest& request);

 private:
    struct STSParseResult {
        bool success = false;
        Aws::Auth::AWSCredentials credentials;
    };

    static STSParseResult
    parseSTSResponse(const Aws::String& responseBody);

    Aws::String m_endpoint;
};
}  // namespace Internal
}  // namespace Aws
