d92502b7创建于 2025年1月17日历史提交
/*
 * Copyright (c) 2025 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 "web_download_delegate.h"

#include <cstring>

#include "js_web_webview.h"
#include "log.h"
#include "nweb_napi_scope.h"
#include "web_download_item.h"
#include "web_download_manager.h"

namespace OHOS {
namespace Plugin {
WebDownloadDelegate::WebDownloadDelegate(napi_env env)
    : delegate_(nullptr),
      download_before_start_callback_(nullptr),
      download_did_update_callback_(nullptr),
      download_did_finish_callback_(nullptr),
      download_did_fail_callback_(nullptr),
      env_(env)
{
}

WebDownloadDelegate::~WebDownloadDelegate()
{
    if (download_before_start_callback_) {
        napi_delete_reference(env_, download_before_start_callback_);
    }
    if (download_did_update_callback_) {
        napi_delete_reference(env_, download_did_update_callback_);
    }
    if (download_did_finish_callback_) {
        napi_delete_reference(env_, download_did_finish_callback_);
    }
    if (download_did_fail_callback_) {
        napi_delete_reference(env_, download_did_fail_callback_);
    }
    WebDownloadManager::RemoveDownloadDelegate(this);
}

void WebDownloadDelegate::RemoveSelfRef()
{
    if (delegate_) {
        napi_delete_reference(env_, delegate_);
        delegate_ = nullptr;
    }
}

void WebDownloadDelegate::DownloadBeforeStart(WebDownloadItem* webDownloadItem)
{
    if (!env_) {
        LOGE("[DOWNLOAD] WebDownloadDelegate::DownloadBeforeStart nil env");
        return;
    }
    size_t paramCount = 1;
    OHOS::NApiScope scope(env_);
    napi_value callbackFunc = nullptr;
    napi_status status;

    if (!download_before_start_callback_) {
        LOGE("[DOWNLOAD] downloadBeforeStart nil env");
        return;
    }
    status = napi_get_reference_value(env_, download_before_start_callback_, &callbackFunc);
    if (status != napi_ok || callbackFunc == nullptr) {
        LOGE("[DOWNLOAD] get downloadBeforeStart func failed.");
        return;
    }
    napi_value webDownloadItemValue = nullptr;
    napi_create_object(env_, &webDownloadItemValue);
    status = napi_wrap(
        env_, webDownloadItemValue, webDownloadItem,
        [](napi_env, void* data, void*) {
            if (data) {
                WebDownloadItem* downloadItem = static_cast<WebDownloadItem*>(data);
                if (downloadItem) {
                    delete downloadItem;
                    downloadItem = nullptr;
                }
            }
        },
        nullptr, nullptr);
    if (status != napi_ok) {
        if (webDownloadItem) {
            delete webDownloadItem;
            webDownloadItem = nullptr;
        }
        return;
    }
    NapiWebDownloadItem::DefineProperties(env_, &webDownloadItemValue);
    napi_value result = nullptr;
    status = napi_call_function(env_, nullptr, callbackFunc, paramCount, &webDownloadItemValue, &result);
    if (status != napi_status::napi_ok) {
        LOGE("[DOWNLOAD] call downloadBeforeStart failed.");
    }
}

void WebDownloadDelegate::DownloadDidUpdate(WebDownloadItem* webDownloadItem)
{
    if (!env_) {
        LOGE("[DOWNLOAD] WebDownloadDelegate::DownloadDidUpdate nil env.");
        return;
    }
    size_t paramCount = 1;
    OHOS::NApiScope scope(env_);
    napi_value callbackFunc = nullptr;
    napi_status status;

    if (!download_did_update_callback_) {
        LOGE("[DOWNLOAD] downloadDidUpdate not exists.");
        return;
    }
    status = napi_get_reference_value(env_, download_did_update_callback_, &callbackFunc);
    if (status != napi_ok || callbackFunc == nullptr) {
        LOGE("[DOWNLOAD] get downloadDidUpdate func failed.");
        return;
    }
    napi_value webDownloadItemValue = nullptr;
    napi_create_object(env_, &webDownloadItemValue);
    status = napi_wrap(
        env_, webDownloadItemValue, webDownloadItem,
        [](napi_env, void* data, void*) {
            if (data) {
                WebDownloadItem* downloadItem = static_cast<WebDownloadItem*>(data);
                if (downloadItem) {
                    delete downloadItem;
                    downloadItem = nullptr;
                }
            }
        },
        nullptr, nullptr);
    if (status != napi_ok) {
        if (webDownloadItem) {
            delete webDownloadItem;
            webDownloadItem = nullptr;
        }
        return;
    } 
    NapiWebDownloadItem::DefineProperties(env_, &webDownloadItemValue);
    napi_value result = nullptr;
    status = napi_call_function(env_, nullptr, callbackFunc, paramCount, &webDownloadItemValue, &result);
    if (status != napi_status::napi_ok) {
        LOGE("[DOWNLOAD] call downloadDidUpdate failed.");
    }
}

void WebDownloadDelegate::DownloadDidFail(WebDownloadItem* webDownloadItem)
{
    if (!env_) {
        LOGE("[DOWNLOAD] WebDownloadDelegate::DownloadDidFail nil env.");
        return;
    }
    size_t paramCount = 1;
    OHOS::NApiScope scope(env_);
    napi_value callbackFunc = nullptr;
    napi_status status;
    if (!download_did_fail_callback_) {
        LOGE("[DOWNLOAD] DownloadDidFail not exists.");
        return;
    }
    status = napi_get_reference_value(env_, download_did_fail_callback_, &callbackFunc);
    if (status != napi_ok || callbackFunc == nullptr) {
        LOGE("[DOWNLOAD] get downloadDidFail func failed.");
        return;
    }
    napi_value webDownloadItemValue = nullptr;
    napi_create_object(env_, &webDownloadItemValue);
    status = napi_wrap(
        env_, webDownloadItemValue, webDownloadItem,
        [](napi_env, void* data, void*) {
            if (data) {
                WebDownloadItem* downloadItem = static_cast<WebDownloadItem*>(data);
                if (downloadItem) {
                    delete downloadItem;
                    downloadItem = nullptr;
                }
            }
        },
        nullptr, nullptr);
    if (status != napi_ok) {
        if (webDownloadItem) {
            delete webDownloadItem;
            webDownloadItem = nullptr;
        }
        return;
    } 
    NapiWebDownloadItem::DefineProperties(env_, &webDownloadItemValue);
    napi_value result = nullptr;
    status = napi_call_function(env_, nullptr, callbackFunc, paramCount, &webDownloadItemValue, &result);
    if (status != napi_status::napi_ok) {
        LOGE("[DOWNLOAD] call downloadDidFail failed.");
    }
}

void WebDownloadDelegate::DownloadDidFinish(WebDownloadItem* webDownloadItem)
{
    if (!env_) {
        LOGE("[DOWNLOAD] WebDownloadDelegate::DownloadDidFinish nil env.");
        return;
    }
    size_t paramCount = 1;
    OHOS::NApiScope scope(env_);
    napi_value callbackFunc = nullptr;
    napi_status status;
    if (!download_did_finish_callback_) {
        LOGE("[DOWNLOAD] downloadDidFinish not exists.");
        return;
    }
    status = napi_get_reference_value(env_, download_did_finish_callback_, &callbackFunc);
    if (status != napi_ok || callbackFunc == nullptr) {
        LOGE("[DOWNLOAD] get downloadDidFinish func failed.");
        return;
    }

    napi_value webDownloadItemValue = nullptr;
    napi_create_object(env_, &webDownloadItemValue);
    status = napi_wrap(
        env_, webDownloadItemValue, webDownloadItem,
        [](napi_env, void* data, void*) {
            if (data) {
                WebDownloadItem* downloadItem = static_cast<WebDownloadItem*>(data);
                if (downloadItem) {
                    delete downloadItem;
                    downloadItem = nullptr;
                }
            }
        },
        nullptr, nullptr);
    if (status != napi_ok) {
        if (webDownloadItem) {
            delete webDownloadItem;
            webDownloadItem = nullptr;
        }
        return;
    } 
    NapiWebDownloadItem::DefineProperties(env_, &webDownloadItemValue);

    napi_value result = nullptr;
    status = napi_call_function(env_, nullptr, callbackFunc, paramCount, &webDownloadItemValue, &result);
    if (status != napi_status::napi_ok) {
        LOGE("[DOWNLOAD] call downloadDidFinish failed.");
    }
}

void WebDownloadDelegate::PutDownloadBeforeStart(napi_env, napi_value callback)
{
    napi_status status = napi_create_reference(env_, callback, 1, &download_before_start_callback_);
    if (status != napi_status::napi_ok) {
        LOGE("[DOWNLOAD] PutDownloadBeforeStart create reference failed.");
    }
}

void WebDownloadDelegate::PutDownloadDidUpdate(napi_env, napi_value callback)
{
    napi_status status = napi_create_reference(env_, callback, 1, &download_did_update_callback_);
    if (status != napi_status::napi_ok) {
        LOGE("[DOWNLOAD] PutDownloadDidUpdate create reference failed.");
    }
}

void WebDownloadDelegate::PutDownloadDidFinish(napi_env, napi_value callback)
{
    napi_status status = napi_create_reference(env_, callback, 1, &download_did_finish_callback_);
    if (status != napi_status::napi_ok) {
        LOGE("[DOWNLOAD] PutDownloadDidFinish create reference failed.");
    }
}

void WebDownloadDelegate::PutDownloadDidFail(napi_env, napi_value callback)
{
    napi_status status = napi_create_reference(env_, callback, 1, &download_did_fail_callback_);
    if (status != napi_status::napi_ok) {
        LOGE("[DOWNLOAD] PutDownloadDidFail create reference failed.");
    }
}

int32_t WebDownloadDelegate::GetNWebId() const
{
    return nwebId_;
}

void WebDownloadDelegate::SetNWebId(int32_t nwebId)
{
    nwebId_ = nwebId;
}

napi_env WebDownloadDelegate::GetEnv()
{
    return env_;
}
} // namespace Plugin
} // namespace OHOS