/*
 * Copyright (c) 2022 Huawei Device Co., Ltd.
 * 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.
 */

#define LOG_TAG "RdbResultSetImpl"

#include <mutex>
#include "log_print.h"
#include "store_types.h"
#include "store/cursor.h"
#include "rdb_result_set_impl.h"

using DistributedDB::DBStatus;
using OHOS::NativeRdb::ColumnType;

namespace OHOS::DistributedRdb {
using OHOS::DistributedData::GeneralError;
using Cursor = OHOS::DistributedData::Cursor;
using namespace OHOS::NativeRdb;
RdbResultSetImpl::RdbResultSetImpl(std::shared_ptr<Cursor> resultSet) : resultSet_(std::move(resultSet))
{
    if (resultSet_ != nullptr) {
        count_ = resultSet_->GetCount();
        resultSet_->GetColumnNames(colNames_);
    }
}

int RdbResultSetImpl::GetAllColumnNames(std::vector<std::string> &columnNames)
{
    std::shared_lock<std::shared_mutex> lock(mutex_);
    if (resultSet_ == nullptr) {
        return NativeRdb::E_ALREADY_CLOSED;
    }
    columnNames = colNames_;
    return NativeRdb::E_OK;
}

int RdbResultSetImpl::GetColumnCount(int &count)
{
    std::shared_lock<std::shared_mutex> lock(mutex_);
    if (resultSet_ == nullptr) {
        return NativeRdb::E_ALREADY_CLOSED;
    }
    count = static_cast<int>(colNames_.size());
    return NativeRdb::E_OK;
}

int RdbResultSetImpl::GetColumnType(int columnIndex, ColumnType &columnType)
{
    std::shared_lock<std::shared_mutex> lock(mutex_);
    if (resultSet_ == nullptr) {
        return NativeRdb::E_ALREADY_CLOSED;
    }
    columnType = ConvertColumnType(resultSet_->GetColumnType(columnIndex));
    return NativeRdb::E_OK;
}

int RdbResultSetImpl::GetColumnIndex(const std::string &columnName, int &columnIndex)
{
    std::shared_lock<std::shared_mutex> lock(mutex_);
    if (resultSet_ == nullptr) {
        return NativeRdb::E_ALREADY_CLOSED;
    }
    for (size_t i = 0; i < colNames_.size(); i++) {
        if (colNames_[i] == columnName) {
            columnIndex = static_cast<int>(i);
            return NativeRdb::E_OK;
        }
    }
    return NativeRdb::E_ERROR;
}

int RdbResultSetImpl::GetColumnName(int columnIndex, std::string &columnName)
{
    std::shared_lock<std::shared_mutex> lock(mutex_);
    if (resultSet_ == nullptr) {
        return NativeRdb::E_ALREADY_CLOSED;
    }
    if (colNames_.size() <= static_cast<uint32_t>(columnIndex) || columnIndex < 0) {
        return NativeRdb::E_ERROR;
    }
    columnName = colNames_[columnIndex];
    return NativeRdb::E_OK;
}

int RdbResultSetImpl::GetRowCount(int &count)
{
    std::shared_lock<std::shared_mutex> lock(mutex_);
    if (resultSet_ == nullptr) {
        return NativeRdb::E_ALREADY_CLOSED;
    }
    count = count_;
    return NativeRdb::E_OK;
}

int RdbResultSetImpl::GetRowIndex(int &position) const
{
    std::shared_lock<std::shared_mutex> lock(mutex_);
    if (resultSet_ == nullptr) {
        return NativeRdb::E_ALREADY_CLOSED;
    }
    position = current_;
    return NativeRdb::E_OK;
}

int RdbResultSetImpl::GoTo(int offset)
{
    int ret = NativeRdb::E_OK;
    while (offset != 0 && ret == NativeRdb::E_OK) {
        if (offset > 0) {
            ret = GoToNextRow();
            offset--;
        } else {
            ret = GoToPreviousRow();
            offset++;
        }
    }
    return ret;
}

int RdbResultSetImpl::GoToRow(int position)
{
    return GoTo(position - current_);
}

int RdbResultSetImpl::GoToFirstRow()
{
    std::unique_lock<std::shared_mutex> lock(mutex_);
    if (resultSet_ == nullptr) {
        return NativeRdb::E_ALREADY_CLOSED;
    }
    auto ret = resultSet_->MoveToFirst();
    current_ = 0;
    return ret == GeneralError::E_OK ?  NativeRdb::E_OK : NativeRdb::E_ERROR;
}

int RdbResultSetImpl::GoToLastRow()
{
    return GoToRow(count_ - 1);
}

int RdbResultSetImpl::GoToNextRow()
{
    std::unique_lock<std::shared_mutex> lock(mutex_);
    if (resultSet_ == nullptr) {
        return NativeRdb::E_ALREADY_CLOSED;
    }
    if (current_ >= count_ - 1) {
        current_ = count_;
        return NativeRdb::E_ERROR;
    }

    auto ret = resultSet_->MoveToNext();
    current_++;
    return ret == GeneralError::E_OK ?  NativeRdb::E_OK : NativeRdb::E_ERROR;
}

int RdbResultSetImpl::GoToPreviousRow()
{
    std::unique_lock<std::shared_mutex> lock(mutex_);
    if (resultSet_ == nullptr) {
        return NativeRdb::E_ALREADY_CLOSED;
    }
    if (current_ <= 0) {
        current_ = -1;
        return NativeRdb::E_ERROR;
    }

    auto ret = resultSet_->MoveToPrev();
    current_--;
    return ret == GeneralError::E_OK ?  NativeRdb::E_OK : NativeRdb::E_ERROR;
}

int RdbResultSetImpl::IsEnded(bool &result)
{
    std::shared_lock<std::shared_mutex> lock(mutex_);
    if (resultSet_ == nullptr) {
        return NativeRdb::E_ALREADY_CLOSED;
    }
    result = current_ >= count_ || count_ <= 0;
    return NativeRdb::E_OK;
}

int RdbResultSetImpl::IsStarted(bool &result) const
{
    std::shared_lock<std::shared_mutex> lock(mutex_);
    if (resultSet_ == nullptr) {
        return NativeRdb::E_ALREADY_CLOSED;
    }
    result = current_ < 0 || count_ <= 0;
    return NativeRdb::E_OK;
}

int RdbResultSetImpl::IsAtFirstRow(bool &result) const
{
    std::shared_lock<std::shared_mutex> lock(mutex_);
    if (resultSet_ == nullptr) {
        return NativeRdb::E_ALREADY_CLOSED;
    }
    result = count_ > 0 && current_ == 0;
    return NativeRdb::E_OK;
}

int RdbResultSetImpl::IsAtLastRow(bool &result)
{
    std::shared_lock<std::shared_mutex> lock(mutex_);
    if (resultSet_ == nullptr) {
        return NativeRdb::E_ALREADY_CLOSED;
    }
    result = count_ > 0 && current_ == count_ - 1;
    return NativeRdb::E_OK;
}

int RdbResultSetImpl::GetBlob(int columnIndex, std::vector<uint8_t> &value)
{
    std::shared_lock<std::shared_mutex> lock(mutex_);
    if (resultSet_ == nullptr) {
        return NativeRdb::E_ALREADY_CLOSED;
    }
    return Get(columnIndex, value);
}

int RdbResultSetImpl::GetString(int columnIndex, std::string &value)
{
    std::shared_lock<std::shared_mutex> lock(mutex_);
    if (resultSet_ == nullptr) {
        return NativeRdb::E_ALREADY_CLOSED;
    }
    return Get(columnIndex, value);
}

int RdbResultSetImpl::GetInt(int columnIndex, int &value)
{
    int64_t tmpValue;
    int status = GetLong(columnIndex, tmpValue);
    if (status == NativeRdb::E_OK) {
        if (tmpValue < INT32_MIN || tmpValue > INT32_MAX) {
            ZLOGE("Get int value overflow.");
            return NativeRdb::E_ERROR;
        }
        value = static_cast<int32_t>(tmpValue);
    }
    return status;
}

int RdbResultSetImpl::GetLong(int columnIndex, int64_t &value)
{
    std::shared_lock<std::shared_mutex> lock(mutex_);
    if (resultSet_ == nullptr) {
        return NativeRdb::E_ALREADY_CLOSED;
    }
    return Get(columnIndex, value);
}

int RdbResultSetImpl::GetDouble(int columnIndex, double &value)
{
    std::shared_lock<std::shared_mutex> lock(mutex_);
    if (resultSet_ == nullptr) {
        return NativeRdb::E_ALREADY_CLOSED;
    }
    return Get(columnIndex, value);
}

int RdbResultSetImpl::IsColumnNull(int columnIndex, bool &isNull)
{
    std::shared_lock<std::shared_mutex> lock(mutex_);
    if (resultSet_ == nullptr) {
        return NativeRdb::E_ALREADY_CLOSED;
    }
    DistributedData::Value var;
    auto status = resultSet_->Get(columnIndex, var);
    if (status != DistributedData::GeneralError::E_OK) {
        return NativeRdb::E_ERROR;
    }
    isNull = var.index() == DistributedData::TYPE_INDEX<std::monostate>;
    return NativeRdb::E_OK;
}

bool RdbResultSetImpl::IsClosed() const
{
    std::shared_lock<std::shared_mutex> lock(mutex_);
    return resultSet_ == nullptr;
}

int RdbResultSetImpl::Close()
{
    std::unique_lock<std::shared_mutex> lock(mutex_);
    if (resultSet_ == nullptr) {
        ZLOGW("Result set has been closed.");
        return NativeRdb::E_OK;
    }
    resultSet_->Close();
    resultSet_ = nullptr;
    return NativeRdb::E_OK;
}

ColumnType RdbResultSetImpl::ConvertColumnType(int32_t columnType) const
{
    if (static_cast<uint32_t>(columnType) >= DistributedData::TYPE_MAX || columnType < 0) {
        return ColumnType::TYPE_NULL;
    }
    return COLUMNTYPES[columnType];
}

int RdbResultSetImpl::GetAsset(int32_t col, NativeRdb::ValueObject::Asset& value)
{
    std::shared_lock<std::shared_mutex> lock(mutex_);
    if (resultSet_ == nullptr) {
        return NativeRdb::E_ALREADY_CLOSED;
    }
    return Get(col, value);
}

int RdbResultSetImpl::GetAssets(int32_t col, NativeRdb::ValueObject::Assets& value)
{
    std::shared_lock<std::shared_mutex> lock(mutex_);
    if (resultSet_ == nullptr) {
        return NativeRdb::E_ALREADY_CLOSED;
    }
    return Get(col, value);
}
int RdbResultSetImpl::Get(int32_t col, NativeRdb::ValueObject& value)
{
    std::shared_lock<std::shared_mutex> lock(mutex_);
    if (resultSet_ == nullptr) {
        return NativeRdb::E_ALREADY_CLOSED;
    }
    auto [errCode, object] = GetValue(col);
    value = std::move(object);
    return errCode;
}

int RdbResultSetImpl::GetRow(NativeRdb::RowEntity& rowEntity)
{
    return NativeRdb::E_NOT_SUPPORT;
}

int RdbResultSetImpl::GetSize(int col, size_t& size)
{
    std::shared_lock<std::shared_mutex> lock(mutex_);
    if (resultSet_ == nullptr) {
        return NativeRdb::E_ALREADY_CLOSED;
    }
    auto [errCode, value] = GetValue(col);
    if (errCode != NativeRdb::E_OK) {
        return errCode;
    }
    auto object = static_cast<ValueObject>(value);
    if (object.GetType() == ValueObject::TYPE_STRING) {
        auto val = std::get_if<std::string>(&object.value);
        if (val != nullptr) {
            size = val->size();
        }
    } else if (object.GetType() == ValueObject::TYPE_BLOB) {
        auto val = std::get_if<std::vector<uint8_t>>(&object.value);
        if (val != nullptr) {
            size = val->size();
        }
    }
    return NativeRdb::E_OK;
}
} // namespace OHOS::DistributedRdb