OpenHarmony-v6.0-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/common/ai/ai_write_adapter.h"

#include "array_wrapper.h"
#include "bool_wrapper.h"
#include "int_wrapper.h"
#include "want.h"
#include "core/components_ng/pattern/text/span/span_string.h"

namespace OHOS::Ace {
const std::pair<std::string, std::string> UI_ENTENSION_TYPE = {"ability.want.params.uiExtensionType", "sys/commonUI"};
const std::u16string BOUNDARY_SYMBOLS = u",.?，。？！";
const std::string API_VERSION = "apiVersion";
const std::string RESULT_BUFFER = "resultBuffer";
const std::string SHEET_DISMISS = "sheetDismiss";
const std::string PROCESS_ID = "processId";
const std::string MAX_CONTENT_LENGTH = "maxContentLength";
const std::string FIRST_HANDLE_RECT = "firstHandleRect";
const std::string SECOND_HANDLE_RECT = "secondHandleRect";
const std::string IS_AI_SUPPORT_METADATA = "isAiSupport";
const std::string SELECT_CONTENT_LENGTH = "selectContentLength";
const std::string REQUEST_LONG_CONTENT = "requestLongContent";
const std::string LONG_SENTENCE_BUFFER = "longSentenceBuffer";
const std::string LONG_SELECT_START = "longSelectStart";
const std::string LONG_SELECT_END = "longSelectEnd";
const std::string KEY_PACKAGE_NAME = "keyPackageApp";
const std::string START_COMPONONT_TYPE = "startComponentType";

bool AIWriteAdapter::IsSentenceBoundary(const char16_t value)
{
    for (char16_t item: BOUNDARY_SYMBOLS) {
        if (value == item) {
            return true;
        }
    }
    return false;
}

uint32_t AIWriteAdapter::GetSelectLengthOnlyText(const std::u16string& content)
{
    uint32_t length = 0;
    for (uint32_t i = 0; i < content.length(); i++) {
        if (content[i] != u' ' && content[i] != u'\n') {
            length++;
        }
    }
    return length;
}

void AIWriteAdapter::CloseModalUIExtension()
{
    TAG_LOGI(AceLogTag::ACE_RICH_TEXT, "CloseModalUIExtension.");
    auto context = pipelineContext_.Upgrade();
    CHECK_NULL_VOID(context);
    auto overlayManager = context->GetOverlayManager();
    CHECK_NULL_VOID(overlayManager);
    overlayManager->CloseModalUIExtension(sessionId_);
    SetAIWrite(false);
}

void AIWriteAdapter::ShowModalUIExtension(const AIWriteInfo& info,
    std::function<void(std::vector<uint8_t>&)> resultCallback)
{
    TAG_LOGI(AceLogTag::ACE_RICH_TEXT, "ShowModalUIExtension.");
    AAFwk::Want want;
    SetWantParams(info, want);
    Ace::ModalUIExtensionCallbacks callbacks;
    BindModalUIExtensionCallback(callbacks, resultCallback);

    auto context = pipelineContext_.Upgrade();
    CHECK_NULL_VOID(context);
    auto overlayManager = context->GetOverlayManager();
    CHECK_NULL_VOID(overlayManager);
    Ace::ModalUIExtensionConfig config;
    sessionId_ = overlayManager->CreateModalUIExtension(want, callbacks, config);
    aiWriteInfo_ = info;
}

void AIWriteAdapter::SetWantParams(const AIWriteInfo& info, AAFwk::Want& want)
{
    auto context = pipelineContext_.Upgrade();
    CHECK_NULL_VOID(context);
    auto apiVersion = AceApplicationInfo::GetInstance().GetApiTargetVersion();
    want.SetElementName(bundleName_, abilityName_);
    want.SetParam(UI_ENTENSION_TYPE.first, UI_ENTENSION_TYPE.second);
    want.SetParam(API_VERSION, static_cast<int>(apiVersion));
    want.SetParam(PROCESS_ID, getpid());
    want.SetParam(MAX_CONTENT_LENGTH, info.maxLength);
    want.SetParam(SELECT_CONTENT_LENGTH, info.selectLength);
    want.SetParam(FIRST_HANDLE_RECT, info.firstHandle);
    want.SetParam(SECOND_HANDLE_RECT, info.secondHandle);
    want.SetParam(KEY_PACKAGE_NAME, context->GetBundleName());
    want.SetParam(START_COMPONONT_TYPE, info.componentType);
}

void AIWriteAdapter::BindModalUIExtensionCallback(
    Ace::ModalUIExtensionCallbacks& callbacks, std::function<void(std::vector<uint8_t>&)> resultCallback)
{
    callbacks.onResult = [](int32_t code, const AAFwk::Want& want) {
        TAG_LOGD(AceLogTag::ACE_UIEXTENSIONCOMPONENT, "UIExtension onResult, code: %{public}d", code);
    };
    callbacks.onDestroy = [weak = WeakClaim(this)]() {
        TAG_LOGD(AceLogTag::ACE_UIEXTENSIONCOMPONENT, "UIExtension onDestroy.");
        auto aiWriteAdapter = weak.Upgrade();
        CHECK_NULL_VOID(aiWriteAdapter);
        aiWriteAdapter->CloseModalUIExtension();
    };
    callbacks.onError = [weak = WeakClaim(this)](int32_t code, const std::string& name, const std::string& message) {
        TAG_LOGE(AceLogTag::ACE_UIEXTENSIONCOMPONENT,
            "UIExtension onError, code: %{public}d, name: %{public}s, message: %{public}s",
            code, name.c_str(), message.c_str());
        auto aiWriteAdapter = weak.Upgrade();
        CHECK_NULL_VOID(aiWriteAdapter);
        aiWriteAdapter->CloseModalUIExtension();
    };
    callbacks.onRelease = [](int32_t code) {
        TAG_LOGD(AceLogTag::ACE_UIEXTENSIONCOMPONENT, "UIExtension onRelease, code: %{public}d", code);
    };
    callbacks.onRemoteReady = [weak = WeakClaim(this)](const std::shared_ptr<Ace::ModalUIExtensionProxy>& proxy) {
        TAG_LOGD(AceLogTag::ACE_UIEXTENSIONCOMPONENT, "UIExtension onRemoteReady.");
        auto aiWriteAdapter = weak.Upgrade();
        CHECK_NULL_VOID(aiWriteAdapter);
        aiWriteAdapter->SetModalUIExtensionProxy(proxy);
    };
    callbacks.onReceive = [weak = WeakClaim(this), cb = std::move(resultCallback)]
        (const AAFwk::WantParams& wantParams) {
        auto aiWriteAdapter = weak.Upgrade();
        CHECK_NULL_VOID(aiWriteAdapter);
        auto isSheetClose = false;
        auto isRequest = false;
        auto result = aiWriteAdapter->GetBufferParam(RESULT_BUFFER, wantParams);
        isSheetClose = aiWriteAdapter->GetBoolParam(SHEET_DISMISS, wantParams);
        isRequest = aiWriteAdapter->GetBoolParam(REQUEST_LONG_CONTENT, wantParams);
        if (isRequest) {
            aiWriteAdapter->SendData();
            return;
        }
        if (cb && !result.empty()) {
            cb(result);
            return;
        }
        if (isSheetClose) {
            aiWriteAdapter->CloseModalUIExtension();
        }
    };
}

void AIWriteAdapter::SendData()
{
    auto proxy = GetModalUIExtensionProxy();
    AAFwk::WantParams wantParams;
    SetArrayParam(wantParams, LONG_SENTENCE_BUFFER, aiWriteInfo_.sentenceBuffer);
    wantParams.SetParam(LONG_SELECT_START, AAFwk::Integer::Box(aiWriteInfo_.start));
    wantParams.SetParam(LONG_SELECT_END, AAFwk::Integer::Box(aiWriteInfo_.end));
    proxy->SendData(wantParams);
}

void AIWriteAdapter::SetArrayParam(
    AAFwk::WantParams& wantParams, const std::string& key, const std::vector<uint8_t>& value)
{
    std::vector<int> valueVec(value.size());
    std::transform(value.begin(), value.end(), valueVec.begin(),
        [](uint8_t x) { return static_cast<int>(x); });
    size_t size = valueVec.size();
    sptr<AAFwk::IArray> ao = new (std::nothrow) AAFwk::Array(size, AAFwk::g_IID_IInteger);
    if (ao == nullptr) {
        return;
    }
    for (size_t i = 0; i < size; i++) {
        ao->Set(i, AAFwk::Integer::Box(value[i]));
    }
    wantParams.SetParam(key, ao);
}

std::vector<uint8_t> AIWriteAdapter::GetBufferParam(const std::string& key, const AAFwk::WantParams& wantParams)
{
    std::vector<uint8_t> array;
    auto value = wantParams.GetParam(key);
    AAFwk::IArray *ao = AAFwk::IArray::Query(value);
    if (ao != nullptr && AAFwk::Array::IsIntegerArray(ao)) {
        auto func = [&](AAFwk::IInterface *object) {
            CHECK_NULL_VOID(object);
            AAFwk::IInteger *value = AAFwk::IInteger::Query(object);
            CHECK_NULL_VOID(value);
            array.emplace_back(AAFwk::Integer::Unbox(value));
        };
        AAFwk::Array::ForEach(ao, func);
    }
    return array;
}

bool AIWriteAdapter::GetBoolParam(const std::string& key, const AAFwk::WantParams& wantParams)
{
    auto value = wantParams.GetParam(key);
    AAFwk::IBoolean *bo = AAFwk::IBoolean::Query(value);
    if (bo != nullptr) {
        return AAFwk::Boolean::Unbox(bo);
    }
    return false;
}
}