// Copyright (c) 2017, Tencent Inc.
// All rights reserved.
//
// Author: sevenyou <sevenyou@tencent.com>
// Created: 07/15/17
// Description:

#ifndef BASEREQ_H
#define BASEREQ_H
#pragma once

#include <stdint.h>
#include <map>
#include <string>
#include "util/string_util.h"

namespace qcloud_cos {

/// \brief 封装Http请求
class BaseReq {
public:
    typedef std::map<std::string, std::string> Str2StrMap;
public:
    BaseReq();
    virtual ~BaseReq() {}

    /// \brief 设置请求头部
    void AddHeader(const std::string& key, const std::string& value);
    void AddHeaders(const Str2StrMap& user_headers);

    /// \brief 获取请求头部
    std::string GetHeader(const std::string& key) const;
    const Str2StrMap& GetHeaders() const { return m_headers_map; }

    /// \brief 清空请求头部
    void ClearHeaders() { m_headers_map.clear(); }

    /// \brief 设置请求参数
    void AddParam(const std::string& key, const std::string& value);
    void AddParams(const Str2StrMap& user_params);

    /// \brief 获取请求参数
    std::string GetParam(const std::string& key) const;
    const Str2StrMap& GetParams() const { return m_params_map; };

    /// \brief 清空请求参数
    void ClearParams() { m_params_map.clear(); }

    /// \brief 获取请求的http方法
    std::string GetMethod() const { return m_method; }

    /// \brief 设置请求的http方法
    void SetMethod(const std::string& method) {
        m_method = StringUtil::StringToUpper(method);
    }

    /// \brief 获取请求的body
    std::string GetBody() const { return m_body; }

    /// \brief 设置请求的body
    void SetBody(const std::string& body) { m_body = body; }

    /// \brief 获取Path
    std::string GetPath() const { return m_path; }

    /// \brief 设置Path
    void SetPath(const std::string& path) { m_path = path; }

    void SetConnTimeoutInms(uint64_t conn_timeout_in_ms) {
        m_conn_timeout_in_ms = conn_timeout_in_ms;
    }

    uint64_t GetConnTimeoutInms() const {
        return m_conn_timeout_in_ms;
    }

    void SetRecvTimeoutInms(uint64_t recv_timeout_in_ms) {
        m_recv_timeout_in_ms = recv_timeout_in_ms;
    }

    uint64_t GetRecvTimeoutInms() const {
        return m_recv_timeout_in_ms;
    }

    /// \brief 设置当前请求是否使用https
    void SetHttps() { m_is_https = true; }
    bool IsHttps() const { return m_is_https; }

    /// \brief set whether check content md5 each request, used for close range download check
    void SetCheckMD5(bool check_md5) { mb_check_md5 = check_md5; }
    bool CheckMD5() const { return mb_check_md5; }

    void SetCheckCRC(bool check_crc) { mb_check_crc = check_crc; }
    bool CheckCRC() const { return mb_check_crc; }

    /// \brief 输出请求的header和param信息
    std::string DebugString() const;

protected:
    // acl相关的请求，供PutObjectACL和PutBucketACL使用
    bool GenerateAclRequestBody(const Owner& owner,
                                const std::vector<Grant>& acl,
                                std::string* body) const;

protected:
    Str2StrMap m_headers_map;
    Str2StrMap m_params_map;
    std::string m_path;
    std::string m_method;
    std::string m_body;

    uint64_t m_conn_timeout_in_ms;
    uint64_t m_recv_timeout_in_ms;
    bool m_is_https;
    bool mb_check_md5; // defaule is true
    bool mb_check_crc; // default is false
};

} // namespaace qcloud_cos

#endif // BASEREQ_H