/*
 * 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.
 */

#include "ecmascript/dfx/hprof/string_hashmap.h"

#include "ecmascript/mem/native_area_allocator.h"

namespace panda::ecmascript {
CString *StringHashMap::FindOrInsertString(const CString *cstr)
{
    StringKey key = GenerateStringKey(cstr);
    auto it = hashmap_.find(key);
    if (it != hashmap_.end()) {
        return it->second;
    } else {  // NOLINT(readability-else-after-return)
        index_++;
        auto *newStr = const_cast<NativeAreaAllocator *>(
            vm_->GetNativeAreaAllocator())->New<CString>(cstr->c_str());
        hashmap_.emplace(key, newStr);
        orderedKey_.emplace_back(key);
        indexMap_.emplace(key, index_);
        return newStr;
    }
}

StringId StringHashMap::GetStringId(const CString *cstr) const
{
    auto it = indexMap_.find(GenerateStringKey(cstr));
    return it != indexMap_.end() ? it->second : 1;  // ""
}

CString *StringHashMap::GetStringByKey(StringKey key) const
{
    auto it = hashmap_.find(key);
    if (it != hashmap_.end()) {
        return FormatString(it->second);
    }
    return nullptr;
}

CString *StringHashMap::FormatString(CString *cstr) const
{
    // remove "\"" | "\r\n" | "\\" | "\t" | "\f"
    int length = static_cast<int>(cstr->length());
    char *charSeq = const_cast<char *>(cstr->c_str());
    for (int i = 0; i < length; i++) {
        if (charSeq[i] == '\"') {         // NOLINT(cppcoreguidelines-pro-bounds-pointer-arithmetic)
            charSeq[i] = '`';             // NOLINT(cppcoreguidelines-pro-bounds-pointer-arithmetic)
        } else if (charSeq[i] == '\r') {  // NOLINT(cppcoreguidelines-pro-bounds-pointer-arithmetic)
            charSeq[i] = '`';             // NOLINT(cppcoreguidelines-pro-bounds-pointer-arithmetic)
        } else if (charSeq[i] == '\n') {  // NOLINT(cppcoreguidelines-pro-bounds-pointer-arithmetic)
            charSeq[i] = '`';             // NOLINT(cppcoreguidelines-pro-bounds-pointer-arithmetic)
        } else if (charSeq[i] == '\\') {  // NOLINT(cppcoreguidelines-pro-bounds-pointer-arithmetic)
            charSeq[i] = '`';             // NOLINT(cppcoreguidelines-pro-bounds-pointer-arithmetic)
        } else if (charSeq[i] == '\t') {  // NOLINT(cppcoreguidelines-pro-bounds-pointer-arithmetic)
            charSeq[i] = '`';             // NOLINT(cppcoreguidelines-pro-bounds-pointer-arithmetic)
        } else if (charSeq[i] == '\f') {  // NOLINT(cppcoreguidelines-pro-bounds-pointer-arithmetic)
            charSeq[i] = '`';             // NOLINT(cppcoreguidelines-pro-bounds-pointer-arithmetic)
        } else if (charSeq[i] < ' ') {    // NOLINT(cppcoreguidelines-pro-bounds-pointer-arithmetic)
            // ctrl chars 0~31
            charSeq[i] = '`';  // NOLINT(cppcoreguidelines-pro-bounds-pointer-arithmetic)
        }
    }
    *cstr = charSeq;
    return cstr;
}

StringKey StringHashMap::GenerateStringKey(const CString *cstr) const
{
    return std::hash<CString>{} (*cstr);
}

CString *StringHashMap::GetString(const CString &cstr)
{
    return FindOrInsertString(&cstr);
}

void StringHashMap::Clear()
{
    auto *alloctor = const_cast<NativeAreaAllocator *>(vm_->GetNativeAreaAllocator());
    for (auto &[_, cstr] : hashmap_) {
        if (cstr != nullptr) {
            alloctor->Delete(cstr);
        }
    }
}
}  // namespace panda::ecmascript