ohos/osal/data_detector_adapter.cpp

/*
 * 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/common/ai/data_detector_adapter.h"

#include "adapter/ohos/entrance/ace_container.h"
#include "base/utils/utf_helper.h"
#include "core/common/ai/data_detector_mgr.h"
#include "core/common/ai/data_url_analyzer_mgr.h"
#include "core/components/text_overlay/text_overlay_theme.h"
#include "core/pipeline_ng/pipeline_context.h"

namespace OHOS::Ace {

constexpr int32_t AI_TEXT_MAX_LENGTH = 500;
constexpr int32_t AI_TEXT_GAP = 100;
constexpr int32_t AI_DELAY_TIME = 100;
constexpr uint32_t SECONDS_TO_MILLISECONDS = 1000;
constexpr uint8_t URL_DETECT_FINISH = (1 << 0);
constexpr uint8_t OTHER_DETECT_FINISH = (1 << 1);
constexpr uint8_t ALL_DETECT_FINISH = URL_DETECT_FINISH | OTHER_DETECT_FINISH;

const std::string ALL_TEXT_DETECT_TYPES = "phoneNum,url,email,location,datetime";
const std::string TEXT_DETECT_TYPES_WITHOUT_URL = "phoneNum,email,location,datetime";
const std::string ASK_CELIA_TAG = "askCelia";

const std::unordered_map<TextDataDetectType, std::string> TEXT_DETECT_MAP = {
    { TextDataDetectType::PHONE_NUMBER, "phoneNum" }, { TextDataDetectType::URL, "url" },
    { TextDataDetectType::EMAIL, "email" }, { TextDataDetectType::ADDRESS, "location" },
    { TextDataDetectType::DATE_TIME, "datetime" }
};
const std::unordered_map<std::string, TextDataDetectType> TEXT_DETECT_MAP_REVERSE = {
    { "phoneNum", TextDataDetectType::PHONE_NUMBER }, { "url", TextDataDetectType::URL },
    { "email", TextDataDetectType::EMAIL }, { "location", TextDataDetectType::ADDRESS },
    { "datetime", TextDataDetectType::DATE_TIME }
};

void DataDetectorAdapter::GetAIEntityMenu()
{
    auto context = PipelineContext::GetCurrentContextSafely();
    CHECK_NULL_VOID(context);
    auto uiTaskExecutor = SingleTaskExecutor::Make(context->GetTaskExecutor(), TaskExecutor::TaskType::UI);
    uiTaskExecutor.PostTask(
        [weak = AceType::WeakClaim(this), instanceId = context->GetInstanceId()] {
            ContainerScope scope(instanceId);
            auto dataDetectorAdapter = weak.Upgrade();
            CHECK_NULL_VOID(dataDetectorAdapter);
            TAG_LOGI(AceLogTag::ACE_TEXT, "Get AI entity menu from ai_engine");
            DataDetectorMgr::GetInstance().GetAIEntityMenu(dataDetectorAdapter->textDetectResult_);
        },
        "ArkUITextInitDataDetect");
}

bool DataDetectorAdapter::ShowAIEntityMenu(
    const AISpan& aiSpan, const NG::RectF& aiRect, const RefPtr<NG::FrameNode>& targetNode, AIMenuInfo info)
{
    std::vector<std::pair<std::string, std::function<void()>>> menuOptions;
    if (!GetAiEntityMenuOptions(aiSpan, targetNode, info, menuOptions)) {
        return false;
    }
    auto pipeline = NG::PipelineContext::GetCurrentContextSafelyWithCheck();
    CHECK_NULL_RETURN(pipeline, false);
    auto overlayManager = pipeline->GetOverlayManager();
    CHECK_NULL_RETURN(overlayManager, false);
    return overlayManager->ShowAIEntityMenu(menuOptions, aiRect, targetNode);
}

bool DataDetectorAdapter::GetAiEntityMenuOptions(const AISpan& aiSpan, const RefPtr<NG::FrameNode>& targetNode,
    AIMenuInfo info, std::vector<std::pair<std::string, std::function<void()>>>& menuOptions)
{
    if (textDetectResult_.menuOptionAndAction.empty()) {
        TAG_LOGW(AceLogTag::ACE_TEXT, "menu option is empty, please try again");
        GetAIEntityMenu();
        return false;
    }

    mainContainerId_ = Container::CurrentId();
    auto menuOptionAndAction = textDetectResult_.menuOptionAndAction[TEXT_DETECT_MAP.at(aiSpan.type)];
    if (menuOptionAndAction.empty()) {
        return false;
    }
    if (!info.isShowSelectText) {
        // delete the last option: selectText.
        menuOptionAndAction.pop_back();
        if (!info.isShowCopy) {
            // delete the last option: copy.
            menuOptionAndAction.pop_back();
        }
    }

    auto index = -1;
    for (auto menuOption : menuOptionAndAction) {
        ++index;
        std::function<void()> onClickEvent = [aiSpan, menuOption, weak = AceType::WeakClaim(this),
                                                 targetNodeWeak = AceType::WeakClaim(AceType::RawPtr(targetNode)),
                                                 mainId = Container::CurrentIdSafelyWithCheck(), index]() {
            ContainerScope scope(mainId);
            auto dataDetectorAdapter = weak.Upgrade();
            CHECK_NULL_VOID(dataDetectorAdapter);
            auto targetNode = targetNodeWeak.Upgrade();
            CHECK_NULL_VOID(targetNode);
            dataDetectorAdapter->OnClickAIMenuOption(aiSpan, menuOption, targetNode);
        };
        menuOptions.push_back(std::make_pair(menuOption.first, onClickEvent));
    }
    return true;
}

std::function<void()> DataDetectorAdapter::GetPreviewMenuOptionCallback(
    TextDataDetectType type, const std::string& content)
{
    return [content, type, mainId = Container::CurrentIdSafelyWithCheck()]() {
        ContainerScope scope(mainId);
        auto pipeline = NG::PipelineContext::GetCurrentContextSafelyWithCheck();
        CHECK_NULL_VOID(pipeline);
        auto fontManager = pipeline->GetFontManager();
        CHECK_NULL_VOID(fontManager);
        fontManager->OnPreviewMenuOptionClick(type, content);
    };
}

RefPtr<NG::FrameNode> DataDetectorAdapter::CreateAIEntityMenu(
    const AISpan& aiSpan, const RefPtr<NG::FrameNode>& targetNode, AIMenuInfo info)
{
    std::vector<std::pair<std::string, std::function<void()>>> menuOptions;
    if (!GetAiEntityMenuOptions(aiSpan, targetNode, info, menuOptions)) {
        return nullptr;
    }
    CHECK_NULL_RETURN(targetNode, nullptr);
    auto pipeline = NG::PipelineContext::GetCurrentContextSafelyWithCheck();
    CHECK_NULL_RETURN(pipeline, nullptr);
    auto overlayManager = pipeline->GetOverlayManager();
    CHECK_NULL_RETURN(overlayManager, nullptr);
    auto theme = pipeline->GetTheme<TextOverlayTheme>();
    CHECK_NULL_RETURN(theme, nullptr);
    auto name = theme->GetAiMenuPreviewOptionName(aiSpan.type);
    if (!menuOptions.empty() && !name.empty()) {
        auto& option = menuOptions.front();
        option.first = name;
        option.second = GetPreviewMenuOptionCallback(aiSpan.type, aiSpan.content);
    }
    auto menuNode = overlayManager->BuildAIEntityMenu(menuOptions);
    CHECK_NULL_RETURN(menuNode, nullptr);
    return menuNode;
}

void DataDetectorAdapter::OnClickAIMenuOption(const AISpan& aiSpan,
    const std::pair<std::string, FuncVariant>& menuOption, const RefPtr<NG::FrameNode>& targetNode)
{
    TAG_LOGI(AceLogTag::ACE_TEXT, "Click AI menu option: %{public}s", menuOption.first.c_str());
    auto pipeline = NG::PipelineContext::GetCurrentContextSafelyWithCheck();
    CHECK_NULL_VOID(pipeline);
    auto overlayManager = pipeline->GetOverlayManager();
    CHECK_NULL_VOID(overlayManager);
    if (targetNode) {
        overlayManager->CloseAIEntityMenu(targetNode->GetId());
    }
    if (mainContainerId_ == -1) {
        mainContainerId_ = Container::CurrentId();
    }
    Container::UpdateCurrent(mainContainerId_);

    auto runtimeContext = Platform::AceContainer::GetRuntimeContext(pipeline->GetInstanceId());
    CHECK_NULL_VOID(runtimeContext);
    auto token = runtimeContext->GetToken();
    auto bundleName = runtimeContext->GetBundleName();

    hasClickedMenuOption_ = true;
    if (aiSpan.type == TextDataDetectType::ASK_CELIA) {
        auto vectorStringFunc = textDetectResult_.menuOptionAndAction.find(ASK_CELIA_TAG);
        if (vectorStringFunc == textDetectResult_.menuOptionAndAction.end()) {
            TAG_LOGW(AceLogTag::ACE_TEXT, "No askCelia option");
        } else {
            auto funcVariant = vectorStringFunc->second.begin()->second;
            if (std::holds_alternative<std::function<void(int, std::string)>>(funcVariant) &&
                std::get<std::function<void(int, std::string)>>(funcVariant)) {
                TAG_LOGI(AceLogTag::ACE_TEXT, "DataDetectorAdapter::OnClickAIMenuOption, call ask celia");
                std::get<std::function<void(int, std::string)>>(funcVariant)(true, aiSpan.content);
            }
        }
        hasClickedMenuOption_ = false;
        return;
    }
    if (onClickMenu_ && std::holds_alternative<std::function<std::string()>>(menuOption.second)) {
        onClickMenu_(std::get<std::function<std::string()>>(menuOption.second)());
    } else if (std::holds_alternative<std::function<void(sptr<IRemoteObject>, std::string)>>(menuOption.second)) {
        std::get<std::function<void(sptr<IRemoteObject>, std::string)>>(menuOption.second)(token, aiSpan.content);
    } else if (std::holds_alternative<std::function<void(int32_t, std::string)>>(menuOption.second)) {
        std::get<std::function<void(int32_t, std::string)>>(menuOption.second)(mainContainerId_, aiSpan.content);
    } else if (std::holds_alternative<std::function<void(int32_t, std::string, std::string, int32_t, std::string)>>(
                   menuOption.second)) {
        std::get<std::function<void(int32_t, std::string, std::string, int32_t, std::string)>>(menuOption.second)(
            mainContainerId_, UtfUtils::Str16DebugToStr8(textForAI_), bundleName, aiSpan.start, aiSpan.content);
        TAG_LOGI(AceLogTag::ACE_TEXT, "textForAI:%{public}d, start:%{public}d, aiSpan.length:%{public}d",
            static_cast<int32_t>(textForAI_.length()), aiSpan.start, static_cast<int32_t>(aiSpan.content.length()));
    } else {
        TAG_LOGW(AceLogTag::ACE_TEXT, "No matching menu option");
    }
    hasClickedMenuOption_ = false;
}

void DataDetectorAdapter::ResponseBestMatchItem(const AISpan& aiSpan)
{
    if (textDetectResult_.menuOptionAndAction.empty()) {
        TAG_LOGW(AceLogTag::ACE_TEXT, "menu option is empty, please try again");
        GetAIEntityMenu();
        return;
    }
    auto menuOptions = textDetectResult_.menuOptionAndAction[TEXT_DETECT_MAP.at(aiSpan.type)];
    if (menuOptions.empty()) {
        TAG_LOGW(AceLogTag::ACE_TEXT, "menu option is empty");
        return;
    }
    OnClickAIMenuOption(aiSpan, menuOptions[0]);
}

void DataDetectorAdapter::SetTextDetectTypes(const std::string& textDetectTypes)
{
    auto types = textDetectTypes;
    if (types.empty()) {
        types.append(ALL_TEXT_DETECT_TYPES);
    }

    std::set<std::string> newTypesSet;
    std::istringstream iss(types);
    std::string type;
    while (std::getline(iss, type, ',')) {
        newTypesSet.insert(type);
    }
    std::string detectTypesResult;
    bool hasUrlType = false;
    std::for_each(newTypesSet.begin(), newTypesSet.end(), [&hasUrlType, &detectTypesResult](const std::string& type) {
        if (type == "url") {
            hasUrlType = true;
        } else {
            detectTypesResult.append(type).append(",");
        }
    });
    if (!detectTypesResult.empty()) {
        detectTypesResult.pop_back();
    }
    hasUrlType_ = hasUrlType;
    textDetectTypes_ = detectTypesResult;
    if (newTypesSet != textDetectTypesSet_) {
        textDetectTypesSet_ = newTypesSet;
        typeChanged_ = true;
        aiDetectInitialized_ = false;
        MarkDirtyNode();
    }
}

bool DataDetectorAdapter::ParseOriText(const std::unique_ptr<JsonValue>& entityJson, std::u16string& text)
{
    TAG_LOGI(AceLogTag::ACE_TEXT, "Parse origin text entry");
    auto runtimeContext = Platform::AceContainer::GetRuntimeContext(Container::CurrentId());
    CHECK_NULL_RETURN(runtimeContext, false);
    if (runtimeContext->GetBundleName() != entityJson->GetString("bundleName")) {
        TAG_LOGW(AceLogTag::ACE_TEXT,
            "Wrong bundleName, the context bundleName is: %{public}s, but your bundleName is: %{public}s",
            runtimeContext->GetBundleName().c_str(), entityJson->GetString("bundleName").c_str());
        return false;
    }
    auto aiSpanArray = entityJson->GetValue("entity");
    if (aiSpanArray->IsNull() || !aiSpanArray->IsArray()) {
        TAG_LOGW(AceLogTag::ACE_TEXT, "Wrong AI entity");
        return false;
    }

    aiSpanMap_.clear();
    aiSpanRects_.clear();
    detectTexts_.clear();
    AISpan aiSpan;
    for (int32_t i = 0; i < aiSpanArray->GetArraySize(); ++i) {
        auto item = aiSpanArray->GetArrayItem(i);
        aiSpan.content = item->GetString("entityContent");
        aiSpan.type = TEXT_DETECT_MAP_REVERSE.at(item->GetString("entityType"));
        aiSpan.start = item->GetInt("start");
        aiSpan.end = item->GetInt("end");
        aiSpanMap_[aiSpan.start] = aiSpan;
    }
    aiDetectInitialized_ = true;
    text = UtfUtils::Str8ToStr16(entityJson->GetString("content"));
    textForAI_ = text;
    lastTextForAI_ = textForAI_;
    if (textDetectResult_.menuOptionAndAction.empty()) {
        GetAIEntityMenu();
    }

    TAG_LOGI(AceLogTag::ACE_TEXT, "Parse origin text successful");
    return true;
}

void DataDetectorAdapter::PreprocessTextDetect()
{
    aiDetectFlag_ = 0;
    if (textDetectTypes_.empty() && !hasUrlType_) {
        textDetectTypes_ = TEXT_DETECT_TYPES_WITHOUT_URL;
        hasUrlType_ = true;
    }
    if (textDetectTypes_.empty()) {
        aiDetectFlag_ |= OTHER_DETECT_FINISH;
    }
    if (!hasUrlType_) {
        aiDetectFlag_ |= URL_DETECT_FINISH;
    }
}

void DataDetectorAdapter::InitTextDetect(int32_t startPos, std::string detectText)
{
    CHECK_NULL_VOID(!textDetectTypes_.empty());
    TextDataDetectInfo info;
    info.text = detectText;
    info.module = textDetectTypes_;

    auto context = PipelineContext::GetCurrentContextSafelyWithCheck();
    CHECK_NULL_VOID(context);
    int32_t instanceID = context->GetInstanceId();
    auto textFunc = [weak = WeakClaim(this),
        instanceID, startPos, detectTypesSet = textDetectTypesSet_](const TextDataDetectResult result) {
        ContainerScope scope(instanceID);
        auto context = PipelineContext::GetCurrentContextSafelyWithCheck();
        CHECK_NULL_VOID(context);
        auto uiTaskExecutor = SingleTaskExecutor::Make(context->GetTaskExecutor(), TaskExecutor::TaskType::UI);
        uiTaskExecutor.PostTask(
            [result, weak, instanceID, startPos, detectTypesSet] {
                ContainerScope scope(instanceID);
                auto dataDetectorAdapter = weak.Upgrade();
                CHECK_NULL_VOID(dataDetectorAdapter);
                if (detectTypesSet != dataDetectorAdapter->textDetectTypesSet_) {
                    return;
                }
                dataDetectorAdapter->ParseAIResult(result, startPos);
                dataDetectorAdapter->MarkDirtyNode();
            },
            "ArkUITextParseAIResult");
    };

    auto backgroundExecutor = SingleTaskExecutor::Make(context->GetTaskExecutor(), TaskExecutor::TaskType::BACKGROUND);
    backgroundExecutor.PostTask(
        [info, textFunc] {
            TAG_LOGI(AceLogTag::ACE_TEXT, "DataDetectorAdapter::InitTextDetect, start AI detect, length: %{public}zu",
                info.text.size());
            DataDetectorMgr::GetInstance().DataDetect(info, textFunc);
        },
        "ArkUITextInitDataDetect");
}

void DataDetectorAdapter::HandleTextUrlDetect()
{
    auto context = PipelineContext::GetCurrentContextSafelyWithCheck();
    CHECK_NULL_VOID(context);
    int32_t instanceID = context->GetInstanceId();
    auto textFunc = [weak = WeakClaim(this),
        instanceID, detectTypesSet = textDetectTypesSet_](const std::vector<UrlEntity>& urlEntities) {
        ContainerScope scope(instanceID);
        auto context = PipelineContext::GetCurrentContextSafelyWithCheck();
        CHECK_NULL_VOID(context);
        auto uiTaskExecutor = SingleTaskExecutor::Make(context->GetTaskExecutor(), TaskExecutor::TaskType::UI);
        uiTaskExecutor.PostTask(
            [urlEntities, weak, instanceID, detectTypesSet] {
                ContainerScope scope(instanceID);
                auto dataDetectorAdapter = weak.Upgrade();
                CHECK_NULL_VOID(dataDetectorAdapter);
                if (detectTypesSet != dataDetectorAdapter->textDetectTypesSet_) {
                    return;
                }
                dataDetectorAdapter->HandleUrlResult(urlEntities);
                dataDetectorAdapter->MarkDirtyNode();
            },
            "ArkUITextUrlParseResult");
    };

    auto backgroundExecutor = SingleTaskExecutor::Make(context->GetTaskExecutor(), TaskExecutor::TaskType::BACKGROUND);
    backgroundExecutor.PostTask(
        [text = UtfUtils::Str16DebugToStr8(textForAI_), func = std::move(textFunc)] {
            TAG_LOGI(AceLogTag::ACE_TEXT, "Start url entity detect using AI");
            func(DataUrlAnalyzerMgr::GetInstance().AnalyzeUrls(text));
        },
        "ArkUITextInitUrlDetect");
}

void DataDetectorAdapter::HandleUrlResult(std::vector<UrlEntity> urlEntities)
{
    for (const auto& entity : urlEntities) {
        auto iter = aiSpanMap_.find(entity.charOffset);
        if (iter != aiSpanMap_.end() && iter->second.content.length() >= entity.text.length()) {
            // both entities start at the same position, leaving the longer one
            continue;
        }
        AISpan aiSpan;
        aiSpan.start = entity.charOffset;
        aiSpan.end = aiSpan.start + static_cast<int32_t>(UtfUtils::Str8DebugToStr16(entity.text).length());
        aiSpan.content = entity.text;
        aiSpan.type = TextDataDetectType::URL;
        aiSpanMap_[aiSpan.start] = aiSpan;
    }
    aiDetectFlag_ |= URL_DETECT_FINISH;
    if (aiDetectFlag_ == ALL_DETECT_FINISH) {
        FireFinalResult();
    }
}

void DataDetectorAdapter::ParseAIResult(const TextDataDetectResult& result, int32_t startPos)
{
    TAG_LOGI(AceLogTag::ACE_TEXT, "DataDetectorAdapter::ParseAIResult, ResultLength: %{public}zu, id: %{public}i",
        result.entity.size(), GetHost() ? GetHost()->GetId() : -1);
    auto entityJson = JsonUtil::ParseJsonString(result.entity);
    CHECK_NULL_VOID(entityJson);
    for (const auto& type : TEXT_DETECT_MAP) {
        auto jsonValue = entityJson->GetValue(type.second);
        ParseAIJson(jsonValue, type.first, startPos);
    }

    if (startPos + AI_TEXT_MAX_LENGTH >= static_cast<int32_t>(textForAI_.length())) {
        SetTextDetectResult(result);
        aiDetectFlag_ |= OTHER_DETECT_FINISH;
        if (aiDetectFlag_ == ALL_DETECT_FINISH) {
            FireFinalResult();
        }
    }
}

void DataDetectorAdapter::FireFinalResult()
{
    aiDetectInitialized_ = true;
    auto entityJsonArray = JsonUtil::CreateArray(true);
    // process with overlapping entities, leaving only the earlier ones
    int32_t preEnd = 0;
    auto aiSpanIterator = aiSpanMap_.begin();
    while (aiSpanIterator != aiSpanMap_.end()) {
        auto aiSpan = aiSpanIterator->second;
        if (aiSpan.start < preEnd) {
            aiSpanIterator = aiSpanMap_.erase(aiSpanIterator);
        } else {
            preEnd = aiSpan.end;
            ++aiSpanIterator;
            auto aiSpanJson = JsonUtil::Create(true);
            aiSpanJson->Put("start", aiSpan.start);
            aiSpanJson->Put("end", aiSpan.end);
            aiSpanJson->Put("entityContent", aiSpan.content.c_str());
            aiSpanJson->Put("entityType", TEXT_DETECT_MAP.at(aiSpan.type).c_str());
            entityJsonArray->Put(aiSpanJson);
        }
    }
    auto resultJson = JsonUtil::Create(true);
    resultJson->Put("entity", entityJsonArray);
    resultJson->Put("code", textDetectTypes_.empty() && hasUrlType_ ? 0: textDetectResult_.code);
    FireOnResult(resultJson->ToString());
    aiDetectFlag_ = 0;
}

void DataDetectorAdapter::ParseAIJson(
    const std::unique_ptr<JsonValue>& jsonValue, TextDataDetectType type, int32_t startPos)
{
    if (!jsonValue || !jsonValue->IsArray()) {
        TAG_LOGW(AceLogTag::ACE_TEXT, "Wrong AI result");
        return;
    }

    for (int32_t i = 0; i < jsonValue->GetArraySize(); ++i) {
        auto item = jsonValue->GetArrayItem(i);
        auto charOffset = item->GetInt("charOffset");
        auto oriText = item->GetString("oriText");
        auto wOriText = UtfUtils::Str8ToStr16(oriText);
        auto startTimestamp = item->GetInt64("startTimestamp", -1);
        int32_t end = startPos + charOffset + static_cast<int32_t>(wOriText.length());
        if (charOffset < 0 || startPos + charOffset >= static_cast<int32_t>(textForAI_.length()) ||
            end >= startPos + AI_TEXT_MAX_LENGTH || oriText.empty()) {
            TAG_LOGW(AceLogTag::ACE_TEXT, "The result of AI is wrong");
            continue;
        }
        if (wOriText != textForAI_.substr(startPos + charOffset, static_cast<int32_t>(wOriText.length()))) {
            TAG_LOGW(AceLogTag::ACE_TEXT, "The charOffset is wrong");
            continue;
        }
        int32_t start = startPos + charOffset;
        auto iter = aiSpanMap_.find(start);
        if (iter != aiSpanMap_.end() && iter->second.content.length() > oriText.length()) {
            // both entities start at the same position, leaving the longer one
            continue;
        }

        TimeStamp currentDetectorTimeStamp = std::chrono::high_resolution_clock::now();
        std::chrono::duration<float, std::ratio<1, SECONDS_TO_MILLISECONDS>> costTime =
            currentDetectorTimeStamp - startDetectorTimeStamp_;
        item->Put("costTime", costTime.count());
        item->Put("resultCode", textDetectResult_.code);
        entityJson_[start] = item->ToString();
        TAG_LOGI(AceLogTag::ACE_TEXT, "size of the entity json is: %{public}zu", entityJson_[start].size());

        AISpan aiSpan;
        aiSpan.start = start;
        aiSpan.end = end;
        aiSpan.content = oriText;
        aiSpan.type = type;
        if (startTimestamp != -1) {
            aiSpan.params["startTimestamp"] = std::to_string(startTimestamp);
        }
        aiSpanMap_[aiSpan.start] = aiSpan;
    }
}

std::function<void()> DataDetectorAdapter::GetDetectDelayTask(const std::map<int32_t, AISpan>& aiSpanMap)
{
    return [aiSpanMap, weak = WeakClaim(this)]() {
        auto dataDetectorAdapter = weak.Upgrade();
        CHECK_NULL_VOID(dataDetectorAdapter && !dataDetectorAdapter->textForAI_.empty());
        TAG_LOGI(AceLogTag::ACE_TEXT, "DataDetectorAdapter, delayed whole task executed, id: %{public}i",
            dataDetectorAdapter->GetHost() ? dataDetectorAdapter->GetHost()->GetId() : -1);
        dataDetectorAdapter->lastTextForAI_ = dataDetectorAdapter->textForAI_;
        size_t detectTextIdx = 0;
        auto aiSpanMapIt = aiSpanMap.begin();
        int32_t startPos = 0;
        bool hasSame = false;
        auto wTextForAI = dataDetectorAdapter->textForAI_;
        auto wTextForAILength = static_cast<int32_t>(wTextForAI.length());
        dataDetectorAdapter->PreprocessTextDetect();
        do {
            std::string detectText = UtfUtils::Str16DebugToStr8(
                wTextForAI.substr(startPos, std::min(AI_TEXT_MAX_LENGTH, wTextForAILength - startPos)));
            bool isSameDetectText = detectTextIdx < dataDetectorAdapter->detectTexts_.size() &&
                                    detectText == dataDetectorAdapter->detectTexts_[detectTextIdx];
            while (!aiSpanMap.empty() && aiSpanMapIt != aiSpanMap.end() && aiSpanMapIt->first >= 0 &&
                   aiSpanMapIt->first < std::min(wTextForAILength, startPos + AI_TEXT_MAX_LENGTH - AI_TEXT_GAP)) {
                auto aiContent = aiSpanMapIt->second.content;
                auto wAIContent = StringUtils::ToWstring(aiContent);
                if (isSameDetectText || aiContent == UtfUtils::Str16DebugToStr8(wTextForAI.substr(aiSpanMapIt->first,
                    std::min(static_cast<int32_t>(wAIContent.length()), wTextForAILength - aiSpanMapIt->first)))) {
                    dataDetectorAdapter->aiSpanMap_[aiSpanMapIt->first] = aiSpanMapIt->second;
                    hasSame = true;
                }
                ++aiSpanMapIt;
            }
            if (!isSameDetectText) {
                dataDetectorAdapter->InitTextDetect(startPos, detectText);
                if (detectTextIdx < dataDetectorAdapter->detectTexts_.size()) {
                    dataDetectorAdapter->detectTexts_[detectTextIdx] = detectText;
                } else {
                    dataDetectorAdapter->detectTexts_.emplace_back(detectText);
                }
            }
            ++detectTextIdx;
            startPos += AI_TEXT_MAX_LENGTH - AI_TEXT_GAP;
        } while (startPos + AI_TEXT_GAP < wTextForAILength && (!dataDetectorAdapter->textDetectTypes_.empty()));
        if (dataDetectorAdapter->hasUrlType_) {
            dataDetectorAdapter->HandleTextUrlDetect();
        }
        if (hasSame) {
            dataDetectorAdapter->MarkDirtyNode();
        }
    };
}

void DataDetectorAdapter::StartAITask()
{
    if (textForAI_.empty() || (!typeChanged_ && lastTextForAI_ == textForAI_)) {
        MarkDirtyNode();
        return;
    }
    std::map<int32_t, AISpan> aiSpanMapCopy;
    if (!typeChanged_) {
        aiSpanMapCopy = aiSpanMap_;
    }
    detectTexts_.clear();
    aiSpanMap_.clear();
    typeChanged_ = false;
    startDetectorTimeStamp_ = std::chrono::high_resolution_clock::now();
    auto context = PipelineContext::GetCurrentContextSafely();
    CHECK_NULL_VOID(context);
    auto taskExecutor = context->GetTaskExecutor();
    CHECK_NULL_VOID(taskExecutor);
    aiDetectDelayTask_.Cancel();
    aiDetectDelayTask_.Reset(GetDetectDelayTask(aiSpanMapCopy));
    TAG_LOGI(AceLogTag::ACE_TEXT, "DataDetectorAdapter::StartAITask, post whole task, id: %{public}i",
        GetHost() ? GetHost()->GetId() : -1);
    taskExecutor->PostDelayedTask(
        aiDetectDelayTask_, TaskExecutor::TaskType::UI, AI_DELAY_TIME, "ArkUITextStartAIDetect");
}

void DataDetectorAdapter::MarkDirtyNode() const
{
    auto host = GetHost();
    CHECK_NULL_VOID(host);
    host->MarkDirtyNode(NG::PROPERTY_UPDATE_MEASURE);
    auto layoutProperty = host->GetLayoutProperty();
    CHECK_NULL_VOID(layoutProperty);
    layoutProperty->OnPropertyChangeMeasure();
}
} // namespace OHOS::Ace