OpenHarmony-v5.0.1-Release/s

/*
 * Copyright (c) 2024 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 "core/components_ng/pattern/text/span/span_string.h"

#include <cstdint>
#include <iterator>
#include <utility>

#include "base/utils/string_utils.h"
#include "base/utils/utils.h"
#include "core/components/common/properties/color.h"
#include "core/components_ng/pattern/text/span/span_object.h"
#include "core/components_ng/pattern/text/span_node.h"
#include "core/text/text_emoji_processor.h"

namespace OHOS::Ace {

const std::unordered_set<SpanType> specailTypes = { SpanType::Image, SpanType::CustomSpan };

std::wstring SpanString::GetWideStringSubstr(const std::wstring& content, int32_t start, int32_t length)
{
    if (start >= static_cast<int32_t>(content.length())) {
        return StringUtils::ToWstring("");
    }
    return content.substr(start, length);
}

std::wstring SpanString::GetWideStringSubstr(const std::wstring& content, int32_t start)
{
    if (start >= static_cast<int32_t>(content.length())) {
        return StringUtils::ToWstring("");
    }
    return content.substr(start);
}

SpanString::SpanString(const std::string& text)
{
    auto spanItem = MakeRefPtr<NG::SpanItem>();
    std::wstring wideText = StringUtils::ToWstring(text);
    if (wideText.length() == 0 && text.length() != 0) {
        text_ = TextEmojiProcessor::ConvertU8stringUnpairedSurrogates(text);
        wideText = StringUtils::ToWstring(text_);
    } else {
        text_ = text;
    }
    spanItem->content = text_;
    spanItem->interval = { 0, wideText.length() };
    spans_.emplace_back(spanItem);
    auto it = spans_.begin();
    SplitSpansAndForward(it);
}

SpanString::SpanString(const ImageSpanOptions& options) : text_(" ")
{
    auto spanItem = MakeRefPtr<NG::ImageSpanItem>();
    spanItem->options = options;
    spanItem->content = " ";
    spanItem->interval = { 0, 1 };
    spans_.emplace_back(spanItem);
    spansMap_[SpanType::Image].emplace_back(MakeRefPtr<ImageSpan>(options));
}

SpanString::SpanString(RefPtr<CustomSpan>& span) : text_(" ")
{
    auto spanItem = MakeRefPtr<NG::CustomSpanItem>();
    spanItem->content = " ";
    spanItem->interval = { 0, 1 };
    spanItem->onMeasure = span->GetOnMeasure();
    spanItem->onDraw = span->GetOnDraw();
    spans_.emplace_back(spanItem);
    spansMap_[SpanType::CustomSpan].emplace_back(span);
}

void SpanString::AddCustomSpan()
{
    auto spanBases = GetSpans(0, GetLength(), SpanType::CustomSpan);
    for (const auto& spanBase : spanBases) {
        if (spanBase->GetSpanType() != SpanType::CustomSpan) {
            continue;
        }
        auto customSpan = DynamicCast<CustomSpan>(spanBase);
        if (!customSpan) {
            continue;
        }
        customSpan->AddStyledString(Referenced::WeakClaim(this));
    }
}

void SpanString::RemoveCustomSpan()
{
    auto spanBases = GetSpans(0, GetLength(), SpanType::CustomSpan);
    for (const auto& spanBase : spanBases) {
        if (spanBase->GetSpanType() != SpanType::CustomSpan) {
            continue;
        }
        auto customSpan = DynamicCast<CustomSpan>(spanBase);
        if (!customSpan) {
            continue;
        }
        customSpan->RemoveStyledString(Referenced::WeakClaim(this));
    }
}
void SpanString::SetFramNode(const WeakPtr<NG::FrameNode>& frameNode)
{
    framNode_ = frameNode;
}

void SpanString::MarkDirtyFrameNode()
{
    auto frameNode = framNode_.Upgrade();
    CHECK_NULL_VOID(frameNode);
    frameNode->MarkDirtyNode(NG::PROPERTY_UPDATE_RENDER);
}

SpanString::~SpanString()
{
    spansMap_.clear();
    spans_.clear();
}

std::list<RefPtr<NG::SpanItem>>::iterator SpanString::SplitSpansAndForward(
    std::list<RefPtr<NG::SpanItem>>::iterator& it)
{
    auto wString = StringUtils::ToWstring((*it)->content);
    auto newlineIndex = static_cast<int32_t>(wString.find(L'\n'));
    int32_t offset = (*it)->interval.first;
    while (newlineIndex != -1 && newlineIndex != static_cast<int32_t>(wString.size()) - 1) {
        auto newSpan = (*it)->GetSameStyleSpanItem();
        newSpan->interval = { offset + newlineIndex + 1, (*it)->interval.second };
        (*it)->interval = { offset, offset + newlineIndex + 1 };
        (*it)->content = StringUtils::ToString(GetWideStringSubstr(wString, 0, newlineIndex + 1));
        wString = GetWideStringSubstr(wString, newlineIndex + 1);
        newSpan->content = StringUtils::ToString(wString);
        newlineIndex = static_cast<int32_t>(wString.find(L'\n'));

        offset = newSpan->interval.first;
        ++it;
        it = spans_.insert(it, newSpan);
    }

    return std::next(it);
}

void SpanString::ApplyToSpans(
    const RefPtr<SpanBase>& span, std::pair<int32_t, int32_t> interval, SpanOperation operation)
{
    for (auto it = spans_.begin(); it != spans_.end(); ++it) {
        auto intersection = (*it)->GetIntersectionInterval(interval);
        if (!intersection) {
            continue;
        }
        auto oldStart = (*it)->interval.first;
        auto oldEnd = (*it)->interval.second;
        if (oldStart == intersection->first && intersection->second == oldEnd) {
            span->ApplyToSpanItem(*it, operation);
            continue;
        }

        auto wContent = StringUtils::ToWstring((*it)->content);
        auto newSpan = (*it)->GetSameStyleSpanItem();
        if (oldStart < intersection->first && intersection->second < oldEnd) {
            (*it)->interval = { oldStart, intersection->first };
            (*it)->content = StringUtils::ToString(wContent.substr(0, intersection->first - oldStart));

            newSpan->interval = { intersection->first, intersection->second };
            newSpan->content = StringUtils::ToString(
                wContent.substr(intersection->first - oldStart, intersection->second - intersection->first));
            span->ApplyToSpanItem(newSpan, operation);

            auto newSpan2 = (*it)->GetSameStyleSpanItem();
            newSpan2->interval = { intersection->second, oldEnd };
            newSpan2->content = StringUtils::ToString(wContent.substr(intersection->second - oldStart));
            it = spans_.insert(std::next(it), newSpan);
            it = spans_.insert(std::next(it), newSpan2);
            continue;
        }

        if (oldEnd > intersection->second) {
            (*it)->content = StringUtils::ToString(wContent.substr(0, intersection->second - oldStart));
            (*it)->interval = { oldStart, intersection->second };
            span->ApplyToSpanItem(*it, operation);
            newSpan->interval = { intersection->second, oldEnd };
            newSpan->content = StringUtils::ToString(wContent.substr(intersection->second - oldStart));
            it = spans_.insert(std::next(it), newSpan);
            continue;
        }

        if (intersection->first > oldStart) {
            (*it)->content = StringUtils::ToString(wContent.substr(0, intersection->first - oldStart));
            (*it)->interval = { oldStart, intersection->first };
            newSpan->interval = { intersection->first, oldEnd };
            newSpan->content = StringUtils::ToString(wContent.substr(intersection->first - oldStart));
            span->ApplyToSpanItem(newSpan, operation);
            it = spans_.insert(std::next(it), newSpan);
        }
    }
}

void SpanString::SplitInterval(std::list<RefPtr<SpanBase>>& spans, std::pair<int32_t, int32_t> interval)
{
    std::list<RefPtr<SpanBase>> newSpans;
    for (auto it = spans.begin(); it != spans.end();) {
        auto intersection = (*it)->GetIntersectionInterval(interval);
        if (!intersection) {
            ++it;
            continue;
        }
        auto oldStart = (*it)->GetStartIndex();
        auto oldEnd = (*it)->GetEndIndex();
        if (intersection->first == oldStart && intersection->second == oldEnd) {
            it = spans.erase(it);
            continue;
        }
        if (oldStart < intersection->first && intersection->second < oldEnd) {
            newSpans.emplace_back((*it)->GetSubSpan(oldStart, intersection->first));
            newSpans.emplace_back((*it)->GetSubSpan(intersection->second, oldEnd));
            it = spans.erase(it);
            continue;
        }
        if (oldEnd > intersection->second) {
            (*it)->UpdateStartIndex(intersection->second);
            ++it;
            continue;
        }
        if (intersection->first > oldStart) {
            (*it)->UpdateEndIndex(intersection->first);
            ++it;
        }
    }
    spans.merge(newSpans);
}

void SpanString::SortSpans(std::list<RefPtr<SpanBase>>& spans)
{
    spans.sort(
        [](const RefPtr<SpanBase>& a, const RefPtr<SpanBase>& b) { return a->GetStartIndex() < b->GetStartIndex(); });
}

bool SpanString::CanMerge(const RefPtr<SpanBase>& a, const RefPtr<SpanBase>& b)
{
    return a->GetEndIndex() >= b->GetStartIndex() && a->IsAttributesEqual(b);
}

void SpanString::MergeIntervals(std::list<RefPtr<SpanBase>>& spans)
{
    auto it = spans.begin();
    while (it != spans.end()) {
        auto spanType = (*it)->GetSpanType();
        if (spanType == SpanType::Image || spanType == SpanType::CustomSpan) {
            return;
        }
        auto current = it++;
        if (it != spans.end() && CanMerge(*current, *it)) {
            (*current)->UpdateStartIndex(std::min((*current)->GetStartIndex(), (*it)->GetStartIndex()));
            (*current)->UpdateEndIndex(std::max((*current)->GetEndIndex(), (*it)->GetEndIndex()));
            spans.erase(it++);
            if (it == spans.end()) {
                break;
            }
            it = current;
        }
    }
}

int32_t SpanString::GetStepsByPosition(int32_t pos)
{
    if (pos == 0) {
        return 0;
    }
    int32_t step = 0;
    for (auto iter = spans_.begin(); iter != spans_.end(); ++iter) {
        if ((*iter)->interval.first == pos) {
            return step;
        }
        if ((*iter)->interval.first < pos && pos < (*iter)->interval.second) {
            auto spanItem = (*iter)->GetSameStyleSpanItem();
            spanItem->interval.first = pos;
            spanItem->interval.second = (*iter)->interval.second;
            auto wStr = StringUtils::ToWstring(spanItem->content);
            auto start = (*iter)->interval.first;
            spanItem->content = StringUtils::ToString(wStr.substr(pos - start));
            spans_.insert(std::next(iter), spanItem);
            (*iter)->interval.second = pos;
            (*iter)->content = StringUtils::ToString(wStr.substr(0, pos - start));
            return step;
        }
        step++;
    }
    return step;
}

void SpanString::AddSpecialSpan(const RefPtr<SpanBase>& span, SpanType type, int32_t start)
{
    auto wStr = GetWideString();
    text_ = StringUtils::ToString(
        wStr.substr(0, start) + StringUtils::ToWstring(" ") + wStr.substr(start));
    auto iter = spans_.begin();
    auto step = GetStepsByPosition(start);
    std::advance(iter, step);
    RefPtr<NG::SpanItem> spanItem;
    if (type == SpanType::Image) {
        auto imageSpan = DynamicCast<ImageSpan>(span);
        CHECK_NULL_VOID(imageSpan);
        spanItem = MakeImageSpanItem(imageSpan);
    } else if (type == SpanType::CustomSpan) {
        auto customSpan = AceType::DynamicCast<CustomSpan>(span);
        CHECK_NULL_VOID(customSpan);
        spanItem = MakeCustomSpanItem(customSpan);
    }
    iter = spans_.insert(iter, spanItem);
    for (++iter; iter != spans_.end(); ++iter) {
        ++(*iter)->interval.first;
        ++(*iter)->interval.second;
    }

    UpdateSpanMapWithOffset(start - 1, 1);
    if (spansMap_.find(type) == spansMap_.end()) {
        spansMap_[type].emplace_back(span);
    } else {
        auto specialList = spansMap_[type];
        int32_t step = 0;
        for (const auto& specialSpan : specialList) {
            if (specialSpan->GetStartIndex() >= start) {
                break;
            }
            ++step;
        }
        auto iter = specialList.begin();
        std::advance(iter, step);
        specialList.insert(iter, span);
        spansMap_[type] = specialList;
    }
}

RefPtr<NG::ImageSpanItem> SpanString::MakeImageSpanItem(const RefPtr<ImageSpan>& imageSpan)
{
    auto spanItem = MakeRefPtr<NG::ImageSpanItem>();
    spanItem->content = " ";
    spanItem->interval.first = imageSpan->GetStartIndex();
    spanItem->interval.second = imageSpan->GetEndIndex();
    spanItem->SetImageSpanOptions(imageSpan->GetImageSpanOptions());
    return spanItem;
}

RefPtr<NG::CustomSpanItem> SpanString::MakeCustomSpanItem(const RefPtr<CustomSpan>& customSpan)
{
    auto spanItem = MakeRefPtr<NG::CustomSpanItem>();
    spanItem->content = " ";
    spanItem->interval.first = customSpan->GetStartIndex();
    spanItem->interval.second = customSpan->GetEndIndex();
    spanItem->onDraw = customSpan->GetOnDraw();
    spanItem->onMeasure = customSpan->GetOnMeasure();
    return spanItem;
}

void SpanString::ChangeStartAndEndToCorrectNum(int32_t& start, int32_t& end)
{
    auto text = GetWideString();
    TextEmojiSubStringRange range = TextEmojiProcessor::CalSubWstringRange(
        start, end-start, text, true);
    int startIndex = range.startIndex;
    int endIndex = range.endIndex;

    if (start == startIndex) {
        ChangeStartToCorrectNum(start);
    } else {
        LOGI("SpanString: Get Emoji, Change Start %{public}d to %{public}d",
            start, startIndex);
        start = startIndex;
    }

    if (end == endIndex) {
        ChangeEndToCorrectNum(end);
    } else {
        LOGI("SpanString: Get Emoji, Change End %{public}d to %{public}d",
            end, endIndex);
        end = endIndex;
    }

    if (end < start) {
        std::swap(start, end);
    }
}

void SpanString::ChangeStartToCorrectNum(int32_t& start)
{
    if (start == 0) {
        return;
    }
    auto text = GetWideString();
    auto textLen = static_cast<int32_t>(text.length());
    if (textLen == 0) {
        return;
    }
    auto tmpStart = start;
    auto substr = StringUtils::ToString(GetWideStringSubstr(text, 0, tmpStart));
    while (substr.length() == 0) {
        if (tmpStart == textLen) {
            break;
        }
        tmpStart --;
        if (tmpStart <= 0) {
            break;
        }
        substr = StringUtils::ToString(GetWideStringSubstr(text, 0, tmpStart));
    }
    if (tmpStart != start) {
        LOGI("SpanString: Get Complex Char, Change Start %{public}d to %{public}d", start, tmpStart);
        start = tmpStart;
    }
}

void SpanString::ChangeEndToCorrectNum(int32_t& end)
{
    auto text = GetWideString();
    auto textLen = static_cast<int32_t>(text.length());
    if (textLen == 0) {
        return;
    }
    auto tmpEnd = end;
    auto substr = StringUtils::ToString(GetWideStringSubstr(text, end));
    while (substr.length() == 0) {
        if (tmpEnd == textLen) {
            break;
        }
        tmpEnd ++;
        if (tmpEnd >= textLen) {
            break;
        }
        substr = StringUtils::ToString(GetWideStringSubstr(text, tmpEnd));
    }
    if (tmpEnd != end) {
        LOGI("SpanString: Get Complex Char, Change End %{public}d to %{public}d", end, tmpEnd);
        end = tmpEnd;
    }
}

void SpanString::AddSpan(const RefPtr<SpanBase>& span)
{
    if (!span || !CheckRange(span)) {
        return;
    }
    auto start = span->GetStartIndex();
    auto end = span->GetEndIndex();
    ChangeStartAndEndToCorrectNum(start, end);
    if (span->GetSpanType() == SpanType::Image || span->GetSpanType() == SpanType::CustomSpan) {
        AddSpecialSpan(span, span->GetSpanType(), start);
        return;
    }
    if (spansMap_.find(span->GetSpanType()) == spansMap_.end()) {
        spansMap_[span->GetSpanType()].emplace_back(span);
        ApplyToSpans(span, { start, end }, SpanOperation::ADD);
        return;
    }
    RemoveSpan(start, end - start, span->GetSpanType());
    auto spans = spansMap_[span->GetSpanType()];
    ApplyToSpans(span, { start, end }, SpanOperation::ADD);
    SplitInterval(spans, { start, end });
    spans.emplace_back(span);
    SortSpans(spans);
    MergeIntervals(spans);
    spansMap_[span->GetSpanType()] = spans;
}

void SpanString::RemoveSpan(int32_t start, int32_t length, SpanType key)
{
    if (!CheckRange(start, length)) {
        return;
    }
    auto end = start + length;
    ChangeStartAndEndToCorrectNum(start, end);
    length = end - start;
    auto it = spansMap_.find(key);
    if (it == spansMap_.end()) {
        return;
    }
    auto spans = spansMap_[key];
    if (key == SpanType::Image) {
        RemoveSpecialSpan(start, end, key);
        return;
    }
    if (key == SpanType::CustomSpan) {
        RemoveSpecialSpan(start, end, key);
        return;
    }
    auto defaultSpan = GetDefaultSpan(key);
    CHECK_NULL_VOID(defaultSpan);
    defaultSpan->UpdateStartIndex(start);
    defaultSpan->UpdateEndIndex(end);
    ApplyToSpans(defaultSpan, { start, end }, SpanOperation::REMOVE);
    SplitInterval(spans, { start, end });
    SortSpans(spans);
    MergeIntervals(spans);
    if (spans.empty()) {
        spansMap_.erase(key);
    } else {
        spansMap_[key] = spans;
    }
}

RefPtr<SpanBase> SpanString::GetDefaultSpan(SpanType type)
{
    switch (type) {
        case SpanType::Font:
            return MakeRefPtr<FontSpan>();
        case SpanType::TextShadow:
            return MakeRefPtr<TextShadowSpan>();
        case SpanType::Gesture:
            return MakeRefPtr<GestureSpan>();
        case SpanType::Decoration:
            return MakeRefPtr<DecorationSpan>();
        case SpanType::BaselineOffset:
            return MakeRefPtr<BaselineOffsetSpan>();
        case SpanType::LetterSpacing:
            return MakeRefPtr<LetterSpacingSpan>();
        case SpanType::ParagraphStyle:
            return MakeRefPtr<ParagraphStyleSpan>();
        case SpanType::LineHeight:
            return MakeRefPtr<LineHeightSpan>();
        case SpanType::ExtSpan:
            return MakeRefPtr<ExtSpan>();
        default:
            return nullptr;
    }
}

bool SpanString::CheckRange(const RefPtr<SpanBase>& spanBase) const
{
    auto start = spanBase->GetStartIndex();
    auto length = spanBase->GetLength();
    if (length <= 0) {
        return false;
    }
    auto spanType = spanBase->GetSpanType();
    auto len = spanType == SpanType::Image || spanType == SpanType::CustomSpan ? GetLength() + 1 : GetLength();
    auto end = start + length;

    if (start > len || end > len) {
        return false;
    }

    if (start < 0) {
        return false;
    }

    return true;
}

bool SpanString::CheckRange(int32_t start, int32_t length, bool allowLengthZero) const
{
    if (length < 0 || (length == 0 && !allowLengthZero)) {
        return false;
    }

    auto len = GetLength();
    auto end = start + length;

    if (start > len || end > len) {
        return false;
    }

    if (start < 0) {
        return false;
    }

    return true;
}

RefPtr<NG::SpanItem> SpanString::GetDefaultSpanItem(const std::string& text)
{
    auto spanItem = MakeRefPtr<NG::SpanItem>();
    spanItem->content = text;
    spanItem->interval = { 0, StringUtils::ToWstring(text).length() };
    return spanItem;
}

void SpanString::SetString(const std::string& text)
{
    text_ = text;
}

void SpanString::SetSpanItems(const std::list<RefPtr<NG::SpanItem>>&& spanItems)
{
    spans_ = spanItems;
}

void SpanString::SetSpanMap(std::unordered_map<SpanType, std::list<RefPtr<SpanBase>>>&& spansMap)
{
    spansMap_ = spansMap;
}

const std::string& SpanString::GetString() const
{
    return text_;
}

std::wstring SpanString::GetWideString()
{
    return StringUtils::ToWstring(text_);
}

int32_t SpanString::GetLength() const
{
    return StringUtils::ToWstring(text_).length();
}

bool SpanString::IsEqualToSpanString(const RefPtr<SpanString>& other) const
{
    return *this == *other;
}

RefPtr<SpanString> SpanString::GetSubSpanString(int32_t start, int32_t length) const
{
    if (!CheckRange(start, length)) {
        RefPtr<SpanString> span = AceType::MakeRefPtr<SpanString>("");
        return span;
    }
    int32_t end = start + length;
    RefPtr<SpanString> span =
        AceType::MakeRefPtr<SpanString>(StringUtils::ToString(StringUtils::ToWstring(text_).substr(start, length)));
    std::unordered_map<SpanType, std::list<RefPtr<SpanBase>>> subMap;
    for (const auto& map : spansMap_) {
        auto subList = GetSubSpanList(start, length, map.second);
        if (!subList.empty()) {
            subMap.insert({ map.first, subList });
        }
    }
    span->spansMap_ = subMap;

    std::list<RefPtr<NG::SpanItem>> subSpans_;
    for (const auto& spanItem : spans_) {
        auto intersection = spanItem->GetIntersectionInterval({start, start+length});
        if (intersection) {
            int32_t oldStart = spanItem->interval.first;
            int32_t oldEnd = spanItem->interval.second;
            auto spanStart = oldStart <= start ? 0 : oldStart - start;
            auto spanEnd = oldEnd < end ? oldEnd - start : end - start;
            auto newSpanItem = spanItem->GetSameStyleSpanItem();
            newSpanItem->interval = { spanStart, spanEnd };
            newSpanItem->content = StringUtils::ToString(
                StringUtils::ToWstring(spanItem->content)
                    .substr(std::max(start - oldStart, 0), std::min(end, oldEnd) - std::max(start, oldStart)));
            subSpans_.emplace_back(newSpanItem);
        }
    }
    span->spans_ = subSpans_;
    return span;
}

std::list<RefPtr<SpanBase>> SpanString::GetSubSpanList(
    int32_t start, int32_t length, const std::list<RefPtr<SpanBase>>& spans) const
{
    std::list<RefPtr<SpanBase>> res;
    int32_t end = start + length;
    for (auto& span : spans) {
        auto intersection = span->GetIntersectionInterval({ start, end });
        if (intersection) {
            int32_t spanStart = span->GetStartIndex();
            int32_t spanEnd = span->GetEndIndex();
            spanStart = spanStart <= start ? 0 : spanStart - start;
            spanEnd = spanEnd < end ? spanEnd - start : end - start;
            if (spanStart == spanEnd) {
                continue;
            }
            res.emplace_back(span->GetSubSpan(spanStart, spanEnd));
        }
    }
    return res;
}

const std::unordered_map<SpanType, std::list<RefPtr<SpanBase>>>& SpanString::GetSpansMap() const
{
    return spansMap_;
}

std::vector<RefPtr<SpanBase>> SpanString::GetSpans(int32_t start, int32_t length) const
{
    std::vector<RefPtr<SpanBase>> res;
    if (!CheckRange(start, length)) {
        return res;
    }

    for (const auto& map : spansMap_) {
        auto spans = GetSpans(start, length, map.first);
        res.insert(res.begin(), spans.begin(), spans.end());
    }
    return res;
}

std::vector<RefPtr<SpanBase>> SpanString::GetSpans(int32_t start, int32_t length, SpanType spanType) const
{
    std::vector<RefPtr<SpanBase>> res;
    if (!CheckRange(start, length)) {
        return res;
    }
    int32_t end = start + length;
    RefPtr<SpanBase> span;
    while ((span = GetSpan(start, length, spanType)) != nullptr) {
        res.emplace_back(span);
        start = span->GetEndIndex();
        length = end - start;
    }
    return res;
}

RefPtr<SpanBase> SpanString::GetSpan(int32_t start, int32_t length, SpanType spanType) const
{
    if (!CheckRange(start, length) || spansMap_.find(spanType) == spansMap_.end()) {
        return nullptr;
    }
    int32_t end = start + length;
    auto spanBaseList = spansMap_.find(spanType)->second;
    for (auto& spanBase : spanBaseList) {
        auto intersection = spanBase->GetIntersectionInterval({ start, end });
        if (intersection) {
            int32_t newStart = intersection->first;
            int32_t newEnd = intersection->second;
            if (newStart == newEnd) {
                continue;
            }
            return spanBase->GetSubSpan(newStart, newEnd);
        }
    }
    return nullptr;
}

bool SpanString::operator==(const SpanString& other) const
{
    if (text_ != other.text_) {
        return false;
    }
    auto size =
        !spansMap_.empty()
            ? (static_cast<int32_t>(spansMap_.size()) - (spansMap_.find(SpanType::Gesture) == spansMap_.end() ? 0 : 1))
            : 0;
    auto sizeOther = !other.spansMap_.empty()
                         ? (static_cast<int32_t>(other.spansMap_.size()) -
                               (other.spansMap_.find(SpanType::Gesture) == other.spansMap_.end() ? 0 : 1))
                         : 0;
    if (size != sizeOther) {
        return false;
    }

    for (const auto& map : spansMap_) {
        if (map.first == SpanType::Gesture) {
            continue;
        }
        auto spansOtherMap = other.spansMap_.find(map.first);
        if (spansOtherMap == other.spansMap_.end()) {
            return false;
        }
        auto spans = map.second;
        auto spansOther = spansOtherMap->second;
        if (spans.size() != spansOther.size()) {
            return false;
        }
        for (auto spansItr = spans.begin(), spansOtherItr = spansOther.begin();
             spansItr != spans.end() && spansOtherItr != spansOther.end(); ++spansItr, ++spansOtherItr) {
            if (!(*spansItr)->IsAttributesEqual(*spansOtherItr) ||
                (*spansItr)->GetEndIndex() != (*spansOtherItr)->GetEndIndex() ||
                (*spansItr)->GetStartIndex() != (*spansOtherItr)->GetStartIndex()) {
                return false;
            }
        }
    }
    return true;
}

std::list<RefPtr<NG::SpanItem>> SpanString::GetSpanItems() const
{
    return spans_;
}

void SpanString::BindWithSpans(const std::vector<RefPtr<SpanBase>>& spans)
{
    for (auto& span : spans) {
        AddSpan(span);
    }
}

void SpanString::UpdateSpansWithOffset(int32_t start, int32_t offset)
{
    for (auto& span : spans_) {
        if (span->interval.second > start && span->interval.first != start) {
            span->interval.second += offset;
        }
        if (span->interval.first > start) {
            span->interval.first += offset;
        }
    }
}

void SpanString::UpdateSpanMapWithOffset(int32_t start, int32_t offset)
{
    for (auto& iter : spansMap_) {
        if (spansMap_.find(iter.first) == spansMap_.end()) {
            continue;
        }
        auto spans = spansMap_[iter.first];
        for (auto& it : spans) {
            UpdateSpanBaseWithOffset(it, start, offset);
        }
        spansMap_[iter.first] = spans;
    }
}

void SpanString::UpdateSpanBaseWithOffset(RefPtr<SpanBase>& span, int32_t start, int32_t offset)
{
    if (span->GetEndIndex() > start && span->GetStartIndex() != start) {
        span->UpdateEndIndex(span->GetEndIndex() + offset);
    }
    if (span->GetStartIndex() > start) {
        span->UpdateStartIndex(span->GetStartIndex() + offset);
    }
}

void SpanString::RemoveSpecialSpan(int32_t start, int32_t end, SpanType type)
{
    auto spans = spansMap_[type];
    int32_t count = 0;
    for (auto iter = spans.begin(); iter != spans.end();) {
        if ((*iter)->GetStartIndex() >= start && (*iter)->GetStartIndex() < end - count) {
            auto wStr = GetWideString();
            wStr.erase((*iter)->GetStartIndex(), 1);
            text_ = StringUtils::ToString(wStr);
            UpdateSpanMapWithOffset((*iter)->GetStartIndex(), -1);
            iter = spans.erase(iter);
            ++count;
            continue;
        }
        ++iter;
    }
    if (spans.empty()) {
        spansMap_.erase(type);
    } else {
        spansMap_[type] = spans;
    }
    count = 0;
    for (auto iter = spans_.begin(); iter != spans_.end();) {
        if ((*iter)->interval.first >= start && (*iter)->interval.first < end - count
            && ((type == SpanType::Image && (*iter)->spanItemType == NG::SpanItemType::IMAGE)
                || (type == SpanType::CustomSpan && (*iter)->spanItemType == NG::SpanItemType::CustomSpan))) {
            UpdateSpansWithOffset((*iter)->interval.first, -1);
            iter = spans_.erase(iter);
            ++count;
            continue;
        }
        ++iter;
    }
}

void SpanString::GetSpecialTypesVector(std::list<int32_t>& indexList, int32_t start, int32_t length)
{
    int32_t end = start + length;
    auto iter = indexList.begin();
    for (const auto& type : specailTypes) {
        auto spans = spansMap_[type];
        for (const auto& span : spans) {
            auto intersection = span->GetIntersectionInterval({ start, end });
            if (!intersection) {
                continue;
            }
            iter = indexList.insert(iter, span->GetStartIndex());
        }
    }
    indexList.sort([](const int32_t& a, const int32_t& b) { return a < b; });
}

void SpanString::GetNormalTypesVector(std::list<std::pair<int32_t, int32_t>>& indexList, int32_t start, int32_t length)
{
    std::list<int32_t> specialList;
    GetSpecialTypesVector(specialList, start, length);
    auto next = start;
    auto iter = indexList.begin();
    for (const auto& index : specialList) {
        if (index > next) {
            iter = indexList.insert(iter, { next, index - next });
        }
        next = index + 1;
    }
    if (next < start + length) {
        indexList.insert(iter, { next, start + length - next });
    }
}

bool SpanString::ContainSpecialNode(int32_t start, int32_t length)
{
    int32_t end = start + length;
    for (const auto& type : specailTypes) {
        auto spans = spansMap_[type];
        for (const auto& span : spans) {
            auto intersection = span->GetIntersectionInterval({ start, end });
            if (intersection) {
                return true;
            }
            if (span->GetStartIndex() >= end) {
                break;
            }
        }
    }
    return false;
}

bool SpanString::IsSpecialNode(RefPtr<SpanBase> span)
{
    auto type = span->GetSpanType();
    if (specailTypes.find(type) == specailTypes.end()) {
        return false;
    }
    return true;
}

void SpanString::ClearSpans()
{
    spans_.clear();
}

void SpanString::AppendSpanItem(const RefPtr<NG::SpanItem>& spanItem)
{
    spans_.emplace_back(spanItem);
}

bool SpanString::EncodeTlv(std::vector<uint8_t>& buff)
{
    TLVUtil::WriteUint8(buff, TLV_SPAN_STRING_SPANS);
    TLVUtil::WriteInt32(buff, spans_.size());
    for (auto it = spans_.begin(); it != spans_.end(); ++it) {
        auto spanItem = (*it);
        if (spanItem->spanItemType == NG::SpanItemType::CustomSpan) {
            TLVUtil::WriteInt32(buff, static_cast<int32_t>(NG::SpanItemType::NORMAL));
            auto placeHolderSpan = AceType::MakeRefPtr<NG::SpanItem>();
            placeHolderSpan->content = " ";
            placeHolderSpan->interval = spanItem->interval;
            placeHolderSpan->EncodeTlv(buff);
            continue;
        }
        TLVUtil::WriteInt32(buff, static_cast<int32_t>(spanItem->spanItemType));
        spanItem->EncodeTlv(buff);
    }
    TLVUtil::WriteUint8(buff, TLV_SPAN_STRING_CONTENT);
    TLVUtil::WriteString(buff, text_);
    TLVUtil::WriteUint8(buff, TLV_END);
    return true;
}

RefPtr<SpanString> SpanString::DecodeTlv(std::vector<uint8_t>& buff)
{
    int32_t cursor = 0;
    RefPtr<SpanString> spanStr = MakeRefPtr<SpanString>("");
    spanStr->ClearSpans();
    for (uint8_t tag = TLVUtil::ReadUint8(buff, cursor); tag != TLV_END; tag = TLVUtil::ReadUint8(buff, cursor)) {
        switch (tag) {
            case TLV_SPAN_STRING_CONTENT: {
                auto str = TLVUtil::ReadString(buff, cursor);
                spanStr->SetString(str);
                break;
            }
            case TLV_SPAN_STRING_SPANS: {
                DecodeSpanItemList(buff, cursor, spanStr);
                break;
            }
            default:
                break;
        }
    }
    return spanStr;
}

void SpanString::DecodeSpanItemList(std::vector<uint8_t>& buff, int32_t& cursor, RefPtr<SpanString>& spanStr)
{
    int32_t spanLength = TLVUtil::ReadInt32(buff, cursor);
    for (auto i = 0; i < spanLength; i++) {
        auto spanItemType = TLVUtil::ReadInt32(buff, cursor);
        if (spanItemType == static_cast<int32_t>(NG::SpanItemType::IMAGE)) {
            auto imageSpanItem = NG::ImageSpanItem::DecodeTlv(buff, cursor);
            spanStr->AppendSpanItem(imageSpanItem);
        } else {
            auto spanItem = NG::SpanItem::DecodeTlv(buff, cursor);
            spanStr->AppendSpanItem(spanItem);
        }
    }
    spanStr->UpdateSpansMap();
}

void SpanString::UpdateSpansMap()
{
    spansMap_.clear();
    for (auto& spanItem : spans_) {
        if (!spanItem) {
            continue;
        }
        auto start = spanItem->interval.first;
        auto end = spanItem->interval.second;
        std::list<RefPtr<SpanBase>> spanBases = {
            ToFontSpan(spanItem, start, end),
            ToDecorationSpan(spanItem, start, end),
            ToBaselineOffsetSpan(spanItem, start, end),
            ToLetterSpacingSpan(spanItem, start, end),
            ToGestureSpan(spanItem, start, end),
            ToImageSpan(spanItem),
            ToParagraphStyleSpan(spanItem, start, end),
            ToLineHeightSpan(spanItem, start, end) };
        for (auto& spanBase : spanBases) {
            if (!spanBase) {
                continue;
            }
            auto it = spansMap_.find(spanBase->GetSpanType());
            if (it == spansMap_.end()) {
                spansMap_.insert({ spanBase->GetSpanType(), { spanBase } });
            } else {
                it->second.emplace_back(std::move(spanBase));
            }
        }
    }
}

RefPtr<FontSpan> SpanString::ToFontSpan(const RefPtr<NG::SpanItem>& spanItem, int32_t start, int32_t end)
{
    CHECK_NULL_RETURN(spanItem && spanItem->fontStyle, nullptr);
    Font font;
    font.fontColor = spanItem->fontStyle->GetTextColor();
    font.fontFamiliesNG = spanItem->fontStyle->GetFontFamily();
    font.fontSize = spanItem->fontStyle->GetFontSize();
    font.fontStyle = spanItem->fontStyle->GetItalicFontStyle();
    font.fontWeight = spanItem->fontStyle->GetFontWeight();
    return AceType::MakeRefPtr<FontSpan>(font, start, end);
}

RefPtr<DecorationSpan> SpanString::ToDecorationSpan(
    const RefPtr<NG::SpanItem>& spanItem, int32_t start, int32_t end)
{
    CHECK_NULL_RETURN(spanItem && spanItem->fontStyle, nullptr);
    TextDecoration type = spanItem->fontStyle->GetTextDecoration().value_or(TextDecoration::NONE);
    std::optional<Color> color = spanItem->fontStyle->GetTextDecorationColor();
    std::optional<TextDecorationStyle> style = spanItem->fontStyle->GetTextDecorationStyle();
    return AceType::MakeRefPtr<DecorationSpan>(type, color, style, start, end);
}

RefPtr<BaselineOffsetSpan> SpanString::ToBaselineOffsetSpan(
    const RefPtr<NG::SpanItem>& spanItem, int32_t start, int32_t end)
{
    CHECK_NULL_RETURN(spanItem && spanItem->textLineStyle, nullptr);
    Dimension baselineOffset;
    if (spanItem->textLineStyle->GetBaselineOffset().has_value()) {
        baselineOffset.SetValue(spanItem->textLineStyle->GetBaselineOffsetValue().ConvertToVp());
    }
    return AceType::MakeRefPtr<BaselineOffsetSpan>(baselineOffset, start, end);
}

RefPtr<LetterSpacingSpan> SpanString::ToLetterSpacingSpan(
    const RefPtr<NG::SpanItem>& spanItem, int32_t start, int32_t end)
{
    CHECK_NULL_RETURN(spanItem && spanItem->fontStyle, nullptr);
    Dimension letterSpacing;
    if (spanItem->fontStyle->GetLetterSpacing().has_value()) {
        letterSpacing.SetValue(spanItem->fontStyle->GetLetterSpacingValue().ConvertToVp());
    }
    return AceType::MakeRefPtr<LetterSpacingSpan>(letterSpacing, start, end);
}

RefPtr<GestureSpan> SpanString::ToGestureSpan(const RefPtr<NG::SpanItem>& spanItem, int32_t start, int32_t end)
{
    GestureStyle gestureInfo;
    if (spanItem->onClick) {
        gestureInfo.onClick = spanItem->onClick;
    }
    if (spanItem->onLongPress) {
        gestureInfo.onLongPress = spanItem->onLongPress;
    }
    return AceType::MakeRefPtr<GestureSpan>(gestureInfo, start, end);
}

RefPtr<TextShadowSpan> SpanString::ToTextShadowSpan(
    const RefPtr<NG::SpanItem>& spanItem, int32_t start, int32_t end)
{
    CHECK_NULL_RETURN(spanItem && spanItem->fontStyle, nullptr);
    std::vector<Shadow> textShadow = spanItem->fontStyle->GetTextShadow().value_or(std::vector<Shadow> { Shadow() });
    return AceType::MakeRefPtr<TextShadowSpan>(textShadow, start, end);
}

RefPtr<ImageSpan> SpanString::ToImageSpan(const RefPtr<NG::SpanItem>& spanItem)
{
    auto imageItem = DynamicCast<NG::ImageSpanItem>(spanItem);
    CHECK_NULL_RETURN(imageItem, nullptr);
    return AceType::MakeRefPtr<ImageSpan>(imageItem->options);
}

RefPtr<ParagraphStyleSpan> SpanString::ToParagraphStyleSpan(
    const RefPtr<NG::SpanItem>& spanItem, int32_t start, int32_t end)
{
    CHECK_NULL_RETURN(spanItem && spanItem->textLineStyle, nullptr);
    SpanParagraphStyle paragraphStyle;
    paragraphStyle.align = spanItem->textLineStyle->GetTextAlign();
    paragraphStyle.maxLines = spanItem->textLineStyle->GetMaxLines();
    paragraphStyle.textOverflow = spanItem->textLineStyle->GetTextOverflow();
    paragraphStyle.leadingMargin = spanItem->textLineStyle->GetLeadingMargin();
    paragraphStyle.wordBreak = spanItem->textLineStyle->GetWordBreak();
    paragraphStyle.textIndent = spanItem->textLineStyle->GetTextIndent();
    return AceType::MakeRefPtr<ParagraphStyleSpan>(paragraphStyle, start, end);
}

RefPtr<LineHeightSpan> SpanString::ToLineHeightSpan(const RefPtr<NG::SpanItem>& spanItem, int32_t start, int32_t end)
{
    CHECK_NULL_RETURN(spanItem && spanItem->textLineStyle, nullptr);
    Dimension lineHeight;
    if (spanItem->textLineStyle->GetLineHeight().has_value()) {
        lineHeight.SetValue(spanItem->textLineStyle->GetLineHeightValue().ConvertToVp());
    }
    return AceType::MakeRefPtr<LineHeightSpan>(lineHeight, start, end);
}
} // namespace OHOS::Ace